Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
J
jinchat-server
概览
概览
详情
活动
周期分析
版本库
存储库
文件
提交
分支
标签
贡献者
分支图
比较
统计图
问题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
aigc-pioneer
jinchat-server
Commits
8477368f
Unverified
提交
8477368f
authored
5月 30, 2023
作者:
移花香
提交者:
GitHub
5月 30, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
delete_docs (#499)
删除文档的时候重新生成知识库(FAISS)内的数据
上级
8d669cdb
显示空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
8 行增加
和
6 行删除
+8
-6
api.py
api.py
+8
-6
没有找到文件。
api.py
浏览文件 @
8477368f
...
@@ -175,7 +175,7 @@ async def list_docs(
...
@@ -175,7 +175,7 @@ async def list_docs(
async
def
delete_docs
(
async
def
delete_docs
(
knowledge_base_id
:
str
=
Query
(
...
,
knowledge_base_id
:
str
=
Query
(
...
,
description
=
"Knowledge Base Name
(注意此方法仅删除上传的文件并不会删除知识库(FAISS)内数据)
"
,
description
=
"Knowledge Base Name"
,
example
=
"kb1"
),
example
=
"kb1"
),
doc_name
:
Optional
[
str
]
=
Query
(
doc_name
:
Optional
[
str
]
=
Query
(
None
,
description
=
"doc name"
,
example
=
"doc_name_1.pdf"
None
,
description
=
"doc name"
,
example
=
"doc_name_1.pdf"
...
@@ -188,18 +188,20 @@ async def delete_docs(
...
@@ -188,18 +188,20 @@ async def delete_docs(
doc_path
=
get_file_path
(
knowledge_base_id
,
doc_name
)
doc_path
=
get_file_path
(
knowledge_base_id
,
doc_name
)
if
os
.
path
.
exists
(
doc_path
):
if
os
.
path
.
exists
(
doc_path
):
os
.
remove
(
doc_path
)
os
.
remove
(
doc_path
)
return
BaseResponse
(
code
=
200
,
msg
=
f
"document {doc_name} delete success"
)
else
:
return
BaseResponse
(
code
=
1
,
msg
=
f
"document {doc_name} not found"
)
# 删除上传的文件后重新生成知识库(FAISS)内的数据
remain_docs
=
await
list_docs
(
knowledge_base_id
)
remain_docs
=
await
list_docs
(
knowledge_base_id
)
remain_docs
=
remain_docs
.
json
()
if
len
(
remain_docs
.
data
)
==
0
:
if
len
(
remain_docs
[
"data"
])
==
0
:
shutil
.
rmtree
(
get_folder_path
(
knowledge_base_id
),
ignore_errors
=
True
)
shutil
.
rmtree
(
get_folder_path
(
knowledge_base_id
),
ignore_errors
=
True
)
else
:
else
:
local_doc_qa
.
init_knowledge_vector_store
(
local_doc_qa
.
init_knowledge_vector_store
(
get_folder_path
(
knowledge_base_id
),
get_vs_path
(
knowledge_base_id
)
get_folder_path
(
knowledge_base_id
),
get_vs_path
(
knowledge_base_id
)
)
)
return
BaseResponse
(
code
=
200
,
msg
=
f
"document {doc_name} delete success"
)
else
:
return
BaseResponse
(
code
=
1
,
msg
=
f
"document {doc_name} not found"
)
else
:
else
:
shutil
.
rmtree
(
get_folder_path
(
knowledge_base_id
))
shutil
.
rmtree
(
get_folder_path
(
knowledge_base_id
))
return
BaseResponse
(
code
=
200
,
msg
=
f
"Knowledge Base {knowledge_base_id} delete success"
)
return
BaseResponse
(
code
=
200
,
msg
=
f
"Knowledge Base {knowledge_base_id} delete success"
)
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论