divi 0.0.1b1__py3-none-any.whl → 0.0.1b2__py3-none-any.whl

divi/__init__.py

@@ -13,5 +13,5 @@ _auth: Optional[Auth] = None
 _datapark: Optional[DataPark] = None
 _evaluator: Optional[Evaluator] = None
 
-__version__ = "0.0.1b1"
+__version__ = "0.0.1b2"
 __all__ = ["obs_openai", "observable"]

divi/signals/span.py

@@ -1,3 +1,4 @@
+import atexit
 import os
 import time
 from enum import Enum
@@ -62,6 +63,8 @@ class Span:
         """Start the span by recording the current time in nanoseconds."""
         self.start_time_unix_nano = time.time_ns()
         self.upsert_span()
+        # Register the end method to be called at exit
+        atexit.register(self.end)
 
     def end(self):
         """End the span by recording the end time in nanoseconds."""
@@ -69,6 +72,8 @@ class Span:
             raise ValueError("Span must be started before ending.")
         self.end_time_unix_nano = time.time_ns()
         self.upsert_span()
+        # Unregister the end method
+        atexit.unregister(self.end)
 
     def _add_node(self, trace_id: UUID4, parent_id: Optional[bytes] = None):
         """Add node for obs tree."""

divi/signals/trace.py

@@ -1,3 +1,4 @@
+import atexit
 from datetime import UTC, datetime
 from typing import Optional
 from uuid import uuid4
@@ -63,6 +64,8 @@ class Trace:
         """Start the trace by recording the current time in nanoseconds."""
         self.start_time = datetime.now(UTC).isoformat()
         self.upsert_trace()
+        # Register the end method to be called on exit
+        atexit.register(self.end)
 
     def end(self):
         """End the trace by recording the end time in nanoseconds."""
@@ -70,6 +73,8 @@ class Trace:
             raise ValueError("Span must be started before ending.")
         self.end_time = datetime.now(UTC).isoformat()
         self.upsert_trace()
+        # Unregister the end method to prevent multiple calls
+        atexit.unregister(self.end)
 
     def upsert_trace(self):
         """Upsert trace with datapark."""

{divi-0.0.1b1.dist-info → divi-0.0.1b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: divi
-Version: 0.0.1b1
+Version: 0.0.1b2
 Summary: The Agent Platform for Observability & Evaluation
 License-File: LICENSE
 Requires-Python: >=3.11

{divi-0.0.1b1.dist-info → divi-0.0.1b2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 divi/README.md,sha256=hdw10Mkebd1SGMeA3tPJtlOwbWZzuQ8J-N9YPEHGuXg,67
-divi/__init__.py,sha256=XTB0By729PUQY9TYgoQXrzfQ2fiNEXRsx2mrA4A-83Y,436
+divi/__init__.py,sha256=FNz-nbMRHLhT-JWowhQTK7zMajy2XY0TFPD3Jjar1sw,436
 divi/utils.py,sha256=fXkjoyo_Lh8AZliKICOP460m0czUcNQjcEcceJbaOVA,1439
 divi/decorators/__init__.py,sha256=HkyWdC1ctTsVFucCWCkj57JB4NmwONus1d2S2dUbvs4,110
 divi/decorators/collect.py,sha256=5iUxAnbHYx4ISkFg64IK_4miGdrWgbOXLJxKz8lGIv8,1074
@@ -44,9 +44,9 @@ divi/session/session.py,sha256=QxtEezI447PbtKG2U6cxL1ACae55e8nFfTufAY8pEYI,811
 divi/session/setup.py,sha256=rC1QdCxpdCOaRXmcLEQs4Yuu5UC_aRzKSaqWRPJN4Og,1390
 divi/session/teardown.py,sha256=YiBz_3yCiljMFEofZ60VmRL5sb8WA5GT7EYF8nFznZ4,133
 divi/signals/__init__.py,sha256=wfSkkCwkRsFP4aLj8aGHk_k6Y50P5yN44WWlO3XyW18,43
-divi/signals/span.py,sha256=FQWql6ivAeXGk1HPZCsCjL5mXW6S6Nn9SmOiKH4aXik,2629
-divi/signals/trace.py,sha256=IoYeTfd6x_Xmxcp4HbFSEne0d48hol4ng2Mb_AO8hZw,2144
-divi-0.0.1b1.dist-info/METADATA,sha256=-qCsxelGi6SeeA6x3aodNtiI_Z4GWcLUagFb5LWrfNQ,493
-divi-0.0.1b1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-divi-0.0.1b1.dist-info/licenses/LICENSE,sha256=5OJuZ4wMMEV0DgF0tofhAlS_KLkaUsZwwwDS2U_GwQ0,1063
-divi-0.0.1b1.dist-info/RECORD,,
+divi/signals/span.py,sha256=AQ24C8EKzGceryybX0rBe9B0z43kKaULdR6eOeOCtno,2804
+divi/signals/trace.py,sha256=US_M2CAYNGUhEa5pppRPJN8dfNA5owgHMFDK2AOrFXA,2345
+divi-0.0.1b2.dist-info/METADATA,sha256=IS0W_tN3EECkWRlX18GiUtX2idT3lYS2sAF2Z4jmoKA,493
+divi-0.0.1b2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+divi-0.0.1b2.dist-info/licenses/LICENSE,sha256=5OJuZ4wMMEV0DgF0tofhAlS_KLkaUsZwwwDS2U_GwQ0,1063
+divi-0.0.1b2.dist-info/RECORD,,