feat: add open ai key

This commit is contained in:
Rafael Faria 2024-10-13 17:37:17 -03:00
parent db3e97e306
commit 167addf427
3 changed files with 6 additions and 0 deletions

View File

@ -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:

View File

@ -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;
}

View File

@ -3,4 +3,5 @@ interface Env {
PROVIDER: string;
MODEL_NAME: string;
GOOGLE_GENERATIVE_AI_API_KEY: string;
OPEN_AI_API_KEY: string;
}