src/gaia/agent/__init__.py

"""
Agent module for GAIA implementation.
Contains the core agent architecture and configuration.
"""

# Import directly from the new structure
from src.gaia.agent.agent import GAIAAgent
from src.gaia.agent.config import (
    get_logging_config,
    get_model_config,
    get_tool_config,
    get_memory_config,
    get_agent_config
)

# For backward compatibility with tests
Agent = GAIAAgent

__all__ = [
    'GAIAAgent',
    'Agent',  # Added for backward compatibility
    'get_logging_config',
    'get_model_config',
    'get_tool_config',
    'get_memory_config',
    'get_agent_config'
]