holetron-lab/fleet-memory
1
0
Fork 0
Code Releases Packages Activity Actions

fixed cast error (#300)

Browse source 
Signed-off-by: hayden.rear <hayden.rear@gmail.com>
This commit is contained in:
Hayden Rear 2026-02-05 03:05:26 -05:00 committed by GitHub
parent 85b9074f43
commit 093ecff48d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
hindsight-api/hindsight_api/engine/reflect/agent.py
View file

@ -871,21 +871,21 @@ async def _execute_tool(
query = args.get("query") query = args.get("query")
if not query: if not query:
return {"error": "search_mental_models requires a query parameter"} 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) return await search_mental_models_fn(query, max_results)
elif tool_name == "search_observations": elif tool_name == "search_observations":
query = args.get("query") query = args.get("query")
if not query: if not query:
return {"error": "search_observations requires a query parameter"} 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) return await search_observations_fn(query, max_tokens)
elif tool_name == "recall": elif tool_name == "recall":
query = args.get("query") query = args.get("query")
if not query: if not query:
return {"error": "recall requires a query parameter"} 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) return await recall_fn(query, max_tokens)
elif tool_name == "expand": 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": if tool_name == "search_mental_models":
query = args.get("query", "") query = args.get("query", "")
query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{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})" return f"(query={query_preview}, max_results={max_results})"
elif tool_name == "search_observations": elif tool_name == "search_observations":
query = args.get("query", "") query = args.get("query", "")
query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{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})" return f"(query={query_preview}, max_tokens={max_tokens})"
elif tool_name == "recall": elif tool_name == "recall":
query = args.get("query", "") query = args.get("query", "")
query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{query}'" query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{query}'"
# Show actual value used (default 2048, min 1000) # 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})" return f"(query={query_preview}, max_tokens={max_tokens})"
elif tool_name == "expand": elif tool_name == "expand":
memory_ids = args.get("memory_ids", []) memory_ids = args.get("memory_ids", [])