Pārlūkot izejas kodu

refine file select code (#1021)

### What problem does this PR solve?

#1015 

### Type of change

- [x] Refactoring
tags/v0.8.0
KevinHuSh pirms 1 gada
vecāks
revīzija
1de31ca9f6
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
2 mainītis faili ar 6 papildinājumiem un 7 dzēšanām
  1. 1
    1
      api/db/init_data.py
  2. 5
    6
      api/db/services/file_service.py

+ 1
- 1
api/db/init_data.py Parādīt failu

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

+ 5
- 6
api/db/services/file_service.py Parādīt failu

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


Notiek ielāde…
Atcelt
Saglabāt