@lobehub/lobehub 2.0.0-next.57 → 2.0.0-next.59

package/packages/model-runtime/src/providers/deepseek/index.test.ts

@@ -20,6 +20,92 @@ testProvider({
 });
 
 describe('LobeDeepSeekAI - custom features', () => {
+  describe('chatCompletion.handlePayload', () => {
+    it('should transform reasoning object to reasoning_content string', () => {
+      const payload = {
+        messages: [
+          { role: 'user', content: 'Hello' },
+          {
+            role: 'assistant',
+            content: 'Hi there',
+            reasoning: { content: 'Let me think...', duration: 1000 },
+          },
+          { role: 'user', content: 'How are you?' },
+        ],
+        model: 'deepseek-r1',
+      };
+
+      const result = params.chatCompletion!.handlePayload!(payload as any);
+
+      expect(result.messages).toEqual([
+        { role: 'user', content: 'Hello' },
+        {
+          role: 'assistant',
+          content: 'Hi there',
+          reasoning_content: 'Let me think...',
+        },
+        { role: 'user', content: 'How are you?' },
+      ]);
+    });
+
+    it('should not modify messages without reasoning field', () => {
+      const payload = {
+        messages: [
+          { role: 'user', content: 'Hello' },
+          { role: 'assistant', content: 'Hi there' },
+        ],
+        model: 'deepseek-chat',
+      };
+
+      const result = params.chatCompletion!.handlePayload!(payload as any);
+
+      expect(result.messages).toEqual(payload.messages);
+    });
+
+    it('should handle empty reasoning content', () => {
+      const payload = {
+        messages: [
+          {
+            role: 'assistant',
+            content: 'Response',
+            reasoning: { duration: 1000 },
+          },
+        ],
+        model: 'deepseek-r1',
+      };
+
+      const result = params.chatCompletion!.handlePayload!(payload as any);
+
+      expect(result.messages[0]).toEqual({
+        role: 'assistant',
+        content: 'Response',
+      });
+    });
+
+    it('should set stream to true by default', () => {
+      const payload = {
+        messages: [{ role: 'user', content: 'Hello' }],
+        model: 'deepseek-chat',
+      };
+
+      const result = params.chatCompletion!.handlePayload!(payload as any);
+
+      expect(result.stream).toBe(true);
+    });
+
+    it('should preserve existing stream value', () => {
+      const payload = {
+        messages: [{ role: 'user', content: 'Hello' }],
+        model: 'deepseek-chat',
+        stream: false,
+      };
+
+      const result = params.chatCompletion!.handlePayload!(payload as any);
+
+      expect(result.stream).toBe(false);
+    });
+  });
+
   describe('Debug Configuration', () => {
     it('should disable debug by default', () => {
       delete process.env.DEBUG_DEEPSEEK_CHAT_COMPLETION;

package/packages/model-runtime/src/providers/deepseek/index.ts

@@ -12,6 +12,30 @@ export interface DeepSeekModelCard {
 
 export const params = {
   baseURL: 'https://api.deepseek.com/v1',
+  chatCompletion: {
+    handlePayload: (payload) => {
+      // Transform reasoning object to reasoning_content string for multi-turn conversations
+      const messages = payload.messages.map((message: any) => {
+        // Only transform if message has reasoning.content
+        if (message.reasoning?.content) {
+          const { reasoning, ...rest } = message;
+          return {
+            ...rest,
+            reasoning_content: reasoning.content,
+          };
+        }
+        // If message has reasoning but no content, remove reasoning field entirely
+        delete message.reasoning;
+        return message;
+      });
+
+      return {
+        ...payload,
+        messages,
+        stream: payload.stream ?? true,
+      } as any;
+    },
+  },
   debug: {
     chatCompletion: () => process.env.DEBUG_DEEPSEEK_CHAT_COMPLETION === '1',
   },