Langchain-Chatchat

1
from server.chat.knowledge_base_chat import knowledge_base_chat
2
from configs import VECTOR_SEARCH_TOP_K, SCORE_THRESHOLD, MAX_TOKENS
3
import json
4
import asyncio
5
from server.agent import model_container
6

7
async def search_knowledge_base_iter(database: str, query: str) -> str:
8
    response = await knowledge_base_chat(query=query,
9
                                         knowledge_base_name=database,
10
                                         model_name=model_container.MODEL.model_name,
11
                                         temperature=0.01,
12
                                         history=[],
13
                                         top_k=VECTOR_SEARCH_TOP_K,
14
                                         max_tokens=MAX_TOKENS,
15
                                         prompt_name="knowledge_base_chat",
16
                                         score_threshold=SCORE_THRESHOLD,
17
                                         stream=False)
18

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

26
def search_knowledgebase_simple(query: str):
27
    return asyncio.run(search_knowledge_base_iter(query))
28

29

30
if __name__ == "__main__":
31
    result = search_knowledgebase_simple("大数据男女比例")
32
    print("答案:",result)