openinference-instrumentation-beeai 0.1.8__py3-none-any.whl → 0.1.9__py3-none-any.whl

openinference/instrumentation/beeai/__init__.py

@@ -2,6 +2,8 @@ import logging
 from importlib.metadata import PackageNotFoundError, version
 from typing import TYPE_CHECKING, Any, Callable, Collection
 
+from opentelemetry.trace import StatusCode
+
 if TYPE_CHECKING:
     from beeai_framework.emitter import EventMeta
 
@@ -82,7 +84,7 @@ class BeeAIInstrumentor(BaseInstrumentor):  # type: ignore
                 self._build_tree(children)
 
             current_span.set_status(node.status)
-            if node.error is not None:
+            if node.error is not None and node.status == StatusCode.ERROR:
                 current_span.record_exception(node.error)
 
             current_span.end(_datetime_to_span_time(node.ended_at) if node.ended_at else None)

openinference/instrumentation/beeai/_span.py

@@ -77,5 +77,20 @@ class SpanWrapper:
     def set_status(self, status: StatusCode) -> None:
         self.status = status
 
+    def reset_exception(self) -> None:
+        self.error = None
+        self.set_status(StatusCode.OK)
+
     def record_exception(self, error: Exception) -> None:
+        from beeai_framework.errors import FrameworkError
+
         self.error = error
+        self.set_status(StatusCode.ERROR)
+        self.set_attributes(
+            {
+                SpanAttributes.OUTPUT_VALUE: error.explain()
+                if isinstance(error, FrameworkError)
+                else str(error),
+                SpanAttributes.OUTPUT_MIME_TYPE: OpenInferenceMimeTypeValues.TEXT.value,
+            }
+        )

openinference/instrumentation/beeai/_utils.py

@@ -22,7 +22,7 @@ def _unpack_object(obj: dict[str, Any] | list[Any] | BaseModel, prefix: str = ""
         obj_ref = obj
         obj = json.loads(stringify(obj))
         if not isinstance(obj, dict) and not isinstance(obj, list):
-            logger.warning(f"Cannot unpack object of type {type(obj_ref)} (prefix={prefix})")
+            logger.debug(f"Cannot unpack object of type {type(obj_ref)} (prefix={prefix})")
             return {"value": str(obj)}
 
     if prefix and prefix.startswith("."):
@@ -34,7 +34,7 @@ def _unpack_object(obj: dict[str, Any] | list[Any] | BaseModel, prefix: str = ""
     for key, value in obj.items() if isinstance(obj, dict) else enumerate(obj):
         if value is None:
             continue
-        if is_primitive(value):
+        if is_primitive(value) or has_custom_str(value):
             output[f"{prefix}{key}"] = str(value)
         else:
             output.update(_unpack_object(value, prefix=f"{prefix}{key}"))
@@ -45,6 +45,13 @@ def is_primitive(value: Any) -> bool:
     return isinstance(value, str | bool | int | float | type(None))
 
 
+def has_custom_str(value: Any) -> bool:
+    if value.__class__.__module__ == "builtins":
+        return False
+
+    return value.__class__.__str__ is not object.__str__
+
+
 def stringify(value: Any, pretty: bool = False) -> str:
     if is_primitive(value):
         return str(value)

openinference/instrumentation/beeai/processors/base.py

@@ -43,6 +43,9 @@ class Processor:
         pass
 
     async def end(self, event: "RunContextFinishEvent", meta: "EventMeta") -> None:
+        if event.error is not None:
+            self.span.record_exception(event.error)
+
         if event.output is not None:
             if SpanAttributes.OUTPUT_VALUE not in self.span.attributes:
                 self.span.attributes.update(
@@ -53,8 +56,4 @@ class Processor:
                 )
             self.span.set_status(StatusCode.OK)
 
-        if event.error is not None:
-            self.span.set_status(StatusCode.ERROR)
-            self.span.record_exception(event.error)
-
         self.span.ended_at = meta.created_at

openinference/instrumentation/beeai/processors/tool.py

@@ -51,6 +51,7 @@ class ToolProcessor(Processor):
             case ToolSuccessEvent():
                 output_cls = type(event.output)
 
+                self.span.reset_exception()
                 self.span.set_attributes(
                     {
                         SpanAttributes.OUTPUT_VALUE: event.output.get_text_content(),
@@ -62,6 +63,7 @@ class ToolProcessor(Processor):
             case ToolErrorEvent():
                 span = self.span.child(meta.name, event=(event, meta))
                 span.record_exception(event.error)
+                self.span.record_exception(event.error)
             case ToolRetryEvent():
                 self.span.child(meta.name, event=(event, meta))
             case _:

openinference/instrumentation/beeai/version.py

@@ -1 +1 @@
-__version__ = "0.1.8"
+__version__ = "0.1.9"

{openinference_instrumentation_beeai-0.1.8.dist-info → openinference_instrumentation_beeai-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openinference-instrumentation-beeai
-Version: 0.1.8
+Version: 0.1.9
 Summary: OpenInference BeeAI Instrumentation
 Project-URL: Homepage, https://github.com/Arize-ai/openinference/tree/main/python/instrumentation/openinference-instrumentation-beeai
 Author: IBM Corp.

{openinference_instrumentation_beeai-0.1.8.dist-info → openinference_instrumentation_beeai-0.1.9.dist-info}/RECORD

@@ -1,21 +1,21 @@
-openinference/instrumentation/beeai/__init__.py,sha256=5_bUn-XMonUXUgbbqU_PrGyW2ojiWQD7LUvjB7B-g0A,4467
-openinference/instrumentation/beeai/_span.py,sha256=VZfoQHDUlAeyHxI7MnQ6QGq2LwhHiI1AQ2Uw_4Dn5Zw,2750
-openinference/instrumentation/beeai/_utils.py,sha256=m4ZXoGLm-iJGAJBjrfSh06OU2EaDMNgStVTLbljmPSA,2328
-openinference/instrumentation/beeai/version.py,sha256=C69ADlbQREQlR15trneyA2sk8x0-oH4rDAX5fsv19_U,22
+openinference/instrumentation/beeai/__init__.py,sha256=tEM7-1EhCKIFsoLeMEa7GcRYPFfjuGeXi2w4CEKA1R4,4547
+openinference/instrumentation/beeai/_span.py,sha256=iVlYou4vnNKtDxpypMdZuD2AKeaDiG1Cu5PXVzgQ8w4,3259
+openinference/instrumentation/beeai/_utils.py,sha256=tfQsQEcevyLJno8WmLTOe936GVTIS2etnAFVbAPyztc,2521
+openinference/instrumentation/beeai/version.py,sha256=XIaxbMbyiP-L3kguR1GhxirFblTXiHR1lMfDVITvHUI,22
 openinference/instrumentation/beeai/processors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-openinference/instrumentation/beeai/processors/base.py,sha256=akaQCsZmbapEC5tctD_VXWiBOu2pYHFRJRZ_pynpApc,2176
+openinference/instrumentation/beeai/processors/base.py,sha256=eAcKq6ip23sp1pjCa6efjDEcfoPJ_Jc3GwaWk5oZU7k,2125
 openinference/instrumentation/beeai/processors/chat.py,sha256=a4Ps6opRb2POrQ8Nla1JHUhkdVFZF5rmht8Xca3F_xA,9285
 openinference/instrumentation/beeai/processors/embedding.py,sha256=T9fZs2M7qEs4SnLYbSXRbhe3P7rCNch-snRQBDSC9Es,2598
 openinference/instrumentation/beeai/processors/locator.py,sha256=G9TFW_HgXM1TrOVdvtRU2Eq3D-atHLAET4fSo4F02X8,3635
 openinference/instrumentation/beeai/processors/requirement.py,sha256=Q9DgHDd-5rmP88Fe00d7cNKQg5zQ7ILuRzVAej5xfmk,2666
-openinference/instrumentation/beeai/processors/tool.py,sha256=03pNF04ELsxRl8EK3JhHktoQl_aXAY1rrXHV4QuXcJk,2533
+openinference/instrumentation/beeai/processors/tool.py,sha256=ddkJP0Y7-pZFQNlKSB-SOOwquAbWL9ram8VS46SU8-Q,2633
 openinference/instrumentation/beeai/processors/workflow.py,sha256=OMwFFHv3mp4M4hFvH7utYd_fiSkTnBcl2oUVaMEdy-A,3815
 openinference/instrumentation/beeai/processors/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 openinference/instrumentation/beeai/processors/agents/base.py,sha256=3fidrUoU9pVixq9YN_y1jkNMozNMX-YG2CMuh855cLk,1244
 openinference/instrumentation/beeai/processors/agents/react.py,sha256=rS3xlvgyZ5G6MyDMeSh4xFLT_66h7GVAYEYlwZCpIdY,3022
 openinference/instrumentation/beeai/processors/agents/requirement_agent.py,sha256=HpleY8pNWojuUqcae2PZgat7Xq2edU9C-uz0YjZQUyc,2774
 openinference/instrumentation/beeai/processors/agents/tool_calling.py,sha256=yaWP5JmGuvZIha9iUSKgv0MJgI0QSbuiJLLQFnbqUZw,1223
-openinference_instrumentation_beeai-0.1.8.dist-info/METADATA,sha256=Jp_y16JgscRpObUhhtwhgRW-nG3MwTDKQIgQA_oYFxM,5491
-openinference_instrumentation_beeai-0.1.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-openinference_instrumentation_beeai-0.1.8.dist-info/entry_points.txt,sha256=ee7EUhbWv-XK1dxhPXuFVy9qstzj-lc-265Phe2Ml9s,183
-openinference_instrumentation_beeai-0.1.8.dist-info/RECORD,,
+openinference_instrumentation_beeai-0.1.9.dist-info/METADATA,sha256=q2qwIddIyMKySNHBL6DRFSCCQi5NqoK9xHuSn1fkEm0,5491
+openinference_instrumentation_beeai-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+openinference_instrumentation_beeai-0.1.9.dist-info/entry_points.txt,sha256=ee7EUhbWv-XK1dxhPXuFVy9qstzj-lc-265Phe2Ml9s,183
+openinference_instrumentation_beeai-0.1.9.dist-info/RECORD,,