feat: set envionment variables in code (#19)

Signed-off-by: Vanish1011 <[email protected]>
dragonflyoss · Dec 25, 2023 · 443779f · 443779f
1 parent 45595b6
commit 443779f
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;