cleanup

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,

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"]

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

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:

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

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()