Reorganize imports in agent.py

Author: jenna-winkler

src/agentstack_agents/agent.py

@@ -1,8 +1,6 @@
 import os
 from typing import Annotated
 
-from beeai_framework.adapters.openai import OpenAIChatModel
-from beeai_framework.backend.types import ChatModelParameters
 from dotenv import load_dotenv
 
 import a2a.types
@@ -11,8 +9,11 @@
 from agentstack_sdk.server.context import RunContext
 from agentstack_sdk.a2a.extensions import AgentDetailExtensionSpec, AgentDetail, CitationExtensionServer, CitationExtensionSpec, TrajectoryExtensionServer, TrajectoryExtensionSpec, LLMServiceExtensionServer, LLMServiceExtensionSpec
 from agentstack_sdk.a2a.extensions.ui.form import TextField, MultiSelectField, OptionItem, FormExtensionServer, FormExtensionSpec, FormRender
-from beeai_framework.agents.experimental import RequirementAgent
-from beeai_framework.agents.experimental.requirements.conditional import ConditionalRequirement
+
+from beeai_framework.adapters.openai import OpenAIChatModel
+from beeai_framework.backend.types import ChatModelParameters
+from beeai_framework.agents.requirement import RequirementAgent
+from beeai_framework.agents.requirement.requirements.conditional import ConditionalRequirement
 from beeai_framework.agents.types import AgentExecutionConfig
 from beeai_framework.backend.message import UserMessage
 from beeai_framework.memory import UnconstrainedMemory