This commit is contained in:
@@ -10,7 +10,6 @@ import asyncio
|
||||
from ..main_graph.utils.main_graph_builder import build_react_main_graph
|
||||
from ..main_graph.tools.graph_tools import AVAILABLE_TOOLS, TOOLS_BY_NAME
|
||||
from ..main_graph.config import set_stream_writer
|
||||
from ..model_services.chat_services import get_all_chat_services, LocalVLLMChatProvider
|
||||
from ..main_graph.utils.rag_initializer import init_rag_tool
|
||||
from ..core.intent_classifier import get_intent_classifier
|
||||
from ..logger import info, warning, error
|
||||
@@ -33,18 +32,10 @@ class AIAgentService:
|
||||
async def initialize(self):
|
||||
# 0. 初始化 Mem0 客户端
|
||||
from ..memory.mem0_client import Mem0Client
|
||||
# 创建一个临时的 LLM 用于 Mem0(用第一个可用的)
|
||||
chat_services = get_all_chat_services()
|
||||
temp_llm = None
|
||||
if chat_services:
|
||||
temp_llm = list(chat_services.values())[0]
|
||||
self.mem0_client = Mem0Client(temp_llm)
|
||||
self.mem0_client = Mem0Client()
|
||||
|
||||
# 1. 初始化 RAG 工具(如果需要)
|
||||
def create_local_llm():
|
||||
provider = LocalVLLMChatProvider()
|
||||
return provider.get_service()
|
||||
rag_tool = await init_rag_tool(create_local_llm)
|
||||
rag_tool = await init_rag_tool()
|
||||
if rag_tool:
|
||||
self.tools.append(rag_tool)
|
||||
self.tools_by_name[rag_tool.name] = rag_tool
|
||||
|
||||
Reference in New Issue
Block a user