diff --git a/api/apps/sdk/doc.py b/api/apps/sdk/doc.py index 2a7d98c44..bd8c60c74 100644 --- a/api/apps/sdk/doc.py +++ b/api/apps/sdk/doc.py @@ -363,6 +363,10 @@ def download(tenant_id, dataset_id, document_id): schema: type: object """ + if not document_id: + return get_error_data_result( + message="Specify document_id please." + ) if not KnowledgebaseService.query(id=dataset_id, tenant_id=tenant_id): return get_error_data_result(message=f"You do not own the dataset {dataset_id}.") doc = DocumentService.query(kb_id=dataset_id, id=document_id) diff --git a/api/db/init_data.py b/api/db/init_data.py index 523805e70..fe993033a 100644 --- a/api/db/init_data.py +++ b/api/db/init_data.py @@ -105,9 +105,10 @@ def init_llm_factory(): factory_llm_infos = settings.FACTORY_LLM_INFOS for factory_llm_info in factory_llm_infos: - llm_infos = factory_llm_info.pop("llm") + info = deepcopy(factory_llm_info) + llm_infos = info.pop("llm") try: - LLMFactoriesService.save(**factory_llm_info) + LLMFactoriesService.save(**info) except Exception: pass LLMService.filter_delete([LLM.fid == factory_llm_info["name"]])