openinference-instrumentation-beeai 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

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/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/version.py

@@ -1 +1 @@
-__version__ = "0.1.12"
+__version__ = "0.1.14"

{openinference_instrumentation_beeai-0.1.12.dist-info → openinference_instrumentation_beeai-0.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openinference-instrumentation-beeai
-Version: 0.1.12
+Version: 0.1.14
 Summary: OpenInference BeeAI Instrumentation
 Project-URL: Homepage, https://github.com/Arize-ai/openinference/tree/main/python/instrumentation/openinference-instrumentation-beeai
 Author: IBM Corp.
@@ -14,10 +14,11 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Python: <3.14,>=3.11
+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

{openinference_instrumentation_beeai-0.1.12.dist-info → openinference_instrumentation_beeai-0.1.14.dist-info}/RECORD

@@ -1,11 +1,11 @@
 openinference/instrumentation/beeai/__init__.py,sha256=OVys0Yf0LCB76P_YvnJp9Ph7J746ZPC8r8qYe3REeYE,5229
 openinference/instrumentation/beeai/_span.py,sha256=iVlYou4vnNKtDxpypMdZuD2AKeaDiG1Cu5PXVzgQ8w4,3259
 openinference/instrumentation/beeai/_utils.py,sha256=tfQsQEcevyLJno8WmLTOe936GVTIS2etnAFVbAPyztc,2521
-openinference/instrumentation/beeai/version.py,sha256=LcIlFjHZFfiF9Rd4UHoakmombOFkxIYk00I181frGBM,23
+openinference/instrumentation/beeai/version.py,sha256=PIBqEOI-nqKFL9oJAWQQwlHuujG9Cd7EmdxDrThNQto,23
 openinference/instrumentation/beeai/processors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-openinference/instrumentation/beeai/processors/base.py,sha256=-h8tx1moKVi-1t3RpN-c7klGRJyHwoYe16rKWcZ_bhs,2204
+openinference/instrumentation/beeai/processors/base.py,sha256=KyS524sQ5NMXH7HxHGdGiqEofLPHulMG6EURz3ab134,2305
 openinference/instrumentation/beeai/processors/chat.py,sha256=Nl0JWThA7f7Oeet6136fARdpNmakMH0SYym3T9kCGFg,9281
-openinference/instrumentation/beeai/processors/embedding.py,sha256=T9fZs2M7qEs4SnLYbSXRbhe3P7rCNch-snRQBDSC9Es,2598
+openinference/instrumentation/beeai/processors/embedding.py,sha256=Vzk-XhQ7cwVwkhjfa92iQ8_qs54rtlkVgFNwgQzPs_M,3350
 openinference/instrumentation/beeai/processors/locator.py,sha256=oleg8Jxg8sJBIzK4cZ7BbfkHuLH9Hu4XRYqDaOGnNBU,3627
 openinference/instrumentation/beeai/processors/requirement.py,sha256=S5OPN6F5V39puumxEo3fKQRoAOZ_8OC5ml0dr_Z9v0U,2662
 openinference/instrumentation/beeai/processors/tool.py,sha256=o5aKAqEHZIk9bmK2rWUqtACiyMltIFCnzaunCKxRtu8,2765
@@ -15,7 +15,7 @@ openinference/instrumentation/beeai/processors/agents/base.py,sha256=3fidrUoU9pV
 openinference/instrumentation/beeai/processors/agents/react.py,sha256=rS3xlvgyZ5G6MyDMeSh4xFLT_66h7GVAYEYlwZCpIdY,3022
 openinference/instrumentation/beeai/processors/agents/requirement_agent.py,sha256=eEiEls6vUET-gEMor1FDyzOpgXCAZVQ3gugSaK3hOHA,2772
 openinference/instrumentation/beeai/processors/agents/tool_calling.py,sha256=yaWP5JmGuvZIha9iUSKgv0MJgI0QSbuiJLLQFnbqUZw,1223
-openinference_instrumentation_beeai-0.1.12.dist-info/METADATA,sha256=YXnEboNDscHrxTZZBkITmi5F1aGh4CXolz3HMwwW15o,5492
-openinference_instrumentation_beeai-0.1.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-openinference_instrumentation_beeai-0.1.12.dist-info/entry_points.txt,sha256=ee7EUhbWv-XK1dxhPXuFVy9qstzj-lc-265Phe2Ml9s,183
-openinference_instrumentation_beeai-0.1.12.dist-info/RECORD,,
+openinference_instrumentation_beeai-0.1.14.dist-info/METADATA,sha256=YrO30Bkk1Wh_PLCDruC_Al1MSQ0yosi6bb1mMHFvvU0,5543
+openinference_instrumentation_beeai-0.1.14.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+openinference_instrumentation_beeai-0.1.14.dist-info/entry_points.txt,sha256=ee7EUhbWv-XK1dxhPXuFVy9qstzj-lc-265Phe2Ml9s,183
+openinference_instrumentation_beeai-0.1.14.dist-info/RECORD,,

{openinference_instrumentation_beeai-0.1.12.dist-info → openinference_instrumentation_beeai-0.1.14.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any