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,