verifiers 0.1.15.dev175__tar.gz → 0.1.15.dev176__tar.gz

{verifiers-0.1.15.dev175 → verifiers-0.1.15.dev176}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: verifiers
-Version: 0.1.15.dev175
+Version: 0.1.15.dev176
 Summary: Verifiers: Environments for LLM Reinforcement Learning
 Project-URL: Homepage, https://github.com/primeintellect-ai/verifiers
 Project-URL: Documentation, https://github.com/primeintellect-ai/verifiers
@@ -40,7 +40,7 @@ Requires-Dist: openai>=1.108.1
 Requires-Dist: pillow
 Requires-Dist: prime-pydantic-config[toml]
 Requires-Dist: prime-sandboxes>=0.2.25
-Requires-Dist: prime-tunnel>=0.1.6
+Requires-Dist: prime-tunnel>=0.1.8
 Requires-Dist: pydantic>=2.11.9
 Requires-Dist: pymupdf
 Requires-Dist: pyzmq>=27.1.0

{verifiers-0.1.15.dev175 → verifiers-0.1.15.dev176}/pyproject.toml

@@ -37,7 +37,7 @@ dependencies = [
     "nest-asyncio>=1.6.0", # for jupyter notebooks
     "openai>=1.108.1",
     "openai-agents>=0.0.7",
-    "prime-tunnel>=0.1.6",
+    "prime-tunnel>=0.1.8",
     "prime-sandboxes>=0.2.25",
     "pydantic>=2.11.9",
     "requests",

{verifiers-0.1.15.dev175 → verifiers-0.1.15.dev176}/verifiers/v1/harness.py

@@ -236,8 +236,10 @@ class Harness(RuntimeOwnerMixin[ConfigT], Generic[ConfigT]):
         return config
 
     def load_endpoint(self) -> Endpoint:
+        sandbox_config = self.program_sandbox_config(self.program_config)
         return Endpoint(
-            use_tunnel=self.program_sandbox_config(self.program_config) is not None
+            use_tunnel=sandbox_config is not None,
+            tunnel_labels=sandbox_config.labels if sandbox_config is not None else None,
         )
 
     def rebuild_runtime(self) -> None:

{verifiers-0.1.15.dev175 → verifiers-0.1.15.dev176}/verifiers/v1/utils/endpoint_utils.py

@@ -146,6 +146,7 @@ class Endpoint:
         secret: str | None = None,
         use_tunnel: bool = False,
         logger: logging.Logger | None = None,
+        tunnel_labels: list[str] | None = None,
     ):
         self.use_tunnel = use_tunnel
         self.logger = logger or logging.getLogger(__name__)
@@ -154,6 +155,7 @@ class Endpoint:
             secret=secret or os.environ.get("ENDPOINT_SECRET"),
         )
         self.secret = self.server.secret
+        self.tunnel_labels = list(tunnel_labels) if tunnel_labels else []
         self._tunnel: TunnelHandle | None = None
         self._tunnel_lock = asyncio.Lock()
         self._tunnel_last_checked = 0.0
@@ -295,7 +297,10 @@ class Endpoint:
                         self._tunnel = None
 
             if self._tunnel is None:
-                tunnel = cast(TunnelHandle, Tunnel(local_port=self.server.port))
+                tunnel = cast(
+                    TunnelHandle,
+                    Tunnel(local_port=self.server.port, labels=self.tunnel_labels),
+                )
                 url = await tunnel.start()
                 self._tunnel = tunnel
                 self._tunnel_last_checked = time.time()