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 affinityMap = { {"PreferredAffinity", core_service::AffinityType::PreferredAffinity}, diff --git a/test/libruntime/invoke_spec_test.cpp b/test/libruntime/invoke_spec_test.cpp index 2b73afa6a76303d0b5fadb449b00c4dae70295e7..6b3dae71a04e9d148eaf56af75f9e33b706f4845 100644 --- a/test/libruntime/invoke_spec_test.cpp +++ b/test/libruntime/invoke_spec_test.cpp @@ -236,9 +236,13 @@ TEST_F(InvokeSpecTest, BuildInstanceCreateRequestTest) spec->opts.createOptions["crateOption"] = "createOption"; spec->functionMeta.name = "name"; spec->functionMeta.functionId = "functionId"; + ResourceGroupOptions resourceGroupOpts; + resourceGroupOpts.resourceGroupName = "testResourceGroup"; + spec->opts.resourceGroupOpts = resourceGroupOpts; LibruntimeConfig config; spec->BuildInstanceCreateRequest(config); ASSERT_EQ(spec->requestCreate.mutable_schedulingops()->mutable_resources()->size() != 0, true); + ASSERT_EQ(spec->opts.envVars["RG_NAME"], "testResourceGroup"); } } // namespace test } // namespace YR \ No newline at end of file