From 1e10287a03b00dde297792f33d1eeefaabaaf1c4 Mon Sep 17 00:00:00 2001 From: SmartDever02 Date: Tue, 9 Dec 2025 06:07:19 -0300 Subject: [PATCH] Fix ImportError about completion --- api/apps/sdk/websocket.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/apps/sdk/websocket.py b/api/apps/sdk/websocket.py index 1ba4c6b17..54466d2d8 100644 --- a/api/apps/sdk/websocket.py +++ b/api/apps/sdk/websocket.py @@ -26,7 +26,7 @@ from quart import websocket from api.db.services.dialog_service import DialogService 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.utils.api_utils import ws_token_required from common.constants import StatusEnum @@ -91,7 +91,7 @@ async def chat_completions_ws(tenant_id, chat_id): try: if stream: - for response_chunk in rag_completion( + async for response_chunk in rag_completion( tenant_id=tenant_id, chat_id=chat_id, 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}") else: response = None - for resp in rag_completion( + async for resp in rag_completion( tenant_id=tenant_id, chat_id=chat_id, question=question,