From 90b3070aa2d721c775db554a53b4b23bbe3e6562 Mon Sep 17 00:00:00 2001 From: Ariel Frischer Date: Thu, 13 Feb 2025 20:14:07 -0800 Subject: [PATCH] chore(agent_utils.py): remove debug print statement for config to clean up output chore(fallback_handler.py): comment out cpm call for tool call result to reduce logging noise --- ra_aid/agent_utils.py | 1 - ra_aid/fallback_handler.py | 3 +-- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/ra_aid/agent_utils.py b/ra_aid/agent_utils.py index 5f35a9a..dd46fe8 100644 --- a/ra_aid/agent_utils.py +++ b/ra_aid/agent_utils.py @@ -270,7 +270,6 @@ def create_agent( """ try: config = _global_memory.get("config", {}) - print(f"config={config}") max_input_tokens = ( get_model_token_limit(config, agent_type) or DEFAULT_TOKEN_LIMIT ) diff --git a/ra_aid/fallback_handler.py b/ra_aid/fallback_handler.py index 63dc742..fdf45c6 100644 --- a/ra_aid/fallback_handler.py +++ b/ra_aid/fallback_handler.py @@ -274,11 +274,10 @@ class FallbackHandler: msg_list = self.construct_prompt_msg_list() response = retry_model.invoke(msg_list) - logger.debug(f"raw llm response={response}") tool_call = self.base_message_to_tool_call_dict(response) tool_call_result = self.invoke_prompt_tool_call(tool_call) - cpm(str(tool_call_result), title="Fallback Tool Call Result") + # cpm(str(tool_call_result), title="Fallback Tool Call Result") logger.debug( f"Fallback call successful with model: {self._format_model(fallback_model)}" )