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

divi/__init__.py

@@ -1,9 +1,16 @@
 from typing import Optional
-from .core import Run, init, finish
+
 from . import proto
+from .decorators import obs_openai, observable
+from .services import Auth, Core, DataPark
+from .session import Session
 
 name: str = "divi"
-run: Optional[Run] = None
 
-__version__ = "0.0.1.dev7"
-__all__ = ["init", "finish", "proto"]
+_session: Optional[Session] = None
+_core: Optional[Core] = None
+_auth: Optional[Auth] = None
+_datapark: Optional[DataPark] = None
+
+__version__ = "0.0.1.dev14"
+__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(kind="llm")
+        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,137 @@
+import contextvars
+import functools
+import inspect
+from typing import (
+    Any,
+    Callable,
+    Generic,
+    List,
+    Mapping,
+    Optional,
+    ParamSpec,
+    Protocol,
+    TypeVar,
+    Union,
+    overload,
+    runtime_checkable,
+)
+
+from openai.types.chat import ChatCompletion
+
+import divi
+from divi.proto.trace.v1.trace_pb2 import ScopeSpans
+from divi.session import SessionExtra
+from divi.session.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 Session and parent Span
+_SESSION_EXTRA = contextvars.ContextVar[Optional[SessionExtra]](
+    "_SESSION_EXTRA", default=None
+)
+
+
+@runtime_checkable
+class WithSessionExtra(Protocol, Generic[P, R]):
+    def __call__(
+        self,
+        *args: P.args,
+        session_extra: Optional[SessionExtra] = None,  # type: ignore[valid-type]
+        **kwargs: P.kwargs,
+    ) -> R: ...
+
+
+@overload
+def observable(func: Callable[P, R]) -> WithSessionExtra[P, R]: ...
+
+
+@overload
+def observable(
+    kind: str = "function",
+    *,
+    name: Optional[str] = None,
+    metadata: Optional[Mapping[str, Any]] = None,
+) -> Callable[[Callable[P, R]], WithSessionExtra[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):
+        @functools.wraps(func)
+        def wrapper(
+            *args, session_extra: Optional[SessionExtra] = None, **kwargs
+        ):
+            span = Span(
+                kind=kind, name=name or func.__name__, metadata=metadata
+            )
+            session_extra = setup(span, _SESSION_EXTRA.get() or session_extra)
+            # set current context
+            token = _SESSION_EXTRA.set(session_extra)
+            # execute the function
+            span.start()
+            result = func(*args, **kwargs)
+            span.end()
+            # recover parent context
+            _SESSION_EXTRA.reset(token)
+
+            # TODO: collect result
+            # create the span if it is the root span
+            if divi._datapark and span.trace_id:
+                divi._datapark.create_spans(
+                    span.trace_id, ScopeSpans(spans=[span.signal])
+                )
+            # end the trace if it is the root span
+            if divi._datapark and not span.parent_span_id:
+                trace = session_extra.get("trace")
+                if trace:
+                    trace.end()
+                    divi._datapark.upsert_traces(
+                        session_id=trace.session_id, traces=[trace.signal]
+                    )
+            if divi._datapark and isinstance(result, ChatCompletion):
+                divi._datapark.create_chat_completion(
+                    span_id=span.span_id, completion=result
+                )
+            return result
+
+        @functools.wraps(func)
+        def generator_wrapper(
+            *args, session_extra: Optional[SessionExtra] = None, **kwargs
+        ):
+            span = Span(
+                kind=kind, name=name or func.__name__, metadata=metadata
+            )
+            session_extra = setup(span, _SESSION_EXTRA.get() or session_extra)
+            # set current context
+            token = _SESSION_EXTRA.set(session_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
+            _SESSION_EXTRA.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,49 @@
+syntax = "proto3";
+
+package divi.proto.common.v1;
+
+option go_package = "services/pb";
+
+// 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;
+}

divi/proto/common/v1/common_pb2.py

@@ -0,0 +1,43 @@
+# -*- 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.AnyValueB\rZ\x0bservices/pbb\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\013services/pb'
+  _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
+# @@protoc_insertion_point(module_scope)

divi/proto/common/v1/common_pb2.pyi

@@ -0,0 +1,44 @@
+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: ...

divi/proto/core/health/v1/health_service.proto

@@ -0,0 +1,17 @@
+syntax = "proto3";
+
+package divi.proto.core.health.v1;
+
+option go_package = "services/pb";
+
+// 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/core/{v1/health_check_response_pb2.py → 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/core/v1/health_check_response.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/core/v1/health_check_response.proto'
+    'divi/proto/core/health/v1/health_service.proto'
 )
 # @@protoc_insertion_point(imports)
 
@@ -24,14 +24,18 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.divi/proto/core/v1/health_check_response.proto\x12\x07\x63ore.v1\"6\n\x13HealthCheckResponse\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\tB2Z0github.com/KaikaikaiFang/divine-agent/core/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\rZ\x0bservices/pbb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'divi.proto.core.v1.health_check_response_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'Z0github.com/KaikaikaiFang/divine-agent/core/proto'
-  _globals['_HEALTHCHECKRESPONSE']._serialized_start=59
-  _globals['_HEALTHCHECKRESPONSE']._serialized_end=113
+  _globals['DESCRIPTOR']._serialized_options = b'Z\013services/pb'
+  _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/{v1/health_check_response_pb2.pyi → health/v1/health_service_pb2.pyi}

@@ -4,6 +4,12 @@ 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]

divi/proto/core/{v1/core_pb2_grpc.py → health/v1/health_service_pb2_grpc.py}

@@ -3,8 +3,7 @@
 import grpc
 import warnings
 
-from divi.proto.core.v1 import health_check_request_pb2 as divi_dot_proto_dot_core_dot_v1_dot_health__check__request__pb2
-from divi.proto.core.v1 import health_check_response_pb2 as divi_dot_proto_dot_core_dot_v1_dot_health__check__response__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__
@@ -19,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/v1/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):
@@ -37,14 +36,14 @@ class CoreStub(object):
             channel: A grpc.Channel.
         """
         self.Check = channel.unary_unary(
-                '/core.v1.Core/Check',
-                request_serializer=divi_dot_proto_dot_core_dot_v1_dot_health__check__request__pb2.HealthCheckRequest.SerializeToString,
-                response_deserializer=divi_dot_proto_dot_core_dot_v1_dot_health__check__response__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):
@@ -54,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_core_dot_v1_dot_health__check__request__pb2.HealthCheckRequest.FromString,
-                    response_serializer=divi_dot_proto_dot_core_dot_v1_dot_health__check__response__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.v1.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.v1.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
@@ -87,9 +86,9 @@ class Core(object):
         return grpc.experimental.unary_unary(
             request,
             target,
-            '/core.v1.Core/Check',
-            divi_dot_proto_dot_core_dot_v1_dot_health__check__request__pb2.HealthCheckRequest.SerializeToString,
-            divi_dot_proto_dot_core_dot_v1_dot_health__check__response__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,26 @@
+syntax = "proto3";
+
+package divi.proto.metric.v1;
+
+import "divi/proto/common/v1/common.proto";
+
+option go_package = "services/pb";
+
+message ScopeMetrics {
+    // 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\"=\n\x0cScopeMetrics\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\rZ\x0bservices/pbb\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\013services/pb'
+  _globals['_SCOPEMETRICS']._serialized_start=94
+  _globals['_SCOPEMETRICS']._serialized_end=155
+  _globals['_METRIC']._serialized_start=158
+  _globals['_METRIC']._serialized_end=297
+# @@protoc_insertion_point(module_scope)

divi/proto/metric/v1/metric_pb2.pyi

@@ -0,0 +1,25 @@
+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__ = ("metrics",)
+    METRICS_FIELD_NUMBER: _ClassVar[int]
+    metrics: _containers.RepeatedCompositeFieldContainer[Metric]
+    def __init__(self, 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/trace/v1/trace.proto

@@ -0,0 +1,47 @@
+syntax = "proto3";
+
+package divi.proto.trace.v1;
+
+import "divi/proto/common/v1/common.proto";
+
+option go_package = "services/pb";
+
+message ScopeSpans {
+    // A list of spans that originate from a resource.
+    repeated Span spans = 2;
+}
+
+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;
+}