Skip to content

Commit 0671587

Browse files
Strip is_server from Session __init__
1 parent fbccf2e commit 0671587

File tree

4 files changed

+3
-7
lines changed

4 files changed

+3
-7
lines changed

src/replit_river/client_transport.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,6 @@ async def _create_new_session(
225225
session_id=hs_request.sessionId,
226226
websocket=new_ws,
227227
transport_options=self._transport_options,
228-
is_server=False,
229228
close_session_callback=self._delete_session,
230229
retry_connection_callback=self._retry_connection,
231230
)

src/replit_river/server_session.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,6 @@ def __init__(
4949
session_id: str,
5050
websocket: websockets.WebSocketCommonProtocol,
5151
transport_options: TransportOptions,
52-
is_server: bool,
5352
handlers: dict[tuple[str, str], tuple[str, GenericRpcHandlerBuilder]],
5453
close_session_callback: Callable[["Session"], Coroutine[Any, Any, Any]],
5554
retry_connection_callback: (
@@ -66,10 +65,10 @@ def __init__(
6665
session_id=session_id,
6766
websocket=websocket,
6867
transport_options=transport_options,
69-
is_server=is_server,
7068
close_session_callback=close_session_callback,
7169
retry_connection_callback=retry_connection_callback,
7270
)
71+
self._is_server = True
7372
self._handlers = handlers
7473

7574
async def start_serve_responses(self) -> None:

src/replit_river/server_transport.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,6 @@ async def _get_or_create_session(
133133
session_id,
134134
websocket,
135135
self._transport_options,
136-
True,
137136
self._handlers,
138137
close_session_callback=self._delete_session,
139138
)
@@ -154,7 +153,6 @@ async def _get_or_create_session(
154153
session_id,
155154
websocket,
156155
self._transport_options,
157-
True,
158156
self._handlers,
159157
close_session_callback=self._delete_session,
160158
)

src/replit_river/session.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737

3838
class Session:
3939
"""Common functionality shared between client_session and server_session"""
40+
_is_server: bool
4041

4142
def __init__(
4243
self,
@@ -45,7 +46,6 @@ def __init__(
4546
session_id: str,
4647
websocket: websockets.WebSocketCommonProtocol,
4748
transport_options: TransportOptions,
48-
is_server: bool,
4949
close_session_callback: Callable[["Session"], Coroutine[Any, Any, Any]],
5050
retry_connection_callback: (
5151
Callable[
@@ -58,7 +58,7 @@ def __init__(
5858
self._transport_id = transport_id
5959
self._to_id = to_id
6060
self.session_id = session_id
61-
self._is_server = is_server
61+
self._is_server = False
6262
self._transport_options = transport_options
6363

6464
# session state, only modified during closing

0 commit comments

Comments
 (0)