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

openinference/instrumentation/beeai/_utils.py

@@ -19,9 +19,11 @@ def _datetime_to_span_time(dt: datetime.datetime) -> int:
 
 def _unpack_object(obj: dict[str, Any] | list[Any] | BaseModel, prefix: str = "") -> dict[str, Any]:
     if not isinstance(obj, dict) and not isinstance(obj, list):
+        obj_ref = obj
         obj = json.loads(stringify(obj))
         if not isinstance(obj, dict) and not isinstance(obj, list):
-            raise ValueError(f"Cannot unpack object of type {type(obj)}")
+            logger.warning(f"Cannot unpack object of type {type(obj_ref)} (prefix={prefix})")
+            return {"value": str(obj)}
 
     if prefix and prefix.startswith("."):
         prefix = prefix[1:]

openinference/instrumentation/beeai/processors/chat.py

@@ -1,4 +1,3 @@
-import json
 from datetime import datetime
 from typing import Any, ClassVar
 
@@ -81,7 +80,7 @@ class ChatModelProcessor(Processor):
                 )
                 self.span.set_attributes(
                     {
-                        SpanAttributes.LLM_TOOLS: json.loads(stringify(event.input.tools or [])),
+                        SpanAttributes.LLM_TOOLS: [t.name for t in (event.input.tools or [])],
                         SpanAttributes.LLM_INVOCATION_PARAMETERS: stringify(
                             meta.creator.parameters.model_dump(
                                 exclude_none=True, exclude_unset=True
@@ -103,18 +102,27 @@ class ChatModelProcessor(Processor):
                     self._add_new_messages(event.value.messages)
 
                 usage = event.value.usage
+                if usage:
+                    self.span.set_attributes(
+                        {
+                            SpanAttributes.LLM_TOKEN_COUNT_TOTAL: usage.total_tokens,
+                            SpanAttributes.LLM_TOKEN_COUNT_PROMPT: usage.prompt_tokens,
+                            SpanAttributes.LLM_TOKEN_COUNT_COMPLETION: usage.completion_tokens,
+                        }
+                    )
+
+                cost = event.value.cost
+                if cost:
+                    self.span.set_attributes(
+                        {
+                            SpanAttributes.LLM_COST_COMPLETION: cost.completion_tokens_cost_usd,
+                            SpanAttributes.LLM_COST_PROMPT: cost.prompt_tokens_usd,
+                            SpanAttributes.LLM_COST_TOTAL: cost.total_cost_usd,
+                        }
+                    )
 
                 self.span.set_attributes(
                     {
-                        **(
-                            {
-                                SpanAttributes.LLM_TOKEN_COUNT_TOTAL: usage.total_tokens,
-                                SpanAttributes.LLM_TOKEN_COUNT_PROMPT: usage.prompt_tokens,
-                                SpanAttributes.LLM_TOKEN_COUNT_COMPLETION: usage.completion_tokens,
-                            }
-                            if usage
-                            else {}
-                        ),
                         SpanAttributes.OPENINFERENCE_SPAN_KIND: type(self).kind,
                         SpanAttributes.OUTPUT_VALUE: event.value.get_text_content(),
                         SpanAttributes.OUTPUT_MIME_TYPE: OpenInferenceMimeTypeValues.TEXT.value,
@@ -149,8 +157,6 @@ def _process_tools(tools: list[AnyTool]) -> list[dict[str, str | Any]]:
         {
             SpanAttributes.TOOL_NAME: t.name,
             SpanAttributes.TOOL_DESCRIPTION: t.description,
-            # TODO: difference between TOOL_PARAMETERS and TOOL_JSON_SCHEMA is not obvious
-            SpanAttributes.TOOL_PARAMETERS: safe_dump_model_schema(t.input_schema),
             ToolAttributes.TOOL_JSON_SCHEMA: safe_dump_model_schema(t.input_schema),
         }
         for t in tools

openinference/instrumentation/beeai/processors/tool.py

@@ -31,10 +31,6 @@ class ToolProcessor(Processor):
             {
                 SpanAttributes.TOOL_NAME: tool.name,
                 SpanAttributes.TOOL_DESCRIPTION: tool.description,
-                # TODO: what's the difference?
-                SpanAttributes.TOOL_PARAMETERS: stringify(
-                    safe_dump_model_schema(tool.input_schema)
-                ),
                 ToolAttributes.TOOL_JSON_SCHEMA: stringify(
                     safe_dump_model_schema(tool.input_schema)
                 ),

openinference/instrumentation/beeai/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openinference-instrumentation-beeai
-Version: 0.1.7
+Version: 0.1.8
 Summary: OpenInference BeeAI Instrumentation
 Project-URL: Homepage, https://github.com/Arize-ai/openinference/tree/main/python/instrumentation/openinference-instrumentation-beeai
 Author: IBM Corp.
@@ -15,17 +15,17 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <3.14,>=3.11
-Requires-Dist: beeai-framework<0.2.0,>=0.1.32
-Requires-Dist: openinference-instrumentation>=0.1.36
+Requires-Dist: beeai-framework<0.2.0,>=0.1.36
+Requires-Dist: openinference-instrumentation>=0.1.37
 Requires-Dist: openinference-semantic-conventions>=0.1.21
 Requires-Dist: opentelemetry-api>=1.36.0
 Requires-Dist: opentelemetry-instrumentation>=0.57b0
 Requires-Dist: opentelemetry-semantic-conventions>=0.57b0
 Requires-Dist: wrapt>=1.17.2
 Provides-Extra: instruments
-Requires-Dist: beeai-framework>=0.1.32; extra == 'instruments'
+Requires-Dist: beeai-framework>=0.1.36; extra == 'instruments'
 Provides-Extra: test
-Requires-Dist: beeai-framework>=0.1.32; extra == 'test'
+Requires-Dist: beeai-framework>=0.1.36; extra == 'test'
 Requires-Dist: opentelemetry-exporter-otlp; extra == 'test'
 Requires-Dist: opentelemetry-sdk; extra == 'test'
 Description-Content-Type: text/markdown

{openinference_instrumentation_beeai-0.1.7.dist-info → openinference_instrumentation_beeai-0.1.8.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=nZQkVv1QTNVYu23EqEaRkYtncmpE4LLqEH_xCoHzM3c,2247
-openinference/instrumentation/beeai/version.py,sha256=YpKDcdV7CqL8n45u267wKtyloM13FSVbOdrqgNZnSLM,22
+openinference/instrumentation/beeai/_utils.py,sha256=m4ZXoGLm-iJGAJBjrfSh06OU2EaDMNgStVTLbljmPSA,2328
+openinference/instrumentation/beeai/version.py,sha256=C69ADlbQREQlR15trneyA2sk8x0-oH4rDAX5fsv19_U,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/chat.py,sha256=yyNjoG3AY2XPNHr6t2JJEwmx5NCibc3ThsWBL8lIuTU,9082
+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=LmYD9Oj9l_jRZEXoTyfMSykZQum0EHdBZOW6AtiPoHE,2720
+openinference/instrumentation/beeai/processors/tool.py,sha256=03pNF04ELsxRl8EK3JhHktoQl_aXAY1rrXHV4QuXcJk,2533
 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.7.dist-info/METADATA,sha256=j0D1RjhUiFGl8lZ8vzgQzt_yGWaZO1-2eQVDe6maZdI,5491
-openinference_instrumentation_beeai-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-openinference_instrumentation_beeai-0.1.7.dist-info/entry_points.txt,sha256=ee7EUhbWv-XK1dxhPXuFVy9qstzj-lc-265Phe2Ml9s,183
-openinference_instrumentation_beeai-0.1.7.dist-info/RECORD,,
+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,,