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)}" )