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

add api request methods for all chat apis

This commit is contained in:
liunux4odoo 2023-08-03 09:33:24 +08:00
parent d0fd6253a3
commit d36d801a58
1 changed files with 60 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
webui_pages/utils.py
View File

@ -231,6 +231,66 @@ class ApiRequest:
response = self.post("/chat/chat", json=f"{query}", stream=True)
return response
def knowledge_base_chat(
self,
query: str,
knowledge_base_name: str,
no_remote_api: bool = False,
):
'''
对应/chat/knowledge_base_chat接口
'''
if no_remote_api:
from server.chat.knowledge_base_chat import knowledge_base_chat
response = knowledge_base_chat(query, knowledge_base_name)
return self._stream2generator(response)
else:
response = self.post(
"/chat/knowledge_base_chat",
json={"query": query, "knowledge_base_name": knowledge_base_name},
stream=True,
)
return response
def duckduckgo_search_chat(
self,
query: str,
no_remote_api: bool = False,
):
'''
对应api.py/chat/duckduckgo_search_chat接口
'''
if no_remote_api:
from server.chat.duckduckgo_search_chat import duckduckgo_search_chat
response = duckduckgo_search_chat(query)
return self._stream2generator(response)
else:
response = self.post(
"/chat/duckduckgo_search_chat",
json=f"{query}",
stream=True,
)
return response
def bing_search_chat(
self,
query: str,
no_remote_api: bool = False,
):
'''
对应api.py/chat/bing_search_chat接口
'''
if no_remote_api:
from server.chat.bing_search_chat import bing_search_chat
response = bing_search_chat(query)
return self._stream2generator(response)
else:
response = self.post(
"/chat/bing_search_chat",
json=f"{query}",
stream=True,
)
return response
if __name__ == "__main__":
api = ApiRequest()