Browse Source

feat: 添加知识库重建请求数据传输对象到索引服务

Jax Docker 1 tháng trước cách đây
mục cha
commit
93c50d21df

+ 2 - 2
services/knowledge-service/app/application/indexing_service.py

@@ -27,6 +27,7 @@ from app.infrastructure.object_storage import (
     ObjectStorageNotFoundError,
     build_document_object_key)
 from app.schemas.knowledge import (
+    KnowledgeBaseReindexRequestDto,
     KnowledgeDocumentCreateRequest,
     KnowledgeDocumentCreateRequestDto,
     KnowledgeDocumentReindexRequestDto,
@@ -291,9 +292,8 @@ class KnowledgeIndexingService(_ObjectStorageMixin):
 
     def reindex_base_from_contract(
         self,
-        payload,
+        payload: KnowledgeBaseReindexRequestDto,
     ) -> list[KnowledgeIndexJobData]:
-        from app.schemas.knowledge import KnowledgeBaseReindexRequestDto
         documents = self.document_repository.list_by_base(
             knowledge_base_id=payload.knowledgeBaseId)
         jobs: list[KnowledgeIndexJobData] = []