merge

cozepy/__init__.py

@@ -1,6 +1,6 @@
 from .audio.rooms import CreateRoomResp
 from .audio.speech import AudioFormat
-from .audio.transcriptions import CreateTranslationResp
+from .audio.transcriptions import CreateTranscriptionsResp
 from .audio.voices import Voice
 from .auth import (
     AsyncDeviceOAuthApp,
@@ -92,27 +92,27 @@
 from .templates import TemplateDuplicateResp, TemplateEntityType
 from .version import VERSION
 from .websockets.audio.speech import (
-    AsyncWebsocketsAudioSpeechCreateClient,
+    AsyncWebsocketsAudioSpeechClient,
     AsyncWebsocketsAudioSpeechEventHandler,
     InputTextBufferAppendEvent,
-    InputTextBufferCommitEvent,
-    InputTextBufferCommittedEvent,
+    InputTextBufferCompletedEvent,
+    InputTextBufferCompleteEvent,
     SpeechAudioCompletedEvent,
     SpeechAudioUpdateEvent,
     SpeechUpdateEvent,
 )
 from .websockets.audio.transcriptions import (
-    AsyncWebsocketsAudioTranscriptionsCreateClient,
+    AsyncWebsocketsAudioTranscriptionsClient,
     AsyncWebsocketsAudioTranscriptionsEventHandler,
     InputAudioBufferAppendEvent,
-    InputAudioBufferCommitEvent,
-    InputAudioBufferCommittedEvent,
+    InputAudioBufferCompletedEvent,
+    InputAudioBufferCompleteEvent,
     TranscriptionsMessageCompletedEvent,
     TranscriptionsMessageUpdateEvent,
     TranscriptionsUpdateEvent,
 )
 from .websockets.chat import (
-    AsyncWebsocketsChatCreateClient,
+    AsyncWebsocketsChatClient,
     AsyncWebsocketsChatEventHandler,
     ConversationAudioDeltaEvent,
     ConversationChatCompletedEvent,
@@ -143,7 +143,7 @@
     "Voice",
     "AudioFormat",
     # audio.transcriptions
-    "CreateTranslationResp",
+    "CreateTranscriptionsResp",
     # auth
     "AsyncDeviceOAuthApp",
     "AsyncJWTOAuthApp",
@@ -217,29 +217,29 @@
     "WebsocketsEvent",
     # websockets.audio.speech
     "InputTextBufferAppendEvent",
-    "InputTextBufferCommitEvent",
+    "InputTextBufferCompleteEvent",
     "SpeechUpdateEvent",
-    "InputTextBufferCommittedEvent",
+    "InputTextBufferCompletedEvent",
     "SpeechAudioUpdateEvent",
     "SpeechAudioCompletedEvent",
     "AsyncWebsocketsAudioSpeechEventHandler",
-    "AsyncWebsocketsAudioSpeechCreateClient",
+    "AsyncWebsocketsAudioSpeechClient",
     # websockets.audio.transcriptions
     "InputAudioBufferAppendEvent",
-    "InputAudioBufferCommitEvent",
+    "InputAudioBufferCompleteEvent",
     "TranscriptionsUpdateEvent",
-    "InputAudioBufferCommittedEvent",
+    "InputAudioBufferCompletedEvent",
     "TranscriptionsMessageUpdateEvent",
     "TranscriptionsMessageCompletedEvent",
     "AsyncWebsocketsAudioTranscriptionsEventHandler",
-    "AsyncWebsocketsAudioTranscriptionsCreateClient",
+    "AsyncWebsocketsAudioTranscriptionsClient",
     # websockets.chat
     "ConversationChatCreatedEvent",
     "ConversationMessageDeltaEvent",
     "ConversationAudioDeltaEvent",
     "ConversationChatCompletedEvent",
     "AsyncWebsocketsChatEventHandler",
-    "AsyncWebsocketsChatCreateClient",
+    "AsyncWebsocketsChatClient",
     # workflows.runs
     "WorkflowRunResult",
     "WorkflowEventType",

cozepy/audio/transcriptions/__init__.py

@@ -7,7 +7,7 @@
 from cozepy.util import remove_url_trailing_slash
 
 
-class CreateTranslationResp(CozeModel):
+class CreateTranscriptionsResp(CozeModel):
     # The text of translation
     text: str
 
@@ -23,18 +23,18 @@ def create(
         *,
         file: FileTypes,
         **kwargs,
-    ) -> CreateTranslationResp:
+    ) -> CreateTranscriptionsResp:
         """
-        create translation
+        create transcriptions
 
         :param file: The file to be translated.
-        :return: create translation result
+        :return: create transcriptions result
         """
         url = f"{self._base_url}/v1/audio/transcriptions"
         headers: Optional[dict] = kwargs.get("headers")
         files = {"file": _try_fix_file(file)}
         return self._requester.request(
-            "post", url, stream=False, cast=CreateTranslationResp, headers=headers, files=files
+            "post", url, stream=False, cast=CreateTranscriptionsResp, headers=headers, files=files
         )
 
 
@@ -53,16 +53,16 @@ async def create(
         *,
         file: FileTypes,
         **kwargs,
-    ) -> CreateTranslationResp:
+    ) -> CreateTranscriptionsResp:
         """
-        create translation
+        create transcriptions
 
         :param file: The file to be translated.
-        :return: create translation result
+        :return: create transcriptions result
         """
         url = f"{self._base_url}/v1/audio/transcriptions"
         files = {"file": _try_fix_file(file)}
         headers: Optional[dict] = kwargs.get("headers")
         return await self._requester.arequest(
-            "post", url, stream=False, cast=CreateTranslationResp, headers=headers, files=files
+            "post", url, stream=False, cast=CreateTranscriptionsResp, headers=headers, files=files
         )

cozepy/websockets/__init__.py

@@ -2,8 +2,31 @@
 from cozepy.request import Requester
 from cozepy.util import http_base_url_to_ws, remove_url_trailing_slash
 
-from .audio import AsyncWebsocketsAudioClient
-from .chat import AsyncWebsocketsChatClient
+from .audio import AsyncWebsocketsAudioClient, WebsocketsAudioClient
+from .chat import AsyncWebsocketsChatBuildClient, WebsocketsChatBuildClient
+
+
+class WebsocketsClient(object):
+    def __init__(self, base_url: str, auth: Auth, requester: Requester):
+        self._base_url = http_base_url_to_ws(remove_url_trailing_slash(base_url))
+        self._auth = auth
+        self._requester = requester
+
+    @property
+    def audio(self) -> WebsocketsAudioClient:
+        return WebsocketsAudioClient(
+            base_url=self._base_url,
+            auth=self._auth,
+            requester=self._requester,
+        )
+
+    @property
+    def chat(self) -> WebsocketsChatBuildClient:
+        return WebsocketsChatBuildClient(
+            base_url=self._base_url,
+            auth=self._auth,
+            requester=self._requester,
+        )
 
 
 class AsyncWebsocketsClient(object):
@@ -21,8 +44,8 @@ def audio(self) -> AsyncWebsocketsAudioClient:
         )
 
     @property
-    def chat(self) -> AsyncWebsocketsChatClient:
-        return AsyncWebsocketsChatClient(
+    def chat(self) -> AsyncWebsocketsChatBuildClient:
+        return AsyncWebsocketsChatBuildClient(
             base_url=self._base_url,
             auth=self._auth,
             requester=self._requester,

cozepy/websockets/audio/__init__.py

@@ -1,8 +1,31 @@
 from cozepy.auth import Auth
 from cozepy.request import Requester
 
-from .speech import AsyncWebsocketsAudioSpeechClient
-from .transcriptions import AsyncWebsocketsAudioTranscriptionsClient
+from .speech import AsyncWebsocketsAudioSpeechBuildClient, WebsocketsAudioSpeechBuildClient
+from .transcriptions import AsyncWebsocketsAudioTranscriptionsBuildClient, WebsocketsAudioTranscriptionsBuildClient
+
+
+class WebsocketsAudioClient(object):
+    def __init__(self, base_url: str, auth: Auth, requester: Requester):
+        self._base_url = base_url
+        self._auth = auth
+        self._requester = requester
+
+    @property
+    def transcriptions(self) -> "WebsocketsAudioTranscriptionsBuildClient":
+        return WebsocketsAudioTranscriptionsBuildClient(
+            base_url=self._base_url,
+            auth=self._auth,
+            requester=self._requester,
+        )
+
+    @property
+    def speech(self) -> "WebsocketsAudioSpeechBuildClient":
+        return WebsocketsAudioSpeechBuildClient(
+            base_url=self._base_url,
+            auth=self._auth,
+            requester=self._requester,
+        )
 
 
 class AsyncWebsocketsAudioClient(object):
@@ -12,16 +35,16 @@ def __init__(self, base_url: str, auth: Auth, requester: Requester):
         self._requester = requester
 
     @property
-    def transcriptions(self) -> "AsyncWebsocketsAudioTranscriptionsClient":
-        return AsyncWebsocketsAudioTranscriptionsClient(
+    def transcriptions(self) -> "AsyncWebsocketsAudioTranscriptionsBuildClient":
+        return AsyncWebsocketsAudioTranscriptionsBuildClient(
             base_url=self._base_url,
             auth=self._auth,
             requester=self._requester,
         )
 
     @property
-    def speech(self) -> "AsyncWebsocketsAudioSpeechClient":
-        return AsyncWebsocketsAudioSpeechClient(
+    def speech(self) -> "AsyncWebsocketsAudioSpeechBuildClient":
+        return AsyncWebsocketsAudioSpeechBuildClient(
             base_url=self._base_url,
             auth=self._auth,
             requester=self._requester,