From aafb62cb2f880b6d256c46712add6a4dda33f6f1 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Wed, 19 Feb 2025 10:11:17 +0800
Subject: [PATCH] feat: Add web search support for OpenRouter provider

---
 src/renderer/src/config/models.ts | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts
index cd76e5b2ac..f965f1aa3a 100644
--- a/src/renderer/src/config/models.ts
+++ b/src/renderer/src/config/models.ts
@@ -1671,12 +1671,17 @@ export function isWebSearchModel(model: Model): boolean {
   if (provider.id === 'zhipu') {
     return model?.id?.startsWith('glm-4-')
   }
+
   if (provider.id === 'dashscope') {
     const models = ['qwen-turbo', 'qwen-max', 'qwen-plus']
     // matches id like qwen-max-0919, qwen-max-latest
     return models.some((i) => model.id.startsWith(i))
   }
 
+  if (provider.id === 'openrouter') {
+    return true
+  }
+
   return false
 }
 
@@ -1688,6 +1693,7 @@ export function getOpenAIWebSearchParams(assistant: Assistant, model: Model): Re
       if (model.provider === 'hunyuan') {
         return { enable_enhancement: true }
       }
+
       if (model.provider === 'dashscope') {
         return {
           enable_search: true,
@@ -1696,6 +1702,13 @@ export function getOpenAIWebSearchParams(assistant: Assistant, model: Model): Re
           }
         }
       }
+
+      if (model.provider === 'openrouter') {
+        return {
+          plugins: [{ id: 'web' }]
+        }
+      }
+
       return {
         tools: webSearchTools
       }