RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-03 21:23:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修复 webui.py 中 llm_history_len 和 vector_search_top_k 显示值与启动设置默认值不一致的问题

Browse Source
This commit is contained in:
imClumsyPanda 2023-04-15 20:01:36 +08:00
parent e679136b25
commit 59375244b6
2 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
chains/local_doc_qa.py
View File

@ -11,14 +11,11 @@ import datetime
from typing import List
# return top-k text chunk from vector store
VECTOR_SEARCH_TOP_K = 10
VECTOR_SEARCH_TOP_K = 6
# LLM input history length
LLM_HISTORY_LEN = 3
# Show reply with source text from input document
REPLY_WITH_SOURCE = True
class LocalDocQA:
llm: object = None

17
webui.py
View File

@ -4,6 +4,11 @@ import shutil
from chains.local_doc_qa import LocalDocQA
from configs.model_config import *
# return top-k text chunk from vector store
VECTOR_SEARCH_TOP_K = 6
# LLM input history length
LLM_HISTORY_LEN = 3
def get_file_list():
if not os.path.exists("content"):
@ -124,7 +129,7 @@ with gr.Blocks(css=block_css) as demo:
interactive=True)
llm_history_len = gr.Slider(0,
10,
value=3,
value=LLM_HISTORY_LEN,
step=1,
label="LLM history len",
interactive=True)
@ -137,7 +142,7 @@ with gr.Blocks(css=block_css) as demo:
interactive=True)
top_k = gr.Slider(1,
20,
value=6,
value=VECTOR_SEARCH_TOP_K,
step=1,
label="向量匹配 top k",
interactive=True)
@ -173,5 +178,9 @@ with gr.Blocks(css=block_css) as demo:
[chatbot, query],
)
demo.queue(concurrency_count=3).launch(
server_name='0.0.0.0', share=False, inbrowser=False)
demo.queue(concurrency_count=3
).launch(server_name='0.0.0.0',
server_port=7860,
show_api=False,
share=False,
inbrowser=False)