dify-oapi2 0.4.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

dify_oapi/api/dify/v1/model/submit_feedback_request.py

@@ -0,0 +1,34 @@
+from dify_oapi.core.enum import HttpMethod
+from dify_oapi.core.model.base_request import BaseRequest
+
+from .submit_feedback_request_body import SubmitFeedbackRequestBody
+
+
+class SubmitFeedbackRequest(BaseRequest):
+    def __init__(self) -> None:
+        super().__init__()
+        self.request_body: SubmitFeedbackRequestBody | None = None
+
+    @staticmethod
+    def builder() -> "SubmitFeedbackRequestBuilder":
+        return SubmitFeedbackRequestBuilder()
+
+
+class SubmitFeedbackRequestBuilder:
+    def __init__(self) -> None:
+        submit_feedback_request = SubmitFeedbackRequest()
+        submit_feedback_request.http_method = HttpMethod.POST
+        submit_feedback_request.uri = "/v1/messages/:message_id/feedbacks"
+        self._submit_feedback_request = submit_feedback_request
+
+    def message_id(self, message_id: str) -> "SubmitFeedbackRequestBuilder":
+        self._submit_feedback_request.paths["message_id"] = message_id
+        return self
+
+    def request_body(self, request_body: SubmitFeedbackRequestBody) -> "SubmitFeedbackRequestBuilder":
+        self._submit_feedback_request.request_body = request_body
+        self._submit_feedback_request.body = request_body.model_dump(exclude_none=True, mode="json")
+        return self
+
+    def build(self) -> SubmitFeedbackRequest:
+        return self._submit_feedback_request

dify_oapi/api/dify/v1/model/submit_feedback_request_body.py

@@ -0,0 +1,35 @@
+from typing import Literal
+
+from pydantic import BaseModel
+
+Rating = Literal["like", "dislike"]
+
+
+class SubmitFeedbackRequestBody(BaseModel):
+    rating: Rating | None = None
+    user: str
+    content: str | None = None
+
+    @staticmethod
+    def builder() -> "SubmitFeedbackRequestBodyBuilder":
+        return SubmitFeedbackRequestBodyBuilder()
+
+
+class SubmitFeedbackRequestBodyBuilder:
+    def __init__(self) -> None:
+        self._submit_feedback_request_body = SubmitFeedbackRequestBody(user="")
+
+    def rating(self, rating: Rating | None) -> "SubmitFeedbackRequestBodyBuilder":
+        self._submit_feedback_request_body.rating = rating
+        return self
+
+    def user(self, user: str) -> "SubmitFeedbackRequestBodyBuilder":
+        self._submit_feedback_request_body.user = user
+        return self
+
+    def content(self, content: str | None) -> "SubmitFeedbackRequestBodyBuilder":
+        self._submit_feedback_request_body.content = content
+        return self
+
+    def build(self) -> SubmitFeedbackRequestBody:
+        return self._submit_feedback_request_body

dify_oapi/api/dify/v1/model/submit_feedback_response.py

@@ -0,0 +1,7 @@
+from dify_oapi.core.model.base_response import BaseResponse
+
+
+class SubmitFeedbackResponse(BaseResponse):
+    """Response for submit feedback API."""
+
+    result: str | None = None

dify_oapi/api/dify/v1/resource/__init__.py

@@ -1,6 +0,0 @@
-from .meta import *  # noqa 403
-from .file import *  # noqa 403
-from .audio import *  # noqa 403
-from .parameter import *  # noqa 403
-from .message import *  # noqa 403
-from .info import *  # noqa 403

dify_oapi/api/dify/v1/resource/audio.py

@@ -2,6 +2,8 @@ from dify_oapi.core.http.transport import ATransport, Transport
 from dify_oapi.core.model.config import Config
 from dify_oapi.core.model.request_option import RequestOption
 
+from ..model.audio_to_text_request import AudioToTextRequest
+from ..model.audio_to_text_response import AudioToTextResponse
 from ..model.text_to_audio_request import TextToAudioRequest
 from ..model.text_to_audio_response import TextToAudioResponse
 
@@ -10,8 +12,18 @@ class Audio:
     def __init__(self, config: Config) -> None:
         self.config: Config = config
 
+    def to_text(self, request: AudioToTextRequest, option: RequestOption | None = None) -> AudioToTextResponse:
+        """Convert audio to text (speech-to-text)"""
+        return Transport.execute(self.config, request, unmarshal_as=AudioToTextResponse, option=option)
+
+    async def ato_text(self, request: AudioToTextRequest, option: RequestOption | None = None) -> AudioToTextResponse:
+        """Convert audio to text (speech-to-text) - async version"""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=AudioToTextResponse, option=option)
+
     def from_text(self, request: TextToAudioRequest, option: RequestOption | None = None) -> TextToAudioResponse:
+        """Convert text to audio (text-to-speech)"""
         return Transport.execute(self.config, request, unmarshal_as=TextToAudioResponse, option=option)
 
     async def afrom_text(self, request: TextToAudioRequest, option: RequestOption | None = None) -> TextToAudioResponse:
+        """Convert text to audio (text-to-speech) - async version"""
         return await ATransport.aexecute(self.config, request, unmarshal_as=TextToAudioResponse, option=option)

dify_oapi/api/dify/v1/resource/feedback.py

@@ -0,0 +1,31 @@
+from dify_oapi.core.http.transport import ATransport, Transport
+from dify_oapi.core.model.config import Config
+from dify_oapi.core.model.request_option import RequestOption
+
+from ..model.get_feedbacks_request import GetFeedbacksRequest
+from ..model.get_feedbacks_response import GetFeedbacksResponse
+from ..model.submit_feedback_request import SubmitFeedbackRequest
+from ..model.submit_feedback_response import SubmitFeedbackResponse
+
+
+class Feedback:
+    def __init__(self, config: Config) -> None:
+        self.config: Config = config
+
+    def submit(self, request: SubmitFeedbackRequest, option: RequestOption | None = None) -> SubmitFeedbackResponse:
+        """Submit feedback for a message"""
+        return Transport.execute(self.config, request, unmarshal_as=SubmitFeedbackResponse, option=option)
+
+    async def asubmit(
+        self, request: SubmitFeedbackRequest, option: RequestOption | None = None
+    ) -> SubmitFeedbackResponse:
+        """Submit feedback for a message - async version"""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=SubmitFeedbackResponse, option=option)
+
+    def list(self, request: GetFeedbacksRequest, option: RequestOption | None = None) -> GetFeedbacksResponse:
+        """Get list of feedbacks"""
+        return Transport.execute(self.config, request, unmarshal_as=GetFeedbacksResponse, option=option)
+
+    async def alist(self, request: GetFeedbacksRequest, option: RequestOption | None = None) -> GetFeedbacksResponse:
+        """Get list of feedbacks - async version"""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetFeedbacksResponse, option=option)

dify_oapi/api/dify/v1/resource/info.py

@@ -4,6 +4,12 @@ from dify_oapi.core.model.request_option import RequestOption
 
 from ..model.get_info_request import GetInfoRequest
 from ..model.get_info_response import GetInfoResponse
+from ..model.get_meta_request import GetMetaRequest
+from ..model.get_meta_response import GetMetaResponse
+from ..model.get_parameters_request import GetParametersRequest
+from ..model.get_parameters_response import GetParametersResponse
+from ..model.get_site_request import GetSiteRequest
+from ..model.get_site_response import GetSiteResponse
 
 
 class Info:
@@ -11,9 +17,35 @@ class Info:
         self.config: Config = config
 
     def get(self, request: GetInfoRequest, option: RequestOption | None = None) -> GetInfoResponse:
-        # Send request
+        """Get application information"""
         return Transport.execute(self.config, request, unmarshal_as=GetInfoResponse, option=option)
 
     async def aget(self, request: GetInfoRequest, option: RequestOption | None = None) -> GetInfoResponse:
-        # Send request
+        """Get application information - async version"""
         return await ATransport.aexecute(self.config, request, unmarshal_as=GetInfoResponse, option=option)
+
+    def parameters(self, request: GetParametersRequest, option: RequestOption | None = None) -> GetParametersResponse:
+        """Get application parameters"""
+        return Transport.execute(self.config, request, unmarshal_as=GetParametersResponse, option=option)
+
+    async def aparameters(
+        self, request: GetParametersRequest, option: RequestOption | None = None
+    ) -> GetParametersResponse:
+        """Get application parameters - async version"""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetParametersResponse, option=option)
+
+    def meta(self, request: GetMetaRequest, option: RequestOption | None = None) -> GetMetaResponse:
+        """Get application metadata"""
+        return Transport.execute(self.config, request, unmarshal_as=GetMetaResponse, option=option)
+
+    async def ameta(self, request: GetMetaRequest, option: RequestOption | None = None) -> GetMetaResponse:
+        """Get application metadata - async version"""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetMetaResponse, option=option)
+
+    def site(self, request: GetSiteRequest, option: RequestOption | None = None) -> GetSiteResponse:
+        """Get site settings"""
+        return Transport.execute(self.config, request, unmarshal_as=GetSiteResponse, option=option)
+
+    async def asite(self, request: GetSiteRequest, option: RequestOption | None = None) -> GetSiteResponse:
+        """Get site settings - async version"""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetSiteResponse, option=option)

dify_oapi/api/dify/v1/version.py

@@ -1,13 +1,14 @@
 from dify_oapi.core.model.config import Config
 
-from .resource import Audio, File, Info, Message, Meta, Parameter
+from .resource.audio import Audio
+from .resource.feedback import Feedback
+from .resource.file import File
+from .resource.info import Info
 
 
 class V1:
     def __init__(self, config: Config):
-        self.meta: Meta = Meta(config)
         self.file: File = File(config)
         self.audio: Audio = Audio(config)
-        self.parameter: Parameter = Parameter(config)
-        self.message: Message = Message(config)
         self.info: Info = Info(config)
+        self.feedback: Feedback = Feedback(config)

dify_oapi/api/knowledge/service.py

@@ -3,6 +3,6 @@ from dify_oapi.core.model.config import Config
 from .v1.version import V1
 
 
-class Knowledge:
-    def __init__(self, config: Config):
-        self.v1 = V1(config)
+class KnowledgeService:
+    def __init__(self, config: Config) -> None:
+        self.v1: V1 = V1(config)

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

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import json
 from io import BytesIO
 
 from dify_oapi.core.enum import HttpMethod
@@ -44,8 +43,7 @@ class CreateDocumentByFileRequestBuilder:
 
     def request_body(self, request_body: CreateDocumentByFileRequestBody) -> CreateDocumentByFileRequestBuilder:
         self._create_document_by_file_request.request_body = request_body
-        data_dict = request_body.model_dump(exclude_none=True, mode="json")
-        self._create_document_by_file_request.body = {"data": json.dumps(data_dict)}
+        self._create_document_by_file_request.body = request_body.model_dump(exclude_none=True, mode="json")
         return self
 
     def file(self, file: BytesIO, file_name: str | None = None) -> CreateDocumentByFileRequestBuilder:

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

@@ -4,23 +4,13 @@ from __future__ import annotations
 
 from pydantic import BaseModel
 
-from .knowledge_types import DocumentForm, IndexingTechnique
-from .process_rule import ProcessRule
-from .retrieval_model import RetrievalModel
+from .create_document_by_file_request_body_data import CreateDocumentByFileRequestBodyData
 
 
 class CreateDocumentByFileRequestBody(BaseModel):
     """Request body model for create document by file API."""
 
-    name: str | None = None
-    original_document_id: str | None = None  # UUID for re-upload/modify
-    indexing_technique: IndexingTechnique | None = None
-    doc_form: DocumentForm | None = None
-    doc_language: str | None = None
-    process_rule: ProcessRule | None = None
-    retrieval_model: RetrievalModel | None = None
-    embedding_model: str | None = None
-    embedding_model_provider: str | None = None
+    data: str | None = None
 
     @staticmethod
     def builder() -> CreateDocumentByFileRequestBodyBuilder:
@@ -36,38 +26,6 @@ class CreateDocumentByFileRequestBodyBuilder:
     def build(self) -> CreateDocumentByFileRequestBody:
         return self._create_document_by_file_request_body
 
-    def name(self, name: str) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.name = name
-        return self
-
-    def indexing_technique(self, indexing_technique: IndexingTechnique) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.indexing_technique = indexing_technique
-        return self
-
-    def process_rule(self, process_rule: ProcessRule) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.process_rule = process_rule
-        return self
-
-    def original_document_id(self, original_document_id: str) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.original_document_id = original_document_id
-        return self
-
-    def doc_form(self, doc_form: DocumentForm) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.doc_form = doc_form
-        return self
-
-    def doc_language(self, doc_language: str) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.doc_language = doc_language
-        return self
-
-    def retrieval_model(self, retrieval_model: RetrievalModel) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.retrieval_model = retrieval_model
-        return self
-
-    def embedding_model(self, embedding_model: str) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.embedding_model = embedding_model
-        return self
-
-    def embedding_model_provider(self, embedding_model_provider: str) -> CreateDocumentByFileRequestBodyBuilder:
-        self._create_document_by_file_request_body.embedding_model_provider = embedding_model_provider
+    def data(self, data: CreateDocumentByFileRequestBodyData) -> CreateDocumentByFileRequestBodyBuilder:
+        self._create_document_by_file_request_body.data = data.model_dump_json(exclude_none=True)
         return self

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

@@ -0,0 +1,73 @@
+"""Create document by file request body model."""
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+from .knowledge_types import DocumentForm, IndexingTechnique
+from .process_rule import ProcessRule
+from .retrieval_model import RetrievalModel
+
+
+class CreateDocumentByFileRequestBodyData(BaseModel):
+    """Request body model for create document by file API."""
+
+    name: str | None = None
+    original_document_id: str | None = None  # UUID for re-upload/modify
+    indexing_technique: IndexingTechnique | None = None
+    doc_form: DocumentForm | None = None
+    doc_language: str | None = None
+    process_rule: ProcessRule | None = None
+    retrieval_model: RetrievalModel | None = None
+    embedding_model: str | None = None
+    embedding_model_provider: str | None = None
+
+    @staticmethod
+    def builder() -> CreateDocumentByFileRequestBodyDataBuilder:
+        return CreateDocumentByFileRequestBodyDataBuilder()
+
+
+class CreateDocumentByFileRequestBodyDataBuilder:
+    """Builder for CreateDocumentByFileRequestBodyData."""
+
+    def __init__(self) -> None:
+        self._create_document_by_file_request_body = CreateDocumentByFileRequestBodyData()
+
+    def build(self) -> CreateDocumentByFileRequestBodyData:
+        return self._create_document_by_file_request_body
+
+    def name(self, name: str) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.name = name
+        return self
+
+    def indexing_technique(self, indexing_technique: IndexingTechnique) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.indexing_technique = indexing_technique
+        return self
+
+    def process_rule(self, process_rule: ProcessRule) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.process_rule = process_rule
+        return self
+
+    def original_document_id(self, original_document_id: str) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.original_document_id = original_document_id
+        return self
+
+    def doc_form(self, doc_form: DocumentForm) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.doc_form = doc_form
+        return self
+
+    def doc_language(self, doc_language: str) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.doc_language = doc_language
+        return self
+
+    def retrieval_model(self, retrieval_model: RetrievalModel) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.retrieval_model = retrieval_model
+        return self
+
+    def embedding_model(self, embedding_model: str) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.embedding_model = embedding_model
+        return self
+
+    def embedding_model_provider(self, embedding_model_provider: str) -> CreateDocumentByFileRequestBodyDataBuilder:
+        self._create_document_by_file_request_body.embedding_model_provider = embedding_model_provider
+        return self

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

@@ -0,0 +1,23 @@
+"""Data source detail models."""
+
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class UploadFileDetail(BaseModel):
+    """Upload file detail information."""
+
+    id: Optional[str] = None
+    name: Optional[str] = None
+    size: Optional[int] = None
+    extension: Optional[str] = None
+    mime_type: Optional[str] = None
+    created_by: Optional[str] = None
+    created_at: Optional[float] = None
+
+
+class DataSourceDetailDict(BaseModel):
+    """Data source detail dictionary."""
+
+    upload_file: Optional[UploadFileDetail] = None

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

@@ -4,9 +4,10 @@ from typing import Optional
 
 from pydantic import BaseModel
 
-from .document_metadata import DocumentMetadata
+from .dataset_metadata import DatasetMetadata
 from .external_knowledge_info import ExternalKnowledgeInfo
-from .knowledge_types import DataSourceType, IndexingTechnique, Permission
+from .external_retrieval_model import ExternalRetrievalModel
+from .knowledge_types import DataSourceType, DocumentForm, IndexingTechnique, Permission
 from .retrieval_model import RetrievalModel
 from .tag_info import TagInfo
 
@@ -25,18 +26,18 @@ class DatasetInfo(BaseModel):
     document_count: Optional[int] = None
     word_count: Optional[int] = None
     created_by: Optional[str] = None
-    created_at: Optional[float] = None
+    created_at: Optional[int] = None
     updated_by: Optional[str] = None
-    updated_at: Optional[float] = None
+    updated_at: Optional[int] = None
     embedding_model: Optional[str] = None
     embedding_model_provider: Optional[str] = None
     embedding_available: Optional[bool] = None
     retrieval_model_dict: Optional[RetrievalModel] = None
     tags: Optional[list[TagInfo]] = None
-    doc_form: Optional[str] = None
+    doc_form: Optional[DocumentForm] = None
     external_knowledge_info: Optional[ExternalKnowledgeInfo] = None
-    external_retrieval_model: Optional[RetrievalModel] = None
-    doc_metadata: Optional[list[DocumentMetadata]] = None
+    external_retrieval_model: Optional[ExternalRetrievalModel] = None
+    doc_metadata: Optional[list[DatasetMetadata]] = None
     built_in_field_enabled: Optional[bool] = None
 
     @staticmethod
@@ -89,7 +90,7 @@ class DatasetInfoBuilder:
         self._dataset_info.created_by = created_by
         return self
 
-    def created_at(self, created_at: float) -> "DatasetInfoBuilder":
+    def created_at(self, created_at: int) -> "DatasetInfoBuilder":
         self._dataset_info.created_at = created_at
         return self
 
@@ -97,7 +98,7 @@ class DatasetInfoBuilder:
         self._dataset_info.updated_by = updated_by
         return self
 
-    def updated_at(self, updated_at: float) -> "DatasetInfoBuilder":
+    def updated_at(self, updated_at: int) -> "DatasetInfoBuilder":
         self._dataset_info.updated_at = updated_at
         return self
 

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

@@ -0,0 +1,14 @@
+"""Dataset metadata model."""
+
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class DatasetMetadata(BaseModel):
+    """Dataset metadata information."""
+
+    # Based on API response, this is typically an empty array
+    # Can be extended when actual metadata structure is known
+    key: Optional[str] = None
+    value: Optional[str] = None

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

@@ -4,7 +4,9 @@ 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
 
 
@@ -15,23 +17,25 @@ class DocumentInfo(BaseModel):
     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[float] = 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[float] = 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
-    updated_at: Optional[float] = None
+    doc_metadata: Optional[DocumentMetadata] = None
+    updated_at: Optional[int] = None
 
     @staticmethod
     def builder() -> "DocumentInfoBuilder":
@@ -63,6 +67,10 @@ class 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
@@ -79,7 +87,7 @@ class DocumentInfoBuilder:
         self._document_info.created_by = created_by
         return self
 
-    def created_at(self, created_at: float) -> "DocumentInfoBuilder":
+    def created_at(self, created_at: int) -> "DocumentInfoBuilder":
         self._document_info.created_at = created_at
         return self
 
@@ -99,7 +107,7 @@ class DocumentInfoBuilder:
         self._document_info.enabled = enabled
         return self
 
-    def disabled_at(self, disabled_at: float) -> "DocumentInfoBuilder":
+    def disabled_at(self, disabled_at: int) -> "DocumentInfoBuilder":
         self._document_info.disabled_at = disabled_at
         return self
 
@@ -127,6 +135,10 @@ class DocumentInfoBuilder:
         self._document_info.doc_form = doc_form
         return self
 
-    def updated_at(self, updated_at: float) -> "DocumentInfoBuilder":
+    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/external_knowledge_info.py

@@ -8,8 +8,10 @@ from pydantic import BaseModel
 class ExternalKnowledgeInfo(BaseModel):
     """External knowledge info model with builder pattern."""
 
-    external_knowledge_api_id: Optional[str] = None
     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":
@@ -32,3 +34,11 @@ class ExternalKnowledgeInfoBuilder:
     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":
+        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":
+        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/knowledge_types.py

@@ -15,7 +15,7 @@ SearchMethod = Literal["hybrid_search", "semantic_search", "full_text_search", "
 DocumentStatus = Literal["indexing", "completed", "error", "paused"]
 
 # Processing mode types
-ProcessingMode = Literal["automatic", "custom"]
+ProcessingMode = Literal["automatic", "custom", "hierarchical"]
 
 # File types
 FileType = Literal["document", "image", "audio", "video", "custom"]
@@ -79,7 +79,7 @@ ExternalKnowledgeProvider = Literal["external_api", "notion", "web_crawler"]
 DocumentCreatedFrom = Literal["api", "web", "upload"]
 
 # Document display status types
-DocumentDisplayStatus = Literal["available", "indexing", "error", "paused", "archived"]
+DocumentDisplayStatus = Literal["available", "indexing", "error", "paused", "archived", "queuing"]
 
 # Batch processing status types
 BatchProcessingStatus = Literal["processing", "completed", "failed", "cancelled"]

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

@@ -6,7 +6,6 @@ from pydantic import BaseModel
 
 from .knowledge_types import ProcessingMode
 from .process_rules import ProcessRules
-from .segmentation_rule import SegmentationRule
 
 
 class ProcessRule(BaseModel):
@@ -14,7 +13,6 @@ class ProcessRule(BaseModel):
 
     mode: Optional[ProcessingMode] = None
     rules: Optional[ProcessRules] = None
-    segmentation: Optional[SegmentationRule] = None
 
     @staticmethod
     def builder() -> "ProcessRuleBuilder":
@@ -37,7 +35,3 @@ class ProcessRuleBuilder:
     def rules(self, rules: ProcessRules) -> "ProcessRuleBuilder":
         self._process_rule.rules = rules
         return self
-
-    def segmentation(self, segmentation: SegmentationRule) -> "ProcessRuleBuilder":
-        self._process_rule.segmentation = segmentation
-        return self