From 8aa1f5fd88a7d15f9bf1ce9130a8d55e5835e28f Mon Sep 17 00:00:00 2001 From: hubian <908234780@qq.com> Date: Mon, 13 Apr 2026 17:16:37 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=90=9C=E7=B4=A2=E7=BB=93=E6=9E=9C?= =?UTF-8?q?=E6=95=B0=E9=87=8F=E6=A0=B9=E6=8D=AE=E5=B7=A5=E5=85=B7=E9=85=8D?= =?UTF-8?q?=E7=BD=AE=E8=AF=BB=E5=8F=96=EF=BC=8C=E4=B8=8D=E5=86=8D=E5=9B=BA?= =?UTF-8?q?=E5=AE=9A5=E6=9D=A1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main_v2.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/main_v2.py b/main_v2.py index 48ed287..994e84c 100644 --- a/main_v2.py +++ b/main_v2.py @@ -839,19 +839,21 @@ async def websocket_endpoint(websocket: WebSocket, user_id: str): if search_result.get("results"): # 构建搜索上下文（给LLM） + max_for_llm = config.get('max_results', 5) search_context = "\n\n【搜索结果】\n" - for i, r in enumerate(search_result["results"][:5], 1): + for i, r in enumerate(search_result["results"][:max_for_llm], 1): search_context += f"{i}. {r.get('title', 'N/A')}\n {r.get('content', r.get('snippet', 'N/A'))[:200]}\n 来源: {r.get('url', 'N/A')}\n" - logger.info(f"搜索完成: {len(search_result['results'])} 条结果") + logger.info(f"搜索完成: {len(search_result['results'])} 条结果，使用 {min(len(search_result['results']), max_for_llm)} 条") - # 发送搜索结果给前端 + # 发送搜索结果给前端（按配置的数量） + max_display = config.get('max_results', 5) search_results_for_client = [ { "title": r.get('title', 'N/A'), "snippet": r.get('content', r.get('snippet', ''))[:150], "url": r.get('url', 'N/A') } - for r in search_result["results"][:5] + for r in search_result["results"][:max_display] ] await websocket.send_json({ "type": "search_results",