Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ st.set_page_config(
|
|
| 30 |
from utils.rag.rag_worker import gen_rag_db
|
| 31 |
from utils.tools import resize_image
|
| 32 |
|
| 33 |
-
from utils.model_loader import RAG_RETRIEVER # isort:skip
|
| 34 |
|
| 35 |
|
| 36 |
@st.experimental_dialog("科室介绍", width="large")
|
|
@@ -495,13 +495,13 @@ def update_product_info(
|
|
| 495 |
yaml.dump(product_info_dict, f, allow_unicode=True)
|
| 496 |
|
| 497 |
st.write("生成数据库...")
|
| 498 |
-
if WEB_CONFIGS.ENABLE_RAG:
|
| 499 |
-
|
| 500 |
-
|
| 501 |
|
| 502 |
-
|
| 503 |
-
|
| 504 |
-
|
| 505 |
|
| 506 |
# 更新状态
|
| 507 |
status.update(label="添加科室信息成功!", state="complete", expanded=False)
|
|
|
|
| 30 |
from utils.rag.rag_worker import gen_rag_db
|
| 31 |
from utils.tools import resize_image
|
| 32 |
|
| 33 |
+
# from utils.model_loader import RAG_RETRIEVER # isort:skip
|
| 34 |
|
| 35 |
|
| 36 |
@st.experimental_dialog("科室介绍", width="large")
|
|
|
|
| 495 |
yaml.dump(product_info_dict, f, allow_unicode=True)
|
| 496 |
|
| 497 |
st.write("生成数据库...")
|
| 498 |
+
# if WEB_CONFIGS.ENABLE_RAG:
|
| 499 |
+
# # 重新生成 RAG 向量数据库
|
| 500 |
+
# gen_rag_db(force_gen=True)
|
| 501 |
|
| 502 |
+
# # 重新加载 retriever
|
| 503 |
+
# RAG_RETRIEVER.pop("default")
|
| 504 |
+
# RAG_RETRIEVER.get(fs_id="default", config_path=WEB_CONFIGS.RAG_CONFIG_PATH, work_dir=WEB_CONFIGS.RAG_VECTOR_DB_DIR)
|
| 505 |
|
| 506 |
# 更新状态
|
| 507 |
status.update(label="添加科室信息成功!", state="complete", expanded=False)
|