dify-oapi2 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

dify_oapi/api/knowledge/v1/model/delete_dataset_request.py

@@ -0,0 +1,28 @@
+from dify_oapi.core.enum import HttpMethod
+from dify_oapi.core.model.base_request import BaseRequest
+
+
+class DeleteDatasetRequest(BaseRequest):
+    def __init__(self) -> None:
+        super().__init__()
+        self.dataset_id: str | None = None
+
+    @staticmethod
+    def builder() -> "DeleteDatasetRequestBuilder":
+        return DeleteDatasetRequestBuilder()
+
+
+class DeleteDatasetRequestBuilder:
+    def __init__(self) -> None:
+        delete_dataset_request = DeleteDatasetRequest()
+        delete_dataset_request.http_method = HttpMethod.DELETE
+        delete_dataset_request.uri = "/v1/datasets/:dataset_id"
+        self._delete_dataset_request = delete_dataset_request
+
+    def build(self) -> DeleteDatasetRequest:
+        return self._delete_dataset_request
+
+    def dataset_id(self, dataset_id: str) -> "DeleteDatasetRequestBuilder":
+        self._delete_dataset_request.dataset_id = dataset_id
+        self._delete_dataset_request.paths["dataset_id"] = dataset_id
+        return self

dify_oapi/api/{knowledge_base/v1/model/tag/unbind_response.py → knowledge/v1/model/delete_dataset_response.py}

@@ -1,7 +1,5 @@
-from __future__ import annotations
-
 from dify_oapi.core.model.base_response import BaseResponse
 
 
-class UnbindResponse(BaseResponse):
+class DeleteDatasetResponse(BaseResponse):
     result: str | None = None

dify_oapi/api/knowledge/v1/model/delete_document_request.py

@@ -0,0 +1,42 @@
+"""Delete document request model."""
+
+from __future__ import annotations
+
+from dify_oapi.core.enum import HttpMethod
+from dify_oapi.core.model.base_request import BaseRequest
+
+
+class DeleteDocumentRequest(BaseRequest):
+    """Request model for delete document API."""
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.dataset_id: str | None = None
+        self.document_id: str | None = None
+
+    @staticmethod
+    def builder() -> DeleteDocumentRequestBuilder:
+        return DeleteDocumentRequestBuilder()
+
+
+class DeleteDocumentRequestBuilder:
+    """Builder for DeleteDocumentRequest."""
+
+    def __init__(self) -> None:
+        delete_document_request = DeleteDocumentRequest()
+        delete_document_request.http_method = HttpMethod.DELETE
+        delete_document_request.uri = "/v1/datasets/:dataset_id/documents/:document_id"
+        self._delete_document_request = delete_document_request
+
+    def build(self) -> DeleteDocumentRequest:
+        return self._delete_document_request
+
+    def dataset_id(self, dataset_id: str) -> DeleteDocumentRequestBuilder:
+        self._delete_document_request.dataset_id = dataset_id
+        self._delete_document_request.paths["dataset_id"] = dataset_id
+        return self
+
+    def document_id(self, document_id: str) -> DeleteDocumentRequestBuilder:
+        self._delete_document_request.document_id = document_id
+        self._delete_document_request.paths["document_id"] = document_id
+        return self

dify_oapi/api/knowledge/v1/model/delete_document_response.py

@@ -0,0 +1,9 @@
+"""Delete document response model."""
+
+from dify_oapi.core.model.base_response import BaseResponse
+
+
+class DeleteDocumentResponse(BaseResponse):
+    """Response model for delete document API."""
+
+    result: str | None = None

dify_oapi/api/knowledge/v1/model/delete_segment_request.py

@@ -0,0 +1,48 @@
+"""Delete segment request model."""
+
+from __future__ import annotations
+
+from dify_oapi.core.enum import HttpMethod
+from dify_oapi.core.model.base_request import BaseRequest
+
+
+class DeleteSegmentRequest(BaseRequest):
+    """Request model for delete segment API."""
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.dataset_id: str | None = None
+        self.document_id: str | None = None
+        self.segment_id: str | None = None
+
+    @staticmethod
+    def builder() -> DeleteSegmentRequestBuilder:
+        return DeleteSegmentRequestBuilder()
+
+
+class DeleteSegmentRequestBuilder:
+    """Builder for DeleteSegmentRequest."""
+
+    def __init__(self) -> None:
+        delete_segment_request = DeleteSegmentRequest()
+        delete_segment_request.http_method = HttpMethod.DELETE
+        delete_segment_request.uri = "/v1/datasets/:dataset_id/documents/:document_id/segments/:segment_id"
+        self._delete_segment_request = delete_segment_request
+
+    def build(self) -> DeleteSegmentRequest:
+        return self._delete_segment_request
+
+    def dataset_id(self, dataset_id: str) -> DeleteSegmentRequestBuilder:
+        self._delete_segment_request.dataset_id = dataset_id
+        self._delete_segment_request.paths["dataset_id"] = dataset_id
+        return self
+
+    def document_id(self, document_id: str) -> DeleteSegmentRequestBuilder:
+        self._delete_segment_request.document_id = document_id
+        self._delete_segment_request.paths["document_id"] = document_id
+        return self
+
+    def segment_id(self, segment_id: str) -> DeleteSegmentRequestBuilder:
+        self._delete_segment_request.segment_id = segment_id
+        self._delete_segment_request.paths["segment_id"] = segment_id
+        return self

dify_oapi/api/knowledge/v1/model/delete_segment_response.py

@@ -0,0 +1,9 @@
+"""Delete segment response model."""
+
+from dify_oapi.core.model.base_response import BaseResponse
+
+
+class DeleteSegmentResponse(BaseResponse):
+    """Response model for delete segment API."""
+
+    result: str | None = None

dify_oapi/api/knowledge/v1/model/delete_tag_request.py

@@ -0,0 +1,30 @@
+from dify_oapi.core.enum import HttpMethod
+from dify_oapi.core.model.base_request import BaseRequest
+
+from .delete_tag_request_body import DeleteTagRequestBody
+
+
+class DeleteTagRequest(BaseRequest):
+    def __init__(self):
+        super().__init__()
+        self.request_body: DeleteTagRequestBody | None = None
+
+    @staticmethod
+    def builder() -> "DeleteTagRequestBuilder":
+        return DeleteTagRequestBuilder()
+
+
+class DeleteTagRequestBuilder:
+    def __init__(self):
+        delete_tag_request = DeleteTagRequest()
+        delete_tag_request.http_method = HttpMethod.DELETE
+        delete_tag_request.uri = "/v1/datasets/tags"
+        self._delete_tag_request = delete_tag_request
+
+    def build(self) -> DeleteTagRequest:
+        return self._delete_tag_request
+
+    def request_body(self, request_body: DeleteTagRequestBody) -> "DeleteTagRequestBuilder":
+        self._delete_tag_request.request_body = request_body
+        self._delete_tag_request.body = request_body.model_dump(exclude_none=True, mode="json")
+        return self

dify_oapi/api/knowledge/v1/model/delete_tag_request_body.py

@@ -0,0 +1,21 @@
+from pydantic import BaseModel
+
+
+class DeleteTagRequestBody(BaseModel):
+    tag_id: str | None = None
+
+    @staticmethod
+    def builder() -> "DeleteTagRequestBodyBuilder":
+        return DeleteTagRequestBodyBuilder()
+
+
+class DeleteTagRequestBodyBuilder:
+    def __init__(self):
+        self._delete_tag_request_body = DeleteTagRequestBody()
+
+    def build(self) -> DeleteTagRequestBody:
+        return self._delete_tag_request_body
+
+    def tag_id(self, tag_id: str) -> "DeleteTagRequestBodyBuilder":
+        self._delete_tag_request_body.tag_id = tag_id
+        return self

dify_oapi/api/knowledge/v1/model/delete_tag_response.py

@@ -0,0 +1,5 @@
+from dify_oapi.core.model.base_response import BaseResponse
+
+
+class DeleteTagResponse(BaseResponse):
+    result: str | None = None

dify_oapi/api/knowledge/v1/model/document_info.py

@@ -0,0 +1,144 @@
+"""Document information model for Knowledge Base API."""
+
+from typing import Optional
+
+from pydantic import BaseModel
+
+from .data_source_detail import DataSourceDetailDict
+from .data_source_info import DataSourceInfo
+from .document_metadata import DocumentMetadata
+from .knowledge_types import DataSourceType, DocumentDisplayStatus, DocumentForm, IndexingStatus
+
+
+class DocumentInfo(BaseModel):
+    """Document information model with builder pattern."""
+
+    id: Optional[str] = None
+    position: Optional[int] = None
+    data_source_type: Optional[DataSourceType] = None
+    data_source_info: Optional["DataSourceInfo"] = None
+    data_source_detail_dict: Optional[DataSourceDetailDict] = None
+    dataset_process_rule_id: Optional[str] = None
+    name: Optional[str] = None
+    created_from: Optional[str] = None
+    created_by: Optional[str] = None
+    created_at: Optional[int] = None
+    tokens: Optional[int] = None
+    indexing_status: Optional[IndexingStatus] = None
+    error: Optional[str] = None
+    enabled: Optional[bool] = None
+    disabled_at: Optional[int] = None
+    disabled_by: Optional[str] = None
+    archived: Optional[bool] = None
+    display_status: Optional[DocumentDisplayStatus] = None
+    word_count: Optional[int] = None
+    hit_count: Optional[int] = None
+    doc_form: Optional[DocumentForm] = None
+    doc_metadata: Optional[DocumentMetadata] = None
+    updated_at: Optional[int] = None
+
+    @staticmethod
+    def builder() -> "DocumentInfoBuilder":
+        return DocumentInfoBuilder()
+
+
+class DocumentInfoBuilder:
+    """Builder for DocumentInfo."""
+
+    def __init__(self):
+        self._document_info = DocumentInfo()
+
+    def build(self) -> DocumentInfo:
+        return self._document_info
+
+    def id(self, id: str) -> "DocumentInfoBuilder":
+        self._document_info.id = id
+        return self
+
+    def position(self, position: int) -> "DocumentInfoBuilder":
+        self._document_info.position = position
+        return self
+
+    def data_source_type(self, data_source_type: DataSourceType) -> "DocumentInfoBuilder":
+        self._document_info.data_source_type = data_source_type
+        return self
+
+    def data_source_info(self, data_source_info: DataSourceInfo) -> "DocumentInfoBuilder":
+        self._document_info.data_source_info = data_source_info
+        return self
+
+    def data_source_detail_dict(self, data_source_detail_dict: DataSourceDetailDict) -> "DocumentInfoBuilder":
+        self._document_info.data_source_detail_dict = data_source_detail_dict
+        return self
+
+    def dataset_process_rule_id(self, dataset_process_rule_id: str) -> "DocumentInfoBuilder":
+        self._document_info.dataset_process_rule_id = dataset_process_rule_id
+        return self
+
+    def name(self, name: str) -> "DocumentInfoBuilder":
+        self._document_info.name = name
+        return self
+
+    def created_from(self, created_from: str) -> "DocumentInfoBuilder":
+        self._document_info.created_from = created_from
+        return self
+
+    def created_by(self, created_by: str) -> "DocumentInfoBuilder":
+        self._document_info.created_by = created_by
+        return self
+
+    def created_at(self, created_at: int) -> "DocumentInfoBuilder":
+        self._document_info.created_at = created_at
+        return self
+
+    def tokens(self, tokens: int) -> "DocumentInfoBuilder":
+        self._document_info.tokens = tokens
+        return self
+
+    def indexing_status(self, indexing_status: IndexingStatus) -> "DocumentInfoBuilder":
+        self._document_info.indexing_status = indexing_status
+        return self
+
+    def error(self, error: str) -> "DocumentInfoBuilder":
+        self._document_info.error = error
+        return self
+
+    def enabled(self, enabled: bool) -> "DocumentInfoBuilder":
+        self._document_info.enabled = enabled
+        return self
+
+    def disabled_at(self, disabled_at: int) -> "DocumentInfoBuilder":
+        self._document_info.disabled_at = disabled_at
+        return self
+
+    def disabled_by(self, disabled_by: str) -> "DocumentInfoBuilder":
+        self._document_info.disabled_by = disabled_by
+        return self
+
+    def archived(self, archived: bool) -> "DocumentInfoBuilder":
+        self._document_info.archived = archived
+        return self
+
+    def display_status(self, display_status: DocumentDisplayStatus) -> "DocumentInfoBuilder":
+        self._document_info.display_status = display_status
+        return self
+
+    def word_count(self, word_count: int) -> "DocumentInfoBuilder":
+        self._document_info.word_count = word_count
+        return self
+
+    def hit_count(self, hit_count: int) -> "DocumentInfoBuilder":
+        self._document_info.hit_count = hit_count
+        return self
+
+    def doc_form(self, doc_form: DocumentForm) -> "DocumentInfoBuilder":
+        self._document_info.doc_form = doc_form
+        return self
+
+    def doc_metadata(self, doc_metadata: DocumentMetadata) -> "DocumentInfoBuilder":
+        self._document_info.doc_metadata = doc_metadata
+        return self
+
+    def updated_at(self, updated_at: int) -> "DocumentInfoBuilder":
+        self._document_info.updated_at = updated_at
+        return self

dify_oapi/api/knowledge/v1/model/document_metadata.py

@@ -0,0 +1,39 @@
+"""Document metadata model for Knowledge Base API."""
+
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class DocumentMetadata(BaseModel):
+    """Document metadata model with builder pattern."""
+
+    id: Optional[str] = None
+    name: Optional[str] = None
+    type: Optional[str] = None
+
+    @staticmethod
+    def builder() -> "DocumentMetadataBuilder":
+        return DocumentMetadataBuilder()
+
+
+class DocumentMetadataBuilder:
+    """Builder for DocumentMetadata."""
+
+    def __init__(self):
+        self._document_metadata = DocumentMetadata()
+
+    def build(self) -> DocumentMetadata:
+        return self._document_metadata
+
+    def id(self, id: str) -> "DocumentMetadataBuilder":
+        self._document_metadata.id = id
+        return self
+
+    def name(self, name: str) -> "DocumentMetadataBuilder":
+        self._document_metadata.name = name
+        return self
+
+    def type(self, type: str) -> "DocumentMetadataBuilder":
+        self._document_metadata.type = type
+        return self

dify_oapi/api/knowledge/v1/model/embedding_model_parameters.py

@@ -0,0 +1,47 @@
+"""Embedding model parameters for Knowledge Base API."""
+
+from typing import Optional
+
+from pydantic import BaseModel
+
+from .model_credentials import ModelCredentials
+from .model_parameters import ModelParameters
+
+
+class EmbeddingModelParameters(BaseModel):
+    """Embedding model parameters with builder pattern."""
+
+    model: Optional[str] = None
+    provider: Optional[str] = None
+    credentials: Optional[ModelCredentials] = None
+    model_parameters: Optional[ModelParameters] = None
+
+    @staticmethod
+    def builder() -> "EmbeddingModelParametersBuilder":
+        return EmbeddingModelParametersBuilder()
+
+
+class EmbeddingModelParametersBuilder:
+    """Builder for EmbeddingModelParameters."""
+
+    def __init__(self):
+        self._embedding_model_parameters = EmbeddingModelParameters()
+
+    def build(self) -> EmbeddingModelParameters:
+        return self._embedding_model_parameters
+
+    def model(self, model: str) -> "EmbeddingModelParametersBuilder":
+        self._embedding_model_parameters.model = model
+        return self
+
+    def provider(self, provider: str) -> "EmbeddingModelParametersBuilder":
+        self._embedding_model_parameters.provider = provider
+        return self
+
+    def credentials(self, credentials: ModelCredentials) -> "EmbeddingModelParametersBuilder":
+        self._embedding_model_parameters.credentials = credentials
+        return self
+
+    def model_parameters(self, model_parameters: ModelParameters) -> "EmbeddingModelParametersBuilder":
+        self._embedding_model_parameters.model_parameters = model_parameters
+        return self

dify_oapi/api/{knowledge_base/v1/model/dataset → knowledge/v1/model}/external_knowledge_info.py

@@ -1,38 +1,44 @@
-from __future__ import annotations
+"""External knowledge info model for Knowledge Base API."""
+
+from typing import Optional
 
 from pydantic import BaseModel
 
 
 class ExternalKnowledgeInfo(BaseModel):
-    external_knowledge_id: str | None = None
-    external_knowledge_api_id: str | None = None
-    external_knowledge_api_name: str | None = None
-    external_knowledge_api_endpoint: str | None = None
+    """External knowledge info model with builder pattern."""
+
+    external_knowledge_id: Optional[str] = None
+    external_knowledge_api_id: Optional[str] = None
+    external_knowledge_api_name: Optional[str] = None
+    external_knowledge_api_endpoint: Optional[str] = None
 
     @staticmethod
-    def builder() -> ExternalKnowledgeInfoBuilder:
+    def builder() -> "ExternalKnowledgeInfoBuilder":
         return ExternalKnowledgeInfoBuilder()
 
 
 class ExternalKnowledgeInfoBuilder:
+    """Builder for ExternalKnowledgeInfo."""
+
     def __init__(self):
         self._external_knowledge_info = ExternalKnowledgeInfo()
 
     def build(self) -> ExternalKnowledgeInfo:
         return self._external_knowledge_info
 
-    def external_knowledge_id(self, external_knowledge_id: str) -> ExternalKnowledgeInfoBuilder:
-        self._external_knowledge_info.external_knowledge_id = external_knowledge_id
+    def external_knowledge_api_id(self, external_knowledge_api_id: str) -> "ExternalKnowledgeInfoBuilder":
+        self._external_knowledge_info.external_knowledge_api_id = external_knowledge_api_id
         return self
 
-    def external_knowledge_api_id(self, external_knowledge_api_id: str) -> ExternalKnowledgeInfoBuilder:
-        self._external_knowledge_info.external_knowledge_api_id = external_knowledge_api_id
+    def external_knowledge_id(self, external_knowledge_id: str) -> "ExternalKnowledgeInfoBuilder":
+        self._external_knowledge_info.external_knowledge_id = external_knowledge_id
         return self
 
-    def external_knowledge_api_name(self, external_knowledge_api_name: str) -> ExternalKnowledgeInfoBuilder:
+    def external_knowledge_api_name(self, external_knowledge_api_name: str) -> "ExternalKnowledgeInfoBuilder":
         self._external_knowledge_info.external_knowledge_api_name = external_knowledge_api_name
         return self
 
-    def external_knowledge_api_endpoint(self, external_knowledge_api_endpoint: str) -> ExternalKnowledgeInfoBuilder:
+    def external_knowledge_api_endpoint(self, external_knowledge_api_endpoint: str) -> "ExternalKnowledgeInfoBuilder":
         self._external_knowledge_info.external_knowledge_api_endpoint = external_knowledge_api_endpoint
         return self

dify_oapi/api/knowledge/v1/model/external_retrieval_model.py

@@ -0,0 +1,13 @@
+"""External retrieval model for Knowledge Base API."""
+
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class ExternalRetrievalModel(BaseModel):
+    """External retrieval model configuration."""
+
+    top_k: Optional[int] = None
+    score_threshold: Optional[float] = None
+    score_threshold_enabled: Optional[bool] = None

dify_oapi/api/knowledge/v1/model/file_info.py

@@ -0,0 +1,71 @@
+"""File information model for Knowledge Base API."""
+
+from typing import Optional
+
+from pydantic import BaseModel
+
+from .knowledge_types import FileType
+
+
+class FileInfo(BaseModel):
+    """File information model with builder pattern."""
+
+    id: Optional[str] = None
+    name: Optional[str] = None
+    size: Optional[int] = None
+    extension: Optional[str] = None
+    mime_type: Optional[str] = None
+    type: Optional[FileType] = None
+    created_by: Optional[str] = None
+    created_at: Optional[float] = None
+    upload_file_id: Optional[str] = None
+
+    @staticmethod
+    def builder() -> "FileInfoBuilder":
+        return FileInfoBuilder()
+
+
+class FileInfoBuilder:
+    """Builder for FileInfo."""
+
+    def __init__(self):
+        self._file_info = FileInfo()
+
+    def build(self) -> FileInfo:
+        return self._file_info
+
+    def id(self, id: str) -> "FileInfoBuilder":
+        self._file_info.id = id
+        return self
+
+    def name(self, name: str) -> "FileInfoBuilder":
+        self._file_info.name = name
+        return self
+
+    def size(self, size: int) -> "FileInfoBuilder":
+        self._file_info.size = size
+        return self
+
+    def extension(self, extension: str) -> "FileInfoBuilder":
+        self._file_info.extension = extension
+        return self
+
+    def mime_type(self, mime_type: str) -> "FileInfoBuilder":
+        self._file_info.mime_type = mime_type
+        return self
+
+    def type(self, type: FileType) -> "FileInfoBuilder":
+        self._file_info.type = type
+        return self
+
+    def created_by(self, created_by: str) -> "FileInfoBuilder":
+        self._file_info.created_by = created_by
+        return self
+
+    def created_at(self, created_at: float) -> "FileInfoBuilder":
+        self._file_info.created_at = created_at
+        return self
+
+    def upload_file_id(self, upload_file_id: str) -> "FileInfoBuilder":
+        self._file_info.upload_file_id = upload_file_id
+        return self

dify_oapi/api/knowledge/v1/model/get_batch_indexing_status_request.py

@@ -0,0 +1,42 @@
+"""Get batch indexing status request model."""
+
+from __future__ import annotations
+
+from dify_oapi.core.enum import HttpMethod
+from dify_oapi.core.model.base_request import BaseRequest
+
+
+class GetBatchIndexingStatusRequest(BaseRequest):
+    """Request model for get batch indexing status API."""
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.dataset_id: str | None = None
+        self.batch: str | None = None
+
+    @staticmethod
+    def builder() -> GetBatchIndexingStatusRequestBuilder:
+        return GetBatchIndexingStatusRequestBuilder()
+
+
+class GetBatchIndexingStatusRequestBuilder:
+    """Builder for GetBatchIndexingStatusRequest."""
+
+    def __init__(self) -> None:
+        get_batch_indexing_status_request = GetBatchIndexingStatusRequest()
+        get_batch_indexing_status_request.http_method = HttpMethod.GET
+        get_batch_indexing_status_request.uri = "/v1/datasets/:dataset_id/documents/:batch/indexing-status"
+        self._get_batch_indexing_status_request = get_batch_indexing_status_request
+
+    def build(self) -> GetBatchIndexingStatusRequest:
+        return self._get_batch_indexing_status_request
+
+    def dataset_id(self, dataset_id: str) -> GetBatchIndexingStatusRequestBuilder:
+        self._get_batch_indexing_status_request.dataset_id = dataset_id
+        self._get_batch_indexing_status_request.paths["dataset_id"] = dataset_id
+        return self
+
+    def batch(self, batch: str) -> GetBatchIndexingStatusRequestBuilder:
+        self._get_batch_indexing_status_request.batch = batch
+        self._get_batch_indexing_status_request.paths["batch"] = batch
+        return self

dify_oapi/api/knowledge/v1/model/get_batch_indexing_status_response.py

@@ -0,0 +1,22 @@
+"""Get batch indexing status response model."""
+
+from typing import Optional
+
+from dify_oapi.core.model.base_response import BaseResponse
+
+
+class GetBatchIndexingStatusResponse(BaseResponse):
+    """Response model for get batch indexing status API."""
+
+    id: Optional[str] = None
+    indexing_status: Optional[str] = None
+    processing_started_at: Optional[float] = None
+    parsing_completed_at: Optional[float] = None
+    cleaning_completed_at: Optional[float] = None
+    splitting_completed_at: Optional[float] = None
+    completed_at: Optional[float] = None
+    paused_at: Optional[float] = None
+    error: Optional[str] = None
+    stopped_at: Optional[float] = None
+    completed_segments: Optional[int] = None
+    total_segments: Optional[int] = None