mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-15 05:04:27 +08:00
Fix agentic chat not working in chatbot (#18035)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user