divi 0.0.1.dev12__py3-none-macosx_11_0_arm64.whl → 0.0.1.dev13__py3-none-macosx_11_0_arm64.whl

divi/__init__.py

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

divi/decorators/observable.py

@@ -1,32 +1,100 @@
+import contextvars
 import functools
 import inspect
-from typing import Any, Callable, List, overload
+from typing import (
+    Any,
+    Callable,
+    Generic,
+    List,
+    Mapping,
+    Optional,
+    ParamSpec,
+    Protocol,
+    TypeVar,
+    Union,
+    overload,
+    runtime_checkable,
+)
+
+from divi.run import RunExtra
+from divi.run.setup import setup
+from divi.signals.trace import Span
+
+R = TypeVar("R", covariant=True)
+P = ParamSpec("P")
+
+# ContextVar to store the extra information
+# from the Run and parent Span
+_RUNEXTRA = contextvars.ContextVar[Optional[RunExtra]](
+    "_RUNEXTRA", default=None
+)
+
+
+@runtime_checkable
+class WithRunExtra(Protocol, Generic[P, R]):
+    def __call__(
+        self,
+        *args: P.args,
+        run_extra: Optional[RunExtra] = None,  # type: ignore[valid-type]
+        **kwargs: P.kwargs,
+    ) -> R: ...
 
 
 @overload
-def observable(func: Callable) -> Callable: ...
+def observable(func: Callable[P, R]) -> WithRunExtra[P, R]: ...
 
 
 @overload
-def observable() -> Callable: ...
+def observable(
+    kind: str = "function",
+    *,
+    name: Optional[str] = None,
+    metadata: Optional[Mapping[str, Any]] = None,
+) -> Callable[[Callable[P, R]], WithRunExtra[P, R]]: ...
 
 
-def observable(*args, **kwargs) -> Callable:
+def observable(
+    *args, **kwargs
+) -> Union[Callable, Callable[[Callable], Callable]]:
     """Observable decorator factory."""
 
-    def decorator(func: Callable):
+    kind = kwargs.pop("kind", "function")
+    name = kwargs.pop("name", None)
+    metadata = kwargs.pop("metadata", None)
+
+    def decorator(func):
+        span = Span(kind=kind, name=name or func.__name__, metadata=metadata)
+
         @functools.wraps(func)
-        def wrapper(*args, **kwargs):
+        def wrapper(*args, run_extra: Optional[RunExtra] = None, **kwargs):
+            run_extra = setup(span, _RUNEXTRA.get() or run_extra)
+            # set current context
+            token = _RUNEXTRA.set(run_extra)
+            # execute the function
+            span.start()
             result = func(*args, **kwargs)
+            span.end()
+            # recover parent context
+            _RUNEXTRA.reset(token)
             # TODO: collect result
             return result
 
         @functools.wraps(func)
-        def generator_wrapper(*args, **kwargs):
+        def generator_wrapper(
+            *args, run_extra: Optional[RunExtra] = None, **kwargs
+        ):
+            run_extra = setup(span, _RUNEXTRA.get() or run_extra)
+            # set current context
+            token = _RUNEXTRA.set(run_extra)
+            # execute the function
             results: List[Any] = []
+            span.start()
             for item in func(*args, **kwargs):
                 results.append(item)
                 yield item
+            span.end()
+            # recover parent context
+            _RUNEXTRA.reset(token)
             # TODO: collect results
 
         if inspect.isgeneratorfunction(func):

divi/proto/common/v1/common.proto

@@ -48,12 +48,12 @@ message KeyValue {
   AnyValue value = 2;
 }
 
-// InstrumentationScope is a message representing the instrumentation scope information
-// such as the fully qualified name and version.
-message InstrumentationScope {
-  // An empty instrumentation scope name means the name is unknown.
-  string name = 1;
-  string version = 2;
+// RunScope specifies the scope of the message
+message RunScope {
+  // The run_id is a unique identifier that represents a run. It is a 16-byte array.
+  bytes run_id = 1;
+  // The name of the run.
+  string run_name = 2;
 
   // Additional attributes that describe the scope. [Optional].
   // Attribute keys MUST be unique (it is not allowed to have more than one

divi/proto/common/v1/common_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: divi/proto/common/v1/common.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'divi/proto/common/v1/common.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!divi/proto/common/v1/common.proto\x12\x14\x64ivi.proto.common.v1\"\xfa\x01\n\x08\x41nyValue\x12\x16\n\x0cstring_value\x18\x01 \x01(\tH\x00\x12\x14\n\nbool_value\x18\x02 \x01(\x08H\x00\x12\x13\n\tint_value\x18\x03 \x01(\x03H\x00\x12\x16\n\x0c\x64ouble_value\x18\x04 \x01(\x01H\x00\x12\x37\n\x0b\x61rray_value\x18\x05 \x01(\x0b\x32 .divi.proto.common.v1.ArrayValueH\x00\x12:\n\x0ckvlist_value\x18\x06 \x01(\x0b\x32\".divi.proto.common.v1.KeyValueListH\x00\x12\x15\n\x0b\x62ytes_value\x18\x07 \x01(\x0cH\x00\x42\x07\n\x05value\"<\n\nArrayValue\x12.\n\x06values\x18\x01 \x03(\x0b\x32\x1e.divi.proto.common.v1.AnyValue\">\n\x0cKeyValueList\x12.\n\x06values\x18\x01 \x03(\x0b\x32\x1e.divi.proto.common.v1.KeyValue\"F\n\x08KeyValue\x12\x0b\n\x03key\x18\x01 \x01(\t\x12-\n\x05value\x18\x02 \x01(\x0b\x32\x1e.divi.proto.common.v1.AnyValue\"\x82\x01\n\x08RunScope\x12\x0e\n\x06run_id\x18\x01 \x01(\x0c\x12\x10\n\x08run_name\x18\x02 \x01(\t\x12\x32\n\nattributes\x18\x03 \x03(\x0b\x32\x1e.divi.proto.common.v1.KeyValue\x12 \n\x18\x64ropped_attributes_count\x18\x04 \x01(\rB\x10Z\x0eservices/protob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'divi.proto.common.v1.common_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\016services/proto'
+  _globals['_ANYVALUE']._serialized_start=60
+  _globals['_ANYVALUE']._serialized_end=310
+  _globals['_ARRAYVALUE']._serialized_start=312
+  _globals['_ARRAYVALUE']._serialized_end=372
+  _globals['_KEYVALUELIST']._serialized_start=374
+  _globals['_KEYVALUELIST']._serialized_end=436
+  _globals['_KEYVALUE']._serialized_start=438
+  _globals['_KEYVALUE']._serialized_end=508
+  _globals['_RUNSCOPE']._serialized_start=511
+  _globals['_RUNSCOPE']._serialized_end=641
+# @@protoc_insertion_point(module_scope)

divi/proto/common/v1/common_pb2.pyi

@@ -0,0 +1,56 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class AnyValue(_message.Message):
+    __slots__ = ("string_value", "bool_value", "int_value", "double_value", "array_value", "kvlist_value", "bytes_value")
+    STRING_VALUE_FIELD_NUMBER: _ClassVar[int]
+    BOOL_VALUE_FIELD_NUMBER: _ClassVar[int]
+    INT_VALUE_FIELD_NUMBER: _ClassVar[int]
+    DOUBLE_VALUE_FIELD_NUMBER: _ClassVar[int]
+    ARRAY_VALUE_FIELD_NUMBER: _ClassVar[int]
+    KVLIST_VALUE_FIELD_NUMBER: _ClassVar[int]
+    BYTES_VALUE_FIELD_NUMBER: _ClassVar[int]
+    string_value: str
+    bool_value: bool
+    int_value: int
+    double_value: float
+    array_value: ArrayValue
+    kvlist_value: KeyValueList
+    bytes_value: bytes
+    def __init__(self, string_value: _Optional[str] = ..., bool_value: bool = ..., int_value: _Optional[int] = ..., double_value: _Optional[float] = ..., array_value: _Optional[_Union[ArrayValue, _Mapping]] = ..., kvlist_value: _Optional[_Union[KeyValueList, _Mapping]] = ..., bytes_value: _Optional[bytes] = ...) -> None: ...
+
+class ArrayValue(_message.Message):
+    __slots__ = ("values",)
+    VALUES_FIELD_NUMBER: _ClassVar[int]
+    values: _containers.RepeatedCompositeFieldContainer[AnyValue]
+    def __init__(self, values: _Optional[_Iterable[_Union[AnyValue, _Mapping]]] = ...) -> None: ...
+
+class KeyValueList(_message.Message):
+    __slots__ = ("values",)
+    VALUES_FIELD_NUMBER: _ClassVar[int]
+    values: _containers.RepeatedCompositeFieldContainer[KeyValue]
+    def __init__(self, values: _Optional[_Iterable[_Union[KeyValue, _Mapping]]] = ...) -> None: ...
+
+class KeyValue(_message.Message):
+    __slots__ = ("key", "value")
+    KEY_FIELD_NUMBER: _ClassVar[int]
+    VALUE_FIELD_NUMBER: _ClassVar[int]
+    key: str
+    value: AnyValue
+    def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AnyValue, _Mapping]] = ...) -> None: ...
+
+class RunScope(_message.Message):
+    __slots__ = ("run_id", "run_name", "attributes", "dropped_attributes_count")
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    RUN_NAME_FIELD_NUMBER: _ClassVar[int]
+    ATTRIBUTES_FIELD_NUMBER: _ClassVar[int]
+    DROPPED_ATTRIBUTES_COUNT_FIELD_NUMBER: _ClassVar[int]
+    run_id: bytes
+    run_name: str
+    attributes: _containers.RepeatedCompositeFieldContainer[KeyValue]
+    dropped_attributes_count: int
+    def __init__(self, run_id: _Optional[bytes] = ..., run_name: _Optional[str] = ..., attributes: _Optional[_Iterable[_Union[KeyValue, _Mapping]]] = ..., dropped_attributes_count: _Optional[int] = ...) -> None: ...

divi/proto/core/health/v1/health_service_pb2.py

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: divi/proto/core/health/v1/health_service.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'divi/proto/core/health/v1/health_service.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.divi/proto/core/health/v1/health_service.proto\x12\x19\x64ivi.proto.core.health.v1\"%\n\x12HealthCheckRequest\x12\x0f\n\x07version\x18\x01 \x01(\t\"6\n\x13HealthCheckResponse\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t2y\n\rHealthService\x12h\n\x05\x43heck\x12-.divi.proto.core.health.v1.HealthCheckRequest\x1a..divi.proto.core.health.v1.HealthCheckResponse\"\x00\x42\x10Z\x0eservices/protob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'divi.proto.core.health.v1.health_service_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\016services/proto'
+  _globals['_HEALTHCHECKREQUEST']._serialized_start=77
+  _globals['_HEALTHCHECKREQUEST']._serialized_end=114
+  _globals['_HEALTHCHECKRESPONSE']._serialized_start=116
+  _globals['_HEALTHCHECKRESPONSE']._serialized_end=170
+  _globals['_HEALTHSERVICE']._serialized_start=172
+  _globals['_HEALTHSERVICE']._serialized_end=293
+# @@protoc_insertion_point(module_scope)

divi/proto/core/health/v1/health_service_pb2.pyi

@@ -0,0 +1,19 @@
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Optional as _Optional
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class HealthCheckRequest(_message.Message):
+    __slots__ = ("version",)
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    version: str
+    def __init__(self, version: _Optional[str] = ...) -> None: ...
+
+class HealthCheckResponse(_message.Message):
+    __slots__ = ("status", "message")
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    status: bool
+    message: str
+    def __init__(self, status: bool = ..., message: _Optional[str] = ...) -> None: ...

divi/proto/core/health/v1/health_service_pb2_grpc.py

@@ -0,0 +1,100 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from divi.proto.core.health.v1 import health_service_pb2 as divi_dot_proto_dot_core_dot_health_dot_v1_dot_health__service__pb2
+
+GRPC_GENERATED_VERSION = '1.69.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in divi/proto/core/health/v1/health_service_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class HealthServiceStub(object):
+    """HealthService is a service that implements health check.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.Check = channel.unary_unary(
+                '/divi.proto.core.health.v1.HealthService/Check',
+                request_serializer=divi_dot_proto_dot_core_dot_health_dot_v1_dot_health__service__pb2.HealthCheckRequest.SerializeToString,
+                response_deserializer=divi_dot_proto_dot_core_dot_health_dot_v1_dot_health__service__pb2.HealthCheckResponse.FromString,
+                _registered_method=True)
+
+
+class HealthServiceServicer(object):
+    """HealthService is a service that implements health check.
+    """
+
+    def Check(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_HealthServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'Check': grpc.unary_unary_rpc_method_handler(
+                    servicer.Check,
+                    request_deserializer=divi_dot_proto_dot_core_dot_health_dot_v1_dot_health__service__pb2.HealthCheckRequest.FromString,
+                    response_serializer=divi_dot_proto_dot_core_dot_health_dot_v1_dot_health__service__pb2.HealthCheckResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'divi.proto.core.health.v1.HealthService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('divi.proto.core.health.v1.HealthService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class HealthService(object):
+    """HealthService is a service that implements health check.
+    """
+
+    @staticmethod
+    def Check(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/divi.proto.core.health.v1.HealthService/Check',
+            divi_dot_proto_dot_core_dot_health_dot_v1_dot_health__service__pb2.HealthCheckRequest.SerializeToString,
+            divi_dot_proto_dot_core_dot_health_dot_v1_dot_health__service__pb2.HealthCheckResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

divi/proto/metric/v1/metric.proto

@@ -0,0 +1,29 @@
+syntax = "proto3";
+
+package divi.proto.metric.v1;
+
+import "divi/proto/common/v1/common.proto";
+
+option go_package = "services/proto";
+
+message ScopeMetrics {
+    // The scope of the message
+    divi.proto.common.v1.RunScope scope = 1;
+
+    // A list of spans that originate from a resource.
+    repeated Metric metrics = 2;
+}
+
+message Metric {
+    // The name of the metric.
+    string name = 1;
+
+    // The description of the metric.
+    string description = 2;
+
+    // The data of the metric.
+    divi.proto.common.v1.AnyValue data = 3;
+
+    // The metadata is a set of attributes that describe the span.
+    repeated divi.proto.common.v1.KeyValue metadata = 4;
+}

divi/proto/metric/v1/metric_pb2.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: divi/proto/metric/v1/metric.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'divi/proto/metric/v1/metric.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from divi.proto.common.v1 import common_pb2 as divi_dot_proto_dot_common_dot_v1_dot_common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!divi/proto/metric/v1/metric.proto\x12\x14\x64ivi.proto.metric.v1\x1a!divi/proto/common/v1/common.proto\"l\n\x0cScopeMetrics\x12-\n\x05scope\x18\x01 \x01(\x0b\x32\x1e.divi.proto.common.v1.RunScope\x12-\n\x07metrics\x18\x02 \x03(\x0b\x32\x1c.divi.proto.metric.v1.Metric\"\x8b\x01\n\x06Metric\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12,\n\x04\x64\x61ta\x18\x03 \x01(\x0b\x32\x1e.divi.proto.common.v1.AnyValue\x12\x30\n\x08metadata\x18\x04 \x03(\x0b\x32\x1e.divi.proto.common.v1.KeyValueB\x10Z\x0eservices/protob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'divi.proto.metric.v1.metric_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\016services/proto'
+  _globals['_SCOPEMETRICS']._serialized_start=94
+  _globals['_SCOPEMETRICS']._serialized_end=202
+  _globals['_METRIC']._serialized_start=205
+  _globals['_METRIC']._serialized_end=344
+# @@protoc_insertion_point(module_scope)

divi/proto/metric/v1/metric_pb2.pyi

@@ -0,0 +1,27 @@
+from divi.proto.common.v1 import common_pb2 as _common_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class ScopeMetrics(_message.Message):
+    __slots__ = ("scope", "metrics")
+    SCOPE_FIELD_NUMBER: _ClassVar[int]
+    METRICS_FIELD_NUMBER: _ClassVar[int]
+    scope: _common_pb2.RunScope
+    metrics: _containers.RepeatedCompositeFieldContainer[Metric]
+    def __init__(self, scope: _Optional[_Union[_common_pb2.RunScope, _Mapping]] = ..., metrics: _Optional[_Iterable[_Union[Metric, _Mapping]]] = ...) -> None: ...
+
+class Metric(_message.Message):
+    __slots__ = ("name", "description", "data", "metadata")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    DATA_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    description: str
+    data: _common_pb2.AnyValue
+    metadata: _containers.RepeatedCompositeFieldContainer[_common_pb2.KeyValue]
+    def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., data: _Optional[_Union[_common_pb2.AnyValue, _Mapping]] = ..., metadata: _Optional[_Iterable[_Union[_common_pb2.KeyValue, _Mapping]]] = ...) -> None: ...

divi/proto/run/v1/run.proto

@@ -0,0 +1,38 @@
+syntax = "proto3";
+
+package divi.proto.run.v1;
+
+import "divi/proto/common/v1/common.proto";
+
+option go_package = "services/proto";
+
+message Run {
+    // The run_id is the unique identifier of the run. It is a 16-byte array.
+    bytes run_id = 1;
+
+    // The name of the run.
+    string name = 2;
+
+    enum RunKind {
+        // Observation represents a run that is used for observation.
+        Observation = 0;
+
+        // Evaluation represents a run that is used for evaluation.
+        Evaluation = 1;
+
+        // Dataset represents a run that is used for creating a dataset.
+        Dataset = 2;
+    }
+
+    // The kind of the run.
+    RunKind kind = 3;
+
+    // The start_time_unix_nano is the start time of the run in Unix nanoseconds.
+    fixed64 start_time_unix_nano = 4;
+
+    // The end_time_unix_nano is the end time of the run in Unix nanoseconds.
+    fixed64 end_time_unix_nano = 5;
+
+    // The metadata is a set of attributes that describe the run.
+    repeated divi.proto.common.v1.KeyValue metadata = 6;
+}

divi/proto/run/v1/run_pb2.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: divi/proto/run/v1/run.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'divi/proto/run/v1/run.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from divi.proto.common.v1 import common_pb2 as divi_dot_proto_dot_common_dot_v1_dot_common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x64ivi/proto/run/v1/run.proto\x12\x11\x64ivi.proto.run.v1\x1a!divi/proto/common/v1/common.proto\"\x87\x02\n\x03Run\x12\x0f\n\x07user_id\x18\x01 \x01(\x0c\x12\x0e\n\x06run_id\x18\x02 \x01(\x0c\x12\x0c\n\x04name\x18\x03 \x01(\t\x12,\n\x04kind\x18\x04 \x01(\x0e\x32\x1e.divi.proto.run.v1.Run.RunKind\x12\x1c\n\x14start_time_unix_nano\x18\x05 \x01(\x06\x12\x1a\n\x12\x65nd_time_unix_nano\x18\x06 \x01(\x06\x12\x30\n\x08metadata\x18\x07 \x03(\x0b\x32\x1e.divi.proto.common.v1.KeyValue\"7\n\x07RunKind\x12\x0f\n\x0bObservation\x10\x00\x12\x0e\n\nEvaluation\x10\x01\x12\x0b\n\x07\x44\x61taset\x10\x02\x42\x10Z\x0eservices/protob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'divi.proto.run.v1.run_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\016services/proto'
+  _globals['_RUN']._serialized_start=86
+  _globals['_RUN']._serialized_end=349
+  _globals['_RUN_RUNKIND']._serialized_start=294
+  _globals['_RUN_RUNKIND']._serialized_end=349
+# @@protoc_insertion_point(module_scope)

divi/proto/run/v1/run_pb2.pyi

@@ -0,0 +1,34 @@
+from divi.proto.common.v1 import common_pb2 as _common_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class Run(_message.Message):
+    __slots__ = ("user_id", "run_id", "name", "kind", "start_time_unix_nano", "end_time_unix_nano", "metadata")
+    class RunKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        Observation: _ClassVar[Run.RunKind]
+        Evaluation: _ClassVar[Run.RunKind]
+        Dataset: _ClassVar[Run.RunKind]
+    Observation: Run.RunKind
+    Evaluation: Run.RunKind
+    Dataset: Run.RunKind
+    USER_ID_FIELD_NUMBER: _ClassVar[int]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    START_TIME_UNIX_NANO_FIELD_NUMBER: _ClassVar[int]
+    END_TIME_UNIX_NANO_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    user_id: bytes
+    run_id: bytes
+    name: str
+    kind: Run.RunKind
+    start_time_unix_nano: int
+    end_time_unix_nano: int
+    metadata: _containers.RepeatedCompositeFieldContainer[_common_pb2.KeyValue]
+    def __init__(self, user_id: _Optional[bytes] = ..., run_id: _Optional[bytes] = ..., name: _Optional[str] = ..., kind: _Optional[_Union[Run.RunKind, str]] = ..., start_time_unix_nano: _Optional[int] = ..., end_time_unix_nano: _Optional[int] = ..., metadata: _Optional[_Iterable[_Union[_common_pb2.KeyValue, _Mapping]]] = ...) -> None: ...

divi/proto/trace/v1/trace.proto

@@ -2,17 +2,49 @@ syntax = "proto3";
 
 package divi.proto.trace.v1;
 
-import "divi/proto/resource/v1/resource.proto";
+import "divi/proto/common/v1/common.proto";
 
 option go_package = "services/proto";
 
-message TracesData {
-    // An array of ResourceSpans.
-    // For data coming from a single resource this array will typically contain one element.
-    repeated ResourceSpans resource_spans = 1;
+message ScopeSpans {
+    // The scope of the message
+    divi.proto.common.v1.RunScope scope = 1;
+
+    // A list of spans that originate from a resource.
+    repeated Span spans = 2;
 }
 
-message ResourceSpans {
-    // The resource for which the spans are reported.
-    divi.proto.resource.v1.Resource resource = 1;
+message Span {
+    // The trace_id is a unique identifier that represents a trace. It is a 16-byte array.
+    bytes trace_id = 1;
+
+    // The span_id is a unique identifier that represents a span. It is an 8-byte array.
+    bytes span_id = 2;
+
+    // The parent_span_id is the span_id of the parent span. It is an 8-byte array.
+    bytes parent_span_id = 3;
+
+    // The name of the span.
+    string name = 4;
+
+    // SpanKind is the type of the span.
+    enum SpanKind {
+        // Function represents a function call.
+        SPAN_KIND_FUNCTION = 0;
+
+        // LLM represents a llm api call.
+        SPAN_KIND_LLM = 1;
+    }
+
+    // The kind of the span.
+    SpanKind kind = 5;
+
+    // The start_time_unix_nano is the start time of the span in Unix nanoseconds.
+    fixed64 start_time_unix_nano = 6;
+
+    // The end_time_unix_nano is the end time of the span in Unix nanoseconds.
+    fixed64 end_time_unix_nano = 7;
+
+    // The metadata is a set of attributes that describe the span.
+    repeated divi.proto.common.v1.KeyValue metadata = 8;
 }

divi/proto/trace/v1/trace_pb2.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: divi/proto/trace/v1/trace.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'divi/proto/trace/v1/trace.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from divi.proto.common.v1 import common_pb2 as divi_dot_proto_dot_common_dot_v1_dot_common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1f\x64ivi/proto/trace/v1/trace.proto\x12\x13\x64ivi.proto.trace.v1\x1a!divi/proto/common/v1/common.proto\"e\n\nScopeSpans\x12-\n\x05scope\x18\x01 \x01(\x0b\x32\x1e.divi.proto.common.v1.RunScope\x12(\n\x05spans\x18\x02 \x03(\x0b\x32\x19.divi.proto.trace.v1.Span\"\xa4\x02\n\x04Span\x12\x10\n\x08trace_id\x18\x01 \x01(\x0c\x12\x0f\n\x07span_id\x18\x02 \x01(\x0c\x12\x16\n\x0eparent_span_id\x18\x03 \x01(\x0c\x12\x0c\n\x04name\x18\x04 \x01(\t\x12\x30\n\x04kind\x18\x05 \x01(\x0e\x32\".divi.proto.trace.v1.Span.SpanKind\x12\x1c\n\x14start_time_unix_nano\x18\x06 \x01(\x06\x12\x1a\n\x12\x65nd_time_unix_nano\x18\x07 \x01(\x06\x12\x30\n\x08metadata\x18\x08 \x03(\x0b\x32\x1e.divi.proto.common.v1.KeyValue\"5\n\x08SpanKind\x12\x16\n\x12SPAN_KIND_FUNCTION\x10\x00\x12\x11\n\rSPAN_KIND_LLM\x10\x01\x42\x10Z\x0eservices/protob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'divi.proto.trace.v1.trace_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\016services/proto'
+  _globals['_SCOPESPANS']._serialized_start=91
+  _globals['_SCOPESPANS']._serialized_end=192
+  _globals['_SPAN']._serialized_start=195
+  _globals['_SPAN']._serialized_end=487
+  _globals['_SPAN_SPANKIND']._serialized_start=434
+  _globals['_SPAN_SPANKIND']._serialized_end=487
+# @@protoc_insertion_point(module_scope)

divi/proto/trace/v1/trace_pb2.pyi

@@ -0,0 +1,42 @@
+from divi.proto.common.v1 import common_pb2 as _common_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class ScopeSpans(_message.Message):
+    __slots__ = ("scope", "spans")
+    SCOPE_FIELD_NUMBER: _ClassVar[int]
+    SPANS_FIELD_NUMBER: _ClassVar[int]
+    scope: _common_pb2.RunScope
+    spans: _containers.RepeatedCompositeFieldContainer[Span]
+    def __init__(self, scope: _Optional[_Union[_common_pb2.RunScope, _Mapping]] = ..., spans: _Optional[_Iterable[_Union[Span, _Mapping]]] = ...) -> None: ...
+
+class Span(_message.Message):
+    __slots__ = ("trace_id", "span_id", "parent_span_id", "name", "kind", "start_time_unix_nano", "end_time_unix_nano", "metadata")
+    class SpanKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        SPAN_KIND_FUNCTION: _ClassVar[Span.SpanKind]
+        SPAN_KIND_LLM: _ClassVar[Span.SpanKind]
+    SPAN_KIND_FUNCTION: Span.SpanKind
+    SPAN_KIND_LLM: Span.SpanKind
+    TRACE_ID_FIELD_NUMBER: _ClassVar[int]
+    SPAN_ID_FIELD_NUMBER: _ClassVar[int]
+    PARENT_SPAN_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    START_TIME_UNIX_NANO_FIELD_NUMBER: _ClassVar[int]
+    END_TIME_UNIX_NANO_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    trace_id: bytes
+    span_id: bytes
+    parent_span_id: bytes
+    name: str
+    kind: Span.SpanKind
+    start_time_unix_nano: int
+    end_time_unix_nano: int
+    metadata: _containers.RepeatedCompositeFieldContainer[_common_pb2.KeyValue]
+    def __init__(self, trace_id: _Optional[bytes] = ..., span_id: _Optional[bytes] = ..., parent_span_id: _Optional[bytes] = ..., name: _Optional[str] = ..., kind: _Optional[_Union[Span.SpanKind, str]] = ..., start_time_unix_nano: _Optional[int] = ..., end_time_unix_nano: _Optional[int] = ..., metadata: _Optional[_Iterable[_Union[_common_pb2.KeyValue, _Mapping]]] = ...) -> None: ...

divi/run/__init__.py

@@ -1,3 +1,3 @@
-from .run import Run
+from .run import Run, RunExtra
 
-__all__ = ["Run"]
+__all__ = ["RunExtra", "Run"]

divi/run/run.py

@@ -1,2 +1,24 @@
+from typing import Optional, TypedDict
+from uuid import uuid4
+
+from pydantic import UUID4
+
+
+class RunExtra(TypedDict, total=False):
+    """Extra information for Run"""
+
+    run_name: Optional[str]
+    """Name of the Run"""
+    trace_id: Optional[UUID4]
+    """Trace ID UUID4"""
+    parent_span_id: Optional[UUID4]
+    """Parent Span ID UUID4"""
+
+
 class Run:
-    pass
+    def __init__(
+        self,
+        name: Optional[str] = None,
+    ):
+        self.id = uuid4()
+        self.name = name

divi/run/setup.py

@@ -0,0 +1,50 @@
+import divi
+from divi.run import Run, RunExtra
+from divi.signals.trace import Span
+
+
+def init(run_extra: RunExtra) -> Run:
+    """init initializes the services and the Run"""
+    # init_services()
+    return Run(name=run_extra.get("run_name"))
+
+
+def setup(
+    span: Span,
+    run_extra: RunExtra | None,
+):
+    """setup trace
+
+    Args:
+        span (Span): Span instance
+        run_extra (RunExtra | None): Extra information from user input
+    """
+    # TOOD: merge run_extra input by user with the one from the context
+    # temp solution: Priority: run_extra_context.get() > run_extra
+    run_extra = run_extra or RunExtra()
+
+    # init the Run if not already initialized
+    if not divi._run:
+        divi._run = init(run_extra=run_extra)
+
+    # setup current span
+    trace_id = run_extra.get("trace_id")
+    parent_span_id = run_extra.get("parent_span_id")
+    if trace_id and parent_span_id:
+        span._add_parent(trace_id, parent_span_id)
+    else:
+        span._as_root()
+
+    # set _RUNEXTRA
+    run_extra = RunExtra(
+        run_name=divi._run.name,
+        trace_id=span.trace_id,
+        # set the parent_span_id to the current span_id
+        parent_span_id=span.span_id,
+    )
+
+    # offer end hook to collect data at whe end of the span ?
+    # offer hook to reset the context with the token
+    # context = copy_context()
+    # context.run(run_extra_context.set, run_extra)
+    return run_extra

divi/run/teardown.py

@@ -0,0 +1,7 @@
+import divi
+from divi.services import finish as clean_up_services
+
+
+def teardown():
+    clean_up_services()
+    divi._run = None

divi/services/auth/init.py

@@ -1,19 +1,16 @@
 import os
 from typing import Optional
 
-import divi
 from divi.services.auth import Auth
 
 DIVI_API_KEY = "DIVI_API_KEY"
 
 
-def init(api_key: Optional[str] = None) -> Optional[Auth]:
+def init(api_key: Optional[str] = None) -> Auth:
     key = api_key if api_key else os.getenv(DIVI_API_KEY)
     if not key:
         raise ValueError("API key is required")
-    divi._auth = Auth(api_key=key)
-    # TODO: Test the token
-    return divi._auth
+    return Auth(api_key=key)
 
 
 if __name__ == "__main__":

divi/services/core/core.py

@@ -4,8 +4,8 @@ from typing import Callable, List, Optional
 import grpc
 
 import divi
-from divi.proto.core_pb2_grpc import CoreStub
-from divi.proto.health_pb2 import HealthCheckRequest
+from divi.proto.core.health.v1.health_service_pb2 import HealthCheckRequest
+from divi.proto.core.health.v1.health_service_pb2_grpc import HealthServiceStub
 from divi.services.service import Service
 
 
@@ -20,7 +20,7 @@ class Core(Service):
     def check_health(self) -> bool:
         """Check the health of the service."""
         with grpc.insecure_channel(self.target) as channel:
-            stub = CoreStub(channel)
+            stub = HealthServiceStub(channel)
             response, call = stub.Check.with_call(
                 HealthCheckRequest(version=divi.__version__),
                 # Note: ((),) notice the `,` at the end of the tuple

divi/services/core/init.py

@@ -2,28 +2,21 @@ import atexit
 import socket
 import subprocess
 import time
-from typing import Optional
 
 import grpc
 
-import divi
 from divi.services.core import Core
 from divi.utils import get_server_path
 
 
-def init(host="localhost", port=50051) -> Optional[Core]:
-    divi._core = Core(host=host, port=port)
-    _start_server()
-    return divi._core
+def init(host="localhost", port=50051) -> Core:
+    core = Core(host=host, port=port)
+    _start_server(core)
+    return core
 
 
-def _start_server():
+def _start_server(core: Core):
     """Start the backend server."""
-    # get the run object
-    core = divi._core
-    if core is None:
-        return
-
     # start the server
     bin_path = get_server_path()
     command = [bin_path]

divi/services/datapark/__init__.py

@@ -1,3 +1,4 @@
 from .datapark import DataPark
+from .init import init
 
-__all__ = ["DataPark"]
+__all__ = ["DataPark", "init"]

divi/services/datapark/init.py

@@ -1,9 +1,5 @@
-from typing import Optional
-
-import divi
 from divi.services.datapark import DataPark
 
 
-def init() -> Optional[DataPark]:
-    divi._datapark = DataPark()
-    return divi._datapark
+def init() -> DataPark:
+    return DataPark()

divi/services/finish.py

@@ -1,5 +1,9 @@
+import divi
 from divi.services.core import finish as clean_up_core
 
 
 def finish():
     clean_up_core()
+    divi._auth = None
+    divi._core = None
+    divi._datapark = None

divi/services/init.py

@@ -1,7 +1,13 @@
+import divi
 from divi.services.auth import init as init_auth
 from divi.services.core import init as init_core
+from divi.services.datapark import init as init_datapark
 
 
 def init():
-    init_core()
-    init_auth()
+    if not divi._auth:
+        divi._auth = init_auth()
+    if not divi._core:
+        divi._core = init_core()
+    if not divi._datapark:
+        divi._datapark = init_datapark()

divi/signals/__init__.py

@@ -1,4 +1,3 @@
-from .metric import Metric
-from .trace import Trace
+from .trace import Span
 
-__all__ = ["Metric", "Trace"]
+__all__ = ["Span"]

divi/signals/trace/__init__.py

@@ -1,3 +1,3 @@
-from .trace import Trace
+from .trace import Span
 
-__all__ = ["Trace"]
+__all__ = ["Span"]

divi/signals/trace/trace.py

@@ -1,2 +1,81 @@
-class Trace:
-    pass
+import time
+from typing import Any, Mapping, Optional
+from uuid import uuid4
+
+from pydantic import UUID4
+
+from divi.proto.common.v1.common_pb2 import KeyValue
+from divi.proto.trace.v1.trace_pb2 import Span as SpanProto
+
+
+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,
+    ):
+        self.span_id: UUID4 = uuid4()
+        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: UUID4 | None = None
+
+    @property
+    def signal(self):
+        signal: SpanProto = SpanProto(
+            name=self.name,
+            span_id=self.span_id.bytes,
+            kind=self._get_kind(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
+
+    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):
+        """Set the span as a root span."""
+        self.trace_id = uuid4()
+        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: UUID4):
+        """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-0.0.1.dev12.dist-info → divi-0.0.1.dev13.dist-info}/METADATA

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

divi-0.0.1.dev13.dist-info/RECORD

@@ -0,0 +1,49 @@
+divi/__init__.py,sha256=gjUnJYCo6ZWzaFn7NsIZWrCPHiu6nS5ASwjWPwWsY60,380
+divi/utils.py,sha256=3iVDogCjqQg0jEjhUKEuQ6vHPFp9w7kXNjSVwXt8KmI,574
+divi/bin/core,sha256=5TH23JdCcV3dmWFAQgDUwjugZ1H6W41asTTnS2JG0pI,14648850
+divi/config/config.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+divi/decorators/__init__.py,sha256=HkyWdC1ctTsVFucCWCkj57JB4NmwONus1d2S2dUbvs4,110
+divi/decorators/obs_openai.py,sha256=2KVF2U0Z1-hcVxwryE5rvDr7LTADqbNaQYrWfswStUs,993
+divi/decorators/observable.py,sha256=nrlZpk4-p_7XBYIdR2P86qVkevURsjWLnJANusU6uYw,2837
+divi/proto/common/v1/common.proto,sha256=VcXK6JVwBXJ-70dTtJxa8If-dU__N26oAPWOAYulsko,2252
+divi/proto/common/v1/common_pb2.py,sha256=e-B4LJ6fDytcY9YO-wlR6Z-_VR0yiLe2FRowVZXbk6Y,2772
+divi/proto/common/v1/common_pb2.pyi,sha256=Eo1DJJBTddi0isywywKh2cEFjoMGsbxGRUW7klhVBW0,2789
+divi/proto/core/health/v1/health_service.proto,sha256=kUkCFAA1cJyQCN_jvdoUJli0pjHSgFKGysqLMs88Dyc,372
+divi/proto/core/health/v1/health_service_pb2.py,sha256=JL34hqA0cogvRZ_Ql9c1RGIcCl4d-71fNtSUFfTl5gk,2007
+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=RMyzWSvH_hNH1MQO17dlS3JAuyLA26xcwkUsoioP5tM,666
+divi/proto/metric/v1/metric_pb2.py,sha256=9zEhcqbzGsO8-3nst46Sjcd27xYMt98JFNpA0x7_RU4,2054
+divi/proto/metric/v1/metric_pb2.pyi,sha256=7ZxtOhqP9sl4hzaFBlPJ0SIk4afEEYQT94fy5an56nM,1447
+divi/proto/run/v1/run.proto,sha256=Znnj8ZvKUEMtDTlMhOTrjD3yFpcWnQ2glR_XCPqttlk,1018
+divi/proto/run/v1/run_pb2.py,sha256=NpBsyZZvEZG90l8DEt0pqzxFZ1HyGP5o57wGvezENNI,2103
+divi/proto/run/v1/run_pb2.pyi,sha256=H6sPhBN8fqsoeb_F_0vGCegH49ZWmPHSShvJ77jojDk,1768
+divi/proto/trace/v1/trace.proto,sha256=a24OcSsMciwQnQx-aPBnF1U3jnQb9nUjHnueRVXxaHA,1365
+divi/proto/trace/v1/trace_pb2.py,sha256=squnWqxwaLv8dJADvJfQhGMw0mTtXdHrAMIr1QSuBls,2407
+divi/proto/trace/v1/trace_pb2.pyi,sha256=E-GMVDvJ6_c69BSeiQ7zFccVu1m4DEyhf8Q3lz6ZJHs,2279
+divi/run/__init__.py,sha256=AiuwvrmmXT0me76bo6rRt3MGxzf3bBhDXH-sVFU2-rw,62
+divi/run/run.py,sha256=6UGALwRRorA74Ak48mv1ZxVkTKu6A55f3qQHAw5vfhA,484
+divi/run/setup.py,sha256=xz9CogPdlLpOqvEkU2Iq0xW7uDApOKM2Ga4YeRtriBE,1418
+divi/run/teardown.py,sha256=Vguc8_x2NTucABIo0lE71yT5zpQI0a974IKz7BrI_No,129
+divi/services/__init__.py,sha256=TcVJ_gKxyPIcwhT9GgttqHeyk0icW44uE285KmUiyh4,185
+divi/services/finish.py,sha256=XKPKGJ5cWd5H95G_VpIOlOZOLrcf9StoTs7ayRic2jY,173
+divi/services/init.py,sha256=ocfMDylWc6_6HdlIr_xyf_3iuP6KZ6M8eOObPdYo76o,361
+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=PXmQ52TGc0sdXm6uy4UbJ_3MAYE0Mkw_LTh4Kr1KgkQ,1941
+divi/services/datapark/__init__.py,sha256=GbV1mwHE07yutgOlCIYHykSEL5KJ-ApgLutGMzu2eUE,86
+divi/services/datapark/datapark.py,sha256=a7db-SQPqiae1Yxs78h7omQq7sItX5UClfniwxN-CUc,159
+divi/services/datapark/init.py,sha256=C32f9t3eLsxcYNqEyheh6nW455G2oR0YhhdqBcbN3ec,92
+divi/signals/__init__.py,sha256=K1PaTAMwyBDsK6jJUg4QWy0xVJ_5MA6dlWiUyJeiSQA,44
+divi/signals/trace/__init__.py,sha256=K1PaTAMwyBDsK6jJUg4QWy0xVJ_5MA6dlWiUyJeiSQA,44
+divi/signals/trace/trace.py,sha256=sWXvUhzgwkLlaI-pGEHL8vim9PHP9Av6Ss0a9reLqXA,2476
+divi-0.0.1.dev13.dist-info/METADATA,sha256=KZ0LLyd51oXZQzNm7oDqPSg48c1AVzzF27x_IAHF24c,457
+divi-0.0.1.dev13.dist-info/WHEEL,sha256=G84COq2WqV__z0k5yV28XkqceijvjyhYltl4MnmpO9g,102
+divi-0.0.1.dev13.dist-info/licenses/LICENSE,sha256=5OJuZ4wMMEV0DgF0tofhAlS_KLkaUsZwwwDS2U_GwQ0,1063
+divi-0.0.1.dev13.dist-info/RECORD,,

divi/proto/resource/v1/resource.proto

@@ -1,19 +0,0 @@
-syntax = "proto3";
-
-package divi.proto.resource.v1;
-
-import "divi/proto/common/v1/common.proto";
-
-option go_package = "services/proto";
-
-// Resource information.
-message Resource {
-  // Set of attributes that describe the resource.
-  // Attribute keys MUST be unique (it is not allowed to have more than one
-  // attribute with the same key).
-  repeated divi.proto.common.v1.KeyValue attributes = 1;
-
-  // dropped_attributes_count is the number of dropped attributes. If the value is 0, then
-  // no attributes were dropped.
-  uint32 dropped_attributes_count = 2;
-}

divi/signals/metric/__init__.py

@@ -1,3 +0,0 @@
-from .metric import Metric
-
-__all__ = ["Metric"]

divi/signals/metric/metric.py

@@ -1,2 +0,0 @@
-class Metric:
-    pass

divi-0.0.1.dev12.dist-info/RECORD

@@ -1,37 +0,0 @@
-divi/__init__.py,sha256=iYB7lJdbqdP1b1Qi_dDljwannxsfxeutg6QxEsi9lIk,380
-divi/utils.py,sha256=3iVDogCjqQg0jEjhUKEuQ6vHPFp9w7kXNjSVwXt8KmI,574
-divi/bin/core,sha256=GMj7vkrAhOajWhtUyHV41AV40kh01L6MhJdbmQrYU2E,14543426
-divi/config/config.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-divi/decorators/__init__.py,sha256=HkyWdC1ctTsVFucCWCkj57JB4NmwONus1d2S2dUbvs4,110
-divi/decorators/obs_openai.py,sha256=2KVF2U0Z1-hcVxwryE5rvDr7LTADqbNaQYrWfswStUs,993
-divi/decorators/observable.py,sha256=d0gqIqPvU7nhO2GtdgLfGkcoZsZ9i0U1PKmd7QS-O_E,1018
-divi/proto/common/v1/common.proto,sha256=gkVT3Ac7oMg-JjUhHiT4JGq7RyjPmvpId9TZpfvmpoM,2309
-divi/proto/core/health/v1/health_service.proto,sha256=kUkCFAA1cJyQCN_jvdoUJli0pjHSgFKGysqLMs88Dyc,372
-divi/proto/resource/v1/resource.proto,sha256=gpB3Vj3BHNAhbN6Yk_oHWwbh16L8VBXpRl2-TKn6R3g,566
-divi/proto/trace/v1/trace.proto,sha256=ZCPEW1haW-fwXbgMpw9HfORLoNeWrrqauT3CL0OTP-I,466
-divi/run/__init__.py,sha256=0DQw0lBBiXRyW74PZkDCQVj4ZmX8lQrHo8_ZixdHNrI,40
-divi/run/run.py,sha256=UOxW0gy9nEu8snjyc5O3Pw5o69ary-b-QpqFqTXybNE,20
-divi/services/__init__.py,sha256=TcVJ_gKxyPIcwhT9GgttqHeyk0icW44uE285KmUiyh4,185
-divi/services/finish.py,sha256=epgUNYdsGSC8j8uj_fhs7P0-Atem_hTlzyPPvbsWEPQ,91
-divi/services/init.py,sha256=JlsKHK-IEQ2xh76gNBqF_eqIFNBqoTnwiDeMXSJg4wQ,144
-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=bScuij1a97sDPRZyrzIkrwX3X25LLY7NG6zZI92C7Xs,554
-divi/services/auth/tokman.py,sha256=V03wcV6TEy1k9o55teDVB9JZ3Ve2cJdwzWstQhWx4BQ,1070
-divi/services/core/__init__.py,sha256=FWl4ShX0AEnOyL24NEc9mNIVoQOeYO4q0qYX27L_Yyw,111
-divi/services/core/core.py,sha256=kwqW1DSkLcOjVVWA5y8cdYo35rJ9nNffs_aVC1894Dk,1198
-divi/services/core/finish.py,sha256=dIGQpVXcJY4-tKe7A1_VV3yoSHNCDPfOlUltvzvk6VI,231
-divi/services/core/init.py,sha256=TiY9Z6Vl93ZMAE07kuw6H2uHdk3jUtzGCV_I1cIdaLI,2073
-divi/services/datapark/__init__.py,sha256=ftKZC1QGegK7SFslC6jrbRD-S4WLXwBpucA4oiMj5so,55
-divi/services/datapark/datapark.py,sha256=a7db-SQPqiae1Yxs78h7omQq7sItX5UClfniwxN-CUc,159
-divi/services/datapark/init.py,sha256=K0GxbZW0qeimpEoKZeDJo-NIbffX7Hy7GlncOqTOj7U,179
-divi/signals/__init__.py,sha256=bN3jKJkY9vlpCjEK5llQddgloQ1KXg1z5xjYWY1JF-c,83
-divi/signals/metric/__init__.py,sha256=Enn1iNvJUxBSIuKEJM2DbBWET719oSFgpbhhLUMc0GE,49
-divi/signals/metric/metric.py,sha256=gFy-iKvFLKUe5Me9jPR_vZX_kIqa0FrgBMI7x9jaFMI,23
-divi/signals/trace/__init__.py,sha256=hE2j94DHT2a-8gUtcuWiFxrhq1sWcqnSjhSZ41q6nBQ,46
-divi/signals/trace/trace.py,sha256=vGdcQ0WzLjmIOH0rtyQFkes30XhD4W2lG6uEP5oznEM,22
-divi-0.0.1.dev12.dist-info/METADATA,sha256=WpXFxou0VM2Rdfw84Ue64EV85GtAEOlZcGEhrEMsQW4,457
-divi-0.0.1.dev12.dist-info/WHEEL,sha256=G84COq2WqV__z0k5yV28XkqceijvjyhYltl4MnmpO9g,102
-divi-0.0.1.dev12.dist-info/licenses/LICENSE,sha256=5OJuZ4wMMEV0DgF0tofhAlS_KLkaUsZwwwDS2U_GwQ0,1063
-divi-0.0.1.dev12.dist-info/RECORD,,