import ai.agent
from database import BasicDatabase

Message = ai.agent.Message
Agent = ai.agent.AiAgent

# Глобальный экземпляр агента
agent_instance: ai.agent.AiAgent


def create_ai_agent(openrouter_token: str, openrouter_model: str,
                    replicate_token: str, tavily_token: str,
                    db: BasicDatabase, platform: str):
    global agent_instance
    agent_instance = ai.agent.AiAgent(openrouter_token, openrouter_model,
                                      replicate_token, tavily_token,
                                      db, platform)


__all__ = ["agent_instance", "Agent", "Message", "create_ai_agent"]