docent-python 0.1.14a0__tar.gz → 0.1.16a0__tar.gz

{docent_python-0.1.14a0 → docent_python-0.1.16a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docent-python
-Version: 0.1.14a0
+Version: 0.1.16a0
 Summary: Docent SDK
 Project-URL: Homepage, https://github.com/TransluceAI/docent
 Project-URL: Issues, https://github.com/TransluceAI/docent/issues
@@ -9,6 +9,8 @@ Author-email: Transluce <info@transluce.org>
 License-Expression: Apache-2.0
 License-File: LICENSE.md
 Requires-Python: >=3.11
+Requires-Dist: backoff>=2.2.1
+Requires-Dist: inspect-ai>=0.3.132
 Requires-Dist: opentelemetry-api>=1.34.1
 Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.34.1
 Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.34.1

{docent_python-0.1.14a0 → docent_python-0.1.16a0}/docent/sdk/agent_run_writer.py

@@ -105,9 +105,17 @@ class AgentRunWriter:
         # Register shutdown hooks
         atexit.register(self.finish)
 
+        def _handle_sigint(s: int, f: object) -> None:
+            self._shutdown()
+            raise KeyboardInterrupt
+
+        def _handle_sigterm(s: int, f: object) -> None:
+            self._shutdown()
+            raise SystemExit(0)
+
         # Register signal handlers for graceful shutdown
-        signal.signal(signal.SIGINT, lambda s, f: self._shutdown())  # Ctrl+C
-        signal.signal(signal.SIGTERM, lambda s, f: self._shutdown())  # Kill signal
+        signal.signal(signal.SIGINT, _handle_sigint)  # Ctrl+C
+        signal.signal(signal.SIGTERM, _handle_sigterm)  # Kill signal
 
     def log_agent_runs(self, agent_runs: list[AgentRun]) -> None:
         """Put a list of AgentRun objects into the queue.

{docent_python-0.1.14a0 → docent_python-0.1.16a0}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "docent-python"
 description = "Docent SDK"
-version = "0.1.14-alpha"
+version = "0.1.16-alpha"
 authors = [
   { name="Transluce", email="info@transluce.org" },
 ]
@@ -14,6 +14,8 @@ dependencies = [
     "pyyaml>=6.0.2",
     "tiktoken>=0.7.0",
     "tqdm>=4.67.1",
+    "backoff>=2.2.1",
+    "inspect-ai>=0.3.132",
     # tracing
     "opentelemetry-api>=1.34.1",
     "opentelemetry-sdk>=1.34.1",