feat: update beeai-framework (#51)

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

Author: Tomas2D

beeai_framework_starter/agent.py

@@ -3,7 +3,6 @@
 import sys
 import traceback
 
-from beeai_framework.agents import AgentExecutionConfig
 from beeai_framework.agents.experimental import RequirementAgent
 from beeai_framework.agents.experimental.requirements.conditional import ConditionalRequirement
 from beeai_framework.backend import ChatModel
@@ -38,11 +37,9 @@ async def main() -> None:
     reader = ConsoleReader({"fallback": "What to do in Boston?"})
 
     for prompt in reader:
-        response = await agent.run(
-            prompt, execution=AgentExecutionConfig(max_iterations=8, max_retries_per_step=3, total_max_retries=10)
-        )
+        response = await agent.run(prompt, max_iterations=8, max_retries_per_step=3, total_max_retries=10)
 
-        reader.write("Agent 🤖 : ", response.result.text)
+        reader.write("Agent 🤖 : ", response.last_message.text)
 
 
 if __name__ == "__main__":

beeai_framework_starter/agent_code_interpreter.py

@@ -3,7 +3,6 @@
 import sys
 import traceback
 
-from beeai_framework.agents import AgentExecutionConfig
 from beeai_framework.agents.react import ReActAgent
 from beeai_framework.backend import ChatModel
 from beeai_framework.errors import FrameworkError
@@ -67,11 +66,11 @@ def get_riddle() -> Optional[Dict[str, str]]:
     reader = ConsoleReader({"fallback": "Generate a random riddle."})
 
     for prompt in reader:
-        response = await agent.run(
-            prompt, execution=AgentExecutionConfig(max_iterations=8, max_retries_per_step=3, total_max_retries=10)
-        ).on("update", lambda data, event: reader.write(f"Agent 🤖 ({data.update.key}) : ", data.update.parsed_value))
+        response = await agent.run(prompt, max_iterations=8, max_retries_per_step=3, total_max_retries=10).on(
+            "update", lambda data, event: reader.write(f"Agent 🤖 ({data.update.key}) : ", data.update.parsed_value)
+        )
 
-        reader.write("Agent 🤖 : ", response.result.text)
+        reader.write("Agent 🤖 : ", response.last_message.text)
 
 
 if __name__ == "__main__":

beeai_framework_starter/agent_observe.py

@@ -3,7 +3,6 @@
 import sys
 import traceback
 
-from beeai_framework.agents import AgentExecutionConfig
 from beeai_framework.agents.experimental import RequirementAgent
 from beeai_framework.agents.experimental.requirements.conditional import ConditionalRequirement
 from beeai_framework.backend import ChatModel
@@ -42,11 +41,9 @@ async def main() -> None:
     reader = ConsoleReader({"fallback": "What to do in Boston?"})
 
     for prompt in reader:
-        response = await agent.run(
-            prompt, execution=AgentExecutionConfig(max_iterations=8, max_retries_per_step=3, total_max_retries=10)
-        )
+        response = await agent.run(prompt, max_iterations=8, max_retries_per_step=3, total_max_retries=10)
 
-        reader.write("Agent 🤖 : ", response.result.text)
+        reader.write("Agent 🤖 : ", response.last_message.text)
 
 
 if __name__ == "__main__":

beeai_framework_starter/agent_requirement.py

@@ -21,7 +21,7 @@ async def main() -> None:
     )
 
     response = await agent.run("What to do in Boston?").middleware(GlobalTrajectoryMiddleware(excluded=[]))
-    print(response.answer.text)
+    print(response.last_message.text)
 
 
 if __name__ == "__main__":

beeai_framework_starter/agent_tool_calling.py

@@ -28,7 +28,7 @@ def on_success(data: ToolCallingAgentSuccessEvent, event: EventMeta) -> None:
 
     for prompt in reader:
         response = await agent.run(prompt).on("success", on_success)
-        reader.write("Agent 🤖 : ", response.result.text)
+        reader.write("Agent 🤖 : ", response.last_message.text)
 
 
 if __name__ == "__main__":

poetry.lock

@@ -15,7 +15,7 @@ repository = "https://github.com/i-am-bee/beeai-framework-py-starter"
 
 [tool.poetry.dependencies]
 python = ">= 3.11,<3.14"
-beeai-framework = {extras = ["all"], version = "^0.1.40"}
+beeai-framework = {extras = ["all"], version = "^0.1.42"}
 openinference-instrumentation-beeai = "^0.1.10"
 opentelemetry-exporter-otlp = "^1.36.0"
 opentelemetry-sdk = "^1.36.0"