From 5d8545242c29dffac21d7282c9b4d8d57dbedc4c Mon Sep 17 00:00:00 2001
From: Matt Guo <matt@mattguo.com>
Date: Sun, 2 Mar 2025 16:50:10 -0800
Subject: [PATCH] support using deepseek models

---
 src/lib/openai.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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