20 lines
668 B
Python
20 lines
668 B
Python
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"]
|