spaces 0.52.0.dev3__tar.gz → 0.52.0.dev4__tar.gz

{spaces-0.52.0.dev3 → spaces-0.52.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spaces
-Version: 0.52.0.dev3
+Version: 0.52.0.dev4
 Summary: Utilities for Hugging Face Spaces
 Author: Charles Bensimon
 Author-email: Charles Bensimon <charles@huggingface.co>

{spaces-0.52.0.dev3 → spaces-0.52.0.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spaces"
-version = "0.52.0.dev3"
+version = "0.52.0.dev4"
 description = "Utilities for Hugging Face Spaces"
 readme = "README.md"
 requires-python = ">=3.10,<3.15"

{spaces-0.52.0.dev3 → spaces-0.52.0.dev4}/spaces/zero/context.py

@@ -4,6 +4,7 @@
 from contextlib import contextmanager
 from contextvars import ContextVar
 from typing import Callable
+from typing import Iterable
 from typing import Literal
 from typing import TypedDict
 from uuid import UUID
@@ -66,7 +67,7 @@ class ZeroGPUContext:
         worker_contextualizer: Callable[[], Picklable[Callable[[], None]]] | None = None,
         disconnect_detector: Callable[[], bool] | None = None,
         connection_state: ConnectionState | None = None,
-        server_port: int | None = None,
+        server_ports: Iterable[int] | None = None,
         headers: dict[str, str] | None = None,
     ):
         self.exception_mapper = exception_mapper or exception_mapper_default
@@ -74,7 +75,7 @@ class ZeroGPUContext:
         self.worker_contextualizer = worker_contextualizer or WorkerContextualizerDefault
         self.disconnect_detector = disconnect_detector
         self.connection_state = connection_state
-        self.server_port = server_port
+        self.server_ports = set(server_ports) if server_ports is not None else set()
         self.headers = headers or {}
 
     def exception(
@@ -108,9 +109,8 @@ class ZeroGPUContext:
     def success(self, title: str, message: str):
         self.log_emitter(ZeroGPULog(title=title, message=message, level='success'))
 
-    def server_ports(self):
-        res = {self.server_port} if self.server_port is not None else set()
-        return res | set(GlobalContext.server_ports.values())
+    def get_server_ports(self):
+        return self.server_ports | set(GlobalContext.server_ports.values())
 
     def disconnected(self):
         if self.disconnect_detector is not None:

{spaces-0.52.0.dev3 → spaces-0.52.0.dev4}/spaces/zero/gradio.py

@@ -24,7 +24,7 @@ from .context import ZeroGPULog
 from .lifecycle import zerogpu_mode_enabled
 from .lifecycle import zerogpu_startup
 from .middleware import ASGIScope
-from .middleware import get_server_port
+from .middleware import get_server_ports
 from .middleware import ZeroGPUMiddleware
 
 
@@ -68,7 +68,7 @@ def error(title: str, message: str): # pragma: no cover
     return gr.Error(message, **kwargs)
 
 
-def log_emitter(log: ZeroGPULog):
+def log_emitter(log: ZeroGPULog): # pragma: no cover
     return info(
         title=log['title'],
         message=log['message'],
@@ -130,7 +130,7 @@ def get_zerogpu_context():
             log_emitter=log_emitter,
             worker_contextualizer=WorkerContextualizer,
             disconnect_detector=disconnect_detector,
-            server_port=get_server_port(scope) if scope is not None else None,
+            server_ports=get_server_ports(scope) if scope is not None else None,
             headers={key.lower(): value for key, value in dict(getattr(request, 'headers', {})).items()},
         )
 

{spaces-0.52.0.dev3 → spaces-0.52.0.dev4}/spaces/zero/middleware.py

@@ -87,18 +87,18 @@ class ZeroGPUMiddleware:
             worker_contextualizer=self.worker_contextualizer,
             disconnect_detector=self.disconnect_detector,
             connection_state=connection_state,
-            server_port=get_server_port(asgi_scope),
+            server_ports=get_server_ports(asgi_scope),
             headers=_decode_headers(asgi_scope.get('headers', [])),
         )):
             await asgi_app(asgi_scope, wrapped_receive, asgi_send)
 
 
-def get_server_port(scope: ASGIScope):
+def get_server_ports(scope: ASGIScope):
     match scope.get('server'):
         case (_, int() as port):
-            return port
+            return {port}
         case _: # pragma: no cover
-            return None
+            return set()
 
 
 def _decode_headers(headers: Iterable[tuple[bytes, bytes]]):

{spaces-0.52.0.dev3 → spaces-0.52.0.dev4}/spaces/zero/wrappers.py

@@ -188,7 +188,7 @@ def regular_function_wrapper(
             assert worker.arg_queue.empty()
             assert worker.res_queue.empty()
         else:
-            worker = Worker(worker_loop, allow_token, nvidia_uuid, ctx.server_ports())
+            worker = Worker(worker_loop, allow_token, nvidia_uuid, ctx.get_server_ports())
 
         worker_context_setup = ctx.worker_contextualizer()
 
@@ -315,7 +315,7 @@ def generator_function_wrapper(
             assert worker.arg_queue.empty()
             assert worker.res_queue.empty()
         else:
-            worker = Worker(worker_loop, allow_token, nvidia_uuid, ctx.server_ports())
+            worker = Worker(worker_loop, allow_token, nvidia_uuid, ctx.get_server_ports())
 
         worker_context_setup = ctx.worker_contextualizer()