from typing import Optional

import ai.agent
from database import BasicDatabase

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

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


def create_ai_agent(openai_url: Optional[str], openai_token: str, openai_model: str,
                    replicate_token: str, tavily_token: str,
                    db: BasicDatabase, platform: str):
    global agent_instance
    agent_instance = ai.agent.AiAgent(openai_url, openai_token, openai_model,
                                      replicate_token, tavily_token,
                                      db, platform)


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