@@ -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