From 17a1e783e3b4e9f2aa8dad802b2eee4d5eec4022 Mon Sep 17 00:00:00 2001 From: Adrian Cole Date: Wed, 1 Oct 2025 12:32:49 +0800 Subject: [PATCH] fix(beeai): handle removal of experimental features Signed-off-by: Adrian Cole --- .../openinference-instrumentation-beeai/pyproject.toml | 6 +++--- .../src/openinference/instrumentation/beeai/__init__.py | 2 +- .../beeai/processors/agents/requirement_agent.py | 4 ++-- .../instrumentation/beeai/processors/locator.py | 4 ++-- .../instrumentation/beeai/processors/requirement.py | 8 ++++---- 5 files changed, 12 insertions(+), 12 deletions(-) diff --git a/python/instrumentation/openinference-instrumentation-beeai/pyproject.toml b/python/instrumentation/openinference-instrumentation-beeai/pyproject.toml index cb203c6c8f..7b556949aa 100644 --- a/python/instrumentation/openinference-instrumentation-beeai/pyproject.toml +++ b/python/instrumentation/openinference-instrumentation-beeai/pyproject.toml @@ -22,7 +22,7 @@ classifiers = [ "Programming Language :: Python :: 3.13", ] dependencies = [ - "beeai-framework (>=0.1.49,<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.49", + "beeai-framework >= 0.1.51", ] test = [ - "beeai-framework >= 0.1.49", + "beeai-framework >= 0.1.51", "opentelemetry-sdk", "opentelemetry-exporter-otlp" ] diff --git a/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/__init__.py b/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/__init__.py index fd14c67d32..d3f3adc6d6 100644 --- a/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/__init__.py +++ b/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/__init__.py @@ -24,7 +24,7 @@ logger = logging.getLogger(__name__) -_instruments = ("beeai-framework >= 0.1.32",) +_instruments = ("beeai-framework >= 0.1.51",) try: __version__ = version("beeai-framework") except PackageNotFoundError: diff --git a/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/agents/requirement_agent.py b/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/agents/requirement_agent.py index d74068418b..8a0ab4a0e9 100644 --- a/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/agents/requirement_agent.py +++ b/python/instrumentation/openinference-instrumentation-beeai/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 diff --git a/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/locator.py b/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/locator.py index 909d9d0a15..35ac3c805d 100644 --- a/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/locator.py +++ b/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/locator.py @@ -47,14 +47,14 @@ def _load() -> None: 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, diff --git a/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/requirement.py b/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/requirement.py index 5472a9f485..85d118b190 100644 --- a/python/instrumentation/openinference-instrumentation-beeai/src/openinference/instrumentation/beeai/processors/requirement.py +++ b/python/instrumentation/openinference-instrumentation-beeai/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