import ai.agent
from database import BasicDatabase

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

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


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


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