fix bug

neu-se · Dec 23, 2024 · 92101fd · 92101fd
1 parent 1c5dbe7
commit 92101fd
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/model/Model.ts b/src/model/Model.ts
@@ -24,9 +24,16 @@ export class Model implements IModel {
     getEnv("LLMORPHEUS_LLM_AUTH_HEADERS")
   );
 
-  protected static LLMORPHEUS_LLM_PROVIDER = JSON.parse(
-    getEnv("LLMORPHEUS_LLM_PROVIDER", false)
-  );
+  protected static LLMORPHEUS_LLM_PROVIDER = Model.getLLMProvider();
+
+  private static getLLMProvider() {
+    const llmProvider = getEnv("LLMORPHEUS_LLM_PROVIDER", false);
+    if (llmProvider) {
+      return JSON.parse(llmProvider);
+    } else {
+      return null;
+    }
+  }
 
   protected instanceOptions: PostOptions;
   protected rateLimiter: RateLimiter;