UniqAI/bolt.diy
1
0
Fork 0
mirror of https://github.com/stackblitz-labs/bolt.diy synced 2025-02-01 23:44:39 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Added deepseek models

Browse Source
This commit is contained in:
zenith110 2024-10-22 18:19:27 -04:00
parent 50a501ecb1
commit 3b90afba48
4 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/lib/.server/llm/api-key.ts
View File

@ -19,6 +19,8 @@ export function getAPIKey(cloudflareEnv: Env, provider: string) {
return env.GROQ_API_KEY || cloudflareEnv.GROQ_API_KEY;
case 'OpenRouter':
return env.OPEN_ROUTER_API_KEY || cloudflareEnv.OPEN_ROUTER_API_KEY;
case 'Deepseek':
return env.DEEPSEEK_API_KEY || cloudflareEnv.DEEPSEEK_API_KEY
default:
return "";
}

10
app/lib/.server/llm/model.ts
View File

@ -40,6 +40,14 @@ export function getGroqModel(apiKey: string, model: string) {
return openai(model);
}
export function getDeepseekModel(apiKey: string, model: string){
const openai = createOpenAI({
baseURL: 'https://api.deepseek.com/beta',
apiKey,
});
return openai(model);
}
export function getOllamaModel(model: string) {
return ollama(model);
}
@ -67,6 +75,8 @@ export function getModel(provider: string, model: string, env: Env) {
return getOpenRouterModel(apiKey, model);
case 'Google':
return getGoogleModel(apiKey, model)
case 'Deepseek':
return getDeepseekModel(apiKey, model)
default:
return getOllamaModel(model);
}

2
app/utils/constants.ts
View File

@ -30,6 +30,8 @@ const staticModels: ModelInfo[] = [
{ name: 'gpt-4-turbo', label: 'GPT-4 Turbo', provider: 'OpenAI' },
{ name: 'gpt-4', label: 'GPT-4', provider: 'OpenAI' },
{ name: 'gpt-3.5-turbo', label: 'GPT-3.5 Turbo', provider: 'OpenAI' },
{ name: 'deepseek-coder', label: 'Deepseek-Coder', provider: 'Deepseek'},
{ name: 'deepseek-chat', label: 'Deepseek-Chat', provider: 'Deepseek'},
];
export let MODEL_LIST: ModelInfo[] = [...staticModels];

1
worker-configuration.d.ts vendored
View File

@ -4,4 +4,5 @@ interface Env {
GROQ_API_KEY: string;
OPEN_ROUTER_API_KEY: string;
OLLAMA_API_BASE_URL: string;
DEEPSEEK_API_KEY: string;
}