Skip to content

Commit 093ecff

Browse files
authored
fixed cast error (#300)
Signed-off-by: hayden.rear <hayden.rear@gmail.com>
1 parent 85b9074 commit 093ecff

File tree

1 file changed

+6
-6
lines changed
  • hindsight-api/hindsight_api/engine/reflect

1 file changed

+6
-6
lines changed

hindsight-api/hindsight_api/engine/reflect/agent.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -871,21 +871,21 @@ async def _execute_tool(
871871
query = args.get("query")
872872
if not query:
873873
return {"error": "search_mental_models requires a query parameter"}
874-
max_results = args.get("max_results") or 5
874+
max_results = int(args.get("max_results") or 5)
875875
return await search_mental_models_fn(query, max_results)
876876

877877
elif tool_name == "search_observations":
878878
query = args.get("query")
879879
if not query:
880880
return {"error": "search_observations requires a query parameter"}
881-
max_tokens = max(args.get("max_tokens") or 5000, 1000) # Default 5000, min 1000
881+
max_tokens = max(int(args.get("max_tokens") or 5000), 1000) # Default 5000, min 1000
882882
return await search_observations_fn(query, max_tokens)
883883

884884
elif tool_name == "recall":
885885
query = args.get("query")
886886
if not query:
887887
return {"error": "recall requires a query parameter"}
888-
max_tokens = max(args.get("max_tokens") or 2048, 1000) # Default 2048, min 1000
888+
max_tokens = max(int(args.get("max_tokens") or 2048), 1000) # Default 2048, min 1000
889889
return await recall_fn(query, max_tokens)
890890

891891
elif tool_name == "expand":
@@ -904,18 +904,18 @@ def _summarize_input(tool_name: str, args: dict[str, Any]) -> str:
904904
if tool_name == "search_mental_models":
905905
query = args.get("query", "")
906906
query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{query}'"
907-
max_results = args.get("max_results") or 5
907+
max_results = int(args.get("max_results") or 5)
908908
return f"(query={query_preview}, max_results={max_results})"
909909
elif tool_name == "search_observations":
910910
query = args.get("query", "")
911911
query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{query}'"
912-
max_tokens = max(args.get("max_tokens") or 5000, 1000)
912+
max_tokens = max(int(args.get("max_tokens") or 5000), 1000)
913913
return f"(query={query_preview}, max_tokens={max_tokens})"
914914
elif tool_name == "recall":
915915
query = args.get("query", "")
916916
query_preview = f"'{query[:30]}...'" if len(query) > 30 else f"'{query}'"
917917
# Show actual value used (default 2048, min 1000)
918-
max_tokens = max(args.get("max_tokens") or 2048, 1000)
918+
max_tokens = max(int(args.get("max_tokens") or 2048), 1000)
919919
return f"(query={query_preview}, max_tokens={max_tokens})"
920920
elif tool_name == "expand":
921921
memory_ids = args.get("memory_ids", [])

0 commit comments

Comments
 (0)