set envionment variables in code

Signed-off-by: Vanish1011 <[email protected]>
dragonflyoss · Dec 25, 2023 · df45ed7 · df45ed7
1 parent 45595b6
commit df45ed7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/dragonfly.cpp b/src/dragonfly.cpp
@@ -43,6 +43,8 @@ TRITONREPOAGENT_ModelAction(
       if (file_path_c_str != nullptr) {
         // Load from credential file
         cred_path = std::string(file_path_c_str);
+      } else {
+        cred_path="/home/triton/cloud_credential.json" ;
       }
 
       const char* config_path_c_str =
@@ -51,6 +53,8 @@ TRITONREPOAGENT_ModelAction(
       if (config_path_c_str != nullptr) {
         // Load from config file
         config_path = std::string(config_path_c_str);
+      } else {
+        config_path="/home/triton/dragonfly_config.json" ;
       }
 
       const char* temp_dir_cstr = nullptr;