Langchain-Chatchat

1
import json
2
from server.chat.search_engine_chat import search_engine_chat
3
from configs import VECTOR_SEARCH_TOP_K, MAX_TOKENS
4
import asyncio
5
from server.agent import model_container
6
from pydantic import BaseModel, Field
7

8
async def search_engine_iter(query: str):
9
    response = await search_engine_chat(query=query,
10
                                         search_engine_name="bing", # 这里切换搜索引擎
11
                                         model_name=model_container.MODEL.model_name,
12
                                         temperature=0.01, # Agent 搜索互联网的时候，温度设置为0.01
13
                                         history=[],
14
                                         top_k = VECTOR_SEARCH_TOP_K,
15
                                         max_tokens= MAX_TOKENS,
16
                                         prompt_name = "default",
17
                                         stream=False)
18

19
    contents = ""
20

21
    async for data in response.body_iterator: # 这里的data是一个json字符串
22
        data = json.loads(data)
23
        contents = data["answer"]
24
        docs = data["docs"]
25

26
    return contents
27

28
def search_internet(query: str):
29
    return asyncio.run(search_engine_iter(query))
30

31
class SearchInternetInput(BaseModel):
32
    location: str = Field(description="Query for Internet search")
33

34

35
if __name__ == "__main__":
36
    result = search_internet("今天星期几")
37
    print("答案:",result)
38