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