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

dify_oapi/api/knowledge_base/v1/model/dataset/__init__.py

@@ -1,49 +0,0 @@
-from ..metadata.metadata_info import MetadataInfo
-from ..tag.tag_info import TagInfo
-from .create_request import CreateRequest
-from .create_request_body import CreateRequestBody
-from .create_response import CreateResponse
-from .dataset_info import DatasetInfo
-from .delete_request import DeleteRequest
-from .delete_response import DeleteResponse
-from .external_knowledge_info import ExternalKnowledgeInfo
-from .filter_condition import FilterCondition
-from .get_request import GetRequest
-from .get_response import GetResponse
-from .list_request import ListRequest
-from .list_response import ListResponse
-from .metadata_filtering_conditions import MetadataFilteringConditions
-from .reranking_model import RerankingModel
-from .retrieval_model import RetrievalModel
-from .retrieve_request import RetrieveRequest
-from .retrieve_request_body import RetrieveRequestBody
-from .retrieve_response import RetrieveResponse
-from .update_request import UpdateRequest
-from .update_request_body import UpdateRequestBody
-from .update_response import UpdateResponse
-
-__all__ = [
-    "CreateRequest",
-    "CreateRequestBody",
-    "CreateResponse",
-    "DatasetInfo",
-    "DeleteRequest",
-    "DeleteResponse",
-    "ExternalKnowledgeInfo",
-    "FilterCondition",
-    "GetRequest",
-    "GetResponse",
-    "ListRequest",
-    "ListResponse",
-    "MetadataFilteringConditions",
-    "MetadataInfo",
-    "RerankingModel",
-    "RetrieveRequest",
-    "RetrieveRequestBody",
-    "RetrieveResponse",
-    "RetrievalModel",
-    "TagInfo",
-    "UpdateRequest",
-    "UpdateRequestBody",
-    "UpdateResponse",
-]

dify_oapi/api/knowledge_base/v1/model/dataset/create_request.py

@@ -1,32 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.core.enum import HttpMethod
-from dify_oapi.core.model.base_request import BaseRequest
-
-from .create_request_body import CreateRequestBody
-
-
-class CreateRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-        self.request_body: CreateRequestBody | None = None
-
-    @staticmethod
-    def builder() -> CreateRequestBuilder:
-        return CreateRequestBuilder()
-
-
-class CreateRequestBuilder:
-    def __init__(self):
-        create_request = CreateRequest()
-        create_request.http_method = HttpMethod.POST
-        create_request.uri = "/v1/datasets"
-        self._create_request = create_request
-
-    def build(self) -> CreateRequest:
-        return self._create_request
-
-    def request_body(self, request_body: CreateRequestBody) -> CreateRequestBuilder:
-        self._create_request.request_body = request_body
-        self._create_request.body = request_body.model_dump(exclude_none=True, mode="json")
-        return self

dify_oapi/api/knowledge_base/v1/model/dataset/create_request_body.py

@@ -1,72 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-from .dataset_types import IndexingTechnique
-from .retrieval_model import RetrievalModel
-
-
-class CreateRequestBody(BaseModel):
-    name: str | None = None
-    description: str | None = None
-    indexing_technique: IndexingTechnique | None = None
-    permission: str | None = None
-    provider: str | None = None
-    external_knowledge_api_id: str | None = None
-    external_knowledge_id: str | None = None
-    embedding_model: str | None = None
-    embedding_model_provider: str | None = None
-    retrieval_model: RetrievalModel | None = None
-
-    @staticmethod
-    def builder() -> CreateRequestBodyBuilder:
-        return CreateRequestBodyBuilder()
-
-
-class CreateRequestBodyBuilder:
-    def __init__(self):
-        create_request_body = CreateRequestBody()
-        self._create_request_body = create_request_body
-
-    def build(self) -> CreateRequestBody:
-        return self._create_request_body
-
-    def name(self, name: str) -> CreateRequestBodyBuilder:
-        self._create_request_body.name = name
-        return self
-
-    def description(self, description: str) -> CreateRequestBodyBuilder:
-        self._create_request_body.description = description
-        return self
-
-    def indexing_technique(self, indexing_technique: IndexingTechnique) -> CreateRequestBodyBuilder:
-        self._create_request_body.indexing_technique = indexing_technique
-        return self
-
-    def permission(self, permission: str) -> CreateRequestBodyBuilder:
-        self._create_request_body.permission = permission
-        return self
-
-    def provider(self, provider: str) -> CreateRequestBodyBuilder:
-        self._create_request_body.provider = provider
-        return self
-
-    def external_knowledge_api_id(self, external_knowledge_api_id: str) -> CreateRequestBodyBuilder:
-        self._create_request_body.external_knowledge_api_id = external_knowledge_api_id
-        return self
-
-    def external_knowledge_id(self, external_knowledge_id: str) -> CreateRequestBodyBuilder:
-        self._create_request_body.external_knowledge_id = external_knowledge_id
-        return self
-
-    def embedding_model(self, embedding_model: str) -> CreateRequestBodyBuilder:
-        self._create_request_body.embedding_model = embedding_model
-        return self
-
-    def embedding_model_provider(self, embedding_model_provider: str) -> CreateRequestBodyBuilder:
-        self._create_request_body.embedding_model_provider = embedding_model_provider
-        return self
-
-    def retrieval_model(self, retrieval_model: RetrievalModel) -> CreateRequestBodyBuilder:
-        self._create_request_body.retrieval_model = retrieval_model
-        return self

dify_oapi/api/knowledge_base/v1/model/dataset/create_response.py

@@ -1,11 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.core.model.base_response import BaseResponse
-
-from .dataset_info import DatasetInfo
-
-
-class CreateResponse(DatasetInfo, BaseResponse):
-    """Response model for dataset creation API"""
-
-    pass

dify_oapi/api/knowledge_base/v1/model/dataset/dataset_info.py

@@ -1,150 +0,0 @@
-from __future__ import annotations
-
-from typing import Literal
-
-from pydantic import BaseModel
-
-from ..metadata.metadata_info import MetadataInfo
-from ..tag.tag_info import TagInfo
-from .external_knowledge_info import ExternalKnowledgeInfo
-from .retrieval_model import RetrievalModel
-
-
-class DatasetInfo(BaseModel):
-    id: str | None = None
-    name: str | None = None
-    description: str | None = None
-    provider: str | None = None
-    permission: str | None = None
-    data_source_type: str | None = None
-    indexing_technique: str | None = None
-    app_count: int | None = None
-    document_count: int | None = None
-    word_count: int | None = None
-    created_by: str | None = None
-    created_at: int | None = None
-    updated_by: str | None = None
-    updated_at: int | None = None
-    embedding_model: str | None = None
-    embedding_model_provider: str | None = None
-    embedding_available: bool | None = None
-    retrieval_model_dict: RetrievalModel | None = None
-    tags: list[TagInfo] | None = None
-    doc_form: str | None = None
-    external_knowledge_info: ExternalKnowledgeInfo | None = None
-    external_retrieval_model: RetrievalModel | None = None
-    partial_member_list: list[str] | None = None
-    doc_metadata: list[MetadataInfo] | None = None
-    built_in_field_enabled: bool | None = None
-
-    @staticmethod
-    def builder() -> DatasetInfoBuilder:
-        return DatasetInfoBuilder()
-
-
-class DatasetInfoBuilder:
-    def __init__(self):
-        self._dataset_info = DatasetInfo()
-
-    def build(self) -> DatasetInfo:
-        return self._dataset_info
-
-    def id(self, id: str) -> DatasetInfoBuilder:
-        self._dataset_info.id = id
-        return self
-
-    def name(self, name: str) -> DatasetInfoBuilder:
-        self._dataset_info.name = name
-        return self
-
-    def description(self, description: str) -> DatasetInfoBuilder:
-        self._dataset_info.description = description
-        return self
-
-    def provider(self, provider: str) -> DatasetInfoBuilder:
-        self._dataset_info.provider = provider
-        return self
-
-    def permission(self, permission: str) -> DatasetInfoBuilder:
-        self._dataset_info.permission = permission
-        return self
-
-    def data_source_type(self, data_source_type: str) -> DatasetInfoBuilder:
-        self._dataset_info.data_source_type = data_source_type
-        return self
-
-    def indexing_technique(self, indexing_technique: Literal["high_quality", "economy"]) -> DatasetInfoBuilder:
-        self._dataset_info.indexing_technique = indexing_technique
-        return self
-
-    def app_count(self, app_count: int) -> DatasetInfoBuilder:
-        self._dataset_info.app_count = app_count
-        return self
-
-    def document_count(self, document_count: int) -> DatasetInfoBuilder:
-        self._dataset_info.document_count = document_count
-        return self
-
-    def word_count(self, word_count: int) -> DatasetInfoBuilder:
-        self._dataset_info.word_count = word_count
-        return self
-
-    def created_by(self, created_by: str) -> DatasetInfoBuilder:
-        self._dataset_info.created_by = created_by
-        return self
-
-    def created_at(self, created_at: int) -> DatasetInfoBuilder:
-        self._dataset_info.created_at = created_at
-        return self
-
-    def updated_by(self, updated_by: str) -> DatasetInfoBuilder:
-        self._dataset_info.updated_by = updated_by
-        return self
-
-    def updated_at(self, updated_at: int) -> DatasetInfoBuilder:
-        self._dataset_info.updated_at = updated_at
-        return self
-
-    def embedding_model(self, embedding_model: str) -> DatasetInfoBuilder:
-        self._dataset_info.embedding_model = embedding_model
-        return self
-
-    def embedding_model_provider(self, embedding_model_provider: str) -> DatasetInfoBuilder:
-        self._dataset_info.embedding_model_provider = embedding_model_provider
-        return self
-
-    def embedding_available(self, embedding_available: bool) -> DatasetInfoBuilder:
-        self._dataset_info.embedding_available = embedding_available
-        return self
-
-    def retrieval_model_dict(self, retrieval_model_dict: RetrievalModel) -> DatasetInfoBuilder:
-        self._dataset_info.retrieval_model_dict = retrieval_model_dict
-        return self
-
-    def tags(self, tags: list[TagInfo]) -> DatasetInfoBuilder:
-        self._dataset_info.tags = tags
-        return self
-
-    def doc_form(self, doc_form: Literal["text_model", "hierarchical_model", "qa_model"]) -> DatasetInfoBuilder:
-        self._dataset_info.doc_form = doc_form
-        return self
-
-    def external_knowledge_info(self, external_knowledge_info: ExternalKnowledgeInfo) -> DatasetInfoBuilder:
-        self._dataset_info.external_knowledge_info = external_knowledge_info
-        return self
-
-    def external_retrieval_model(self, external_retrieval_model: RetrievalModel) -> DatasetInfoBuilder:
-        self._dataset_info.external_retrieval_model = external_retrieval_model
-        return self
-
-    def partial_member_list(self, partial_member_list: list[str]) -> DatasetInfoBuilder:
-        self._dataset_info.partial_member_list = partial_member_list
-        return self
-
-    def doc_metadata(self, doc_metadata: list[MetadataInfo]) -> DatasetInfoBuilder:
-        self._dataset_info.doc_metadata = doc_metadata
-        return self
-
-    def built_in_field_enabled(self, built_in_field_enabled: bool) -> DatasetInfoBuilder:
-        self._dataset_info.built_in_field_enabled = built_in_field_enabled
-        return self

dify_oapi/api/knowledge_base/v1/model/dataset/dataset_types.py

@@ -1,36 +0,0 @@
-from __future__ import annotations
-
-from typing import Literal
-
-# Indexing technique types
-IndexingTechnique = Literal["high_quality", "economy"]
-
-# Search method types
-SearchMethod = Literal["keyword_search", "semantic_search", "full_text_search", "hybrid_search"]
-
-# Reranking model types
-RerankingModelType = Literal["rerank-model"]
-
-# Processing rule mode types
-ProcessingRuleMode = Literal["automatic", "custom"]
-
-# Data source types
-DataSourceType = Literal["upload_file", "notion_import", "website_crawl"]
-
-# Document status types
-DocumentStatus = Literal["indexing", "completed", "error", "paused"]
-
-# Metadata field types
-MetadataFieldType = Literal["text", "number", "select"]
-
-# Tag types
-TagType = Literal["knowledge", "custom"]
-
-# Built-in metadata actions
-BuiltinMetadataAction = Literal["enable", "disable"]
-
-# Filter operator types
-FilterOperator = Literal["contains", "not_contains", "is", "is_not", "is_empty", "is_not_empty"]
-
-# Reranking enable types
-RerankingEnable = Literal[True, False]

dify_oapi/api/knowledge_base/v1/model/dataset/delete_request.py

@@ -1,30 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.core.enum import HttpMethod
-from dify_oapi.core.model.base_request import BaseRequest
-
-
-class DeleteRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-        self.dataset_id: str | None = None
-
-    @staticmethod
-    def builder() -> DeleteRequestBuilder:
-        return DeleteRequestBuilder()
-
-
-class DeleteRequestBuilder:
-    def __init__(self):
-        delete_request = DeleteRequest()
-        delete_request.http_method = HttpMethod.DELETE
-        delete_request.uri = "/v1/datasets/:dataset_id"
-        self._delete_request = delete_request
-
-    def build(self) -> DeleteRequest:
-        return self._delete_request
-
-    def dataset_id(self, dataset_id: str) -> DeleteRequestBuilder:
-        self._delete_request.dataset_id = dataset_id
-        self._delete_request.paths["dataset_id"] = dataset_id
-        return self

dify_oapi/api/knowledge_base/v1/model/dataset/filter_condition.py

@@ -1,72 +0,0 @@
-from __future__ import annotations
-
-from typing import Literal
-
-from pydantic import BaseModel
-
-
-class FilterCondition(BaseModel):
-    name: str
-    comparison_operator: Literal[
-        "contains",
-        "not contains",
-        "start with",
-        "end with",
-        "is",
-        "is not",
-        "empty",
-        "not empty",
-        "=",
-        "≠",
-        ">",
-        "<",
-        "≥",
-        "≤",
-        "before",
-        "after",
-    ]
-    value: str | int | float | None = None
-
-    @staticmethod
-    def builder() -> FilterConditionBuilder:
-        return FilterConditionBuilder()
-
-
-class FilterConditionBuilder:
-    def __init__(self):
-        self._filter_condition = FilterCondition(name="", comparison_operator="contains")
-
-    def build(self) -> FilterCondition:
-        return self._filter_condition
-
-    def name(self, name: str) -> FilterConditionBuilder:
-        self._filter_condition.name = name
-        return self
-
-    def comparison_operator(
-        self,
-        comparison_operator: Literal[
-            "contains",
-            "not contains",
-            "start with",
-            "end with",
-            "is",
-            "is not",
-            "empty",
-            "not empty",
-            "=",
-            "≠",
-            ">",
-            "<",
-            "≥",
-            "≤",
-            "before",
-            "after",
-        ],
-    ) -> FilterConditionBuilder:
-        self._filter_condition.comparison_operator = comparison_operator
-        return self
-
-    def value(self, value: str | int | float | None) -> FilterConditionBuilder:
-        self._filter_condition.value = value
-        return self

dify_oapi/api/knowledge_base/v1/model/dataset/get_request.py

@@ -1,30 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.core.enum import HttpMethod
-from dify_oapi.core.model.base_request import BaseRequest
-
-
-class GetRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-        self.dataset_id: str | None = None
-
-    @staticmethod
-    def builder() -> GetRequestBuilder:
-        return GetRequestBuilder()
-
-
-class GetRequestBuilder:
-    def __init__(self):
-        get_request = GetRequest()
-        get_request.http_method = HttpMethod.GET
-        get_request.uri = "/v1/datasets/:dataset_id"
-        self._get_request = get_request
-
-    def build(self) -> GetRequest:
-        return self._get_request
-
-    def dataset_id(self, dataset_id: str) -> GetRequestBuilder:
-        self._get_request.dataset_id = dataset_id
-        self._get_request.paths["dataset_id"] = dataset_id
-        return self

dify_oapi/api/knowledge_base/v1/model/dataset/get_response.py

@@ -1,11 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.core.model.base_response import BaseResponse
-
-from .dataset_info import DatasetInfo
-
-
-class GetResponse(DatasetInfo, BaseResponse):
-    """Response model for get dataset details API"""
-
-    pass

dify_oapi/api/knowledge_base/v1/model/dataset/list_request.py

@@ -1,45 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.core.enum import HttpMethod
-from dify_oapi.core.model.base_request import BaseRequest
-
-
-class ListRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-
-    @staticmethod
-    def builder() -> ListRequestBuilder:
-        return ListRequestBuilder()
-
-
-class ListRequestBuilder:
-    def __init__(self):
-        list_request = ListRequest()
-        list_request.http_method = HttpMethod.GET
-        list_request.uri = "/v1/datasets"
-        self._list_request = list_request
-
-    def build(self) -> ListRequest:
-        return self._list_request
-
-    def keyword(self, keyword: str) -> ListRequestBuilder:
-        self._list_request.add_query("keyword", keyword)
-        return self
-
-    def tag_ids(self, tag_ids: list[str]) -> ListRequestBuilder:
-        for tag_id in tag_ids:
-            self._list_request.add_query("tag_ids", tag_id)
-        return self
-
-    def page(self, page: int) -> ListRequestBuilder:
-        self._list_request.add_query("page", page)
-        return self
-
-    def limit(self, limit: str) -> ListRequestBuilder:
-        self._list_request.add_query("limit", limit)
-        return self
-
-    def include_all(self, include_all: bool) -> ListRequestBuilder:
-        self._list_request.add_query("include_all", include_all)
-        return self

dify_oapi/api/knowledge_base/v1/model/dataset/metadata_filtering_conditions.py

@@ -1,30 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-from .filter_condition import FilterCondition
-
-
-class MetadataFilteringConditions(BaseModel):
-    logical_operator: str
-    conditions: list[FilterCondition]
-
-    @staticmethod
-    def builder() -> MetadataFilteringConditionsBuilder:
-        return MetadataFilteringConditionsBuilder()
-
-
-class MetadataFilteringConditionsBuilder:
-    def __init__(self):
-        self._metadata_filtering_conditions = MetadataFilteringConditions(logical_operator="", conditions=[])
-
-    def build(self) -> MetadataFilteringConditions:
-        return self._metadata_filtering_conditions
-
-    def logical_operator(self, logical_operator: str) -> MetadataFilteringConditionsBuilder:
-        self._metadata_filtering_conditions.logical_operator = logical_operator
-        return self
-
-    def conditions(self, conditions: list[FilterCondition]) -> MetadataFilteringConditionsBuilder:
-        self._metadata_filtering_conditions.conditions = conditions
-        return self

dify_oapi/api/knowledge_base/v1/model/dataset/reranking_model.py

@@ -1,28 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-
-class RerankingModel(BaseModel):
-    reranking_provider_name: str | None = None
-    reranking_model_name: str | None = None
-
-    @staticmethod
-    def builder() -> RerankingModelBuilder:
-        return RerankingModelBuilder()
-
-
-class RerankingModelBuilder:
-    def __init__(self):
-        self._reranking_model = RerankingModel()
-
-    def build(self) -> RerankingModel:
-        return self._reranking_model
-
-    def reranking_provider_name(self, reranking_provider_name: str) -> RerankingModelBuilder:
-        self._reranking_model.reranking_provider_name = reranking_provider_name
-        return self
-
-    def reranking_model_name(self, reranking_model_name: str) -> RerankingModelBuilder:
-        self._reranking_model.reranking_model_name = reranking_model_name
-        return self