openinference-instrumentation-beeai 0.1.10__tar.gz → 0.1.12__tar.gz

{openinference_instrumentation_beeai-0.1.10 → openinference_instrumentation_beeai-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openinference-instrumentation-beeai
-Version: 0.1.10
+Version: 0.1.12
 Summary: OpenInference BeeAI Instrumentation
 Project-URL: Homepage, https://github.com/Arize-ai/openinference/tree/main/python/instrumentation/openinference-instrumentation-beeai
 Author: IBM Corp.
@@ -15,7 +15,7 @@ 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.36
+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: opentelemetry-api>=1.36.0
@@ -23,9 +23,9 @@ 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.36; extra == 'instruments'
+Requires-Dist: beeai-framework>=0.1.51; extra == 'instruments'
 Provides-Extra: test
-Requires-Dist: beeai-framework>=0.1.36; extra == 'test'
+Requires-Dist: beeai-framework>=0.1.51; 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.10 → openinference_instrumentation_beeai-0.1.12}/pyproject.toml

@@ -22,7 +22,7 @@ classifiers = [
   "Programming Language :: Python :: 3.13",
 ]
 dependencies = [
-    "beeai-framework (>=0.1.36,<0.2.0)",
+    "beeai-framework (>=0.1.51,<0.2.0)",
     "openinference-instrumentation>=0.1.37",
     "openinference-semantic-conventions>=0.1.21",
     "opentelemetry-api>=1.36.0",
@@ -33,10 +33,10 @@ dependencies = [
 
 [project.optional-dependencies]
 instruments = [
-  "beeai-framework >= 0.1.36",
+  "beeai-framework >= 0.1.51",
 ]
 test = [
-  "beeai-framework >= 0.1.36",
+  "beeai-framework >= 0.1.51",
   "opentelemetry-sdk",
   "opentelemetry-exporter-otlp"
 ]

{openinference_instrumentation_beeai-0.1.10 → openinference_instrumentation_beeai-0.1.12}/src/openinference/instrumentation/beeai/__init__.py

@@ -24,7 +24,7 @@ from openinference.instrumentation.beeai.processors.locator import ProcessorLoca
 
 logger = logging.getLogger(__name__)
 
-_instruments = ("beeai-framework >= 0.1.32",)
+_instruments = ("beeai-framework >= 0.1.51",)
 try:
     __version__ = version("beeai-framework")
 except PackageNotFoundError:

{openinference_instrumentation_beeai-0.1.10 → openinference_instrumentation_beeai-0.1.12}/src/openinference/instrumentation/beeai/processors/agents/requirement_agent.py

@@ -1,10 +1,10 @@
 from typing import Any
 
-from beeai_framework.agents.experimental.events import (
+from beeai_framework.agents.requirement.events import (
     RequirementAgentStartEvent,
     RequirementAgentSuccessEvent,
 )
-from beeai_framework.agents.experimental.types import RequirementAgentRunStateStep
+from beeai_framework.agents.requirement.types import RequirementAgentRunStateStep
 from beeai_framework.context import RunContextStartEvent
 from beeai_framework.emitter import EventMeta
 from typing_extensions import override

{openinference_instrumentation_beeai-0.1.10 → openinference_instrumentation_beeai-0.1.12}/src/openinference/instrumentation/beeai/processors/chat.py

@@ -99,7 +99,7 @@ class ChatModelProcessor(Processor):
 
             case ChatModelSuccessEvent():
                 if not self._messages:  # only when no streaming
-                    self._add_new_messages(event.value.messages)
+                    self._add_new_messages(event.value.output)
 
                 usage = event.value.usage
                 if usage:
@@ -126,7 +126,7 @@ class ChatModelProcessor(Processor):
                         SpanAttributes.OPENINFERENCE_SPAN_KIND: type(self).kind,
                         SpanAttributes.OUTPUT_VALUE: event.value.get_text_content(),
                         SpanAttributes.OUTPUT_MIME_TYPE: OpenInferenceMimeTypeValues.TEXT.value,
-                        f"{SpanAttributes.METADATA}.chunks_count": len(event.value.messages),
+                        f"{SpanAttributes.METADATA}.chunks_count": len(event.value.output),
                         **_unpack_object(
                             usage.model_dump(exclude_none=True) if usage else {},
                             prefix=f"{SpanAttributes.METADATA}.usage",

{openinference_instrumentation_beeai-0.1.10 → openinference_instrumentation_beeai-0.1.12}/src/openinference/instrumentation/beeai/processors/locator.py

@@ -47,14 +47,14 @@ class ProcessorLocator:
             ProcessorLocator.entries[ToolCallingAgent] = ToolCallingAgentProcessor
 
         with contextlib.suppress(ImportError):
-            from beeai_framework.agents.experimental.agent import RequirementAgent
+            from beeai_framework.agents.requirement import RequirementAgent
 
             from .agents.requirement_agent import RequirementAgentProcessor
 
             ProcessorLocator.entries[RequirementAgent] = RequirementAgentProcessor
 
         with contextlib.suppress(ImportError):
-            from beeai_framework.agents.experimental.requirements.requirement import Requirement
+            from beeai_framework.agents.requirement.requirements.requirement import Requirement
 
             from openinference.instrumentation.beeai.processors.requirement import (
                 RequirementProcessor,

{openinference_instrumentation_beeai-0.1.10 → openinference_instrumentation_beeai-0.1.12}/src/openinference/instrumentation/beeai/processors/requirement.py

@@ -1,9 +1,9 @@
 from typing import Any, ClassVar
 
-from beeai_framework.agents.experimental.requirements.ask_permission import AskPermissionRequirement
-from beeai_framework.agents.experimental.requirements.conditional import ConditionalRequirement
-from beeai_framework.agents.experimental.requirements.events import RequirementInitEvent
-from beeai_framework.agents.experimental.requirements.requirement import Requirement
+from beeai_framework.agents.requirement.requirements.ask_permission import AskPermissionRequirement
+from beeai_framework.agents.requirement.requirements.conditional import ConditionalRequirement
+from beeai_framework.agents.requirement.requirements.events import RequirementInitEvent
+from beeai_framework.agents.requirement.requirements.requirement import Requirement
 from beeai_framework.context import RunContext, RunContextStartEvent
 from beeai_framework.emitter import EventMeta
 from typing_extensions import override

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

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

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

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