add tool support for deepseek-chat、deepseek-reasoner(Only for deepseek official API )

Author: xiaohuanxiong3

core/llm/autodetect.ts

@@ -10,7 +10,6 @@ import {
   chatmlTemplateMessages,
   codeLlama70bTemplateMessages,
   codestralTemplateMessages,
-  deepseekTemplateMessages,
   gemmaTemplateMessage,
   graniteTemplateMessages,
   llama2TemplateMessages,
@@ -293,7 +292,8 @@ function autodetectTemplateFunction(
       zephyr: zephyrTemplateMessages,
       anthropic: anthropicTemplateMessages,
       chatml: chatmlTemplateMessages,
-      deepseek: deepseekTemplateMessages,
+      // deepseek: deepseekTemplateMessages,
+      deepseek: null,
       openchat: openchatTemplateMessages,
       "xwin-coder": xWinCoderTemplateMessages,
       "neural-chat": neuralChatTemplateMessages,

core/llm/llms/Deepseek.ts

@@ -1,7 +1,14 @@
 import { streamSse } from "@continuedev/fetch";
-import { CompletionOptions, LLMOptions } from "../../index.js";
+import {
+  AssistantChatMessage,
+  ChatMessage,
+  CompletionOptions,
+  LLMOptions,
+  ThinkingChatMessage,
+} from "../../index.js";
 import { osModelsEditPrompt } from "../templates/edit.js";
 
+import { LlmApiRequestType } from "../openaiTypeConverters";
 import OpenAI from "./OpenAI.js";
 
 class Deepseek extends OpenAI {
@@ -16,6 +23,8 @@ class Deepseek extends OpenAI {
   };
   maxStopWords: number | undefined = 16;
 
+  protected useOpenAIAdapterFor: (LlmApiRequestType | "*")[] = [];
+
   supportsFim(): boolean {
     return true;
   }
@@ -52,6 +61,110 @@ class Deepseek extends OpenAI {
       yield chunk.choices[0].text;
     }
   }
+
+  protected async *_streamChat(
+    messages: ChatMessage[],
+    signal: AbortSignal,
+    options: CompletionOptions,
+  ): AsyncGenerator<ChatMessage, any, any> {
+    const body = this._convertArgs(options, messages);
+
+    const response = await this.fetch(this._getEndpoint("chat/completions"), {
+      method: "POST",
+      headers: this._getHeaders(),
+      body: JSON.stringify({
+        ...body,
+        ...this.extraBodyProperties(),
+      }),
+      signal,
+    });
+
+    // Handle non-streaming response
+    if (body.stream === false) {
+      if (response.status === 499) {
+        return; // Aborted by user
+      }
+      const data = await response.json();
+      yield data.choices[0].message;
+      return;
+    }
+
+    let message: AssistantChatMessage | ThinkingChatMessage | undefined;
+    let myArguments: string | undefined;
+    let lastMessageRole: "assistant" | "thinking" | undefined;
+
+    function fromChatCompletionChunk(chunk: any): ChatMessage | undefined {
+      const delta = chunk.choices?.[0]?.delta;
+
+      if (delta?.content) {
+        lastMessageRole = "assistant";
+        return {
+          role: "assistant",
+          content: delta.content,
+        };
+      } else if (delta?.reasoning_content) {
+        lastMessageRole = "thinking";
+        return {
+          role: "thinking",
+          content: delta.reasoning_content,
+        };
+      } else if (delta?.tool_calls) {
+        if (!message) {
+          message = {
+            role: "assistant",
+            content: "",
+            toolCalls: delta?.tool_calls.map((tool_call: any) => ({
+              id: tool_call.id,
+              type: tool_call.type,
+              function: {
+                name: tool_call.function?.name,
+                arguments: tool_call.function?.arguments,
+              },
+            })),
+          };
+          myArguments = "";
+          return message;
+        } else {
+          // @ts-ignore
+          myArguments += delta?.tool_calls[0].function.arguments;
+        }
+        return undefined;
+      }
+
+      if (chunk.choices?.[0]?.finish_reason === "tool_calls") {
+        if (message) {
+          message = {
+            role: message.role,
+            content: message.content,
+            toolCalls: [
+              {
+                id: message.toolCalls?.[0].id,
+                type: message.toolCalls?.[0].type,
+                function: {
+                  name: message.toolCalls?.[0].function?.name,
+                  arguments: myArguments,
+                },
+              },
+            ],
+          };
+          const tempMessage = message;
+          message = undefined;
+          return tempMessage;
+        } else {
+          return undefined;
+        }
+      } else {
+        return undefined;
+      }
+    }
+
+    for await (const value of streamSse(response)) {
+      const chunk = fromChatCompletionChunk(value);
+      if (chunk) {
+        yield chunk;
+      }
+    }
+  }
 }
 
 export default Deepseek;

core/llm/toolSupport.ts

@@ -179,7 +179,11 @@ export const PROVIDER_TOOL_SUPPORT: Record<string, (model: string) => boolean> =
       return false;
     },
     deepseek: (model) => {
-      if (model !== "deepseek-reasoner") {
+      if (
+        model == "deepseek-reasoner" ||
+        model == "deepseek-chat" ||
+        model == "deepseek-coder"
+      ) {
         return true;
       }