Add missing exports

Author: AHGIJMKLKKZNPJKQR

fishjam/__init__.py

@@ -22,7 +22,6 @@
     PeerOptions,
     Room,
     RoomOptions,
-    SubscribeOptions,
 )
 
 __all__ = [
@@ -31,7 +30,6 @@
     "receive_binary",
     "PeerMetadata",
     "PeerOptions",
-    "SubscribeOptions",
     "RoomOptions",
     "Room",
     "Peer",

fishjam/peer/__init__.py

@@ -1,3 +1,17 @@
-from fishjam._openapi_client.models import PeerStatus, PeerType
+from fishjam._openapi_client.models import (
+    PeerMetadata,
+    PeerStatus,
+    PeerType,
+    SubscribeOptions,
+    SubscribeOptionsAudioFormat,
+    SubscribeOptionsAudioSampleRate,
+)
 
-__all__ = ["PeerStatus", "PeerType"]
+__all__ = [
+    "PeerMetadata",
+    "PeerStatus",
+    "PeerType",
+    "SubscribeOptions",
+    "SubscribeOptionsAudioFormat",
+    "SubscribeOptionsAudioSampleRate",
+]

tests/agent/test_agent.py

@@ -5,10 +5,9 @@
 import pytest
 import pytest_asyncio
 
-from fishjam import FishjamClient, FishjamNotifier
+from fishjam import FishjamClient, FishjamNotifier, Room
 from fishjam.agent.agent import Agent
 from fishjam.agent.errors import AgentAuthError
-from fishjam.api._fishjam_client import Room
 from fishjam.events._protos.fishjam import (
     ServerMessagePeerDisconnected,
     ServerMessagePeerMetadataUpdated,

tests/test_room_api.py

@@ -7,24 +7,30 @@
 
 from fishjam import (
     FishjamClient,
+    Peer,
     PeerOptions,
+    Room,
     RoomOptions,
 )
-from fishjam._openapi_client.models import (
+from fishjam.errors import (
+    BadRequestError,
+    NotFoundError,
+    ServiceUnavailableError,
+    UnauthorizedError,
+)
+from fishjam.peer import (
     PeerMetadata,
     PeerStatus,
     PeerType,
+    SubscribeOptions,
+    SubscribeOptionsAudioFormat,
+    SubscribeOptionsAudioSampleRate,
+)
+from fishjam.room import (
     RoomConfig,
     RoomConfigRoomType,
     RoomConfigVideoCodec,
 )
-from fishjam.api._fishjam_client import Peer, Room
-from fishjam.errors import (
-    BadRequestError,
-    NotFoundError,
-    ServiceUnavailableError,
-    UnauthorizedError,
-)
 
 HOST = "fishjam" if os.getenv("DOCKER_TEST") == "TRUE" else "localhost"
 FISHJAM_URL = f"http://{HOST}:5002"
@@ -196,7 +202,13 @@ def _assert_peer_created(
         assert peer in room.peers
 
     def test_with_specified_options(self, room_api: FishjamClient):
-        options = PeerOptions(enable_simulcast=True)
+        options = PeerOptions(
+            enable_simulcast=True,
+            subscribe=SubscribeOptions(
+                audio_format=SubscribeOptionsAudioFormat.PCM16,
+                audio_sample_rate=SubscribeOptionsAudioSampleRate.VALUE_16000,
+            ),
+        )
 
         room = room_api.create_room()
         peer, _token = room_api.create_peer(room.id, options=options)