mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-26 00:33:35 +08:00
d0846f88cc
- 重构 api.py：
- 按模块划分为不同的 router
- 添加 openai 兼容的转发接口，项目默认使用该接口以实现模型负载均衡
- 添加 /tools 接口，可以获取/调用编写的 agent tools
- 移除所有 EmbeddingFuncAdapter，统一改用 get_Embeddings
- 待办：
- /chat/chat 接口改为 openai 兼容
- 添加 /chat/kb_chat 接口，openai 兼容
- 改变 ntlk/knowledge_base/logs 等数据目录位置
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from urllib.parse import urlencode
|
|
from server.pydantic_types import BaseModel, Field
|
|
|
|
from server.knowledge_base.kb_doc_api import search_docs
|
|
from configs import TOOL_CONFIG
|
|
|
|
|
|
def search_knowledgebase(query: str, database: str, config: dict):
|
|
docs = search_docs(
|
|
query=query,
|
|
knowledge_base_name=database,
|
|
top_k=config["top_k"],
|
|
score_threshold=config["score_threshold"])
|
|
context = ""
|
|
source_documents = []
|
|
for inum, doc in enumerate(docs):
|
|
filename = doc.metadata.get("source")
|
|
parameters = urlencode({"knowledge_base_name": database, "file_name": filename})
|
|
url = f"download_doc?" + parameters
|
|
text = f"""出处 [{inum + 1}] [{filename}]({url}) \n\n{doc.page_content}\n\n"""
|
|
source_documents.append(text)
|
|
|
|
if len(source_documents) == 0:
|
|
context = "没有找到相关文档,请更换关键词重试"
|
|
else:
|
|
for doc in source_documents:
|
|
context += doc + "\n"
|
|
|
|
return context
|
|
|
|
|
|
class SearchKnowledgeInput(BaseModel):
|
|
database: str = Field(description="Database for Knowledge Search")
|
|
query: str = Field(description="Query for Knowledge Search")
|
|
|
|
|
|
def search_local_knowledgebase(database: str, query: str):
|
|
tool_config = TOOL_CONFIG["search_local_knowledgebase"]
|
|
return search_knowledgebase(query=query, database=database, config=tool_config)
|