From e75b92e112d193460cfdc51b635268416bc984fd Mon Sep 17 00:00:00 2001
From: zcr <cgzhou@aidlab.hk>
Date: Tue, 19 May 2026 16:27:46 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8A=A5=E5=91=8A=E6=A8=A1=E5=9E=8B=E5=8D=87?=
 =?UTF-8?q?=E7=BA=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/server/deep_agent/agents/researcher.py   |  2 ++
 src/server/deep_agent/init_llm.py            | 12 +++++++++++-
 src/server/deep_agent/tools/research_tool.py | 10 ++++++++--
 3 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/src/server/deep_agent/agents/researcher.py b/src/server/deep_agent/agents/researcher.py
index ccf00df..2d1f495 100755
--- a/src/server/deep_agent/agents/researcher.py
+++ b/src/server/deep_agent/agents/researcher.py
@@ -1,5 +1,6 @@
 from langchain.agents.middleware import dynamic_prompt, ModelRequest
 
+from src.server.deep_agent.init_llm import latest_llm, qwen_plus_llm
 from src.server.deep_agent.init_prompt import build_researcher_prompt
 from src.server.deep_agent.tools.crawl_tool import create_crawl4ai_batch_tool
 from src.server.deep_agent.tools.report_generator_tool import create_report_generator_tool
@@ -79,6 +80,7 @@ Do NOT use this sub-agent for:
 - User profile collection 
 - Image generation or editing tasks
         """,
+        "model": qwen_plus_llm,
         "system_prompt": build_researcher_prompt(),
         "middleware": [language_control],
         "tools": [
diff --git a/src/server/deep_agent/init_llm.py b/src/server/deep_agent/init_llm.py
index b56999e..48f9b28 100755
--- a/src/server/deep_agent/init_llm.py
+++ b/src/server/deep_agent/init_llm.py
@@ -3,7 +3,7 @@ from langchain_qwq import ChatQwen
 from src.core.config import settings
 
 llm = ChatQwen(
-    model="qwen3.5-flash",
+    model="qwen3.6-plus",
     max_tokens=3_000,
     timeout=None,
     max_retries=2,
@@ -21,6 +21,16 @@ qwen_plus_llm = ChatQwen(
     top_p=0.8,
     api_key=settings.QWEN_API_KEY
 )
+latest_llm = ChatQwen(
+    model="qwen3.6-plus",
+    max_tokens=3_000,
+    timeout=None,
+    max_retries=2,
+    streaming=False,
+    temperature=0.25,
+    top_p=0.8,
+    api_key=settings.QWEN_API_KEY
+)
 
 
 def build_main_llm(enable_thinking):
diff --git a/src/server/deep_agent/tools/research_tool.py b/src/server/deep_agent/tools/research_tool.py
index 46f70aa..5dacf71 100755
--- a/src/server/deep_agent/tools/research_tool.py
+++ b/src/server/deep_agent/tools/research_tool.py
@@ -74,9 +74,16 @@ from typing import List, Set
 
 from ddgs import DDGS
 from langchain.tools import tool
+from pydantic import BaseModel, Field
 
 
-@tool
+class TopicResearchInput(BaseModel):
+    """Input for topic research tool."""
+    topic: List[str] = Field(description="List of separate research keyword strings. Example: ['modern sofa design', 'sustainable wood furniture']")
+    max_urls: int = Field(default=5, description="Maximum number of URLs to return")
+
+
+@tool(args_schema=TopicResearchInput)
 async def topic_research(topic: List[str], max_urls: int = 5) -> str:
     """
     In-depth research tool (DuckDuckGo version).
@@ -127,7 +134,6 @@ async def topic_research(topic: List[str], max_urls: int = 5) -> str:
     print(f" topic research !!!!!!!!!!!!!!!!!!!!! {selected_urls}")
     return json.dumps(selected_urls, ensure_ascii=False)
 
-
 # import asyncio
 #
 #