修复 title 总结失败bug,新增enable_thinking参数

This commit is contained in:
zcr
2026-03-23 16:52:11 +08:00
parent d5ef985e52
commit 7c23d16ea6
5 changed files with 20 additions and 15 deletions

View File

@@ -9,7 +9,7 @@ from src.core.config import MONGO_URI
from src.server.deep_agent.agents.painter import build_painter_subagent
from src.server.deep_agent.agents.researcher import build_researcher_subagent
from src.server.deep_agent.agents.user_profile import user_profile_subagent
from src.server.deep_agent.init_llm import main_llm
from src.server.deep_agent.init_llm import build_main_llm
from src.server.deep_agent.init_prompt import build_system_prompt
client = MongoClient(MONGO_URI)
@@ -39,7 +39,7 @@ class CanvasMiddleware:
return state, agent_input
def build_main_agent(use_report, workspace_dir):
def build_main_agent(use_report, workspace_dir, enable_thinking):
research_subagent = build_researcher_subagent(workspace_dir)
painter_subagent = build_painter_subagent(workspace_dir)
subagents = [
@@ -48,7 +48,7 @@ def build_main_agent(use_report, workspace_dir):
user_profile_subagent
]
main_agent = create_deep_agent(
model=main_llm,
model=build_main_llm(enable_thinking=enable_thinking),
system_prompt=build_system_prompt(use_report=use_report),
subagents=subagents,
checkpointer=checkpointer,
@@ -58,7 +58,7 @@ def build_main_agent(use_report, workspace_dir):
),
middleware=[
SummarizationMiddleware(
model=main_llm,
model=build_main_llm(enable_thinking=enable_thinking),
trigger=("tokens", 3000),
keep=("messages", 100),
),

View File

@@ -22,13 +22,18 @@ title_llm = ChatQwen(
api_key=settings.QWEN_API_KEY
)
main_llm = ChatQwen(
model="qwen3.5-flash",
temperature=0.2,
max_tokens=3_000,
timeout=None,
max_retries=2,
api_key=settings.QWEN_API_KEY)
def build_main_llm(enable_thinking):
main_llm = ChatQwen(
enable_thinking=enable_thinking,
model="qwen3.5-flash",
temperature=0.2,
max_tokens=3_000,
timeout=None,
max_retries=2,
api_key=settings.QWEN_API_KEY)
return main_llm
suggested_llm = ChatQwen(
model="qwen-plus",

View File

@@ -22,8 +22,6 @@ async def conversation_title(agent, config):
if user_msg and ai_msg:
break
if not user_msg or not ai_msg:
return None
prompt = f"""
请根据以下首次对话内容生成一个简洁、精准的标题2-15个字