divi 0.0.1.dev17__tar.gz → 0.0.1.dev21__tar.gz

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

@@ -1,10 +1,11 @@
 Metadata-Version: 2.4
 Name: divi
-Version: 0.0.1.dev17
+Version: 0.0.1.dev21
 Summary: The Agent Platform for Observability & Evaluation
 License-File: LICENSE
 Requires-Python: >=3.11
 Requires-Dist: grpcio>=1.71.0
+Requires-Dist: openai>=1.65.2
 Requires-Dist: protobuf<6.0.dev0,>=5.26.1
 Requires-Dist: pyjwt>=2.10.1
 Requires-Dist: requests>=2.32.3

{divi-0.0.1.dev17 → 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.dev17"
+__version__ = "0.0.1.dev21"
 __all__ = ["proto", "obs_openai", "observable"]

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

@@ -1,4 +1,5 @@
-from typing import Any, Dict
+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,6 +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 import NotGiven
 
 
 class DataPark(Service):
@@ -22,6 +24,17 @@ class DataPark(Service):
     def headers(self) -> Dict[str, str]:
         return {"Authorization": f"Bearer {self.token}"}
 
+    @staticmethod
+    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)
 
@@ -40,12 +53,11 @@ class DataPark(Service):
         completion: ChatCompletion,
     ) -> None:
         hex_span_id = span_id.hex()
-        print(inputs)
         self.post_concurrent(
             {
                 "/api/v1/chat/completions/input": {
                     "span_id": hex_span_id,
-                    "data": inputs,
+                    "data": self.strip_not_given(inputs),
                 },
                 "/api/v1/chat/completions": {
                     "span_id": hex_span_id,

{divi-0.0.1.dev17 → divi-0.0.1.dev21}/pyproject.toml

@@ -6,9 +6,11 @@ readme = "docs/package_readme.md"
 requires-python = ">=3.11"
 dependencies = [
     "grpcio>=1.71.0",
+    "openai>=1.65.2",
     "protobuf>=5.26.1,<6.0.dev0",
     "pyjwt>=2.10.1",
     "requests>=2.32.3",
+    "openai>=1.65.2",
 ]
 
 [build-system]
@@ -20,7 +22,6 @@ dev = [
     "cryptography>=44.0.0",
     "grpcio-tools>=1.71.0",
     "hatchling>=1.27.0",
-    "openai>=1.65.2",
     "pytest>=8.3.4",
     "python-dotenv>=1.1.0",
 ]

{divi-0.0.1.dev17 → divi-0.0.1.dev21}/uv.lock

@@ -181,6 +181,7 @@ name = "divi"
 source = { editable = "." }
 dependencies = [
     { name = "grpcio" },
+    { name = "openai" },
     { name = "protobuf" },
     { name = "pyjwt" },
     { name = "requests" },
@@ -199,6 +200,7 @@ dev = [
 [package.metadata]
 requires-dist = [
     { name = "grpcio", specifier = ">=1.71.0" },
+    { name = "openai", specifier = ">=1.65.2" },
     { name = "protobuf", specifier = ">=5.26.1,<6.0.dev0" },
     { name = "pyjwt", specifier = ">=2.10.1" },
     { name = "requests", specifier = ">=2.32.3" },