diff --git a/api.py b/api.py
index be6ea52..822288f 100644
--- a/api.py
+++ b/api.py
@@ -252,11 +252,18 @@ async def no_knowledge_chat(
         ),
 ):
 
-    for resp, history in local_doc_qa._call(
+    for resp, history in local_doc_qa.llm._call(
             query=question, chat_history=history, streaming=True
     ):
         pass
 
+    return ChatMessage(
+        question=question,
+        response=resp["result"],
+        history=history,
+        source_documents=None,
+    )
+
 async def stream_chat(websocket: WebSocket, knowledge_base_id: str):
     await websocket.accept()
     vs_path = os.path.join(VS_ROOT_PATH, knowledge_base_id)