diff --git a/hindsight-api/hindsight_api/engine/reflect/agent.py b/hindsight-api/hindsight_api/engine/reflect/agent.py index 289f0b6b..750df75c 100644 --- a/hindsight-api/hindsight_api/engine/reflect/agent.py +++ b/hindsight-api/hindsight_api/engine/reflect/agent.py @@ -871,21 +871,21 @@ async def _execute_tool( query = args.get("query") if not query: return {"error": "search_mental_models requires a query parameter"} - max_results = args.get("max_results") or 5 + max_results = int(args.get("max_results") or 5) return await search_mental_models_fn(query, max_results) elif tool_name == "search_observations": query = args.get("query") if not query: return {"error": "search_observations requires a query parameter"} - max_tokens = max(args.get("max_tokens") or 5000, 1000) # Default 5000, min 1000 + max_tokens = max(int(args.get("max_tokens") or 5000), 1000) # Default 5000, min 1000 return await search_observations_fn(query, max_tokens) elif tool_name == "recall": query = args.get("query") if not query: return {"error": "recall requires a query parameter"} - max_tokens = max(args.get("max_tokens") or 2048, 1000) # Default 2048, min 1000 + max_tokens = max(int(args.get("max_tokens") or 2048), 1000) # Default 2048, min 1000 return await recall_fn(query, max_tokens) elif tool_name == "expand": @@ -904,18 +904,18 @@ def _summarize_input(tool_name: str, args: dict[str, Any]) -> str: if tool_name == "search_mental_models": query = args.get("query", "") query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{query}'" - max_results = args.get("max_results") or 5 + max_results = int(args.get("max_results") or 5) return f"(query={query_preview}, max_results={max_results})" elif tool_name == "search_observations": query = args.get("query", "") query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{query}'" - max_tokens = max(args.get("max_tokens") or 5000, 1000) + max_tokens = max(int(args.get("max_tokens") or 5000), 1000) return f"(query={query_preview}, max_tokens={max_tokens})" elif tool_name == "recall": query = args.get("query", "") query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{query}'" # Show actual value used (default 2048, min 1000) - max_tokens = max(args.get("max_tokens") or 2048, 1000) + max_tokens = max(int(args.get("max_tokens") or 2048), 1000) return f"(query={query_preview}, max_tokens={max_tokens})" elif tool_name == "expand": memory_ids = args.get("memory_ids", [])