From be0b566edb3e5d3294e0486b0d8251e1ec4855b2 Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Thu, 13 Mar 2025 07:11:32 -0400 Subject: [PATCH] fix ERROR - Error getting expert guidance for planning: module 'ra_aid.agent_utils' has no attribute 'process_thinking_content' --- ra_aid/agents/planning_agent.py | 3 ++- ra_aid/agents/research_agent.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/ra_aid/agents/planning_agent.py b/ra_aid/agents/planning_agent.py index 8b02a38..ec5736b 100644 --- a/ra_aid/agents/planning_agent.py +++ b/ra_aid/agents/planning_agent.py @@ -33,6 +33,7 @@ from ra_aid.logging_config import get_logger from ra_aid.model_formatters import format_key_facts_dict from ra_aid.model_formatters.key_snippets_formatter import format_key_snippets_dict from ra_aid.model_formatters.research_notes_formatter import format_research_notes_dict +from ra_aid.text.processing import process_thinking_content from ra_aid.models_params import models_params from ra_aid.project_info import format_project_info, get_project_info from ra_aid.prompts.expert_prompts import EXPERT_PROMPT_SECTION_PLANNING @@ -286,7 +287,7 @@ def run_planning_agent( content = "\n".join(str(item) for item in content) elif supports_think_tag or supports_thinking: # Process thinking content using the centralized function - content, _ = agent_utils.process_thinking_content( + content, _ = process_thinking_content( content=content, supports_think_tag=supports_think_tag, supports_thinking=supports_thinking, diff --git a/ra_aid/agents/research_agent.py b/ra_aid/agents/research_agent.py index 966321f..77608e5 100644 --- a/ra_aid/agents/research_agent.py +++ b/ra_aid/agents/research_agent.py @@ -34,6 +34,7 @@ from ra_aid.logging_config import get_logger from ra_aid.model_formatters import format_key_facts_dict from ra_aid.model_formatters.key_snippets_formatter import format_key_snippets_dict from ra_aid.model_formatters.research_notes_formatter import format_research_notes_dict +from ra_aid.text.processing import process_thinking_content from ra_aid.models_params import models_params from ra_aid.project_info import display_project_status, format_project_info, get_project_info from ra_aid.prompts.expert_prompts import EXPERT_PROMPT_SECTION_RESEARCH @@ -293,7 +294,7 @@ def run_research_agent( content = "\n".join(str(item) for item in content) elif supports_think_tag or supports_thinking: # Process thinking content using the centralized function - content, _ = agent_utils.process_thinking_content( + content, _ = process_thinking_content( content=content, supports_think_tag=supports_think_tag, supports_thinking=supports_thinking,