updated ollama to use defined base URL for model calls

This commit is contained in:
noobydp
2024-10-24 16:49:11 +08:00
parent 94fa108142
commit 8ce82b5a1c
4 changed files with 9 additions and 5 deletions

View File

@@ -30,6 +30,8 @@ export function getBaseURL(cloudflareEnv: Env, provider: string) {
switch (provider) {
case 'OpenAILike':
return env.OPENAI_LIKE_API_BASE_URL || cloudflareEnv.OPENAI_LIKE_API_BASE_URL;
case 'Ollama':
return env.OLLAMA_API_BASE_URL || cloudflareEnv.OLLAMA_API_BASE_URL;
default:
return "";
}