initial simple version openhands instrumentor (#179)

* wip

* initial simple version openhands instrumentor

* fix undefined var

* rename id var to a more descriptive name

* small refactor; store input and output on parent span

* restore dev deps

* small fixes around

* bump version to 0.7.7,revert line change

Author: dinmukhamedm

pyproject.toml

@@ -6,7 +6,7 @@
 
 [project]
 name = "lmnr"
-version = "0.7.6"
+version = "0.7.7"
 description = "Python SDK for Laminar"
 authors = [
   { name = "lmnr.ai", email = "[email protected]" }

src/lmnr/opentelemetry_lib/decorators/__init__.py

@@ -1,5 +1,4 @@
 from functools import wraps
-import logging
 import pydantic
 import orjson
 import types
@@ -60,7 +59,7 @@ def json_dumps(data: dict) -> str:
         ).decode("utf-8")
     except Exception:
         # Log the exception and return a placeholder if serialization completely fails
-        logging.warning("Failed to serialize data to JSON, type: %s", type(data))
+        logger.info("Failed to serialize data to JSON, type: %s", type(data))
         return "{}"  # Return an empty JSON object as a fallback
 
 

src/lmnr/opentelemetry_lib/litellm/__init__.py

@@ -7,14 +7,13 @@
 from lmnr.opentelemetry_lib.litellm.utils import model_as_dict, set_span_attribute
 from lmnr.opentelemetry_lib.tracing import TracerWrapper
 
-from lmnr.opentelemetry_lib.tracing.context import get_event_attributes_from_context
+from lmnr.opentelemetry_lib.tracing.context import (
+    get_current_context,
+    get_event_attributes_from_context,
+)
 from lmnr.opentelemetry_lib.utils.package_check import is_package_installed
 from lmnr.sdk.log import get_default_logger
 
-from lmnr.opentelemetry_lib.opentelemetry.instrumentation.openai import (
-    OpenAIInstrumentor,
-)
-
 logger = get_default_logger(__name__)
 
 SUPPORTED_CALL_TYPES = ["completion", "acompletion"]
@@ -46,6 +45,10 @@ def __init__(self, **kwargs):
                 raise ValueError("Laminar must be initialized before LiteLLM callback")
 
             if is_package_installed("openai"):
+                from lmnr.opentelemetry_lib.opentelemetry.instrumentation.openai import (
+                    OpenAIInstrumentor,
+                )
+
                 openai_instrumentor = OpenAIInstrumentor()
                 if (
                     openai_instrumentor
@@ -117,6 +120,7 @@ def _create_span(
                 attributes={
                     "lmnr.internal.provider": "litellm",
                 },
+                context=get_current_context(),
             )
             try:
                 model = kwargs.get("model", "unknown")

src/lmnr/opentelemetry_lib/opentelemetry/instrumentation/openai/utils.py

@@ -12,6 +12,7 @@
 
 import openai
 
+
 _OPENAI_VERSION = version("openai")
 
 LMNR_TRACE_CONTENT = "LMNR_TRACE_CONTENT"
@@ -22,6 +23,7 @@ def is_openai_v1():
 
 
 def is_azure_openai(instance):
+
     return is_openai_v1() and isinstance(
         instance._client, (openai.AsyncAzureOpenAI, openai.AzureOpenAI)
     )