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

dify_oapi/api/chat/v1/model/usage_info.py

@@ -0,0 +1,84 @@
+"""Usage information model for Chat API."""
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class UsageInfo(BaseModel):
+    """Usage information model."""
+
+    prompt_tokens: int | None = None
+    prompt_unit_price: str | None = None
+    prompt_price_unit: str | None = None
+    prompt_price: str | None = None
+    completion_tokens: int | None = None
+    completion_unit_price: str | None = None
+    completion_price_unit: str | None = None
+    completion_price: str | None = None
+    total_tokens: int | None = None
+    total_price: str | None = None
+    currency: str | None = None
+    latency: float | None = None
+
+    @staticmethod
+    def builder() -> UsageInfoBuilder:
+        return UsageInfoBuilder()
+
+
+class UsageInfoBuilder:
+    """Builder for UsageInfo."""
+
+    def __init__(self):
+        self._usage_info = UsageInfo()
+
+    def build(self) -> UsageInfo:
+        return self._usage_info
+
+    def prompt_tokens(self, prompt_tokens: int) -> UsageInfoBuilder:
+        self._usage_info.prompt_tokens = prompt_tokens
+        return self
+
+    def prompt_unit_price(self, prompt_unit_price: str) -> UsageInfoBuilder:
+        self._usage_info.prompt_unit_price = prompt_unit_price
+        return self
+
+    def prompt_price_unit(self, prompt_price_unit: str) -> UsageInfoBuilder:
+        self._usage_info.prompt_price_unit = prompt_price_unit
+        return self
+
+    def prompt_price(self, prompt_price: str) -> UsageInfoBuilder:
+        self._usage_info.prompt_price = prompt_price
+        return self
+
+    def completion_tokens(self, completion_tokens: int) -> UsageInfoBuilder:
+        self._usage_info.completion_tokens = completion_tokens
+        return self
+
+    def completion_unit_price(self, completion_unit_price: str) -> UsageInfoBuilder:
+        self._usage_info.completion_unit_price = completion_unit_price
+        return self
+
+    def completion_price_unit(self, completion_price_unit: str) -> UsageInfoBuilder:
+        self._usage_info.completion_price_unit = completion_price_unit
+        return self
+
+    def completion_price(self, completion_price: str) -> UsageInfoBuilder:
+        self._usage_info.completion_price = completion_price
+        return self
+
+    def total_tokens(self, total_tokens: int) -> UsageInfoBuilder:
+        self._usage_info.total_tokens = total_tokens
+        return self
+
+    def total_price(self, total_price: str) -> UsageInfoBuilder:
+        self._usage_info.total_price = total_price
+        return self
+
+    def currency(self, currency: str) -> UsageInfoBuilder:
+        self._usage_info.currency = currency
+        return self
+
+    def latency(self, latency: float) -> UsageInfoBuilder:
+        self._usage_info.latency = latency
+        return self

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

@@ -1,4 +1,8 @@
+from .annotation import *  # noqa F403
+from .app import *  # noqa F403
+from .audio import *  # noqa F403
 from .chat import *  # noqa F403
-from .conversation import *  # noqa 403
-from .message import *  # noqa 403
-from .audio import *  # noqa 403
+from .conversation import *  # noqa F403
+from .feedback import *  # noqa F403
+from .file import *  # noqa F403
+from .message import *  # noqa F403

dify_oapi/api/chat/v1/resource/annotation.py

@@ -0,0 +1,77 @@
+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.configure_annotation_reply_request import ConfigureAnnotationReplyRequest
+from ..model.configure_annotation_reply_response import ConfigureAnnotationReplyResponse
+from ..model.create_annotation_request import CreateAnnotationRequest
+from ..model.create_annotation_response import CreateAnnotationResponse
+from ..model.delete_annotation_request import DeleteAnnotationRequest
+from ..model.delete_annotation_response import DeleteAnnotationResponse
+from ..model.get_annotation_reply_status_request import GetAnnotationReplyStatusRequest
+from ..model.get_annotation_reply_status_response import GetAnnotationReplyStatusResponse
+from ..model.list_annotations_request import ListAnnotationsRequest
+from ..model.list_annotations_response import ListAnnotationsResponse
+from ..model.update_annotation_request import UpdateAnnotationRequest
+from ..model.update_annotation_response import UpdateAnnotationResponse
+
+
+class Annotation:
+    def __init__(self, config: Config) -> None:
+        self.config: Config = config
+
+    def list(self, request: ListAnnotationsRequest, option: RequestOption | None = None) -> ListAnnotationsResponse:
+        return Transport.execute(self.config, request, unmarshal_as=ListAnnotationsResponse, option=option)
+
+    async def alist(
+        self, request: ListAnnotationsRequest, option: RequestOption | None = None
+    ) -> ListAnnotationsResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=ListAnnotationsResponse, option=option)
+
+    def create(self, request: CreateAnnotationRequest, option: RequestOption | None = None) -> CreateAnnotationResponse:
+        return Transport.execute(self.config, request, unmarshal_as=CreateAnnotationResponse, option=option)
+
+    async def acreate(
+        self, request: CreateAnnotationRequest, option: RequestOption | None = None
+    ) -> CreateAnnotationResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=CreateAnnotationResponse, option=option)
+
+    def update(self, request: UpdateAnnotationRequest, option: RequestOption | None = None) -> UpdateAnnotationResponse:
+        return Transport.execute(self.config, request, unmarshal_as=UpdateAnnotationResponse, option=option)
+
+    async def aupdate(
+        self, request: UpdateAnnotationRequest, option: RequestOption | None = None
+    ) -> UpdateAnnotationResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=UpdateAnnotationResponse, option=option)
+
+    def delete(self, request: DeleteAnnotationRequest, option: RequestOption | None = None) -> DeleteAnnotationResponse:
+        return Transport.execute(self.config, request, unmarshal_as=DeleteAnnotationResponse, option=option)
+
+    async def adelete(
+        self, request: DeleteAnnotationRequest, option: RequestOption | None = None
+    ) -> DeleteAnnotationResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=DeleteAnnotationResponse, option=option)
+
+    def configure(
+        self, request: ConfigureAnnotationReplyRequest, option: RequestOption | None = None
+    ) -> ConfigureAnnotationReplyResponse:
+        return Transport.execute(self.config, request, unmarshal_as=ConfigureAnnotationReplyResponse, option=option)
+
+    async def aconfigure(
+        self, request: ConfigureAnnotationReplyRequest, option: RequestOption | None = None
+    ) -> ConfigureAnnotationReplyResponse:
+        return await ATransport.aexecute(
+            self.config, request, unmarshal_as=ConfigureAnnotationReplyResponse, option=option
+        )
+
+    def status(
+        self, request: GetAnnotationReplyStatusRequest, option: RequestOption | None = None
+    ) -> GetAnnotationReplyStatusResponse:
+        return Transport.execute(self.config, request, unmarshal_as=GetAnnotationReplyStatusResponse, option=option)
+
+    async def astatus(
+        self, request: GetAnnotationReplyStatusRequest, option: RequestOption | None = None
+    ) -> GetAnnotationReplyStatusResponse:
+        return await ATransport.aexecute(
+            self.config, request, unmarshal_as=GetAnnotationReplyStatusResponse, option=option
+        )

dify_oapi/api/chat/v1/resource/app.py

@@ -0,0 +1,47 @@
+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_app_info_request import GetAppInfoRequest
+from ..model.get_app_info_response import GetAppInfoResponse
+from ..model.get_app_meta_request import GetAppMetaRequest
+from ..model.get_app_meta_response import GetAppMetaResponse
+from ..model.get_app_parameters_request import GetAppParametersRequest
+from ..model.get_app_parameters_response import GetAppParametersResponse
+from ..model.get_site_settings_request import GetSiteSettingsRequest
+from ..model.get_site_settings_response import GetSiteSettingsResponse
+
+
+class App:
+    def __init__(self, config: Config) -> None:
+        self.config: Config = config
+
+    def info(self, request: GetAppInfoRequest, option: RequestOption | None = None) -> GetAppInfoResponse:
+        return Transport.execute(self.config, request, unmarshal_as=GetAppInfoResponse, option=option)
+
+    async def ainfo(self, request: GetAppInfoRequest, option: RequestOption | None = None) -> GetAppInfoResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetAppInfoResponse, option=option)
+
+    def parameters(
+        self, request: GetAppParametersRequest, option: RequestOption | None = None
+    ) -> GetAppParametersResponse:
+        return Transport.execute(self.config, request, unmarshal_as=GetAppParametersResponse, option=option)
+
+    async def aparameters(
+        self, request: GetAppParametersRequest, option: RequestOption | None = None
+    ) -> GetAppParametersResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetAppParametersResponse, option=option)
+
+    def meta(self, request: GetAppMetaRequest, option: RequestOption | None = None) -> GetAppMetaResponse:
+        return Transport.execute(self.config, request, unmarshal_as=GetAppMetaResponse, option=option)
+
+    async def ameta(self, request: GetAppMetaRequest, option: RequestOption | None = None) -> GetAppMetaResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetAppMetaResponse, option=option)
+
+    def site(self, request: GetSiteSettingsRequest, option: RequestOption | None = None) -> GetSiteSettingsResponse:
+        return Transport.execute(self.config, request, unmarshal_as=GetSiteSettingsResponse, option=option)
+
+    async def asite(
+        self, request: GetSiteSettingsRequest, option: RequestOption | None = None
+    ) -> GetSiteSettingsResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetSiteSettingsResponse, option=option)

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

@@ -4,6 +4,7 @@ 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
 
 
 class Audio:
@@ -15,3 +16,15 @@ class Audio:
 
     async def ato_text(self, request: AudioToTextRequest, option: RequestOption | None = None) -> AudioToTextResponse:
         return await ATransport.aexecute(self.config, request, unmarshal_as=AudioToTextResponse, option=option)
+
+    def to_audio(self, request: TextToAudioRequest, option: RequestOption | None = None) -> bytes:
+        response = Transport.execute(self.config, request, option=option)
+        if hasattr(response, "content") and response.content is not None:
+            return bytes(response.content)
+        return b""
+
+    async def ato_audio(self, request: TextToAudioRequest, option: RequestOption | None = None) -> bytes:
+        response = await ATransport.aexecute(self.config, request, option=option)
+        if hasattr(response, "content") and response.content is not None:
+            return bytes(response.content)
+        return b""

dify_oapi/api/chat/v1/resource/chat.py

@@ -7,6 +7,8 @@ from dify_oapi.core.model.request_option import RequestOption
 
 from ..model.chat_request import ChatRequest
 from ..model.chat_response import ChatResponse
+from ..model.get_suggested_questions_request import GetSuggestedQuestionsRequest
+from ..model.get_suggested_questions_response import GetSuggestedQuestionsResponse
 from ..model.stop_chat_request import StopChatRequest
 from ..model.stop_chat_response import StopChatResponse
 
@@ -66,3 +68,15 @@ class Chat:
 
     async def astop(self, request: StopChatRequest, option: RequestOption | None = None) -> StopChatResponse:
         return await ATransport.aexecute(self.config, request, unmarshal_as=StopChatResponse, option=option)
+
+    def suggested(
+        self, request: GetSuggestedQuestionsRequest, option: RequestOption | None = None
+    ) -> GetSuggestedQuestionsResponse:
+        return Transport.execute(self.config, request, unmarshal_as=GetSuggestedQuestionsResponse, option=option)
+
+    async def asuggested(
+        self, request: GetSuggestedQuestionsRequest, option: RequestOption | None = None
+    ) -> GetSuggestedQuestionsResponse:
+        return await ATransport.aexecute(
+            self.config, request, unmarshal_as=GetSuggestedQuestionsResponse, option=option
+        )

dify_oapi/api/chat/v1/resource/conversation.py

@@ -4,8 +4,12 @@ from dify_oapi.core.model.request_option import RequestOption
 
 from ..model.delete_conversation_request import DeleteConversationRequest
 from ..model.delete_conversation_response import DeleteConversationResponse
-from ..model.get_conversation_list_request import GetConversationListRequest
-from ..model.get_conversation_list_response import GetConversationListResponse
+from ..model.get_conversation_list_request import GetConversationsRequest
+from ..model.get_conversation_list_response import GetConversationsResponse
+from ..model.get_conversation_variables_request import GetConversationVariablesRequest
+from ..model.get_conversation_variables_response import GetConversationVariablesResponse
+from ..model.message_history_request import GetMessageHistoryRequest
+from ..model.message_history_response import GetMessageHistoryResponse
 from ..model.rename_conversation_request import RenameConversationRequest
 from ..model.rename_conversation_response import RenameConversationResponse
 
@@ -14,25 +18,67 @@ class Conversation:
     def __init__(self, config: Config) -> None:
         self.config: Config = config
 
-    def list(
-        self, request: GetConversationListRequest, option: RequestOption | None = None
-    ) -> GetConversationListResponse:
+    def list(self, request: GetConversationsRequest, option: RequestOption | None = None) -> GetConversationsResponse:
         # Send request
         return Transport.execute(
             self.config,
             request,
-            unmarshal_as=GetConversationListResponse,
+            unmarshal_as=GetConversationsResponse,
             option=option,
         )
 
     async def alist(
-        self, request: GetConversationListRequest, option: RequestOption | None = None
-    ) -> GetConversationListResponse:
+        self, request: GetConversationsRequest, option: RequestOption | None = None
+    ) -> GetConversationsResponse:
         # Send request
         return await ATransport.aexecute(
             self.config,
             request,
-            unmarshal_as=GetConversationListResponse,
+            unmarshal_as=GetConversationsResponse,
+            option=option,
+        )
+
+    def history(
+        self, request: GetMessageHistoryRequest, option: RequestOption | None = None
+    ) -> GetMessageHistoryResponse:
+        # Send request
+        return Transport.execute(
+            self.config,
+            request,
+            unmarshal_as=GetMessageHistoryResponse,
+            option=option,
+        )
+
+    async def ahistory(
+        self, request: GetMessageHistoryRequest, option: RequestOption | None = None
+    ) -> GetMessageHistoryResponse:
+        # Send request
+        return await ATransport.aexecute(
+            self.config,
+            request,
+            unmarshal_as=GetMessageHistoryResponse,
+            option=option,
+        )
+
+    def variables(
+        self, request: GetConversationVariablesRequest, option: RequestOption | None = None
+    ) -> GetConversationVariablesResponse:
+        # Send request
+        return Transport.execute(
+            self.config,
+            request,
+            unmarshal_as=GetConversationVariablesResponse,
+            option=option,
+        )
+
+    async def avariables(
+        self, request: GetConversationVariablesRequest, option: RequestOption | None = None
+    ) -> GetConversationVariablesResponse:
+        # Send request
+        return await ATransport.aexecute(
+            self.config,
+            request,
+            unmarshal_as=GetConversationVariablesResponse,
             option=option,
         )
 

dify_oapi/api/chat/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):
+        self.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 asynchronously."""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=SubmitFeedbackResponse, option=option)
+
+    def list(self, request: GetFeedbacksRequest, option: RequestOption | None = None) -> GetFeedbacksResponse:
+        """Get application feedbacks."""
+        return Transport.execute(self.config, request, unmarshal_as=GetFeedbacksResponse, option=option)
+
+    async def alist(self, request: GetFeedbacksRequest, option: RequestOption | None = None) -> GetFeedbacksResponse:
+        """Get application feedbacks asynchronously."""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetFeedbacksResponse, option=option)

dify_oapi/api/chat/v1/resource/file.py

@@ -0,0 +1,25 @@
+"""File resource for chat API."""
+
+from __future__ import annotations
+
+from dify_oapi.api.chat.v1.model.upload_file_request import UploadFileRequest
+from dify_oapi.api.chat.v1.model.upload_file_response import UploadFileResponse
+from dify_oapi.core.http.transport import Transport
+from dify_oapi.core.http.transport.async_transport import ATransport
+from dify_oapi.core.model.config import Config
+from dify_oapi.core.model.request_option import RequestOption
+
+
+class File:
+    """File resource for handling file upload operations."""
+
+    def __init__(self, config: Config) -> None:
+        self.config = config
+
+    def upload(self, request: UploadFileRequest, option: RequestOption | None = None) -> UploadFileResponse:
+        """Upload a file for use in chat messages."""
+        return Transport.execute(self.config, request, unmarshal_as=UploadFileResponse, option=option)
+
+    async def aupload(self, request: UploadFileRequest, option: RequestOption | None = None) -> UploadFileResponse:
+        """Upload a file for use in chat messages (async)."""
+        return await ATransport.aexecute(self.config, request, unmarshal_as=UploadFileResponse, option=option)

dify_oapi/api/chat/v1/resource/message.py

@@ -2,10 +2,10 @@ 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.message_history_request import MessageHistoryRequest
-from ..model.message_history_response import MessageHistoryResponse
-from ..model.message_suggested_request import MessageSuggestedRequest
-from ..model.message_suggested_response import MessageSuggestedResponse
+from ..model.get_suggested_questions_request import GetSuggestedQuestionsRequest
+from ..model.get_suggested_questions_response import GetSuggestedQuestionsResponse
+from ..model.message_history_request import GetMessageHistoryRequest
+from ..model.message_history_response import GetMessageHistoryResponse
 
 
 class Message:
@@ -13,23 +13,27 @@ class Message:
         self.config: Config = config
 
     def suggested(
-        self, request: MessageSuggestedRequest, option: RequestOption | None = None
-    ) -> MessageSuggestedResponse:
+        self, request: GetSuggestedQuestionsRequest, option: RequestOption | None = None
+    ) -> GetSuggestedQuestionsResponse:
         # Send request
-        return Transport.execute(self.config, request, unmarshal_as=MessageSuggestedResponse, option=option)
+        return Transport.execute(self.config, request, unmarshal_as=GetSuggestedQuestionsResponse, option=option)
 
     async def asuggested(
-        self, request: MessageSuggestedRequest, option: RequestOption | None = None
-    ) -> MessageSuggestedResponse:
+        self, request: GetSuggestedQuestionsRequest, option: RequestOption | None = None
+    ) -> GetSuggestedQuestionsResponse:
         # Send request
-        return await ATransport.aexecute(self.config, request, unmarshal_as=MessageSuggestedResponse, option=option)
+        return await ATransport.aexecute(
+            self.config, request, unmarshal_as=GetSuggestedQuestionsResponse, option=option
+        )
 
-    def history(self, request: MessageHistoryRequest, option: RequestOption | None = None) -> MessageHistoryResponse:
+    def history(
+        self, request: GetMessageHistoryRequest, option: RequestOption | None = None
+    ) -> GetMessageHistoryResponse:
         # Send request
-        return Transport.execute(self.config, request, unmarshal_as=MessageHistoryResponse, option=option)
+        return Transport.execute(self.config, request, unmarshal_as=GetMessageHistoryResponse, option=option)
 
     async def ahistory(
-        self, request: MessageHistoryRequest, option: RequestOption | None = None
-    ) -> MessageHistoryResponse:
+        self, request: GetMessageHistoryRequest, option: RequestOption | None = None
+    ) -> GetMessageHistoryResponse:
         # Send request
-        return await ATransport.aexecute(self.config, request, unmarshal_as=MessageHistoryResponse, option=option)
+        return await ATransport.aexecute(self.config, request, unmarshal_as=GetMessageHistoryResponse, option=option)

dify_oapi/api/chat/v1/version.py

@@ -1,11 +1,17 @@
 from dify_oapi.core.model.config import Config
 
-from .resource import Audio, Chat, Conversation, Message
+from .resource import Annotation, App, Audio, Chat, Conversation, Feedback, File, Message
 
 
 class V1:
     def __init__(self, config: Config):
         self.chat: Chat = Chat(config)
+        self.file: File = File(config)
+        self.feedback: Feedback = Feedback(config)
         self.conversation: Conversation = Conversation(config)
-        self.message: Message = Message(config)
         self.audio: Audio = Audio(config)
+        self.app: App = App(config)
+        self.annotation: Annotation = Annotation(config)
+
+        # DEPRECATED: Keep for backward compatibility
+        self.message: Message = Message(config)

dify_oapi/api/chatflow/service.py

@@ -0,0 +1,25 @@
+from dify_oapi.core.model.config import Config
+
+from .v1.version import V1
+
+
+class ChatflowService:
+    """Chatflow API service providing access to all chatflow-related operations.
+
+    This service provides access to 17 chatflow APIs across 6 resource categories:
+    - Chatflow: 3 APIs (send, stop, suggested)
+    - File: 1 API (upload)
+    - Feedback: 2 APIs (message, list)
+    - Conversation: 5 APIs (messages, list, delete, rename, variables)
+    - TTS: 2 APIs (speech_to_text, text_to_audio)
+    - Application: 4 APIs (info, parameters, meta, site)
+    - Annotation: 6 APIs (list, create, update, delete, reply_settings, reply_status)
+    """
+
+    def __init__(self, config: Config):
+        """Initialize the Chatflow service with configuration.
+
+        Args:
+            config: The configuration object containing API settings
+        """
+        self.v1 = V1(config)

dify_oapi/api/chatflow/v1/model/annotation_info.py

@@ -0,0 +1,43 @@
+from pydantic import BaseModel
+
+
+class AnnotationInfo(BaseModel):
+    """Annotation information model."""
+
+    id: str | None = None
+    question: str | None = None
+    answer: str | None = None
+    hit_count: int | None = None
+    created_at: int | None = None
+
+    @staticmethod
+    def builder() -> "AnnotationInfoBuilder":
+        return AnnotationInfoBuilder()
+
+
+class AnnotationInfoBuilder:
+    def __init__(self):
+        self._annotation_info = AnnotationInfo()
+
+    def build(self) -> AnnotationInfo:
+        return self._annotation_info
+
+    def id(self, id: str) -> "AnnotationInfoBuilder":
+        self._annotation_info.id = id
+        return self
+
+    def question(self, question: str) -> "AnnotationInfoBuilder":
+        self._annotation_info.question = question
+        return self
+
+    def answer(self, answer: str) -> "AnnotationInfoBuilder":
+        self._annotation_info.answer = answer
+        return self
+
+    def hit_count(self, hit_count: int) -> "AnnotationInfoBuilder":
+        self._annotation_info.hit_count = hit_count
+        return self
+
+    def created_at(self, created_at: int) -> "AnnotationInfoBuilder":
+        self._annotation_info.created_at = created_at
+        return self

dify_oapi/api/chatflow/v1/model/annotation_reply_settings_request.py

@@ -0,0 +1,37 @@
+from dify_oapi.core.enum import HttpMethod
+from dify_oapi.core.model.base_request import BaseRequest
+
+from .annotation_reply_settings_request_body import AnnotationReplySettingsRequestBody
+from .chatflow_types import AnnotationAction
+
+
+class AnnotationReplySettingsRequest(BaseRequest):
+    def __init__(self):
+        super().__init__()
+        self.request_body: AnnotationReplySettingsRequestBody | None = None
+        self.action: AnnotationAction | None = None
+
+    @staticmethod
+    def builder() -> "AnnotationReplySettingsRequestBuilder":
+        return AnnotationReplySettingsRequestBuilder()
+
+
+class AnnotationReplySettingsRequestBuilder:
+    def __init__(self):
+        annotation_reply_settings_request = AnnotationReplySettingsRequest()
+        annotation_reply_settings_request.http_method = HttpMethod.POST
+        annotation_reply_settings_request.uri = "/v1/apps/annotation-reply/:action"
+        self._annotation_reply_settings_request = annotation_reply_settings_request
+
+    def build(self) -> AnnotationReplySettingsRequest:
+        return self._annotation_reply_settings_request
+
+    def action(self, action: AnnotationAction) -> "AnnotationReplySettingsRequestBuilder":
+        self._annotation_reply_settings_request.action = action
+        self._annotation_reply_settings_request.paths["action"] = action
+        return self
+
+    def request_body(self, request_body: AnnotationReplySettingsRequestBody) -> "AnnotationReplySettingsRequestBuilder":
+        self._annotation_reply_settings_request.request_body = request_body
+        self._annotation_reply_settings_request.body = request_body.model_dump(exclude_none=True, mode="json")
+        return self

dify_oapi/api/chatflow/v1/model/annotation_reply_settings_request_body.py

@@ -0,0 +1,33 @@
+from pydantic import BaseModel
+
+
+class AnnotationReplySettingsRequestBody(BaseModel):
+    """Request body for annotation reply settings."""
+
+    embedding_provider_name: str | None = None
+    embedding_model_name: str | None = None
+    score_threshold: float | None = None
+
+    @staticmethod
+    def builder() -> "AnnotationReplySettingsRequestBodyBuilder":
+        return AnnotationReplySettingsRequestBodyBuilder()
+
+
+class AnnotationReplySettingsRequestBodyBuilder:
+    def __init__(self):
+        self._annotation_reply_settings_request_body = AnnotationReplySettingsRequestBody()
+
+    def build(self) -> AnnotationReplySettingsRequestBody:
+        return self._annotation_reply_settings_request_body
+
+    def embedding_provider_name(self, embedding_provider_name: str) -> "AnnotationReplySettingsRequestBodyBuilder":
+        self._annotation_reply_settings_request_body.embedding_provider_name = embedding_provider_name
+        return self
+
+    def embedding_model_name(self, embedding_model_name: str) -> "AnnotationReplySettingsRequestBodyBuilder":
+        self._annotation_reply_settings_request_body.embedding_model_name = embedding_model_name
+        return self
+
+    def score_threshold(self, score_threshold: float) -> "AnnotationReplySettingsRequestBodyBuilder":
+        self._annotation_reply_settings_request_body.score_threshold = score_threshold
+        return self

dify_oapi/api/chatflow/v1/model/annotation_reply_settings_response.py

@@ -0,0 +1,10 @@
+from dify_oapi.core.model.base_response import BaseResponse
+
+from .chatflow_types import JobStatus
+
+
+class AnnotationReplySettingsResponse(BaseResponse):
+    """Response for annotation reply settings."""
+
+    job_id: str | None = None
+    job_status: JobStatus | None = None