diff --git a/agent/agents/product.py b/agent/agents/product.py index beae1e9..44cfe57 100644 --- a/agent/agents/product.py +++ b/agent/agents/product.py @@ -165,12 +165,18 @@ async def product_agent(state: AgentState) -> AgentState: if action == "call_tool": arguments = result.get("arguments", {}) - + + # Inject context for SPU product search (Mall API) + if result["tool_name"] == "search_spu_products": + arguments["user_token"] = state.get("user_token") + arguments["user_id"] = state["user_id"] + arguments["account_id"] = state["account_id"] + # Inject context for recommendation if result["tool_name"] == "recommend_products": arguments["user_id"] = state["user_id"] arguments["account_id"] = state["account_id"] - + # Inject context for quote if result["tool_name"] == "get_quote": arguments["account_id"] = state["account_id"]