feat: Support more params for create room (#237)

Author: chyroc

cozepy/audio/rooms/__init__.py

@@ -2,7 +2,36 @@
 
 from cozepy.model import CozeModel
 from cozepy.request import Requester
-from cozepy.util import remove_url_trailing_slash
+from cozepy.util import remove_none_values, remove_url_trailing_slash
+
+
+class RoomAudioConfig(CozeModel):
+    # 房间音频编码格式，支持设置为：
+    # AACLC: AAC-LC 编码格式。
+    # G711A: G711A 编码格式。
+    # OPUS: （默认）Opus 编码格式。
+    # G722: G.722 编码格式。
+    codec: Optional[str]
+
+
+class RoomVideoConfig(CozeModel):
+    # 房间视频编码格式，支持设置为：
+    # H264:（默认）H264 编码格式。
+    # BYTEVC1: 火山引擎自研的视频编码格式。
+    codec: Optional[str]
+    # 房间视频流类型, 支持 main 和 screen。
+    # main: 主流，包括通过摄像头/麦克风的内部采集机制获取的流，以及通过自定义采集方式获取的流。
+    # screen: 屏幕流，用于屏幕共享或屏幕录制的视频流。
+    stream_video_type: Optional[str]
+
+
+class RoomConfig(CozeModel):
+    # The audio config of the room.
+    audio_config: Optional[RoomAudioConfig]
+    # The video config of the room.
+    video_config: Optional[RoomVideoConfig]
+    # 自定义开场白
+    prologue_content: Optional[str]
 
 
 class CreateRoomResp(CozeModel):
@@ -32,6 +61,8 @@ def create(
         voice_id: Optional[str] = None,
         conversation_id: Optional[str] = None,
         uid: Optional[str] = None,
+        workflow_id: Optional[str] = None,
+        config: Optional[RoomConfig] = None,
     ) -> CreateRoomResp:
         """
         create rtc room
@@ -40,15 +71,21 @@ def create(
         :param voice_id: The voice id of the voice.
         :param conversation_id: The id of the conversation.
         :param uid: The id of the user. if not provided, it will be generated by the server.
+        :param workflow_id: The id of the workflow.
+        :param config: The config of the room.
         :return: create room result
         """
         url = f"{self._base_url}/v1/audio/rooms"
-        body = {
-            "bot_id": bot_id,
-            "voice_id": voice_id,
-            "conversation_id": conversation_id,
-            "uid": uid,
-        }
+        body = remove_none_values(
+            {
+                "bot_id": bot_id,
+                "voice_id": voice_id,
+                "conversation_id": conversation_id,
+                "uid": uid,
+                "workflow_id": workflow_id,
+                "config": config.model_dump() if config else None,
+            }
+        )
         return self._requester.request("post", url, stream=False, cast=CreateRoomResp, body=body)
 
 
@@ -68,6 +105,8 @@ async def create(
         voice_id: Optional[str] = None,
         conversation_id: Optional[str] = None,
         uid: Optional[str] = None,
+        workflow_id: Optional[str] = None,
+        config: Optional[RoomConfig] = None,
     ) -> CreateRoomResp:
         """
         create rtc room
@@ -76,13 +115,19 @@ async def create(
         :param voice_id: The voice id of the voice.
         :param conversation_id: The id of the conversation.
         :param uid: The id of the user. if not provided, it will be generated by the server.
+        :param workflow_id: The id of the workflow.
+        :param config: The config of the room.
         :return: create room result
         """
         url = f"{self._base_url}/v1/audio/rooms"
-        body = {
-            "bot_id": bot_id,
-            "voice_id": voice_id,
-            "conversation_id": conversation_id,
-            "uid": uid,
-        }
+        body = remove_none_values(
+            {
+                "bot_id": bot_id,
+                "voice_id": voice_id,
+                "conversation_id": conversation_id,
+                "uid": uid,
+                "workflow_id": workflow_id,
+                "config": config.model_dump() if config else None,
+            }
+        )
         return await self._requester.arequest("post", url, stream=False, cast=CreateRoomResp, body=body)