提交 8635f63c 作者: imClumsyPanda

update api.py

上级 f173d8b6
...@@ -252,11 +252,18 @@ async def no_knowledge_chat( ...@@ -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)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论