Skip to content

Commit a174e0a

Browse files
authored
Merge pull request #18 from Samge0/patch-allow-samge-filename
update: 允许不同的知识库上传同名文件
2 parents dc1a326 + c89ad98 commit a174e0a

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

ragflows/ragflowdb.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,14 @@ def get_doc_item(doc_id):
4848
return results[0] if results else None
4949

5050
@timeutils.monitor
51-
def get_doc_item_by_name(doc_id):
51+
def get_doc_item_by_name(name):
5252
db = get_db()
53-
sql = f"select id,name,progress from document where name = '{doc_id}'"
53+
kb_id = configs.DIFY_DOC_KB_ID
54+
if kb_id:
55+
# 这里同时查询kb_id和name,如果document表中的数据量很大,需要增加kb_id和name的组合索引:CREATE INDEX document_kb_id_name ON document(kb_id, name);
56+
sql = f"select id,name,progress from document where kb_id = '{kb_id}' and name = '{name}'"
57+
else:
58+
sql = f"select id,name,progress from document where name = '{name}'"
5459
results = db.query_list(sql)
5560
return results[0] if results else None
5661

0 commit comments

Comments
 (0)