diff --git a/agent/agents/product.py b/agent/agents/product.py index 44cfe57..5cc0201 100644 --- a/agent/agents/product.py +++ b/agent/agents/product.py @@ -19,13 +19,23 @@ PRODUCT_AGENT_PROMPT = """你是一个专业的 B2B 商品顾问助手。 - 库存查询 - 商品详情 +## ⚠️ 重要:商品搜索工具选择 + +**商品搜索必须优先使用 `search_spu_products` 工具!** + +- ✅ **search_spu_products**:使用 Mall API,支持用户认证,返回精美卡片展示(推荐) +- ⚠️ **search_products**:仅用于高级搜索(需要复杂过滤条件时) + +**普通商品搜索(如 "ring"、"手机"、"iPhone")必须使用 `search_spu_products`** + ## 可用工具 -1. **search_spu_products** - 搜索商品(使用 Mall API,推荐) +1. **search_spu_products** - 搜索商品(使用 Mall API,推荐)⭐ - keyword: 搜索关键词(商品名称、编号等) - page_size: 每页数量(默认 60,最大 100) - page: 页码(默认 1) - 说明:此工具使用 Mall API 搜索商品 SPU,支持用户 token 认证,返回卡片格式展示 + - **适用于所有普通商品搜索请求** 2. **search_products** - 搜索商品(使用 Hyperf API) - query: 搜索关键词 @@ -34,6 +44,7 @@ PRODUCT_AGENT_PROMPT = """你是一个专业的 B2B 商品顾问助手。 - page: 页码 - page_size: 每页数量 - 说明:此工具用于高级搜索,支持多维度过滤 + - **仅在需要复杂过滤条件时使用** 3. **get_product_detail** - 获取商品详情 - product_id: 商品ID @@ -64,6 +75,19 @@ PRODUCT_AGENT_PROMPT = """你是一个专业的 B2B 商品顾问助手。 } ``` +**示例**: +用户说:"搜索 ring" +返回: +```json +{ + "action": "call_tool", + "tool_name": "search_spu_products", + "arguments": { + "keyword": "ring" + } +} +``` + 当需要向用户询问更多信息时: ```json { @@ -152,15 +176,47 @@ async def product_agent(state: AgentState) -> AgentState: try: llm = get_llm_client() response = await llm.chat(messages, temperature=0.7) - + # Parse response content = response.content.strip() + + # Log raw LLM response for debugging + logger.info( + "Product agent LLM response", + response_length=len(content), + response_preview=content[:200], + conversation_id=state["conversation_id"] + ) + if content.startswith("```"): content = content.split("```")[1] if content.startswith("json"): content = content[4:] - - result = json.loads(content) + + # Handle non-JSON format: "tool_name\n{args}" + if '\n' in content and not content.startswith('{'): + lines = content.split('\n', 1) + tool_name = lines[0].strip() + args_json = lines[1].strip() if len(lines) > 1 else '{}' + + try: + arguments = json.loads(args_json) if args_json else {} + result = { + "action": "call_tool", + "tool_name": tool_name, + "arguments": arguments + } + except json.JSONDecodeError: + # If args parsing fails, use empty dict + result = { + "action": "call_tool", + "tool_name": tool_name, + "arguments": {} + } + else: + # Standard JSON format + result = json.loads(content) + action = result.get("action") if action == "call_tool":