Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
J
jinchat-server
概览
概览
详情
活动
周期分析
版本库
存储库
文件
提交
分支
标签
贡献者
分支图
比较
统计图
问题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
aigc-pioneer
jinchat-server
Commits
875dae64
Unverified
提交
875dae64
authored
5月 30, 2023
作者:
zhoutongqing
提交者:
GitHub
5月 30, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
zhoutongqing:添加重新构建知识库功能 (#418)
Co-authored-by: imClumsyPanda <littlepanda0716@gmail.com>
上级
f7044f6f
显示空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
19 行增加
和
1 行删除
+19
-1
webui.py
webui.py
+19
-1
没有找到文件。
webui.py
浏览文件 @
875dae64
...
...
@@ -230,6 +230,22 @@ def add_vs_name(vs_name, chatbot):
return
gr
.
update
(
visible
=
True
,
choices
=
get_vs_list
(),
value
=
vs_name
),
gr
.
update
(
visible
=
False
),
gr
.
update
(
visible
=
False
),
gr
.
update
(
visible
=
True
),
chatbot
# 自动化加载固定文件间中文件
def
init_set_vector_store
(
content_dir
,
vs_id
,
history
):
try
:
shutil
.
rmtree
(
VS_ROOT_PATH
)
vs_path
=
os
.
path
.
join
(
VS_ROOT_PATH
,
vs_id
)
sentence_size
=
gr
.
Number
(
value
=
SENTENCE_SIZE
,
precision
=
0
,
label
=
"文本入库分句长度限制"
,
interactive
=
True
,
visible
=
True
)
vs_path
,
loaded_files
=
local_doc_qa
.
init_knowledge_vector_store
(
content_dir
,
vs_path
,
sentence_size
)
model_status
=
"""知识库构建成功"""
except
Exception
as
e
:
logger
.
error
(
e
)
model_status
=
"""知识库构建未成功"""
logger
.
info
(
model_status
)
return
history
+
[[
None
,
model_status
]]
def
refresh_vs_list
():
return
gr
.
update
(
choices
=
get_vs_list
())
...
...
@@ -470,7 +486,9 @@ with gr.Blocks(css=block_css, theme=gr.themes.Default(**default_theme_args)) as
load_model_button
.
click
(
reinit_model
,
show_progress
=
True
,
inputs
=
[
llm_model
,
embedding_model
,
llm_history_len
,
no_remote_model
,
use_ptuning_v2
,
use_lora
,
top_k
,
chatbot
],
outputs
=
chatbot
)
load_knowlege_button
=
gr
.
Button
(
"重新构建知识库"
)
load_knowlege_button
.
click
(
init_set_vector_store
,
show_progress
=
True
,
inputs
=
[
UPLOAD_ROOT_PATH
,
select_vs
,
chatbot
],
outputs
=
chatbot
)
(
demo
.
queue
(
concurrency_count
=
3
)
.
launch
(
server_name
=
'0.0.0.0'
,
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论