AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 10:06:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

refine file select code (#1021)

Browse Source 
### What problem does this PR solve?

#1015 

### Type of change

- [x] Refactoring
This commit is contained in:
KevinHuSh 2024-05-31 19:44:33 +08:00 committed by GitHub
parent 4ec845c0a6
commit 1de31ca9f6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/db/init_data.py
View File

@ -385,7 +385,7 @@ def init_llm_factory():
{ {
"fid": factory_infos[7]["name"], "fid": factory_infos[7]["name"],
"llm_name": "maidalun1020/bce-reranker-base_v1", "llm_name": "maidalun1020/bce-reranker-base_v1",
"tags": "RE-RANK, 8K", "tags": "RE-RANK, 512",
"max_tokens": 512, "max_tokens": 512,
"model_type": LLMType.RERANK.value "model_type": LLMType.RERANK.value
}, },

11
api/db/services/file_service.py
View File

@ -164,12 +164,11 @@ class FileService(CommonService):
@classmethod @classmethod
@DB.connection_context() @DB.connection_context()
def get_kb_folder(cls, tenant_id): def get_kb_folder(cls, tenant_id):
for root in cls.model.select().where(cls.model.tenant_id == tenant_id and for root in cls.model.select().where(
cls.model.parent_id == cls.model.id): (cls.model.tenant_id == tenant_id), (cls.model.parent_id == cls.model.id)):
for folder in cls.model.select().where(cls.model.tenant_id == tenant_id and for folder in cls.model.select().where(
cls.model.parent_id == root.id and (cls.model.tenant_id == tenant_id), (cls.model.parent_id == root.id),
cls.model.name == KNOWLEDGEBASE_FOLDER_NAME (cls.model.name == KNOWLEDGEBASE_FOLDER_NAME)):
):
return folder.to_dict() return folder.to_dict()
assert False, "Can't find the KB folder. Database init error." assert False, "Can't find the KB folder. Database init error."