weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update api.py

This commit is contained in:
imClumsyPanda 2023-05-12 18:13:35 +08:00
parent f173d8b6b1
commit 8635f63c31
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api.py
View File

@ -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 query=question, chat_history=history, streaming=True
): ):
pass pass
return ChatMessage(
question=question,
response=resp["result"],
history=history,
source_documents=None,
)
async def stream_chat(websocket: WebSocket, knowledge_base_id: str): async def stream_chat(websocket: WebSocket, knowledge_base_id: str):
await websocket.accept() await websocket.accept()
vs_path = os.path.join(VS_ROOT_PATH, knowledge_base_id) vs_path = os.path.join(VS_ROOT_PATH, knowledge_base_id)