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

View File

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

View File

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