divi 0.0.1.dev0__py3-none-any.whl → 0.0.1.dev14__py3-none-any.whl

divi/signals/trace/trace.py

@@ -0,0 +1,151 @@
+import os
+import time
+from datetime import UTC, datetime
+from typing import Any, Mapping, Optional
+from uuid import uuid4
+
+from pydantic import UUID4
+from typing_extensions import TypedDict
+
+from divi.proto.common.v1.common_pb2 import KeyValue
+from divi.proto.trace.v1.trace_pb2 import Span as SpanProto
+
+
+class NullTime(TypedDict, total=False):
+    """Null time"""
+
+    Time: str
+    """Time in iso format"""
+    Valid: bool
+    """Valid"""
+
+
+class TraceSignal(TypedDict, total=False):
+    """Trace request"""
+
+    id: str
+    """Trace ID UUID4"""
+    start_time: str
+    """Start time in iso format"""
+    end_time: NullTime
+    """End time in iso format"""
+
+
+class Trace:
+    def __init__(self, session_id: UUID4):
+        self.trace_id: UUID4 = uuid4()
+        self.start_time: str | None = None
+        self.end_time: str | None = None
+        self.session_id: UUID4 = session_id
+
+    @property
+    def signal(self) -> TraceSignal:
+        if self.start_time is None:
+            raise ValueError("Trace must be started.")
+        signal = TraceSignal(
+            id=str(self.trace_id),
+            start_time=self.start_time,
+        )
+        if self.end_time is not None:
+            signal["end_time"] = NullTime(
+                Time=self.end_time,
+                Valid=True,
+            )
+        return signal
+
+    @staticmethod
+    def unix_nano_to_iso(unix_nano: int) -> str:
+        return datetime.utcfromtimestamp(unix_nano / 1e9).isoformat()
+
+    def start(self):
+        """Start the trace by recording the current time in nanoseconds."""
+        self.start_time = datetime.now(UTC).isoformat()
+
+    def end(self):
+        """End the trace by recording the end time in nanoseconds."""
+        if self.start_time is None:
+            raise ValueError("Span must be started before ending.")
+        self.end_time = datetime.now(UTC).isoformat()
+
+
+class Span:
+    KIND_MAP = {
+        "function": SpanProto.SpanKind.SPAN_KIND_FUNCTION,
+        "llm": SpanProto.SpanKind.SPAN_KIND_LLM,
+    }
+
+    def __init__(
+        self,
+        kind: str = "function",
+        name: Optional[str] = None,
+        metadata: Optional[Mapping[str, Any]] = None,
+    ):
+        # span_id is a FixedString(8)
+        self.span_id: bytes = self._generate_span_id()
+        self.name = name
+        self.kind = kind
+        self.metadata = metadata
+        self.start_time_unix_nano: int | None = None
+        self.end_time_unix_nano: int | None = None
+
+        self.trace_id: UUID4 | None = None
+        self.parent_span_id: bytes | None = None
+
+    @property
+    def signal(self) -> SpanProto:
+        signal: SpanProto = SpanProto(
+            name=self.name,
+            span_id=self.span_id,
+            kind=self._get_kind(self.kind),
+            start_time_unix_nano=self.start_time_unix_nano,
+            end_time_unix_nano=self.end_time_unix_nano,
+            trace_id=self.trace_id.bytes if self.trace_id else None,
+            parent_span_id=self.parent_span_id,
+        )
+        print(self.kind)
+        signal.metadata.extend(
+            KeyValue(key=k, value=v)
+            for k, v in (self.metadata or dict()).items()
+        )
+        return signal
+
+    @classmethod
+    def _get_kind(cls, kind: str) -> SpanProto.SpanKind:
+        if (k := cls.KIND_MAP.get(kind)) is None:
+            raise ValueError(
+                f"Unknown kind: {kind}. Now allowed: {cls.KIND_MAP.keys()}"
+            )
+        return k
+
+    @classmethod
+    def _generate_span_id(cls) -> bytes:
+        return os.urandom(8)
+
+    def start(self):
+        """Start the span by recording the current time in nanoseconds."""
+        self.start_time_unix_nano = time.time_ns()
+
+    def end(self):
+        """End the span by recording the end time in nanoseconds."""
+        if self.start_time_unix_nano is None:
+            raise ValueError("Span must be started before ending.")
+        self.end_time_unix_nano = time.time_ns()
+
+    def _as_root(self, trace_id: UUID4):
+        """Set the span as a root span."""
+        self.trace_id = trace_id
+        print("as root")
+        print(f"name: {self.name}")
+        print(f"trace_id: {self.trace_id}")
+        print(f"span_id: {self.span_id}")
+
+    def _add_parent(self, trace_id: UUID4, parent_id: bytes):
+        """Set the parent span ID."""
+        self.trace_id = trace_id
+        self.parent_span_id = parent_id
+
+        print("add parent")
+        print(f"name: {self.name}")
+        print(f"trace_id: {trace_id}")
+        print(f"span_id: {self.span_id}")
+        print(f"parent_span_id: {parent_id}")

divi/utils.py

@@ -0,0 +1,22 @@
+import inspect
+import pathlib
+from typing import Callable
+
+
+def get_server_path() -> str:
+    """Get the path to the server binary."""
+    path = pathlib.Path(__file__).parent / "bin" / "core"
+    if not path.exists():
+        raise FileNotFoundError(f"Server binary not found: {path}")
+
+    return str(path)
+
+
+def is_async(func: Callable) -> bool:
+    """Inspect function or wrapped function to see if it is async."""
+    unwrapped_func = inspect.unwrap(func)
+    return inspect.iscoroutinefunction(unwrapped_func)
+
+
+if __name__ == "__main__":
+    print(get_server_path())

divi-0.0.1.dev14.dist-info/METADATA

@@ -0,0 +1,17 @@
+Metadata-Version: 2.4
+Name: divi
+Version: 0.0.1.dev14
+Summary: The Agent Platform for Observability & Evaluation
+License-File: LICENSE
+Requires-Python: >=3.11
+Requires-Dist: grpcio>=1.69.0
+Requires-Dist: protobuf>=5.29.3
+Requires-Dist: pyjwt>=2.10.1
+Requires-Dist: requests>=2.32.3
+Description-Content-Type: text/markdown
+
+# Divine Agent
+
+Agent Platform for Observability • Evaluation • Playground
+
+> The project is still in the development stage. 😇

divi-0.0.1.dev14.dist-info/RECORD

@@ -0,0 +1,45 @@
+divi/__init__.py,sha256=RnfnWUt0BjmN6ngPQQrUIY6TGJX20NtgbB7fsq2elnQ,396
+divi/utils.py,sha256=3iVDogCjqQg0jEjhUKEuQ6vHPFp9w7kXNjSVwXt8KmI,574
+divi/config/config.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+divi/decorators/__init__.py,sha256=HkyWdC1ctTsVFucCWCkj57JB4NmwONus1d2S2dUbvs4,110
+divi/decorators/obs_openai.py,sha256=4LtQPRgJPPyTOw1318Fi04WYqjIwGX8EP0YiWrTW0dk,1003
+divi/decorators/observable.py,sha256=N_MCX_HbMGpD7v_dxnK6ewhgONGxH4XBO-i-6ybGpkc,4038
+divi/proto/common/v1/common.proto,sha256=Rx8wr0_tOtQ1NseTMnsav4ApD1MDALzQDBA2IvLRTU0,1775
+divi/proto/common/v1/common_pb2.py,sha256=br61OHQVAi6SI3baFcb5xJv2Xd-AZ04A19xeSjLNMXo,2442
+divi/proto/common/v1/common_pb2.pyi,sha256=LmTpFFLxHg2a_qPIdNXXwGEMkbiDcTJdarR9eC-6Fq8,2133
+divi/proto/core/health/v1/health_service.proto,sha256=y39MNiwYiSqxb_JPQI7AmPfNNtP6pbX-WKTseMkjWUM,369
+divi/proto/core/health/v1/health_service_pb2.py,sha256=PKAxy_z7lU1HrwDNIa7AFHha9i8gICCVrcEhpOM79E0,1999
+divi/proto/core/health/v1/health_service_pb2.pyi,sha256=VbYHHTcp3epUVg0lTSxo3xpG52ZGhA7KKMRL5LR5e1c,691
+divi/proto/core/health/v1/health_service_pb2_grpc.py,sha256=6owvQ41KPqSOQ8WjXDjYsunoDVydJfP93vZ4uVSG59s,3953
+divi/proto/metric/v1/metric.proto,sha256=YHRMLUW-MtakHuibR3PJ0s2w5KgV12kc4737iHw0DTk,585
+divi/proto/metric/v1/metric_pb2.py,sha256=uvBhyy8QpaES3Jl82yVfsGazW5654XpRnsdGlpVgIRE,1974
+divi/proto/metric/v1/metric_pb2.pyi,sha256=S7ipsojkD7QZAYefDE4b3PO99Yzc6mOdtSLxH3-b67A,1304
+divi/proto/trace/v1/trace.proto,sha256=mh1nzEgufzRTJx3p8NNute-ozEwEYwClWJTdWUGGVA8,1284
+divi/proto/trace/v1/trace_pb2.py,sha256=N2sVI20CP2NtJLiNt376bxyYrwiqhQwGHO0hJKciUTo,2650
+divi/proto/trace/v1/trace_pb2.pyi,sha256=aTkzh1Txje5ghuAjYpowDLWz8LTNBrJZej5tUHfVIow,2805
+divi/services/__init__.py,sha256=TcVJ_gKxyPIcwhT9GgttqHeyk0icW44uE285KmUiyh4,185
+divi/services/finish.py,sha256=XKPKGJ5cWd5H95G_VpIOlOZOLrcf9StoTs7ayRic2jY,173
+divi/services/init.py,sha256=JVzRQ1m1DTHXFVGUMYnsv-vRvzCO8XFdR6MjIwOL_NY,433
+divi/services/service.py,sha256=1c7JQ49BSdBipGLfVIxTHaNxTuyvVAgrvxV7lyYv_68,285
+divi/services/auth/__init__.py,sha256=PIQ9rQ0jcRqcy03a3BOY7wbzwluIRG_4kI_H4J4mRFk,74
+divi/services/auth/auth.py,sha256=dTpFnNhxbEjzOkBLPQvRblFSO5tedJsdObWTXt84MaE,631
+divi/services/auth/init.py,sha256=UXsOweatrWnuFO9ivSjEsJnHrK9YtStKYX2Nup64KEU,477
+divi/services/auth/tokman.py,sha256=V03wcV6TEy1k9o55teDVB9JZ3Ve2cJdwzWstQhWx4BQ,1070
+divi/services/core/__init__.py,sha256=FWl4ShX0AEnOyL24NEc9mNIVoQOeYO4q0qYX27L_Yyw,111
+divi/services/core/core.py,sha256=PRwPtLgrgmCrejUfKf7HJNrAhGS0paFNZ7JwDToEUAk,1264
+divi/services/core/finish.py,sha256=dIGQpVXcJY4-tKe7A1_VV3yoSHNCDPfOlUltvzvk6VI,231
+divi/services/core/init.py,sha256=e7-fgpOPglBXyEoPkgOAnpJk2ApdFbo7LPupxOb8N-w,1966
+divi/services/datapark/__init__.py,sha256=GbV1mwHE07yutgOlCIYHykSEL5KJ-ApgLutGMzu2eUE,86
+divi/services/datapark/datapark.py,sha256=J96C-K9RCaLT7oWXnMQOM1g7btfIlU7BMWReF3ePOlc,2055
+divi/services/datapark/init.py,sha256=C32f9t3eLsxcYNqEyheh6nW455G2oR0YhhdqBcbN3ec,92
+divi/session/__init__.py,sha256=6lYemv21VQCIHx-xIdi7BxXcPNxVdvE60--8ArReUew,82
+divi/session/session.py,sha256=ID7bQ4CuRPUcZ9S-DSjndiduslYFJukRLuCtChYm3wk,838
+divi/session/setup.py,sha256=jlc3ICR5hRDdrcxoSxi66FYS2TYxqrIJOMgI4i1tEJs,1972
+divi/session/teardown.py,sha256=YiBz_3yCiljMFEofZ60VmRL5sb8WA5GT7EYF8nFznZ4,133
+divi/signals/__init__.py,sha256=K1PaTAMwyBDsK6jJUg4QWy0xVJ_5MA6dlWiUyJeiSQA,44
+divi/signals/trace/__init__.py,sha256=K1PaTAMwyBDsK6jJUg4QWy0xVJ_5MA6dlWiUyJeiSQA,44
+divi/signals/trace/trace.py,sha256=y8Xio8Tp_e1tO8l_DAkUuzMrmENNKfIdklF0iWrjGYk,4547
+divi-0.0.1.dev14.dist-info/METADATA,sha256=SPHVAVGgsvGorPvooFCb2MiHdMP2leko0X17Vzvy8xU,457
+divi-0.0.1.dev14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+divi-0.0.1.dev14.dist-info/licenses/LICENSE,sha256=5OJuZ4wMMEV0DgF0tofhAlS_KLkaUsZwwwDS2U_GwQ0,1063
+divi-0.0.1.dev14.dist-info/RECORD,,

divi-0.0.1.dev14.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Kaikai
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

divi/hello.py

@@ -1,6 +0,0 @@
-def hello():
-    print("Hello from divine-agent!")
-
-
-if __name__ == "__main__":
-    hello()

divi-0.0.1.dev0.dist-info/METADATA

@@ -1,19 +0,0 @@
-Metadata-Version: 2.4
-Name: divi
-Version: 0.0.1.dev0
-Summary: The Agent Platform for Observability & Evaluation
-Requires-Python: >=3.11
-Description-Content-Type: text/markdown
-
-# SDK
-
-> Python SDK for Divine Agent
-
-## Version
-
-Use `hatch` to manage the version of the SDK, follow the progression in [supported segments](https://hatch.pypa.io/latest/version/#supported-segments)
-
-## Thanks
-
-1. [uv](https://github.com/astral-sh/uv): An extremely fast Python package and project manager, written in Rust.
-2. [hatch](https://github.com/pypa/hatch): Hatch is a modern, extensible Python project manager.

divi-0.0.1.dev0.dist-info/RECORD

@@ -1,5 +0,0 @@
-divi/__init__.py,sha256=q0MFp_9IarnNwKmPXqIsym2glyxwDg_KrcgOEMpxpYQ,93
-divi/hello.py,sha256=wMV7kaInsn9slGGrsuRhV_b1Xd7-ey53ZOY4vyIQuAw,92
-divi-0.0.1.dev0.dist-info/METADATA,sha256=ptw3iXFWm9HBgG0aQgh-rotFdVqNBxJPifCpnoUnC9o,600
-divi-0.0.1.dev0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-divi-0.0.1.dev0.dist-info/RECORD,,