fix websockets API

Author: ChristianBeilschmidt

geoengine/workflow.py

@@ -25,6 +25,7 @@
 import requests as req
 import rioxarray
 import websockets
+import websockets.asyncio.client
 import xarray as xr
 from owslib.util import Authentication, ResponseWrapper
 from owslib.wcs import WebCoverageService
@@ -622,7 +623,7 @@ async def raster_stream(
 
         async with websockets.asyncio.client.connect(
             uri=self.__replace_http_with_ws(url),
-            extra_headers=session.auth_header,
+            additional_headers=session.auth_header,
             open_timeout=open_timeout,
             max_size=None,
         ) as websocket:

pyproject.toml

@@ -29,7 +29,7 @@ dependencies = [
     "rioxarray >=0.9.1, <0.20",
     "StrEnum >=0.4.6,<0.5",               # TODO: use from stdlib when `python_requires = >=3.11`
     "vega >= 3.5,<4.2",
-    "websockets >= 14.0,<16",
+    "websockets >= 14.2,<16",
     "xarray >=0.19,<2025.8",
     "urllib3 >= 2.1, < 2.6",
     "pydantic >= 2.10.6, < 2.12",