22 lines
720 B
Python
22 lines
720 B
Python
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"]
|