Fix agentic chat not working in chatbot (#18035)

This commit is contained in:
Wang Qi
2026-08-10 13:07:15 +08:00
committed by GitHub
parent b5bffa0fa3
commit 373050c7ca

View File

@@ -22,7 +22,7 @@ from common.constants import StatusEnum
from api.db.db_models import Conversation, DB
from api.db.services.api_service import API4ConversationService
from api.db.services.common_service import CommonService
from api.db.services.dialog_service import DialogService, async_chat
from api.db.services.dialog_service import DialogService, async_chat, rag_agent
from common.misc_utils import get_uuid
import json
@@ -361,7 +361,7 @@ async def async_iframe_completion(dialog_id, question, session_id=None, stream=T
if stream:
try:
async for ans in async_chat(dia, msg, True, session_id=session_id, **kwargs):
async for ans in rag_agent(dia, msg, True, session_id=session_id, **kwargs):
ans = structure_answer(conv, ans, message_id, session_id)
yield "data:" + json.dumps({"code": 0, "message": "", "data": ans}, ensure_ascii=False) + "\n\n"
API4ConversationService.append_message(conv.id, conv.to_dict())
@@ -371,7 +371,7 @@ async def async_iframe_completion(dialog_id, question, session_id=None, stream=T
else:
answer = None
async for ans in async_chat(dia, msg, False, session_id=session_id, **kwargs):
async for ans in rag_agent(dia, msg, False, session_id=session_id, **kwargs):
answer = structure_answer(conv, ans, message_id, session_id)
API4ConversationService.append_message(conv.id, conv.to_dict())
break