verifiers 0.2.2.dev12__py3-none-any.whl → 0.2.2.dev13__py3-none-any.whl

verifiers/v1/agent.py

@@ -147,13 +147,15 @@ class Agent:
             harness_config = harness_config_type("bash")(id="bash")
         self.config = config
         self.harness = load_harness(harness_config)
-        if client is None:
+        self._owns_client = client is None
+        if self._owns_client:
             client = resolve_client(config.client or EvalClientConfig())
         self.ctx = ModelContext(
             model=config.model,
             client=client,
             sampling=config.sampling if config.sampling is not None else Sampling(),
         )
+        self._closed = False
         self.runtime_config: RuntimeConfig = self.harness.config.runtime
         self.interception = interception
         self.limits = RolloutLimits(
@@ -173,6 +175,8 @@ class Agent:
     async def __aenter__(self) -> "Agent":
         if self._entered:
             raise RuntimeError("Agent is already entered; enter it once and share it")
+        if self._closed:
+            raise RuntimeError("Agent is closed; create a new agent")
         self._entered = True
         if self.interception is None:
             # Sized to the runtime policy (remote needs the tunnel); runs the
@@ -186,14 +190,22 @@ class Agent:
                 # A failed __aenter__ gets no __aexit__ from `async with`: unwind
                 # here, or the agent stays "already entered" forever.
                 self._entered, self._server = False, None
+                if self._owns_client:
+                    self._closed = True
+                    await self.ctx.client.close()
                 raise
         return self
 
     async def __aexit__(self, *exc) -> None:
         self._entered = False
         server, self._server = self._server, None
-        if server is not None:
-            await server.__aexit__(*exc)
+        try:
+            if server is not None:
+                await server.__aexit__(*exc)
+        finally:
+            if self._owns_client:
+                self._closed = True
+                await self.ctx.client.close()
 
     def _interception_for(
         self, run_is_local: bool, task: Task, shared_tools: Mapping
@@ -230,6 +242,8 @@ class Agent:
         `on_trace` observes the trace the moment it's minted, before any I/O.
         Retries whole while the trace ends with a retryable error (`config.retries`)
         — never into a borrowed box; the final trace keeps earlier attempts' errors."""
+        if self._closed:
+            raise RuntimeError("Agent is closed; create a new agent")
         retry = self.config.retries
         history: list = []
         for attempt in range(retry.max_retries + 1):

{verifiers-0.2.2.dev12.dist-info → verifiers-0.2.2.dev13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: verifiers
-Version: 0.2.2.dev12
+Version: 0.2.2.dev13
 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

{verifiers-0.2.2.dev12.dist-info → verifiers-0.2.2.dev13.dist-info}/RECORD

@@ -168,7 +168,7 @@ verifiers/utils/tool_utils.py,sha256=gInWZODWQUZN4TyEMuuITyOUm9qlHJxtcvr1zZl_zJs
 verifiers/utils/usage_utils.py,sha256=GPLC0xGY_Obrr8X7huWY-2iODZ7tgX-MtO8WSDN4rXI,3904
 verifiers/utils/version_utils.py,sha256=am3hZLnlaUWTFdllGZR1VMN91hEhiiE8FDmPf1cOG1k,2642
 verifiers/v1/__init__.py,sha256=r8khgi7_KSY4AfevEefSslyxytScmMzzCBA6kjqvd1I,6653
-verifiers/v1/agent.py,sha256=149ex6npYxa0Jba3zJARiS4bt-56zAR5YfSdbCmHmmA,19378
+verifiers/v1/agent.py,sha256=EapDEWf0yeCaSYjD7vZPECbO2L4pd3FAYwhYzcSdzXU,19925
 verifiers/v1/decorators.py,sha256=XRMkUQSyvXCYP5fOwzBYV5qEOlxLg6rzYhnhrHHHTIQ,3838
 verifiers/v1/env.py,sha256=XQHzcTCNLmAbupkFp3edvUJqtSxs6U3mJHVz3hAI6Y4,24734
 verifiers/v1/episode.py,sha256=a-8IwYecfaBAD0NPbPREQWUybx-KrgLev2JHC-shjgU,2301
@@ -315,8 +315,8 @@ verifiers/v1/utils/logging.py,sha256=OcMHA6NsYux3oIzjPuI95rDWmFBHNcHDjZeNIhXTX-Y
 verifiers/v1/utils/memory.py,sha256=ZkIvGk6uITAH5sKon65LifKPbvZr8mJ__PVc23FZpOQ,1835
 verifiers/v1/utils/sampling.py,sha256=JczGzBn6s3wsIrSY2Hy3hE8m6NreNgsNzhSjDikQqX0,1037
 verifiers/v1/utils/version.py,sha256=75ZtI2NHBmlb52KpcKLUpiSXp8q4bASr7uKeXoCKlT8,1582
-verifiers-0.2.2.dev12.dist-info/METADATA,sha256=BeDnNkyFfjrP8k6-vAVe8-xauUQKgE7aYy4sAmFielc,4541
-verifiers-0.2.2.dev12.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
-verifiers-0.2.2.dev12.dist-info/entry_points.txt,sha256=dF82JUYEFslR1AOW8LZfuBWeZeQoyX4wCRrduklg8-I,551
-verifiers-0.2.2.dev12.dist-info/licenses/LICENSE,sha256=v0RrUsdV3IDoZhrRce297IXS3xMHNJ-_LdLpFAUWb9k,1072
-verifiers-0.2.2.dev12.dist-info/RECORD,,
+verifiers-0.2.2.dev13.dist-info/METADATA,sha256=COLxgdKb_Q39fmAy3uwjXxCy4YGakvCyrZFF2W0p1W4,4541
+verifiers-0.2.2.dev13.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+verifiers-0.2.2.dev13.dist-info/entry_points.txt,sha256=dF82JUYEFslR1AOW8LZfuBWeZeQoyX4wCRrduklg8-I,551
+verifiers-0.2.2.dev13.dist-info/licenses/LICENSE,sha256=v0RrUsdV3IDoZhrRce297IXS3xMHNJ-_LdLpFAUWb9k,1072
+verifiers-0.2.2.dev13.dist-info/RECORD,,