gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix ImportError about completion

Browse source
This commit is contained in:
SmartDever02 2025-12-09 06:07:19 -03:00
parent 5ee639fa5a
commit 1e10287a03
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
api/apps/sdk/websocket.py
View file

@ -26,7 +26,7 @@ from quart import websocket
from api.db.services.dialog_service import DialogService from api.db.services.dialog_service import DialogService
from api.db.services.canvas_service import UserCanvasService from api.db.services.canvas_service import UserCanvasService
from api.db.services.conversation_service import completion as rag_completion from api.db.services.conversation_service import async_completion as rag_completion
from api.db.services.canvas_service import completion as agent_completion from api.db.services.canvas_service import completion as agent_completion
from api.utils.api_utils import ws_token_required from api.utils.api_utils import ws_token_required
from common.constants import StatusEnum from common.constants import StatusEnum
@ -91,7 +91,7 @@ async def chat_completions_ws(tenant_id, chat_id):
try: try:
if stream: if stream:
for response_chunk in rag_completion( async for response_chunk in rag_completion(
tenant_id=tenant_id, tenant_id=tenant_id,
chat_id=chat_id, chat_id=chat_id,
question=question, question=question,
@ -110,7 +110,7 @@ async def chat_completions_ws(tenant_id, chat_id):
logging.info(f"Chat completion streamed successfully for chat_id: {chat_id}") logging.info(f"Chat completion streamed successfully for chat_id: {chat_id}")
else: else:
response = None response = None
for resp in rag_completion( async for resp in rag_completion(
tenant_id=tenant_id, tenant_id=tenant_id,
chat_id=chat_id, chat_id=chat_id,
question=question, question=question,