diff --git a/src/libruntime/invoke_spec.cpp b/src/libruntime/invoke_spec.cpp
index b407c20cb6c6fa6fe5342933a4a2a592ab9d5dd1..fe2ddddbe46b5588820ef06e6a97fbe98c9126b9 100644
--- a/src/libruntime/invoke_spec.cpp
+++ b/src/libruntime/invoke_spec.cpp
@@ -138,6 +138,7 @@ void InvokeSpec::BuildRequestPbScheduleOptions(InvokeOptions &opts, const Librun
     }
     if (ResourceGroupEnabled(opts.resourceGroupOpts)) {
         schedulingOps->set_rgroupname(opts.resourceGroupOpts.resourceGroupName);
+        this->opts.envVars["RG_NAME"] = opts.resourceGroupOpts.resourceGroupName;
     }
     static std::unordered_map<std::string, core_service::AffinityType> affinityMap = {
         {"PreferredAffinity", core_service::AffinityType::PreferredAffinity},