✨ feat: Support include_reasoning for OpenRouter provider's models.

Solves lobehub#5766
deephbz · Feb 8, 2025 · 7c7b5ba · 7c7b5ba
1 parent 29d14b6
commit 7c7b5ba
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 7 deletions.
diff --git a/src/config/modelProviders/openrouter.ts b/src/config/modelProviders/openrouter.ts
@@ -181,6 +181,19 @@ const OpenRouter: ModelProviderCard = {
       },
       releasedAt: '2024-09-05',
     },
+    {
+      contextWindowTokens: 163_840,
+      description: 'DeepSeek-R1',
+      displayName: 'DeepSeek R1',
+      enabled: true,
+      functionCall: true,
+      id: 'deepseek/deepseek-r1',
+      pricing: {
+        input: 3,
+        output: 8,
+      },
+      releasedAt: '2025-01-20',
+    },
     {
       contextWindowTokens: 131_072,
       description:

diff --git a/src/libs/agent-runtime/openrouter/index.ts b/src/libs/agent-runtime/openrouter/index.ts
@@ -6,6 +6,14 @@ import { OpenRouterModelCard } from './type';
 
 export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
   baseURL: 'https://openrouter.ai/api/v1',
+  chatCompletion: {
+    handlePayload: (payload) => {
+      return {
+        ...payload,
+        include_reasoning: true,
+      } as any;
+    },
+  },
   constructorOptions: {
     defaultHeaders: {
       'HTTP-Referer': 'https://chat-preview.lobehub.com',
@@ -17,18 +25,15 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
   },
   models: {
     transformModel: (m) => {
-      const visionKeywords = [
-        'qwen/qvq',
-        'vision',
-      ];
+      const visionKeywords = ['qwen/qvq', 'vision'];
 
       const reasoningKeywords = [
         'deepseek/deepseek-r1',
         'openai/o1',
         'openai/o3',
         'qwen/qvq',
         'qwen/qwq',
-        'thinking'
+        'thinking',
       ];
 
       const model = m as unknown as OpenRouterModelCard;
@@ -45,11 +50,11 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
           typeof model.top_provider.max_completion_tokens === 'number'
             ? model.top_provider.max_completion_tokens
             : undefined,
-        reasoning: reasoningKeywords.some(keyword => model.id.toLowerCase().includes(keyword)),
+        reasoning: reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)),
         vision:
           model.description.includes('vision') ||
           model.description.includes('multimodal') ||
-          visionKeywords.some(keyword => model.id.toLowerCase().includes(keyword)),
+          visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)),
       };
     },
   },