feat: update beeai-framework, add streaming (#1313)

* feat: update beeai-framework, add streaming

Signed-off-by: Tomas Dvorak <[email protected]>

* fixup! feat: update beeai-framework, add streaming

Signed-off-by: Radek Ježek <[email protected]>

* fixup! fixup! feat: update beeai-framework, add streaming

Signed-off-by: Lukáš Janeček <[email protected]>

* chore: upgrade framework

Signed-off-by: Tomas Weiss <[email protected]>

* chore: upgrading fw versions

Signed-off-by: Tomas Weiss <[email protected]>

---------

Signed-off-by: Tomas Dvorak <[email protected]>
Signed-off-by: Radek Ježek <[email protected]>
Signed-off-by: Lukáš Janeček <[email protected]>
Signed-off-by: Tomas Weiss <[email protected]>
Co-authored-by: Radek Ježek <[email protected]>
Co-authored-by: Lukáš Janeček <[email protected]>
Co-authored-by: Tomas Weiss <[email protected]>

Author: 4 people

agents/chat/pyproject.toml

@@ -7,10 +7,10 @@ authors = [
 ]
 requires-python = ">=3.13,<3.14"
 dependencies = [
-    "beeai-framework[duckduckgo,wikipedia,beeai-platform]>=0.1.50",
+    "beeai-framework[beeai-platform,duckduckgo,wikipedia]>=0.1.56",
     "openai>=1.107.1",
     "beeai-sdk",
-    "openinference-instrumentation-beeai>=0.1.11",
+    "openinference-instrumentation-beeai>=0.1.12",
     "pydantic-settings>=2.9.0",
     "uvicorn>=0.35.0",
 ]

agents/chat/src/chat/agent.py

@@ -10,16 +10,20 @@
     Message,
 )
 from beeai_framework.adapters.beeai_platform.backend.chat import BeeAIPlatformChatModel
-from beeai_framework.agents.experimental import RequirementAgent
-from beeai_framework.agents.experimental.events import RequirementAgentSuccessEvent
-from beeai_framework.agents.experimental.utils._tool import FinalAnswerTool
+from beeai_framework.agents.requirement import RequirementAgent
+from beeai_framework.agents.requirement.events import (
+    RequirementAgentSuccessEvent,
+    RequirementAgentFinalAnswerEvent,
+)
+from beeai_framework.agents.requirement.utils._tool import FinalAnswerTool
 from beeai_framework.emitter import EventMeta
 from beeai_framework.memory import UnconstrainedMemory
 from beeai_framework.middleware.trajectory import GlobalTrajectoryMiddleware
 from beeai_framework.tools import Tool
 from beeai_framework.tools.search.duckduckgo import DuckDuckGoSearchTool
 from beeai_framework.tools.search.wikipedia import WikipediaTool
 from beeai_framework.tools.weather import OpenMeteoTool
+from beeai_framework.backend import ChatModelParameters
 from beeai_sdk.a2a.extensions import (
     AgentDetail,
     AgentDetailContributor,
@@ -214,7 +218,8 @@ async def chat(
         ActAlwaysFirstRequirement(),  #  Enforces the ActTool to be used before any other tool execution.
     ]
 
-    llm = BeeAIPlatformChatModel()
+    use_streaming = True
+    llm = BeeAIPlatformChatModel(parameters=ChatModelParameters(stream=use_streaming))
     llm.set_context(llm_ext)
 
     # Build dynamic instructions based on available files
@@ -288,6 +293,9 @@ async def chat(
     new_messages = [to_framework_message(item, extracted_files) for item in history]
 
     async for event, meta in agent.run(new_messages):
+        if isinstance(event, RequirementAgentFinalAnswerEvent) and use_streaming:
+            yield event.delta
+
         if not isinstance(event, RequirementAgentSuccessEvent):
             continue
 
@@ -325,7 +333,8 @@ async def chat(
             text=clean_text,
             metadata=(citation.citation_metadata(citations=citations) if citations else None),
         )
-        yield message
+        if not use_streaming:
+            yield message
         await context.store(message)
 
 

agents/chat/uv.lock

@@ -7,10 +7,10 @@ authors = [
 ]
 requires-python = ">=3.13,<3.14"
 dependencies = [
-    "beeai-framework[duckduckgo,wikipedia]>=0.1.50",
+    "beeai-framework[duckduckgo,wikipedia]>=0.1.56",
     "openai==1.99.9",
     "beeai-sdk",
-    "openinference-instrumentation-beeai>=0.1.11",
+    "openinference-instrumentation-beeai>=0.1.12",
     "pydantic-settings>=2.9.0",
     "uvicorn>=0.35.0",
 ]

agents/form/pyproject.toml

@@ -7,10 +7,10 @@ authors = [
 ]
 requires-python = ">=3.13,<3.14"
 dependencies = [
-    "beeai-framework[duckduckgo,wikipedia]>=0.1.50",
+    "beeai-framework[duckduckgo,wikipedia]>=0.1.56",
     "beeai-sdk",
     "openai>=1.107.1",
-    "openinference-instrumentation-beeai>=0.1.11",
+    "openinference-instrumentation-beeai>=0.1.12",
     "pydantic-settings>=2.9.0",
     "tenacity>=9.1.2",
     "langchain-text-splitters>=0.3.0",