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

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

@@ -1,40 +0,0 @@
-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
-        self.document_id: str | None = None
-        self.segment_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/documents/:document_id/segments/:segment_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
-
-    def document_id(self, document_id: str) -> "GetRequestBuilder":
-        self._get_request.document_id = document_id
-        self._get_request.paths["document_id"] = document_id
-        return self
-
-    def segment_id(self, segment_id: str) -> "GetRequestBuilder":
-        self._get_request.segment_id = segment_id
-        self._get_request.paths["segment_id"] = segment_id
-        return self

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

@@ -1,50 +0,0 @@
-from dify_oapi.core.enum import HttpMethod
-from dify_oapi.core.model.base_request import BaseRequest
-
-
-class ListRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-        self.dataset_id: str | None = None
-        self.document_id: str | None = None
-
-    @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/:dataset_id/documents/:document_id/segments"
-        self._list_request = list_request
-
-    def build(self) -> ListRequest:
-        return self._list_request
-
-    def dataset_id(self, dataset_id: str) -> "ListRequestBuilder":
-        self._list_request.dataset_id = dataset_id
-        self._list_request.paths["dataset_id"] = dataset_id
-        return self
-
-    def document_id(self, document_id: str) -> "ListRequestBuilder":
-        self._list_request.document_id = document_id
-        self._list_request.paths["document_id"] = document_id
-        return self
-
-    def keyword(self, keyword: str) -> "ListRequestBuilder":
-        self._list_request.add_query("keyword", keyword)
-        return self
-
-    def status(self, status: str) -> "ListRequestBuilder":
-        self._list_request.add_query("status", status)
-        return self
-
-    def page(self, page: int) -> "ListRequestBuilder":
-        self._list_request.add_query("page", page)
-        return self
-
-    def limit(self, limit: int) -> "ListRequestBuilder":
-        self._list_request.add_query("limit", limit)
-        return self

dify_oapi/api/knowledge_base/v1/model/segment/segment_data.py

@@ -1,43 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-
-class SegmentData(BaseModel):
-    content: str | None = None
-    answer: str | None = None
-    keywords: list[str] | None = None
-    enabled: bool | None = None
-    regenerate_child_chunks: bool | None = None
-
-    @staticmethod
-    def builder() -> SegmentDataBuilder:
-        return SegmentDataBuilder()
-
-
-class SegmentDataBuilder:
-    def __init__(self):
-        self._segment_data = SegmentData()
-
-    def build(self) -> SegmentData:
-        return self._segment_data
-
-    def content(self, content: str) -> SegmentDataBuilder:
-        self._segment_data.content = content
-        return self
-
-    def answer(self, answer: str) -> SegmentDataBuilder:
-        self._segment_data.answer = answer
-        return self
-
-    def keywords(self, keywords: list[str]) -> SegmentDataBuilder:
-        self._segment_data.keywords = keywords
-        return self
-
-    def enabled(self, enabled: bool) -> SegmentDataBuilder:
-        self._segment_data.enabled = enabled
-        return self
-
-    def regenerate_child_chunks(self, regenerate_child_chunks: bool) -> SegmentDataBuilder:
-        self._segment_data.regenerate_child_chunks = regenerate_child_chunks
-        return self

dify_oapi/api/knowledge_base/v1/model/segment/segment_info.py

@@ -1,123 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-
-class SegmentInfo(BaseModel):
-    id: str | None = None
-    position: int | None = None
-    document_id: str | None = None
-    content: str | None = None
-    answer: str | None = None
-    word_count: int | None = None
-    tokens: int | None = None
-    keywords: list[str] | None = None
-    index_node_id: str | None = None
-    index_node_hash: str | None = None
-    hit_count: int | None = None
-    enabled: bool | None = None
-    disabled_at: int | None = None
-    disabled_by: str | None = None
-    status: str | None = None
-    created_by: str | None = None
-    created_at: int | None = None
-    indexing_at: int | None = None
-    completed_at: int | None = None
-    error: str | None = None
-    stopped_at: int | None = None
-
-    @staticmethod
-    def builder() -> SegmentInfoBuilder:
-        return SegmentInfoBuilder()
-
-
-class SegmentInfoBuilder:
-    def __init__(self):
-        self._segment_info = SegmentInfo()
-
-    def build(self) -> SegmentInfo:
-        return self._segment_info
-
-    def id(self, id: str) -> SegmentInfoBuilder:
-        self._segment_info.id = id
-        return self
-
-    def position(self, position: int) -> SegmentInfoBuilder:
-        self._segment_info.position = position
-        return self
-
-    def document_id(self, document_id: str) -> SegmentInfoBuilder:
-        self._segment_info.document_id = document_id
-        return self
-
-    def content(self, content: str) -> SegmentInfoBuilder:
-        self._segment_info.content = content
-        return self
-
-    def answer(self, answer: str) -> SegmentInfoBuilder:
-        self._segment_info.answer = answer
-        return self
-
-    def word_count(self, word_count: int) -> SegmentInfoBuilder:
-        self._segment_info.word_count = word_count
-        return self
-
-    def tokens(self, tokens: int) -> SegmentInfoBuilder:
-        self._segment_info.tokens = tokens
-        return self
-
-    def keywords(self, keywords: list[str]) -> SegmentInfoBuilder:
-        self._segment_info.keywords = keywords
-        return self
-
-    def index_node_id(self, index_node_id: str) -> SegmentInfoBuilder:
-        self._segment_info.index_node_id = index_node_id
-        return self
-
-    def index_node_hash(self, index_node_hash: str) -> SegmentInfoBuilder:
-        self._segment_info.index_node_hash = index_node_hash
-        return self
-
-    def hit_count(self, hit_count: int) -> SegmentInfoBuilder:
-        self._segment_info.hit_count = hit_count
-        return self
-
-    def enabled(self, enabled: bool) -> SegmentInfoBuilder:
-        self._segment_info.enabled = enabled
-        return self
-
-    def disabled_at(self, disabled_at: int) -> SegmentInfoBuilder:
-        self._segment_info.disabled_at = disabled_at
-        return self
-
-    def disabled_by(self, disabled_by: str) -> SegmentInfoBuilder:
-        self._segment_info.disabled_by = disabled_by
-        return self
-
-    def status(self, status: str) -> SegmentInfoBuilder:
-        self._segment_info.status = status
-        return self
-
-    def created_by(self, created_by: str) -> SegmentInfoBuilder:
-        self._segment_info.created_by = created_by
-        return self
-
-    def created_at(self, created_at: int) -> SegmentInfoBuilder:
-        self._segment_info.created_at = created_at
-        return self
-
-    def indexing_at(self, indexing_at: int) -> SegmentInfoBuilder:
-        self._segment_info.indexing_at = indexing_at
-        return self
-
-    def completed_at(self, completed_at: int) -> SegmentInfoBuilder:
-        self._segment_info.completed_at = completed_at
-        return self
-
-    def error(self, error: str) -> SegmentInfoBuilder:
-        self._segment_info.error = error
-        return self
-
-    def stopped_at(self, stopped_at: int) -> SegmentInfoBuilder:
-        self._segment_info.stopped_at = stopped_at
-        return self

dify_oapi/api/knowledge_base/v1/model/segment/segment_types.py

@@ -1,24 +0,0 @@
-from __future__ import annotations
-
-from typing import Literal
-
-# Segment status types
-SegmentStatus = Literal["waiting", "indexing", "completed", "error", "paused"]
-
-# Segment enabled status types
-SegmentEnabledStatus = Literal["enabled", "disabled"]
-
-# Child chunk status types
-ChildChunkStatus = Literal["waiting", "indexing", "completed", "error"]
-
-# Segment search status types
-SearchStatus = Literal["all", "enabled", "disabled"]
-
-# Sort order types
-SortOrder = Literal["created_at", "position", "word_count", "hit_count"]
-
-# Sort direction types
-SortDirection = Literal["asc", "desc"]
-
-# Segment enabled boolean
-SegmentEnabled = Literal[True, False]

dify_oapi/api/knowledge_base/v1/model/segment/update_request.py

@@ -1,48 +0,0 @@
-from dify_oapi.core.enum import HttpMethod
-from dify_oapi.core.model.base_request import BaseRequest
-
-from .update_request_body import UpdateRequestBody
-
-
-class UpdateRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-        self.dataset_id: str | None = None
-        self.document_id: str | None = None
-        self.segment_id: str | None = None
-        self.request_body: UpdateRequestBody | None = None
-
-    @staticmethod
-    def builder() -> "UpdateRequestBuilder":
-        return UpdateRequestBuilder()
-
-
-class UpdateRequestBuilder:
-    def __init__(self):
-        update_request = UpdateRequest()
-        update_request.http_method = HttpMethod.POST
-        update_request.uri = "/v1/datasets/:dataset_id/documents/:document_id/segments/:segment_id"
-        self._update_request = update_request
-
-    def build(self) -> UpdateRequest:
-        return self._update_request
-
-    def dataset_id(self, dataset_id: str) -> "UpdateRequestBuilder":
-        self._update_request.dataset_id = dataset_id
-        self._update_request.paths["dataset_id"] = dataset_id
-        return self
-
-    def document_id(self, document_id: str) -> "UpdateRequestBuilder":
-        self._update_request.document_id = document_id
-        self._update_request.paths["document_id"] = document_id
-        return self
-
-    def segment_id(self, segment_id: str) -> "UpdateRequestBuilder":
-        self._update_request.segment_id = segment_id
-        self._update_request.paths["segment_id"] = segment_id
-        return self
-
-    def request_body(self, request_body: UpdateRequestBody) -> "UpdateRequestBuilder":
-        self._update_request.request_body = request_body
-        self._update_request.body = request_body.model_dump(exclude_none=True, mode="json")
-        return self

dify_oapi/api/knowledge_base/v1/model/segment/update_request_body.py

@@ -1,25 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-from .segment_data import SegmentData
-
-
-class UpdateRequestBody(BaseModel):
-    segment: SegmentData | None = None
-
-    @staticmethod
-    def builder() -> UpdateRequestBodyBuilder:
-        return UpdateRequestBodyBuilder()
-
-
-class UpdateRequestBodyBuilder:
-    def __init__(self):
-        self._update_request_body = UpdateRequestBody()
-
-    def build(self) -> UpdateRequestBody:
-        return self._update_request_body
-
-    def segment(self, segment: SegmentData) -> UpdateRequestBodyBuilder:
-        self._update_request_body.segment = segment
-        return self

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

@@ -1,41 +0,0 @@
-from .bind_request import BindRequest
-from .bind_request_body import BindRequestBody
-from .bind_response import BindResponse
-from .create_request import CreateRequest
-from .create_request_body import CreateRequestBody
-from .create_response import CreateResponse
-from .delete_request import DeleteRequest
-from .delete_request_body import DeleteRequestBody
-from .delete_response import DeleteResponse
-from .list_request import ListRequest
-from .list_response import ListResponse
-from .query_bound_request import QueryBoundRequest
-from .query_bound_response import QueryBoundResponse
-from .unbind_request import UnbindRequest
-from .unbind_request_body import UnbindRequestBody
-from .unbind_response import UnbindResponse
-from .update_request import UpdateRequest
-from .update_request_body import UpdateRequestBody
-from .update_response import UpdateResponse
-
-__all__ = [
-    "BindRequest",
-    "BindRequestBody",
-    "BindResponse",
-    "CreateRequest",
-    "CreateRequestBody",
-    "CreateResponse",
-    "DeleteRequest",
-    "DeleteRequestBody",
-    "DeleteResponse",
-    "ListRequest",
-    "ListResponse",
-    "QueryBoundRequest",
-    "QueryBoundResponse",
-    "UnbindRequest",
-    "UnbindRequestBody",
-    "UnbindResponse",
-    "UpdateRequest",
-    "UpdateRequestBody",
-    "UpdateResponse",
-]

dify_oapi/api/knowledge_base/v1/model/tag/bind_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 .bind_request_body import BindRequestBody
-
-
-class BindRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-        self.request_body: BindRequestBody | None = None
-
-    @staticmethod
-    def builder() -> BindRequestBuilder:
-        return BindRequestBuilder()
-
-
-class BindRequestBuilder:
-    def __init__(self):
-        bind_request = BindRequest()
-        bind_request.http_method = HttpMethod.POST
-        bind_request.uri = "/v1/datasets/tags/binding"
-        self._request = bind_request
-
-    def build(self) -> BindRequest:
-        return self._request
-
-    def request_body(self, request_body: BindRequestBody) -> BindRequestBuilder:
-        self._request.request_body = request_body
-        self._request.body = request_body.model_dump(exclude_none=True, mode="json")
-        return self

dify_oapi/api/knowledge_base/v1/model/tag/bind_request_body.py

@@ -1,29 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-
-class BindRequestBody(BaseModel):
-    tag_ids: list[str] = []
-    target_id: str = ""
-
-    @staticmethod
-    def builder() -> BindRequestBodyBuilder:
-        return BindRequestBodyBuilder()
-
-
-class BindRequestBodyBuilder:
-    def __init__(self):
-        bind_request_body = BindRequestBody()
-        self._request_body = bind_request_body
-
-    def build(self) -> BindRequestBody:
-        return self._request_body
-
-    def tag_ids(self, tag_ids: list[str]) -> BindRequestBodyBuilder:
-        self._request_body.tag_ids = tag_ids
-        return self
-
-    def target_id(self, target_id: str) -> BindRequestBodyBuilder:
-        self._request_body.target_id = target_id
-        return self

dify_oapi/api/knowledge_base/v1/model/tag/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/tags"
-        self._request = create_request
-
-    def build(self) -> CreateRequest:
-        return self._request
-
-    def request_body(self, request_body: CreateRequestBody) -> CreateRequestBuilder:
-        self._request.request_body = request_body
-        self._request.body = request_body.model_dump(exclude_none=True, mode="json")
-        return self

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

@@ -1,24 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-
-class CreateRequestBody(BaseModel):
-    name: str = ""
-
-    @staticmethod
-    def builder() -> CreateRequestBodyBuilder:
-        return CreateRequestBodyBuilder()
-
-
-class CreateRequestBodyBuilder:
-    def __init__(self):
-        create_request_body = CreateRequestBody()
-        self._request_body = create_request_body
-
-    def build(self) -> CreateRequestBody:
-        return self._request_body
-
-    def name(self, name: str) -> CreateRequestBodyBuilder:
-        self._request_body.name = name
-        return self

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

@@ -1,16 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.api.knowledge_base.v1.model.tag.tag_info import TagInfo
-from dify_oapi.core.model.base_response import BaseResponse
-
-
-class CreateResponse(BaseResponse):
-    id: str | None = None
-    name: str | None = None
-    type: str | None = None
-    binding_count: int | None = None
-
-    def to_tag_info(self) -> TagInfo:
-        return TagInfo(
-            id=self.id or "", name=self.name or "", type=self.type or "", binding_count=self.binding_count or 0
-        )

dify_oapi/api/knowledge_base/v1/model/tag/delete_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 .delete_request_body import DeleteRequestBody
-
-
-class DeleteRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-        self.request_body: DeleteRequestBody | 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/tags"
-        self._request = delete_request
-
-    def build(self) -> DeleteRequest:
-        return self._request
-
-    def request_body(self, request_body: DeleteRequestBody) -> DeleteRequestBuilder:
-        self._request.request_body = request_body
-        self._request.body = request_body.model_dump(exclude_none=True, mode="json")
-        return self

dify_oapi/api/knowledge_base/v1/model/tag/delete_request_body.py

@@ -1,24 +0,0 @@
-from __future__ import annotations
-
-from pydantic import BaseModel
-
-
-class DeleteRequestBody(BaseModel):
-    tag_id: str = ""
-
-    @staticmethod
-    def builder() -> DeleteRequestBodyBuilder:
-        return DeleteRequestBodyBuilder()
-
-
-class DeleteRequestBodyBuilder:
-    def __init__(self):
-        delete_request_body = DeleteRequestBody()
-        self._request_body = delete_request_body
-
-    def build(self) -> DeleteRequestBody:
-        return self._request_body
-
-    def tag_id(self, tag_id: str) -> DeleteRequestBodyBuilder:
-        self._request_body.tag_id = tag_id
-        return self

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

@@ -1,24 +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/tags"
-        self._request = list_request
-
-    def build(self) -> ListRequest:
-        return self._request

dify_oapi/api/knowledge_base/v1/model/tag/list_response.py

@@ -1,19 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.api.knowledge_base.v1.model.tag.tag_info import TagInfo
-from dify_oapi.core.model.base_response import BaseResponse
-
-
-class ListResponse(BaseResponse):
-    # For tag list API, the response is directly an array
-    # We'll handle this in the transport layer by wrapping the array
-    data: list[TagInfo] = []
-
-    def __iter__(self):
-        return iter(self.data)
-
-    def __getitem__(self, item):
-        return self.data[item]
-
-    def __len__(self):
-        return len(self.data)

dify_oapi/api/knowledge_base/v1/model/tag/query_bound_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 QueryBoundRequest(BaseRequest):
-    def __init__(self):
-        super().__init__()
-        self.dataset_id: str | None = None
-
-    @staticmethod
-    def builder() -> QueryBoundRequestBuilder:
-        return QueryBoundRequestBuilder()
-
-
-class QueryBoundRequestBuilder:
-    def __init__(self):
-        query_bound_request = QueryBoundRequest()
-        query_bound_request.http_method = HttpMethod.GET
-        query_bound_request.uri = "/v1/datasets/:dataset_id/tags"
-        self._request = query_bound_request
-
-    def build(self) -> QueryBoundRequest:
-        return self._request
-
-    def dataset_id(self, dataset_id: str) -> QueryBoundRequestBuilder:
-        self._request.dataset_id = dataset_id
-        self._request.paths["dataset_id"] = dataset_id
-        return self

dify_oapi/api/knowledge_base/v1/model/tag/query_bound_response.py

@@ -1,9 +0,0 @@
-from __future__ import annotations
-
-from dify_oapi.api.knowledge_base.v1.model.tag.tag_info import TagInfo
-from dify_oapi.core.model.base_response import BaseResponse
-
-
-class QueryBoundResponse(BaseResponse):
-    data: list[TagInfo] = []
-    total: int = 0