divi 0.0.1.dev19__tar.gz → 0.0.1.dev21__tar.gz

{divi-0.0.1.dev19 → divi-0.0.1.dev21}/PKG-INFO

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

{divi-0.0.1.dev19 → divi-0.0.1.dev21}/divi/__init__.py

@@ -12,5 +12,5 @@ _core: Optional[Core] = None
 _auth: Optional[Auth] = None
 _datapark: Optional[DataPark] = None
 
-__version__ = "0.0.1.dev19"
+__version__ = "0.0.1.dev21"
 __all__ = ["proto", "obs_openai", "observable"]

{divi-0.0.1.dev19 → divi-0.0.1.dev21}/divi/services/datapark/datapark.py

@@ -1,4 +1,5 @@
 from typing import Any, Dict, TYPE_CHECKING
+from typing_extensions import Mapping
 
 from google.protobuf.json_format import MessageToDict
 from openai.types.chat import ChatCompletion
@@ -9,8 +10,7 @@ from divi.proto.trace.v1.trace_pb2 import ScopeSpans
 from divi.services.service import Service
 from divi.session.session import SessionSignal
 from divi.signals.trace.trace import TraceSignal
-from openai._utils import maybe_transform
-from openai.types.chat import completion_create_params
+from openai import NotGiven
 
 
 class DataPark(Service):
@@ -25,9 +25,15 @@ class DataPark(Service):
         return {"Authorization": f"Bearer {self.token}"}
 
     @staticmethod
-    def _not_given_to_none(value: Any) -> Any:
-        """Convert NotGiven to None."""
-        return
+    def strip_not_given(obj: object | None) -> object:
+        """Remove all top-level keys where their values are instances of `NotGiven`"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, Mapping):
+            return obj
+
+        return {key: value for key, value in obj.items() if not isinstance(value, NotGiven)}
 
     def create_session(self, session: SessionSignal) -> None:
         self.post("/api/session/", payload=session)
@@ -51,7 +57,7 @@ class DataPark(Service):
             {
                 "/api/v1/chat/completions/input": {
                     "span_id": hex_span_id,
-                    "data": maybe_transform(inputs, completion_create_params.CompletionCreateParams)
+                    "data": self.strip_not_given(inputs),
                 },
                 "/api/v1/chat/completions": {
                     "span_id": hex_span_id,