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

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

@@ -1,37 +1,137 @@
-"""Chat API model package."""
-
-# Annotation Management Models
-from .configure_annotation_reply_request import ConfigureAnnotationReplyRequest
-from .configure_annotation_reply_request_body import ConfigureAnnotationReplyRequestBody
-from .configure_annotation_reply_response import ConfigureAnnotationReplyResponse
-from .create_annotation_request import CreateAnnotationRequest
-from .create_annotation_request_body import CreateAnnotationRequestBody
-from .create_annotation_response import CreateAnnotationResponse
-from .delete_annotation_request import DeleteAnnotationRequest
-from .delete_annotation_response import DeleteAnnotationResponse
-from .get_annotation_reply_status_request import GetAnnotationReplyStatusRequest
-from .get_annotation_reply_status_response import GetAnnotationReplyStatusResponse
-from .list_annotations_request import ListAnnotationsRequest
-from .list_annotations_response import ListAnnotationsResponse
-from .update_annotation_request import UpdateAnnotationRequest
-from .update_annotation_request_body import UpdateAnnotationRequestBody
-from .update_annotation_response import UpdateAnnotationResponse
-
-__all__ = [
-    # Annotation Management Models
-    "ConfigureAnnotationReplyRequest",
-    "ConfigureAnnotationReplyRequestBody",
-    "ConfigureAnnotationReplyResponse",
-    "CreateAnnotationRequest",
-    "CreateAnnotationRequestBody",
-    "CreateAnnotationResponse",
-    "DeleteAnnotationRequest",
-    "DeleteAnnotationResponse",
-    "GetAnnotationReplyStatusRequest",
-    "GetAnnotationReplyStatusResponse",
-    "ListAnnotationsRequest",
-    "ListAnnotationsResponse",
-    "UpdateAnnotationRequest",
-    "UpdateAnnotationRequestBody",
-    "UpdateAnnotationResponse",
-]
+from .agent_thought import AgentThought as AgentThought
+from .agent_thought import AgentThoughtBuilder as AgentThoughtBuilder
+from .annotation_info import AnnotationInfo as AnnotationInfo
+from .annotation_info import AnnotationInfoBuilder as AnnotationInfoBuilder
+from .app_info import AppInfo as AppInfo
+from .app_info import AppInfoBuilder as AppInfoBuilder
+from .app_parameters import AnnotationReplyConfig as AnnotationReplyConfig
+from .app_parameters import AnnotationReplyConfigBuilder as AnnotationReplyConfigBuilder
+from .app_parameters import AppParameters as AppParameters
+from .app_parameters import AppParametersBuilder as AppParametersBuilder
+from .app_parameters import FileUploadConfig as FileUploadConfig
+from .app_parameters import FileUploadConfigBuilder as FileUploadConfigBuilder
+from .app_parameters import FileUploadSystemConfig as FileUploadSystemConfig
+from .app_parameters import FileUploadSystemConfigBuilder as FileUploadSystemConfigBuilder
+from .app_parameters import ImageUploadConfig as ImageUploadConfig
+from .app_parameters import ImageUploadConfigBuilder as ImageUploadConfigBuilder
+from .app_parameters import MoreLikeThisConfig as MoreLikeThisConfig
+from .app_parameters import MoreLikeThisConfigBuilder as MoreLikeThisConfigBuilder
+from .app_parameters import ParagraphConfig as ParagraphConfig
+from .app_parameters import RetrieverResourceConfig as RetrieverResourceConfig
+from .app_parameters import RetrieverResourceConfigBuilder as RetrieverResourceConfigBuilder
+from .app_parameters import SelectConfig as SelectConfig
+from .app_parameters import SensitiveWordAvoidanceConfig as SensitiveWordAvoidanceConfig
+from .app_parameters import SensitiveWordAvoidanceConfigBuilder as SensitiveWordAvoidanceConfigBuilder
+from .app_parameters import SpeechToTextConfig as SpeechToTextConfig
+from .app_parameters import SpeechToTextConfigBuilder as SpeechToTextConfigBuilder
+from .app_parameters import SuggestedQuestionsAfterAnswerConfig as SuggestedQuestionsAfterAnswerConfig
+from .app_parameters import SuggestedQuestionsAfterAnswerConfigBuilder as SuggestedQuestionsAfterAnswerConfigBuilder
+from .app_parameters import SystemParameters as SystemParameters
+from .app_parameters import SystemParametersBuilder as SystemParametersBuilder
+from .app_parameters import TextInputConfig as TextInputConfig
+from .app_parameters import TextToSpeechConfig as TextToSpeechConfig
+from .app_parameters import TextToSpeechConfigBuilder as TextToSpeechConfigBuilder
+from .app_parameters import UserInputFormItem as UserInputFormItem
+from .app_parameters import UserInputFormItemBuilder as UserInputFormItemBuilder
+from .chat_file import ChatFile as ChatFile
+from .chat_file import ChatFileBuilder as ChatFileBuilder
+from .chat_request import ChatRequest as ChatRequest
+from .chat_request import ChatRequestBuilder as ChatRequestBuilder
+from .chat_request_body import ChatRequestBody as ChatRequestBody
+from .chat_request_body import ChatRequestBodyBuilder as ChatRequestBodyBuilder
+from .chat_response import ChatResponse as ChatResponse
+from .chat_response import ChatResponseMetadata as ChatResponseMetadata
+from .configure_annotation_reply_request import ConfigureAnnotationReplyRequest as ConfigureAnnotationReplyRequest
+from .configure_annotation_reply_request import (
+    ConfigureAnnotationReplyRequestBuilder as ConfigureAnnotationReplyRequestBuilder,
+)
+from .configure_annotation_reply_request_body import (
+    ConfigureAnnotationReplyRequestBody as ConfigureAnnotationReplyRequestBody,
+)
+from .configure_annotation_reply_request_body import (
+    ConfigureAnnotationReplyRequestBodyBuilder as ConfigureAnnotationReplyRequestBodyBuilder,
+)
+from .configure_annotation_reply_response import ConfigureAnnotationReplyResponse as ConfigureAnnotationReplyResponse
+from .conversation_info import ConversationInfo as ConversationInfo
+from .conversation_info import ConversationInfoBuilder as ConversationInfoBuilder
+from .conversation_variable import ConversationVariable as ConversationVariable
+from .conversation_variable import ConversationVariableBuilder as ConversationVariableBuilder
+from .create_annotation_request import CreateAnnotationRequest as CreateAnnotationRequest
+from .create_annotation_request import CreateAnnotationRequestBuilder as CreateAnnotationRequestBuilder
+from .create_annotation_request_body import CreateAnnotationRequestBody as CreateAnnotationRequestBody
+from .create_annotation_request_body import CreateAnnotationRequestBodyBuilder as CreateAnnotationRequestBodyBuilder
+from .create_annotation_response import CreateAnnotationResponse as CreateAnnotationResponse
+from .delete_annotation_request import DeleteAnnotationRequest as DeleteAnnotationRequest
+from .delete_annotation_request import DeleteAnnotationRequestBuilder as DeleteAnnotationRequestBuilder
+from .delete_annotation_response import DeleteAnnotationResponse as DeleteAnnotationResponse
+from .delete_conversation_request import DeleteConversationRequest as DeleteConversationRequest
+from .delete_conversation_request import DeleteConversationRequestBuilder as DeleteConversationRequestBuilder
+from .delete_conversation_request_body import DeleteConversationRequestBody as DeleteConversationRequestBody
+from .delete_conversation_request_body import (
+    DeleteConversationRequestBodyBuilder as DeleteConversationRequestBodyBuilder,
+)
+from .delete_conversation_response import DeleteConversationResponse as DeleteConversationResponse
+from .feedback_info import FeedbackInfo as FeedbackInfo
+from .feedback_info import FeedbackInfoBuilder as FeedbackInfoBuilder
+from .file_info import FileInfo as FileInfo
+from .file_info import FileInfoBuilder as FileInfoBuilder
+from .get_annotation_reply_status_request import GetAnnotationReplyStatusRequest as GetAnnotationReplyStatusRequest
+from .get_annotation_reply_status_request import (
+    GetAnnotationReplyStatusRequestBuilder as GetAnnotationReplyStatusRequestBuilder,
+)
+from .get_annotation_reply_status_response import GetAnnotationReplyStatusResponse as GetAnnotationReplyStatusResponse
+from .get_conversation_list_request import GetConversationsListRequest as GetConversationsListRequest
+from .get_conversation_list_request import GetConversationsListRequestBuilder as GetConversationsListRequestBuilder
+from .get_conversation_list_response import GetConversationsResponse as GetConversationsResponse
+from .get_conversation_variables_request import GetConversationVariablesRequest as GetConversationVariablesRequest
+from .get_conversation_variables_request import (
+    GetConversationVariablesRequestBuilder as GetConversationVariablesRequestBuilder,
+)
+from .get_conversation_variables_response import GetConversationVariablesResponse as GetConversationVariablesResponse
+from .get_conversations_request import GetConversationsRequest as GetConversationsRequest
+from .get_conversations_request import GetConversationsRequestBuilder as GetConversationsRequestBuilder
+from .get_suggested_questions_request import GetSuggestedQuestionsRequest as GetSuggestedQuestionsRequest
+from .get_suggested_questions_request import GetSuggestedQuestionsRequestBuilder as GetSuggestedQuestionsRequestBuilder
+from .get_suggested_questions_response import GetSuggestedQuestionsResponse as GetSuggestedQuestionsResponse
+from .list_annotations_request import ListAnnotationsRequest as ListAnnotationsRequest
+from .list_annotations_request import ListAnnotationsRequestBuilder as ListAnnotationsRequestBuilder
+from .list_annotations_response import ListAnnotationsResponse as ListAnnotationsResponse
+from .message_file import MessageFile as MessageFile
+from .message_file import MessageFileBuilder as MessageFileBuilder
+from .message_history_request import GetMessageHistoryRequest as GetMessageHistoryRequest
+from .message_history_request import GetMessageHistoryRequestBuilder as GetMessageHistoryRequestBuilder
+from .message_history_response import GetMessageHistoryResponse as GetMessageHistoryResponse
+from .message_info import MessageInfo as MessageInfo
+from .message_info import MessageInfoBuilder as MessageInfoBuilder
+from .pagination_info import PaginationInfo as PaginationInfo
+from .pagination_info import PaginationInfoBuilder as PaginationInfoBuilder
+from .rename_conversation_request import RenameConversationRequest as RenameConversationRequest
+from .rename_conversation_request import RenameConversationRequestBuilder as RenameConversationRequestBuilder
+from .rename_conversation_request_body import RenameConversationRequestBody as RenameConversationRequestBody
+from .rename_conversation_request_body import (
+    RenameConversationRequestBodyBuilder as RenameConversationRequestBodyBuilder,
+)
+from .rename_conversation_response import RenameConversationResponse as RenameConversationResponse
+from .retriever_resource import RetrieverResource as RetrieverResource
+from .retriever_resource import RetrieverResourceBuilder as RetrieverResourceBuilder
+from .site_settings import SiteSettings as SiteSettings
+from .site_settings import SiteSettingsBuilder as SiteSettingsBuilder
+from .stop_chat_request import StopChatRequest as StopChatRequest
+from .stop_chat_request import StopChatRequestBuilder as StopChatRequestBuilder
+from .stop_chat_request_body import StopChatRequestBody as StopChatRequestBody
+from .stop_chat_request_body import StopChatRequestBodyBuilder as StopChatRequestBodyBuilder
+from .stop_chat_response import StopChatResponse as StopChatResponse
+from .text_to_audio_response import TextToAudioResponse as TextToAudioResponse
+from .tool_icon import ToolIcon as ToolIcon
+from .tool_icon import ToolIconBuilder as ToolIconBuilder
+from .tool_icon import ToolIconDetail as ToolIconDetail
+from .tool_icon import ToolIconDetailBuilder as ToolIconDetailBuilder
+from .update_annotation_request import UpdateAnnotationRequest as UpdateAnnotationRequest
+from .update_annotation_request import UpdateAnnotationRequestBuilder as UpdateAnnotationRequestBuilder
+from .update_annotation_request_body import UpdateAnnotationRequestBody as UpdateAnnotationRequestBody
+from .update_annotation_request_body import UpdateAnnotationRequestBodyBuilder as UpdateAnnotationRequestBodyBuilder
+from .update_annotation_response import UpdateAnnotationResponse as UpdateAnnotationResponse
+from .upload_file_request_body import UploadFileRequestBody as UploadFileRequestBody
+from .upload_file_request_body import UploadFileRequestBodyBuilder as UploadFileRequestBodyBuilder
+from .usage_info import UsageInfo as UsageInfo
+from .usage_info import UsageInfoBuilder as UsageInfoBuilder

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

@@ -25,5 +25,6 @@ class ChatResponse(BaseResponse):
 class ChatResponseMetadata(BaseModel):
     """Chat response metadata model."""
 
+    annotation_reply: str | None = None
     usage: UsageInfo | None = None
     retriever_resources: list[RetrieverResource] | None = None

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

@@ -30,6 +30,7 @@ class ConfigureAnnotationReplyRequestBuilder:
         self, request_body: ConfigureAnnotationReplyRequestBody
     ) -> "ConfigureAnnotationReplyRequestBuilder":
         self._configure_annotation_reply_request.request_body = request_body
+        self._configure_annotation_reply_request.body = request_body.model_dump(exclude_none=True, mode="json")
         return self
 
     def build(self) -> ConfigureAnnotationReplyRequest:

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

@@ -23,6 +23,7 @@ class CreateAnnotationRequestBuilder:
 
     def request_body(self, request_body: CreateAnnotationRequestBody) -> "CreateAnnotationRequestBuilder":
         self._create_annotation_request.request_body = request_body
+        self._create_annotation_request.body = request_body.model_dump(exclude_none=True, mode="json")
         return self
 
     def build(self) -> CreateAnnotationRequest:

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

@@ -5,7 +5,7 @@ from dify_oapi.core.enum import HttpMethod
 from dify_oapi.core.model.base_request import BaseRequest
 
 
-class GetConversationsRequest(BaseRequest):
+class GetConversationsListRequest(BaseRequest):
     def __init__(self):
         super().__init__()
         self.user: str | None = None
@@ -14,36 +14,36 @@ class GetConversationsRequest(BaseRequest):
         self.sort_by: SortBy | None = None
 
     @staticmethod
-    def builder() -> GetConversationsRequestBuilder:
-        return GetConversationsRequestBuilder()
+    def builder() -> GetConversationsListRequestBuilder:
+        return GetConversationsListRequestBuilder()
 
 
-class GetConversationsRequestBuilder:
+class GetConversationsListRequestBuilder:
     def __init__(self):
-        get_conversations_request = GetConversationsRequest()
+        get_conversations_request = GetConversationsListRequest()
         get_conversations_request.http_method = HttpMethod.GET
         get_conversations_request.uri = "/v1/conversations"
         self._get_conversations_request = get_conversations_request
 
-    def user(self, user: str) -> GetConversationsRequestBuilder:
+    def user(self, user: str) -> GetConversationsListRequestBuilder:
         self._get_conversations_request.user = user
         self._get_conversations_request.add_query("user", user)
         return self
 
-    def last_id(self, last_id: str) -> GetConversationsRequestBuilder:
+    def last_id(self, last_id: str) -> GetConversationsListRequestBuilder:
         self._get_conversations_request.last_id = last_id
         self._get_conversations_request.add_query("last_id", last_id)
         return self
 
-    def limit(self, limit: int) -> GetConversationsRequestBuilder:
+    def limit(self, limit: int) -> GetConversationsListRequestBuilder:
         self._get_conversations_request.limit = limit
         self._get_conversations_request.add_query("limit", limit)
         return self
 
-    def sort_by(self, sort_by: SortBy) -> GetConversationsRequestBuilder:
+    def sort_by(self, sort_by: SortBy) -> GetConversationsListRequestBuilder:
         self._get_conversations_request.sort_by = sort_by
         self._get_conversations_request.add_query("sort_by", sort_by)
         return self
 
-    def build(self) -> GetConversationsRequest:
+    def build(self) -> GetConversationsListRequest:
         return self._get_conversations_request

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

@@ -0,0 +1,49 @@
+from __future__ import annotations
+
+from dify_oapi.api.chat.v1.model.chat_types import SortBy
+from dify_oapi.core.enum import HttpMethod
+from dify_oapi.core.model.base_request import BaseRequest
+
+
+class GetConversationsRequest(BaseRequest):
+    def __init__(self):
+        super().__init__()
+        self.user: str | None = None
+        self.last_id: str | None = None
+        self.limit: int | None = None
+        self.sort_by: SortBy | None = None
+
+    @staticmethod
+    def builder() -> GetConversationsRequestBuilder:
+        return GetConversationsRequestBuilder()
+
+
+class GetConversationsRequestBuilder:
+    def __init__(self):
+        get_conversations_request = GetConversationsRequest()
+        get_conversations_request.http_method = HttpMethod.GET
+        get_conversations_request.uri = "/v1/conversations"
+        self._get_conversations_request = get_conversations_request
+
+    def user(self, user: str) -> GetConversationsRequestBuilder:
+        self._get_conversations_request.user = user
+        self._get_conversations_request.add_query("user", user)
+        return self
+
+    def last_id(self, last_id: str) -> GetConversationsRequestBuilder:
+        self._get_conversations_request.last_id = last_id
+        self._get_conversations_request.add_query("last_id", last_id)
+        return self
+
+    def limit(self, limit: int) -> GetConversationsRequestBuilder:
+        self._get_conversations_request.limit = limit
+        self._get_conversations_request.add_query("limit", limit)
+        return self
+
+    def sort_by(self, sort_by: SortBy) -> GetConversationsRequestBuilder:
+        self._get_conversations_request.sort_by = sort_by
+        self._get_conversations_request.add_query("sort_by", sort_by)
+        return self
+
+    def build(self) -> GetConversationsRequest:
+        return self._get_conversations_request

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

@@ -27,6 +27,7 @@ class UpdateAnnotationRequestBuilder:
 
     def request_body(self, request_body: UpdateAnnotationRequestBody) -> "UpdateAnnotationRequestBuilder":
         self._update_annotation_request.request_body = request_body
+        self._update_annotation_request.body = request_body.model_dump(exclude_none=True, mode="json")
         return self
 
     def build(self) -> UpdateAnnotationRequest:

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

@@ -1,8 +1 @@
-from .annotation import *  # noqa F403
-from .app import *  # noqa F403
-from .audio import *  # noqa F403
-from .chat import *  # noqa F403
-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

@@ -18,60 +18,70 @@ from ..model.update_annotation_response import UpdateAnnotationResponse
 
 class Annotation:
     def __init__(self, config: Config) -> None:
-        self.config: Config = config
+        self.config = config
 
-    def list(self, request: ListAnnotationsRequest, option: RequestOption | None = None) -> ListAnnotationsResponse:
-        return Transport.execute(self.config, request, unmarshal_as=ListAnnotationsResponse, option=option)
+    def list(self, request: ListAnnotationsRequest, request_option: RequestOption) -> ListAnnotationsResponse:
+        return Transport.execute(self.config, request, unmarshal_as=ListAnnotationsResponse, option=request_option)
 
-    async def alist(
-        self, request: ListAnnotationsRequest, option: RequestOption | None = None
-    ) -> ListAnnotationsResponse:
-        return await ATransport.aexecute(self.config, request, unmarshal_as=ListAnnotationsResponse, option=option)
+    async def alist(self, request: ListAnnotationsRequest, request_option: RequestOption) -> ListAnnotationsResponse:
+        return await ATransport.aexecute(
+            self.config, request, unmarshal_as=ListAnnotationsResponse, option=request_option
+        )
 
-    def create(self, request: CreateAnnotationRequest, option: RequestOption | None = None) -> CreateAnnotationResponse:
-        return Transport.execute(self.config, request, unmarshal_as=CreateAnnotationResponse, option=option)
+    def create(self, request: CreateAnnotationRequest, request_option: RequestOption) -> CreateAnnotationResponse:
+        return Transport.execute(self.config, request, unmarshal_as=CreateAnnotationResponse, option=request_option)
 
     async def acreate(
-        self, request: CreateAnnotationRequest, option: RequestOption | None = None
+        self, request: CreateAnnotationRequest, request_option: RequestOption
     ) -> CreateAnnotationResponse:
-        return await ATransport.aexecute(self.config, request, unmarshal_as=CreateAnnotationResponse, option=option)
+        return await ATransport.aexecute(
+            self.config, request, unmarshal_as=CreateAnnotationResponse, option=request_option
+        )
 
-    def update(self, request: UpdateAnnotationRequest, option: RequestOption | None = None) -> UpdateAnnotationResponse:
-        return Transport.execute(self.config, request, unmarshal_as=UpdateAnnotationResponse, option=option)
+    def update(self, request: UpdateAnnotationRequest, request_option: RequestOption) -> UpdateAnnotationResponse:
+        return Transport.execute(self.config, request, unmarshal_as=UpdateAnnotationResponse, option=request_option)
 
     async def aupdate(
-        self, request: UpdateAnnotationRequest, option: RequestOption | None = None
+        self, request: UpdateAnnotationRequest, request_option: RequestOption
     ) -> UpdateAnnotationResponse:
-        return await ATransport.aexecute(self.config, request, unmarshal_as=UpdateAnnotationResponse, option=option)
+        return await ATransport.aexecute(
+            self.config, request, unmarshal_as=UpdateAnnotationResponse, option=request_option
+        )
 
-    def delete(self, request: DeleteAnnotationRequest, option: RequestOption | None = None) -> DeleteAnnotationResponse:
-        return Transport.execute(self.config, request, unmarshal_as=DeleteAnnotationResponse, option=option)
+    def delete(self, request: DeleteAnnotationRequest, request_option: RequestOption) -> DeleteAnnotationResponse:
+        return Transport.execute(self.config, request, unmarshal_as=DeleteAnnotationResponse, option=request_option)
 
     async def adelete(
-        self, request: DeleteAnnotationRequest, option: RequestOption | None = None
+        self, request: DeleteAnnotationRequest, request_option: RequestOption
     ) -> DeleteAnnotationResponse:
-        return await ATransport.aexecute(self.config, request, unmarshal_as=DeleteAnnotationResponse, option=option)
+        return await ATransport.aexecute(
+            self.config, request, unmarshal_as=DeleteAnnotationResponse, option=request_option
+        )
 
     def configure(
-        self, request: ConfigureAnnotationReplyRequest, option: RequestOption | None = None
+        self, request: ConfigureAnnotationReplyRequest, request_option: RequestOption
     ) -> ConfigureAnnotationReplyResponse:
-        return Transport.execute(self.config, request, unmarshal_as=ConfigureAnnotationReplyResponse, option=option)
+        return Transport.execute(
+            self.config, request, unmarshal_as=ConfigureAnnotationReplyResponse, option=request_option
+        )
 
     async def aconfigure(
-        self, request: ConfigureAnnotationReplyRequest, option: RequestOption | None = None
+        self, request: ConfigureAnnotationReplyRequest, request_option: RequestOption
     ) -> ConfigureAnnotationReplyResponse:
         return await ATransport.aexecute(
-            self.config, request, unmarshal_as=ConfigureAnnotationReplyResponse, option=option
+            self.config, request, unmarshal_as=ConfigureAnnotationReplyResponse, option=request_option
         )
 
     def status(
-        self, request: GetAnnotationReplyStatusRequest, option: RequestOption | None = None
+        self, request: GetAnnotationReplyStatusRequest, request_option: RequestOption
     ) -> GetAnnotationReplyStatusResponse:
-        return Transport.execute(self.config, request, unmarshal_as=GetAnnotationReplyStatusResponse, option=option)
+        return Transport.execute(
+            self.config, request, unmarshal_as=GetAnnotationReplyStatusResponse, option=request_option
+        )
 
     async def astatus(
-        self, request: GetAnnotationReplyStatusRequest, option: RequestOption | None = None
+        self, request: GetAnnotationReplyStatusRequest, request_option: RequestOption
     ) -> GetAnnotationReplyStatusResponse:
         return await ATransport.aexecute(
-            self.config, request, unmarshal_as=GetAnnotationReplyStatusResponse, option=option
+            self.config, request, unmarshal_as=GetAnnotationReplyStatusResponse, option=request_option
         )

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

@@ -15,68 +15,76 @@ from ..model.stop_chat_response import StopChatResponse
 
 class Chat:
     def __init__(self, config: Config) -> None:
-        self.config: Config = config
+        self.config = config
 
     @overload
     def chat(
-        self, request: ChatRequest, option: RequestOption | None, stream: Literal[True]
+        self,
+        request: ChatRequest,
+        request_option: RequestOption,
+        stream: Literal[True],
     ) -> Generator[bytes, None, None]: ...
 
     @overload
-    def chat(self, request: ChatRequest, option: RequestOption | None, stream: Literal[False]) -> ChatResponse: ...
-
-    @overload
-    def chat(self, request: ChatRequest, option: RequestOption | None) -> ChatResponse: ...
+    def chat(
+        self,
+        request: ChatRequest,
+        request_option: RequestOption,
+        stream: Literal[False] = False,
+    ) -> ChatResponse: ...
 
     def chat(
         self,
         request: ChatRequest,
-        option: RequestOption | None = None,
+        request_option: RequestOption,
         stream: bool = False,
-    ):
+    ) -> ChatResponse | Generator[bytes, None, None]:
         if stream:
-            return Transport.execute(self.config, request, option=option, stream=True)
-        else:
-            return Transport.execute(self.config, request, unmarshal_as=ChatResponse, option=option)
+            return Transport.execute(self.config, request, stream=True, option=request_option)
+        return Transport.execute(self.config, request, unmarshal_as=ChatResponse, option=request_option)
 
     @overload
     async def achat(
-        self, request: ChatRequest, option: RequestOption | None, stream: Literal[True]
+        self,
+        request: ChatRequest,
+        request_option: RequestOption,
+        stream: Literal[True],
     ) -> AsyncGenerator[bytes, None]: ...
 
     @overload
     async def achat(
-        self, request: ChatRequest, option: RequestOption | None, stream: Literal[False]
+        self,
+        request: ChatRequest,
+        request_option: RequestOption,
+        stream: Literal[False] = False,
     ) -> ChatResponse: ...
 
-    @overload
-    async def achat(self, request: ChatRequest, option: RequestOption | None) -> ChatResponse: ...
-
     async def achat(
         self,
         request: ChatRequest,
-        option: RequestOption | None = None,
+        request_option: RequestOption,
         stream: bool = False,
-    ):
+    ) -> ChatResponse | AsyncGenerator[bytes, None]:
         if stream:
-            return await ATransport.aexecute(self.config, request, option=option, stream=True)
-        else:
-            return await ATransport.aexecute(self.config, request, unmarshal_as=ChatResponse, option=option)
+            return await ATransport.aexecute(self.config, request, stream=True, option=request_option)
+        return await ATransport.aexecute(self.config, request, unmarshal_as=ChatResponse, option=request_option)
 
-    def stop(self, request: StopChatRequest, option: RequestOption | None = None) -> StopChatResponse:
-        return Transport.execute(self.config, request, unmarshal_as=StopChatResponse, option=option)
+    def stop(self, request: StopChatRequest, request_option: RequestOption) -> StopChatResponse:
+        return Transport.execute(self.config, request, unmarshal_as=StopChatResponse, option=request_option)
 
-    async def astop(self, request: StopChatRequest, option: RequestOption | None = None) -> StopChatResponse:
-        return await ATransport.aexecute(self.config, request, unmarshal_as=StopChatResponse, option=option)
+    async def astop(self, request: StopChatRequest, request_option: RequestOption) -> StopChatResponse:
+        return await ATransport.aexecute(self.config, request, unmarshal_as=StopChatResponse, option=request_option)
 
     def suggested(
-        self, request: GetSuggestedQuestionsRequest, option: RequestOption | None = None
+        self, request: GetSuggestedQuestionsRequest, request_option: RequestOption
     ) -> GetSuggestedQuestionsResponse:
-        return Transport.execute(self.config, request, unmarshal_as=GetSuggestedQuestionsResponse, option=option)
+        return Transport.execute(
+            self.config, request, unmarshal_as=GetSuggestedQuestionsResponse, option=request_option
+        )
 
     async def asuggested(
-        self, request: GetSuggestedQuestionsRequest, option: RequestOption | None = None
+        self, request: GetSuggestedQuestionsRequest, request_option: RequestOption
     ) -> GetSuggestedQuestionsResponse:
         return await ATransport.aexecute(
-            self.config, request, unmarshal_as=GetSuggestedQuestionsResponse, option=option
+            self.config, request, unmarshal_as=GetSuggestedQuestionsResponse, option=request_option
         )