Compare commits
2 Commits
4a881ea32d
...
598f40ef74
| Author | SHA1 | Date | |
|---|---|---|---|
| 598f40ef74 | |||
| f3bcb01777 |
@@ -14,6 +14,7 @@ from ..model_services.chat_services import get_all_chat_services, LocalVLLMChatP
|
||||
from app.main_graph.utils.rag_initializer import init_rag_tool
|
||||
from app.core.intent_classifier import get_intent_classifier
|
||||
from app.logger import info, warning
|
||||
from app.main_graph.state import MainGraphState, CurrentAction
|
||||
|
||||
class AIAgentService:
|
||||
def __init__(self, checkpointer):
|
||||
@@ -120,8 +121,12 @@ class AIAgentService:
|
||||
"configurable": {"thread_id": thread_id},
|
||||
"metadata": {"user_id": user_id}
|
||||
}
|
||||
input_state = {"messages": [{"role": "user", "content": message}]}
|
||||
context = GraphContext(user_id=user_id)
|
||||
input_state = {
|
||||
"user_query": message,
|
||||
"messages": [{"role": "user", "content": message}],
|
||||
"user_id": user_id,
|
||||
"current_action": CurrentAction.NONE
|
||||
}
|
||||
|
||||
# ========== 新增:混合路由 ==========
|
||||
intent_result = await self.intent_classifier.classify(message)
|
||||
@@ -161,7 +166,6 @@ class AIAgentService:
|
||||
async for chunk in graph.astream(
|
||||
input_state,
|
||||
config=config,
|
||||
context=context,
|
||||
stream_mode=["messages", "updates", "custom"],
|
||||
version="v2",
|
||||
subgraphs=True
|
||||
|
||||
@@ -15,11 +15,12 @@ services:
|
||||
# =========================================================================
|
||||
# PostgreSQL 数据库配置
|
||||
# =========================================================================
|
||||
- DB_HOST=ai-postgres
|
||||
- DB_PORT=5432
|
||||
- DB_USER=postgres
|
||||
- DB_PASSWORD=${DB_PASSWORD:?请配置 DB_PASSWORD(本地:.env 文件 | CI/CD:Secrets)} # ⭐ 敏感密钥配置
|
||||
- DB_NAME=langgraph_db
|
||||
- DB_HOST=${DB_HOST}
|
||||
- DB_PORT=${DB_PORT}
|
||||
- DB_USER=${DB_USER}
|
||||
- DB_PASSWORD=${DB_PASSWORD}
|
||||
- DB_NAME=${DB_NAME}
|
||||
- DB_URI=${DB_URI}
|
||||
|
||||
# =========================================================================
|
||||
# Qdrant 向量数据库配置(URL + API密钥 配对)
|
||||
|
||||
Reference in New Issue
Block a user