From f8b05a0a81b802dc91ec8da60d70d8b32cf03c7e Mon Sep 17 00:00:00 2001
From: Aidan Timson <contact@timmo.xyz>
Date: Fri, 8 Mar 2024 11:11:17 +0000
Subject: [PATCH] Update systembridgeconnector to version 4.0.3 (#112608)

---
 homeassistant/components/system_bridge/config_flow.py | 3 ++-
 homeassistant/components/system_bridge/coordinator.py | 7 +++----
 homeassistant/components/system_bridge/manifest.json  | 2 +-
 requirements_all.txt                                  | 2 +-
 requirements_test_all.txt                             | 2 +-
 5 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/homeassistant/components/system_bridge/config_flow.py b/homeassistant/components/system_bridge/config_flow.py
index 0042d9c647e..1406088f3d1 100644
--- a/homeassistant/components/system_bridge/config_flow.py
+++ b/homeassistant/components/system_bridge/config_flow.py
@@ -61,11 +61,12 @@ async def _validate_input(
         data[CONF_HOST],
         data[CONF_PORT],
         data[CONF_TOKEN],
+        session=async_get_clientsession(hass),
     )
 
     try:
         async with asyncio.timeout(15):
-            await websocket_client.connect(session=async_get_clientsession(hass))
+            await websocket_client.connect()
             hass.async_create_task(
                 websocket_client.listen(callback=_async_handle_module)
             )
diff --git a/homeassistant/components/system_bridge/coordinator.py b/homeassistant/components/system_bridge/coordinator.py
index ca475dc5863..df0a39a877d 100644
--- a/homeassistant/components/system_bridge/coordinator.py
+++ b/homeassistant/components/system_bridge/coordinator.py
@@ -55,6 +55,7 @@ class SystemBridgeDataUpdateCoordinator(DataUpdateCoordinator[SystemBridgeData])
             entry.data[CONF_HOST],
             entry.data[CONF_PORT],
             entry.data[CONF_TOKEN],
+            session=async_get_clientsession(hass),
         )
 
         super().__init__(
@@ -106,7 +107,7 @@ class SystemBridgeDataUpdateCoordinator(DataUpdateCoordinator[SystemBridgeData])
         self,
         base: str,
         path: str,
-    ) -> MediaFile:
+    ) -> MediaFile | None:
         """Get media file."""
         return await self.websocket_client.get_file(
             MediaGetFile(
@@ -168,9 +169,7 @@ class SystemBridgeDataUpdateCoordinator(DataUpdateCoordinator[SystemBridgeData])
         """Use WebSocket for updates."""
         try:
             async with asyncio.timeout(20):
-                await self.websocket_client.connect(
-                    session=async_get_clientsession(self.hass),
-                )
+                await self.websocket_client.connect()
 
             self.hass.async_create_background_task(
                 self._listen_for_data(),
diff --git a/homeassistant/components/system_bridge/manifest.json b/homeassistant/components/system_bridge/manifest.json
index 8355935162d..b4365fda778 100644
--- a/homeassistant/components/system_bridge/manifest.json
+++ b/homeassistant/components/system_bridge/manifest.json
@@ -10,6 +10,6 @@
   "iot_class": "local_push",
   "loggers": ["systembridgeconnector"],
   "quality_scale": "silver",
-  "requirements": ["systembridgeconnector==4.0.2"],
+  "requirements": ["systembridgeconnector==4.0.3"],
   "zeroconf": ["_system-bridge._tcp.local."]
 }
diff --git a/requirements_all.txt b/requirements_all.txt
index 07e276f7c2f..f38f474b1cd 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -2647,7 +2647,7 @@ switchbot-api==2.0.0
 synology-srm==0.2.0
 
 # homeassistant.components.system_bridge
-systembridgeconnector==4.0.2
+systembridgeconnector==4.0.3
 
 # homeassistant.components.tailscale
 tailscale==0.6.0
diff --git a/requirements_test_all.txt b/requirements_test_all.txt
index 5646c80699f..c4a038e88f5 100644
--- a/requirements_test_all.txt
+++ b/requirements_test_all.txt
@@ -2036,7 +2036,7 @@ surepy==0.9.0
 switchbot-api==2.0.0
 
 # homeassistant.components.system_bridge
-systembridgeconnector==4.0.2
+systembridgeconnector==4.0.3
 
 # homeassistant.components.tailscale
 tailscale==0.6.0