diff --git a/ra_aid/agent_utils.py b/ra_aid/agent_utils.py index 2d2b6e0..35d5f07 100644 --- a/ra_aid/agent_utils.py +++ b/ra_aid/agent_utils.py @@ -85,7 +85,7 @@ from ra_aid.tool_configs import ( ) from ra_aid.tools.handle_user_defined_test_cmd_execution import execute_test_command from ra_aid.database.repositories.key_fact_repository import KeyFactRepository -from ra_aid.text.key_facts_formatter import format_key_facts_dict +from ra_aid.model_formatters import format_key_facts_dict from ra_aid.tools.memory import ( _global_memory, get_memory_value, diff --git a/ra_aid/model_formatters/__init__.py b/ra_aid/model_formatters/__init__.py new file mode 100644 index 0000000..e3e1572 --- /dev/null +++ b/ra_aid/model_formatters/__init__.py @@ -0,0 +1,10 @@ +""" +This module provides formatting functions for model data for display or output. + +It includes functions to format key facts in a consistent, readable way for +presentation to users and other parts of the system. +""" + +from ra_aid.model_formatters.key_facts_formatter import format_key_fact, format_key_facts_dict + +__all__ = ["format_key_fact", "format_key_facts_dict"] \ No newline at end of file diff --git a/ra_aid/text/key_facts_formatter.py b/ra_aid/model_formatters/key_facts_formatter.py similarity index 86% rename from ra_aid/text/key_facts_formatter.py rename to ra_aid/model_formatters/key_facts_formatter.py index f30caca..5411c08 100644 --- a/ra_aid/text/key_facts_formatter.py +++ b/ra_aid/model_formatters/key_facts_formatter.py @@ -1,8 +1,8 @@ """ -Key facts text formatting module. +Key facts model formatter module. -This module provides utility functions for formatting key facts -with consistent markdown styling. +This module provides utility functions for formatting key facts from database models +into consistent markdown styling for display or output purposes. """ from typing import Dict, Optional diff --git a/ra_aid/tests/ra_aid/model_formatters/__init__.py b/ra_aid/tests/ra_aid/model_formatters/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/ra_aid/tests/ra_aid/text/test_key_facts_formatter.py b/ra_aid/tests/ra_aid/model_formatters/test_key_facts_formatter.py similarity index 94% rename from ra_aid/tests/ra_aid/text/test_key_facts_formatter.py rename to ra_aid/tests/ra_aid/model_formatters/test_key_facts_formatter.py index 49eed41..c74ed68 100644 --- a/ra_aid/tests/ra_aid/text/test_key_facts_formatter.py +++ b/ra_aid/tests/ra_aid/model_formatters/test_key_facts_formatter.py @@ -1,6 +1,6 @@ """Unit tests for the key_facts_formatter module.""" -from ra_aid.text.key_facts_formatter import format_key_fact, format_key_facts_dict +from ra_aid.model_formatters import format_key_fact, format_key_facts_dict class TestKeyFactsFormatter: diff --git a/ra_aid/tools/agent.py b/ra_aid/tools/agent.py index d5bd47d..05fe1dc 100644 --- a/ra_aid/tools/agent.py +++ b/ra_aid/tools/agent.py @@ -14,7 +14,7 @@ from ra_aid.agent_context import ( from ra_aid.console.formatting import print_error from ra_aid.database.repositories.key_fact_repository import KeyFactRepository from ra_aid.exceptions import AgentInterrupt -from ra_aid.text.key_facts_formatter import format_key_facts_dict +from ra_aid.model_formatters import format_key_facts_dict from ra_aid.tools.memory import _global_memory from ..console import print_task_header diff --git a/ra_aid/tools/expert.py b/ra_aid/tools/expert.py index 420cb04..1c2ed8e 100644 --- a/ra_aid/tools/expert.py +++ b/ra_aid/tools/expert.py @@ -8,7 +8,7 @@ from rich.panel import Panel from ..database.repositories.key_fact_repository import KeyFactRepository from ..llm import initialize_expert_llm -from ..text.key_facts_formatter import format_key_facts_dict +from ..model_formatters import format_key_facts_dict from .memory import _global_memory, get_memory_value console = Console()