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

update server.chat.*: set default value [] to history parameter.

This commit is contained in:
liunux4odoo 2023-08-09 10:48:37 +08:00
parent b98f5fd0b9
commit 9883341924
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server/chat/chat.py
View File

@ -13,15 +13,15 @@ from server.chat.utils import History
def chat(query: str = Body(..., description="用户输入", example="恼羞成怒"), def chat(query: str = Body(..., description="用户输入", example="恼羞成怒"),
history: Optional[List[History]] = Body(..., history: List[History] = Body([],
description="历史对话", description="历史对话",
example=[ example=[
{"role": "user", "content": "我们来玩成语接龙，我先来，生龙活虎"}, {"role": "user", "content": "我们来玩成语接龙，我先来，生龙活虎"},
{"role": "assistant", "content": "虎头虎脑"}] {"role": "assistant", "content": "虎头虎脑"}]
), ),
): ):
async def chat_iterator(query: str, async def chat_iterator(query: str,
history: Optional[List[History]] history: List[History] = [],
) -> AsyncIterable[str]: ) -> AsyncIterable[str]:
callback = AsyncIteratorCallbackHandler() callback = AsyncIteratorCallbackHandler()

16
server/chat/knowledge_base_chat.py
View File

@ -19,14 +19,14 @@ import json
def knowledge_base_chat(query: str = Body(..., description="用户输入", example="你好"), def knowledge_base_chat(query: str = Body(..., description="用户输入", example="你好"),
knowledge_base_name: str = Body(..., description="知识库名称", example="samples"), knowledge_base_name: str = Body(..., description="知识库名称", example="samples"),
top_k: int = Body(VECTOR_SEARCH_TOP_K, description="匹配向量数"), top_k: int = Body(VECTOR_SEARCH_TOP_K, description="匹配向量数"),
history: Optional[List[History]] = Body(..., history: List[History] = Body([],
description="历史对话", description="历史对话",
example=[ example=[
{"role": "user", {"role": "user",
"content": "我们来玩成语接龙，我先来，生龙活虎"}, "content": "我们来玩成语接龙，我先来，生龙活虎"},
{"role": "assistant", {"role": "assistant",
"content": "虎头虎脑"}] "content": "虎头虎脑"}]
), ),
): ):
kb = KBServiceFactory.get_service_by_name(knowledge_base_name) kb = KBServiceFactory.get_service_by_name(knowledge_base_name)
if kb is None: if kb is None: