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

fix download original issue when chating

This commit is contained in:
wvivi2023 2024-03-28 10:55:52 +08:00
parent 0814d5e4ea
commit df219c6ae6
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
configs/model_config.py
View File

@ -1,5 +1,6 @@
import os import os
DOWNLOAD_BASE_URL = "http://10.138.20.199:17861"
# 可以指定一个绝对路径，统一存放所有的Embedding和LLM模型。 # 可以指定一个绝对路径，统一存放所有的Embedding和LLM模型。
# 每个模型可以是一个单独的目录，也可以是某个目录下的二级子目录。 # 每个模型可以是一个单独的目录，也可以是某个目录下的二级子目录。
# 如果模型目录名称和 MODEL_PATH 中的 key 或 value 相同，程序会自动检测加载，无需修改 MODEL_PATH 中的路径。 # 如果模型目录名称和 MODEL_PATH 中的 key 或 value 相同，程序会自动检测加载，无需修改 MODEL_PATH 中的路径。

10
server/chat/knowledge_base_chat.py
View File

@ -10,7 +10,8 @@ from configs import (LLM_MODELS,
USE_RERANKER, USE_RERANKER,
RERANKER_MODEL, RERANKER_MODEL,
RERANKER_MAX_LENGTH, RERANKER_MAX_LENGTH,
MODEL_PATH) MODEL_PATH,
DOWNLOAD_BASE_URL)
from server.utils import wrap_done, get_ChatOpenAI from server.utils import wrap_done, get_ChatOpenAI
from server.utils import BaseResponse, get_prompt_template from server.utils import BaseResponse, get_prompt_template
from langchain.chains import LLMChain from langchain.chains import LLMChain
@ -124,8 +125,11 @@ async def knowledge_base_chat(query: str = Body(..., description="用户输入",
for inum, doc in enumerate(docs): for inum, doc in enumerate(docs):
filename = doc.metadata.get("source") filename = doc.metadata.get("source")
parameters = urlencode({"knowledge_base_name": knowledge_base_name, "file_name": filename}) parameters = urlencode({"knowledge_base_name": knowledge_base_name, "file_name": filename})
base_url = request.base_url #base_url = request.base_url
url = f"{base_url}knowledge_base/download_doc?" + parameters #base_url = "http://10.138.20.199:17861/"
base_url = DOWNLOAD_BASE_URL
print(f"base_url:{base_url}")
url = f"{base_url}/knowledge_base/download_doc?" + parameters
text = f"""出处 [{inum + 1}] [{filename}]({url}) \n\n{doc.page_content}\n\n""" text = f"""出处 [{inum + 1}] [{filename}]({url}) \n\n{doc.page_content}\n\n"""
source_documents.append(text) source_documents.append(text)