mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-27 01:03:35 +08:00
dea468ab0f
* 支持了agentlm * 支持了agentlm和相关提示词 * 修改了Agent的一些功能，加入了Embed方面的一个优化 * 修改了部分Agent的工具 * 增加一些Langchain的自带工具 * 修复一些兼容性的bug * 恢复知识库 * 恢复知识库 * 1 * 修复Azure问题 * 修复zilliz的问题 * 修复Agent历史记录问题，优化知识库问答搜索的uoji * 修复讯飞大模型token问题 * 修复讯飞大模型token问题和配置文件 --------- Co-authored-by: zR <zRzRzRzRzRzRzR>
35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
import json
|
||
from server.chat.search_engine_chat import search_engine_chat
|
||
from configs import VECTOR_SEARCH_TOP_K, MAX_TOKENS
|
||
import asyncio
|
||
from server.agent import model_container
|
||
|
||
async def search_engine_iter(query: str):
|
||
response = await search_engine_chat(query=query,
|
||
search_engine_name="bing", # 这里切换搜索引擎
|
||
model_name=model_container.MODEL.model_name,
|
||
temperature=0.01, # Agent 搜索互联网的时候，温度设置为0.01
|
||
history=[],
|
||
top_k = VECTOR_SEARCH_TOP_K,
|
||
max_tokens= MAX_TOKENS,
|
||
prompt_name = "default",
|
||
stream=False)
|
||
|
||
contents = ""
|
||
|
||
async for data in response.body_iterator: # 这里的data是一个json字符串
|
||
data = json.loads(data)
|
||
contents = data["answer"]
|
||
docs = data["docs"]
|
||
|
||
return contents
|
||
|
||
def search_internet(query: str):
|
||
|
||
return asyncio.run(search_engine_iter(query))
|
||
|
||
|
||
if __name__ == "__main__":
|
||
result = search_internet("今天星期几")
|
||
print("答案:",result)
|