zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
This commit is contained in:
Ariel Frischer 2025-02-13 20:14:07 -08:00
parent efec91579a
commit 90b3070aa2
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ra_aid/agent_utils.py
View File

@ -270,7 +270,6 @@ def create_agent(
""" """
try: try:
config = _global_memory.get("config", {}) config = _global_memory.get("config", {})
print(f"config={config}")
max_input_tokens = ( max_input_tokens = (
get_model_token_limit(config, agent_type) or DEFAULT_TOKEN_LIMIT get_model_token_limit(config, agent_type) or DEFAULT_TOKEN_LIMIT
) )

3
ra_aid/fallback_handler.py
View File

@ -274,11 +274,10 @@ class FallbackHandler:
msg_list = self.construct_prompt_msg_list() msg_list = self.construct_prompt_msg_list()
response = retry_model.invoke(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 = self.base_message_to_tool_call_dict(response)
tool_call_result = self.invoke_prompt_tool_call(tool_call) 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( logger.debug(
f"Fallback call successful with model: {self._format_model(fallback_model)}" f"Fallback call successful with model: {self._format_model(fallback_model)}"
) )