Fix imports

Author: AHGIJMKLKKZNPJKQR

fishjam/agent/__init__.py

@@ -1,4 +1,10 @@
-from .agent import Agent, TrackDataHandler
+from .agent import Agent, AgentResponseTrackData, TrackDataHandler
 from .errors import AgentAuthError, AgentError
 
-__all__ = ["Agent", "AgentError", "AgentAuthError", "TrackDataHandler"]
+__all__ = [
+    "Agent",
+    "AgentError",
+    "AgentAuthError",
+    "TrackDataHandler",
+    "AgentResponseTrackData",
+]

fishjam/events/__init__.py

@@ -15,15 +15,13 @@
     ServerMessageRoomDeleted,
     ServerMessageStreamConnected,
     ServerMessageStreamDisconnected,
-    ServerMessageTrack,
     ServerMessageTrackAdded,
-    ServerMessageTrackData,
     ServerMessageTrackMetadataUpdated,
     ServerMessageTrackRemoved,
-    ServerMessageTrackType,
     ServerMessageViewerConnected,
     ServerMessageViewerDisconnected,
 )
+from fishjam.events._protos.fishjam.notifications import Track, TrackEncoding, TrackType
 
 __all__ = [
     "ServerMessageRoomCreated",
@@ -37,12 +35,12 @@
     "ServerMessagePeerCrashed",
     "ServerMessageStreamConnected",
     "ServerMessageStreamDisconnected",
-    "ServerMessageTrack",
-    "ServerMessageTrackType",
     "ServerMessageTrackAdded",
-    "ServerMessageTrackData",
     "ServerMessageTrackMetadataUpdated",
     "ServerMessageTrackRemoved",
     "ServerMessageViewerConnected",
     "ServerMessageViewerDisconnected",
+    "Track",
+    "TrackEncoding",
+    "TrackType",
 ]