openinference-instrumentation-beeai 0.1.13__tar.gz → 0.1.15__tar.gz

{openinference_instrumentation_beeai-0.1.13 → openinference_instrumentation_beeai-0.1.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openinference-instrumentation-beeai
-Version: 0.1.13
+Version: 0.1.15
 Summary: OpenInference BeeAI Instrumentation
 Project-URL: Homepage, https://github.com/Arize-ai/openinference/tree/main/python/instrumentation/openinference-instrumentation-beeai
 Author: IBM Corp.
@@ -18,7 +18,7 @@ Classifier: Programming Language :: Python :: 3.14
 Requires-Python: <3.15,>=3.11
 Requires-Dist: beeai-framework<0.2.0,>=0.1.51
 Requires-Dist: openinference-instrumentation>=0.1.37
-Requires-Dist: openinference-semantic-conventions>=0.1.21
+Requires-Dist: openinference-semantic-conventions>=0.1.25
 Requires-Dist: opentelemetry-api>=1.36.0
 Requires-Dist: opentelemetry-instrumentation>=0.57b0
 Requires-Dist: opentelemetry-semantic-conventions>=0.57b0
@@ -27,8 +27,13 @@ Provides-Extra: instruments
 Requires-Dist: beeai-framework>=0.1.51; extra == 'instruments'
 Provides-Extra: test
 Requires-Dist: beeai-framework>=0.1.51; extra == 'test'
+Requires-Dist: beeai-framework[duckduckgo]; extra == 'test'
 Requires-Dist: opentelemetry-exporter-otlp; extra == 'test'
 Requires-Dist: opentelemetry-sdk; extra == 'test'
+Requires-Dist: pytest; extra == 'test'
+Requires-Dist: pytest-asyncio; extra == 'test'
+Requires-Dist: pytest-recording; extra == 'test'
+Requires-Dist: vcrpy<8.0.0,>=5.0.0; extra == 'test'
 Description-Content-Type: text/markdown
 
 # OpenInference Instrumentation for BeeAI

{openinference_instrumentation_beeai-0.1.13 → openinference_instrumentation_beeai-0.1.15}/pyproject.toml

@@ -25,7 +25,7 @@ classifiers = [
 dependencies = [
     "beeai-framework (>=0.1.51,<0.2.0)",
     "openinference-instrumentation>=0.1.37",
-    "openinference-semantic-conventions>=0.1.21",
+    "openinference-semantic-conventions>=0.1.25",
     "opentelemetry-api>=1.36.0",
     "opentelemetry-instrumentation>=0.57b0",
     "opentelemetry-semantic-conventions>=0.57b0",
@@ -39,7 +39,12 @@ instruments = [
 test = [
   "beeai-framework >= 0.1.51",
   "opentelemetry-sdk",
-  "opentelemetry-exporter-otlp"
+  "opentelemetry-exporter-otlp",
+  "pytest-recording",
+  "vcrpy>=5.0.0,<8.0.0",
+  "beeai-framework[duckduckgo]",
+  "pytest-asyncio",
+  "pytest",
 ]
 
 [project.entry-points.opentelemetry_instrumentor]

{openinference_instrumentation_beeai-0.1.13 → openinference_instrumentation_beeai-0.1.15}/src/openinference/instrumentation/beeai/processors/base.py

@@ -18,6 +18,9 @@ from openinference.semconv.trace import (
 class Processor:
     kind: ClassVar[OpenInferenceSpanKindValues] = OpenInferenceSpanKindValues.UNKNOWN
 
+    def get_span_name(self, target_cls: type) -> str:
+        return target_cls.__name__
+
     def __init__(self, event: "RunContextStartEvent", meta: "EventMeta"):
         from beeai_framework.context import RunContext
 
@@ -27,7 +30,7 @@ class Processor:
         assert meta.trace is not None
         self.run_id = meta.trace.run_id
 
-        self.span = SpanWrapper(name=target_cls.__name__, kind=type(self).kind)
+        self.span = SpanWrapper(name=self.get_span_name(target_cls), kind=type(self).kind)
         self.span.started_at = meta.created_at
         self.span.attributes.update(
             {

{openinference_instrumentation_beeai-0.1.13 → openinference_instrumentation_beeai-0.1.15}/src/openinference/instrumentation/beeai/processors/chat.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Any, ClassVar
+from typing import Any, ClassVar, Dict, Generator, Tuple
 
 from beeai_framework.backend import (
     AnyMessage,
@@ -19,6 +19,7 @@ from beeai_framework.tools import AnyTool
 from beeai_framework.utils.lists import remove_falsy
 from typing_extensions import override
 
+from openinference.instrumentation import safe_json_dumps
 from openinference.instrumentation.beeai._utils import (
     _unpack_object,
     safe_dump_model_schema,
@@ -36,6 +37,24 @@ from openinference.semconv.trace import (
 )
 
 
+def get_tool_parameters(tool: AnyTool) -> Dict[str, Any]:
+    tool_dict = tool.to_json_safe()
+    if "input_schema" in tool_dict:
+        input_schema = tool_dict.pop("input_schema")
+        tool_dict["parameters"] = input_schema or {}
+        tool_dict["parameters"]["type"] = "object"
+    return tool_dict
+
+
+def get_tools(tools: list[AnyTool]) -> Generator[Tuple[str, str], None, None]:
+    for index, tool in enumerate(tools):
+        function = {"type": "function", "function": get_tool_parameters(tool)}
+        yield (
+            f"{SpanAttributes.LLM_TOOLS}.{index}.{ToolAttributes.TOOL_JSON_SCHEMA}",
+            safe_json_dumps(function),
+        )
+
+
 class ChatModelProcessor(Processor):
     kind: ClassVar[OpenInferenceSpanKindValues] = OpenInferenceSpanKindValues.LLM
 
@@ -80,7 +99,7 @@ class ChatModelProcessor(Processor):
                 )
                 self.span.set_attributes(
                     {
-                        SpanAttributes.LLM_TOOLS: [t.name for t in (event.input.tools or [])],
+                        **dict(get_tools(event.input.tools or [])),
                         SpanAttributes.LLM_INVOCATION_PARAMETERS: stringify(
                             meta.creator.parameters.model_dump(
                                 exclude_none=True, exclude_unset=True

{openinference_instrumentation_beeai-0.1.13 → openinference_instrumentation_beeai-0.1.15}/src/openinference/instrumentation/beeai/processors/embedding.py

@@ -12,6 +12,7 @@ from beeai_framework.backend.events import (
 from beeai_framework.context import RunContext
 from typing_extensions import override
 
+from openinference.instrumentation import safe_json_dumps
 from openinference.instrumentation.beeai.processors.base import Processor
 from openinference.semconv.trace import (
     EmbeddingAttributes,
@@ -23,6 +24,10 @@ from openinference.semconv.trace import (
 class EmbeddingModelProcessor(Processor):
     kind: ClassVar[OpenInferenceSpanKindValues] = OpenInferenceSpanKindValues.EMBEDDING
 
+    @override
+    def get_span_name(self, target_cls: type) -> str:
+        return "CreateEmbeddings"
+
     def __init__(self, event: "RunContextStartEvent", meta: "EventMeta"):
         super().__init__(event, meta)
 
@@ -33,10 +38,22 @@ class EmbeddingModelProcessor(Processor):
         self.span.set_attributes(
             {
                 SpanAttributes.EMBEDDING_MODEL_NAME: llm.model_id,
-                SpanAttributes.LLM_PROVIDER: llm.provider_id,
             }
         )
 
+        # Extract invocation parameters (exclude input values)
+        if hasattr(event, "input") and hasattr(event.input, "__dict__"):
+            invocation_params = {
+                k: v
+                for k, v in event.input.__dict__.items()
+                if k not in {"values", "api_key", "token"} and not k.startswith("_")
+            }
+            if invocation_params:
+                self.span.set_attribute(
+                    SpanAttributes.EMBEDDING_INVOCATION_PARAMETERS,
+                    safe_json_dumps(invocation_params),
+                )
+
     @override
     async def update(
         self,
@@ -56,9 +73,10 @@ class EmbeddingModelProcessor(Processor):
                 )
         elif isinstance(event, EmbeddingModelSuccessEvent):
             for idx, embedding in enumerate(event.value.embeddings):
+                vector = list(embedding) if not isinstance(embedding, list) else embedding
                 self.span.set_attribute(
                     f"{SpanAttributes.EMBEDDING_EMBEDDINGS}.{idx}.{EmbeddingAttributes.EMBEDDING_VECTOR}",
-                    embedding,
+                    vector,
                 )
 
             if event.value.usage:

openinference_instrumentation_beeai-0.1.15/src/openinference/instrumentation/beeai/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.15"

openinference_instrumentation_beeai-0.1.13/src/openinference/instrumentation/beeai/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.13"