diff --git a/src/lib/openai.ts b/src/lib/openai.ts index 2fb499d..b3ed508 100644 --- a/src/lib/openai.ts +++ b/src/lib/openai.ts @@ -144,7 +144,7 @@ export async function openAI( baseURL: options.completionEndpoint }); try { - if (engine.startsWith("gpt-3.5") || engine.startsWith("gpt-4")) { + if (engine.startsWith("gpt-3.5") || engine.startsWith("gpt-4") || engine.startsWith("deepseek")) { const inputMessages:OpenAI.Chat.CreateChatCompletionRequestMessage[] = [{ role: "user", content: input }]; if (openAiOptions.chatPrompt && openAiOptions.chatPrompt.length > 0) { inputMessages.unshift({ role: "system", content: openAiOptions.chatPrompt }); @@ -220,7 +220,7 @@ export async function openAIWithStream( const engine = options.completionEngine!; try { - if (engine.startsWith("gpt-3.5") || engine.startsWith("gpt-4")) { + if (engine.startsWith("gpt-3.5") || engine.startsWith("gpt-4") || engine.startsWith("deepseek")) { const inputMessages: OpenAI.Chat.CreateChatCompletionRequestMessage[] = [{ role: "user", content: input }]; if (openAiOptions.chatPrompt && openAiOptions.chatPrompt.length > 0) { inputMessages.unshift({ role: "system", content: openAiOptions.chatPrompt });