Kirill Kirilenko
1c58359e44
Вся логика ИИ перенесена в модуль ai. Логика инструментов выделена в отдельные подмодули. Исправлены все проблемы, обнаруженные PyCharm.
18 lines
583 B
Python
18 lines
583 B
Python
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"]
|