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

divi/__init__.py

@@ -1,15 +1,16 @@
 from typing import Optional
 
 from . import proto
+from .decorators import obs_openai, observable
+from .run import Run
 from .services import Auth, Core, DataPark
 
 name: str = "divi"
 
-
-name: str = "divi"
+_run: Optional[Run] = None
 _core: Optional[Core] = None
 _auth: Optional[Auth] = None
 _datapark: Optional[DataPark] = None
 
-__version__ = "0.0.1.dev11"
-__all__ = ["proto"]
+__version__ = "0.0.1.dev13"
+__all__ = ["proto", "obs_openai", "observable"]

divi/decorators/__init__.py

@@ -0,0 +1,4 @@
+from .obs_openai import obs_openai
+from .observable import observable
+
+__all__ = ["observable", "obs_openai"]

divi/decorators/obs_openai.py

@@ -0,0 +1,33 @@
+import functools
+from collections.abc import Callable
+from typing import TYPE_CHECKING, TypeVar, Union
+
+from divi.decorators.observable import observable
+from divi.utils import is_async
+
+if TYPE_CHECKING:
+    from openai import AsyncOpenAI, OpenAI
+
+C = TypeVar("C", bound=Union["OpenAI", "AsyncOpenAI"])
+
+
+def _get_observable_create(create: Callable) -> Callable:
+    @functools.wraps(create)
+    def observable_create(*args, stream: bool = False, **kwargs):
+        decorator = observable()
+        return decorator(create)(*args, stream=stream, **kwargs)
+
+    # TODO Async Observable Create
+    print("Is async", is_async(create))
+    return observable_create if not is_async(create) else create
+
+
+def obs_openai(client: C) -> C:
+    """Make OpenAI client observable."""
+    client.chat.completions.create = _get_observable_create(
+        client.chat.completions.create
+    )
+    client.completions.create = _get_observable_create(
+        client.completions.create
+    )
+    return client

divi/decorators/observable.py

@@ -0,0 +1,108 @@
+import contextvars
+import functools
+import inspect
+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[P, R]) -> WithRunExtra[P, R]: ...
+
+
+@overload
+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
+) -> Union[Callable, Callable[[Callable], Callable]]:
+    """Observable decorator factory."""
+
+    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, 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, 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):
+            return generator_wrapper
+        return wrapper
+
+    # Function Decorator
+    if len(args) == 1 and callable(args[0]) and not kwargs:
+        return decorator(args[0])
+    # Factory Decorator
+    return decorator

divi/proto/common/v1/common.proto

@@ -0,0 +1,63 @@
+syntax = "proto3";
+
+package divi.proto.common.v1;
+
+option go_package = "services/proto";
+
+// AnyValue is used to represent any type of attribute value. AnyValue may contain a
+// primitive value such as a string or integer or it may contain an arbitrary nested
+// object containing arrays, key-value lists and primitives.
+message AnyValue {
+  // The value is one of the listed fields. It is valid for all values to be unspecified
+  // in which case this AnyValue is considered to be "empty".
+  oneof value {
+    string string_value = 1;
+    bool bool_value = 2;
+    int64 int_value = 3;
+    double double_value = 4;
+    ArrayValue array_value = 5;
+    KeyValueList kvlist_value = 6;
+    bytes bytes_value = 7;
+  }
+}
+
+// ArrayValue is a list of AnyValue messages. We need ArrayValue as a message
+// since oneof in AnyValue does not allow repeated fields.
+message ArrayValue {
+  // Array of values. The array may be empty (contain 0 elements).
+  repeated AnyValue values = 1;
+}
+
+// KeyValueList is a list of KeyValue messages. We need KeyValueList as a message
+// since `oneof` in AnyValue does not allow repeated fields. Everywhere else where we need
+// a list of KeyValue messages (e.g. in Span) we use `repeated KeyValue` directly to
+// avoid unnecessary extra wrapping (which slows down the protocol). The 2 approaches
+// are semantically equivalent.
+message KeyValueList {
+  // A collection of key/value pairs of key-value pairs. The list may be empty (may
+  // contain 0 elements).
+  // The keys MUST be unique (it is not allowed to have more than one
+  // value with the same key).
+  repeated KeyValue values = 1;
+}
+
+// KeyValue is a key-value pair that is used to store Span attributes, Link
+// attributes, etc.
+message KeyValue {
+  string key = 1;
+  AnyValue value = 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
+  // attribute with the same key).
+  repeated KeyValue attributes = 3;
+  uint32 dropped_attributes_count = 4;
+}

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.proto

@@ -0,0 +1,17 @@
+syntax = "proto3";
+
+package divi.proto.core.health.v1;
+
+option go_package = "services/proto";
+
+// HealthService is a service that implements health check.
+service HealthService {
+  rpc Check(HealthCheckRequest) returns (HealthCheckResponse) {}
+}
+
+message HealthCheckRequest { string version = 1; }
+
+message HealthCheckResponse {
+  bool status = 1;
+  string message = 2;
+}

divi/proto/{health_pb2.py → core/health/v1/health_service_pb2.py}

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
-# source: divi/proto/health.proto
+# 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
@@ -15,7 +15,7 @@ _runtime_version.ValidateProtobufRuntimeVersion(
     29,
     0,
     '',
-    'divi/proto/health.proto'
+    'divi/proto/core/health/v1/health_service.proto'
 )
 # @@protoc_insertion_point(imports)
 
@@ -24,16 +24,18 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17\x64ivi/proto/health.proto\x12\x04\x63ore\"%\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(\tB\x10Z\x0eservices/protob\x06proto3')
+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.health_pb2', _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=33
-  _globals['_HEALTHCHECKREQUEST']._serialized_end=70
-  _globals['_HEALTHCHECKRESPONSE']._serialized_start=72
-  _globals['_HEALTHCHECKRESPONSE']._serialized_end=126
+  _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_pb2_grpc.py → core/health/v1/health_service_pb2_grpc.py}

@@ -3,7 +3,7 @@
 import grpc
 import warnings
 
-from divi.proto import health_pb2 as divi_dot_proto_dot_health__pb2
+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__
@@ -18,15 +18,15 @@ except ImportError:
 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_pb2_grpc.py depends on'
+        + 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 CoreStub(object):
-    """Health is a service that implements health check.
+class HealthServiceStub(object):
+    """HealthService is a service that implements health check.
     """
 
     def __init__(self, channel):
@@ -36,14 +36,14 @@ class CoreStub(object):
             channel: A grpc.Channel.
         """
         self.Check = channel.unary_unary(
-                '/core.Core/Check',
-                request_serializer=divi_dot_proto_dot_health__pb2.HealthCheckRequest.SerializeToString,
-                response_deserializer=divi_dot_proto_dot_health__pb2.HealthCheckResponse.FromString,
+                '/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 CoreServicer(object):
-    """Health is a service that implements health check.
+class HealthServiceServicer(object):
+    """HealthService is a service that implements health check.
     """
 
     def Check(self, request, context):
@@ -53,23 +53,23 @@ class CoreServicer(object):
         raise NotImplementedError('Method not implemented!')
 
 
-def add_CoreServicer_to_server(servicer, server):
+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_health__pb2.HealthCheckRequest.FromString,
-                    response_serializer=divi_dot_proto_dot_health__pb2.HealthCheckResponse.SerializeToString,
+                    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(
-            'core.Core', rpc_method_handlers)
+            'divi.proto.core.health.v1.HealthService', rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
-    server.add_registered_method_handlers('core.Core', rpc_method_handlers)
+    server.add_registered_method_handlers('divi.proto.core.health.v1.HealthService', rpc_method_handlers)
 
 
  # This class is part of an EXPERIMENTAL API.
-class Core(object):
-    """Health is a service that implements health check.
+class HealthService(object):
+    """HealthService is a service that implements health check.
     """
 
     @staticmethod
@@ -86,9 +86,9 @@ class Core(object):
         return grpc.experimental.unary_unary(
             request,
             target,
-            '/core.Core/Check',
-            divi_dot_proto_dot_health__pb2.HealthCheckRequest.SerializeToString,
-            divi_dot_proto_dot_health__pb2.HealthCheckResponse.FromString,
+            '/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,

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;
+}