diff --git a/examples/transcription/main.py b/examples/transcription/main.py index 5f5530e..7a7e374 100644 --- a/examples/transcription/main.py +++ b/examples/transcription/main.py @@ -6,7 +6,8 @@ from transcription.room import RoomService, fishjam from transcription.worker import async_worker -from fishjam import PeerOptions, SubscribeOptions +from fishjam import PeerOptions +from fishjam.peer import SubscribeOptions _room_service: RoomService | None = None diff --git a/fishjam/__init__.py b/fishjam/__init__.py index b5a817a..c597338 100644 --- a/fishjam/__init__.py +++ b/fishjam/__init__.py @@ -22,7 +22,6 @@ PeerOptions, Room, RoomOptions, - SubscribeOptions, ) __all__ = [ @@ -31,7 +30,6 @@ "receive_binary", "PeerMetadata", "PeerOptions", - "SubscribeOptions", "RoomOptions", "Room", "Peer", diff --git a/fishjam/peer/__init__.py b/fishjam/peer/__init__.py index 1870ad5..9e921cf 100644 --- a/fishjam/peer/__init__.py +++ b/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", +] diff --git a/tests/agent/test_agent.py b/tests/agent/test_agent.py index 4d78fa3..c662cd7 100644 --- a/tests/agent/test_agent.py +++ b/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, diff --git a/tests/test_room_api.py b/tests/test_room_api.py index 48ade7f..d30ecac 100644 --- a/tests/test_room_api.py +++ b/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" @@ -179,7 +185,12 @@ def test_id_not_found(self, room_api): class TestCreatePeer: def _assert_peer_created( - self, room_api, webrtc_peer, room_id, server_metadata=None + self, + room_api, + webrtc_peer, + room_id, + server_metadata=None, + subscribe=None, ): server_metadata = server_metadata or {} @@ -189,19 +200,26 @@ def _assert_peer_created( status=PeerStatus("disconnected"), tracks=[], metadata=PeerMetadata.from_dict({"peer": {}, "server": server_metadata}), - subscribe=None, + subscribe=subscribe, ) room = room_api.get_room(room_id) assert peer in room.peers def test_with_specified_options(self, room_api: FishjamClient): - options = PeerOptions(enable_simulcast=True) + subscribe = SubscribeOptions( + audio_format=SubscribeOptionsAudioFormat.PCM16, + audio_sample_rate=SubscribeOptionsAudioSampleRate.VALUE_16000, + ) + options = PeerOptions( + enable_simulcast=True, + subscribe=subscribe, + ) room = room_api.create_room() peer, _token = room_api.create_peer(room.id, options=options) - self._assert_peer_created(room_api, peer, room.id) + self._assert_peer_created(room_api, peer, room.id, subscribe=subscribe) def test_with_metadata(self, room_api: FishjamClient): options = PeerOptions(metadata={"is_test": True})