feat: add open ai key

stackblitz · Oct 13, 2024 · 167addf · 167addf
1 parent db3e97e
commit 167addf
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
@@ -63,6 +63,7 @@ ANTHROPIC_API_KEY=XXX
 PROVIDER= gemini | antrophic | openai | ollama
 MODEL_NAME=XXX
 GOOGLE_GENERATIVE_AI_API_KEY=XXX
+OPEN_AI_API_KEY=XXX
 ```
 
 Optionally, you can set the debug level:

diff --git a/app/lib/.server/llm/api-key.ts b/app/lib/.server/llm/api-key.ts
@@ -11,5 +11,9 @@ export function getAPIKey(cloudflareEnv: Env) {
     return cloudflareEnv.GOOGLE_GENERATIVE_AI_API_KEY || (env.GOOGLE_GENERATIVE_AI_API_KEY as string);
   }
 
+  if (provider === 'openai') {
+    return cloudflareEnv.OPEN_AI_API_KEY || (env.OPEN_AI_API_KEY as string);
+  }
+
   return env.ANTHROPIC_API_KEY || cloudflareEnv.ANTHROPIC_API_KEY;
 }
diff --git a/worker-configuration.d.ts b/worker-configuration.d.ts
@@ -3,4 +3,5 @@ interface Env {
   PROVIDER: string;
   MODEL_NAME: string;
   GOOGLE_GENERATIVE_AI_API_KEY: string;
+  OPEN_AI_API_KEY: string;
 }