From ef6c2aede70d155bc2f05b39a016f600d6179389 Mon Sep 17 00:00:00 2001 From: Philippe Page Date: Tue, 21 Jan 2025 09:17:49 -0500 Subject: [PATCH 1/2] patch initial_response handling in task.py --- packages/python/src/mainframe_orchestra/task.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/packages/python/src/mainframe_orchestra/task.py b/packages/python/src/mainframe_orchestra/task.py index 7cae01d..689cd1e 100644 --- a/packages/python/src/mainframe_orchestra/task.py +++ b/packages/python/src/mainframe_orchestra/task.py @@ -806,6 +806,8 @@ async def callback_wrapper(event): # Consume the stream async for chunk in initial_stream: pass # The callback will handle the chunks + else: + await self._direct_llm_call(callback) finally: self.require_json_output = original_json_requirement self.messages = original_messages From 9b9123c7cdd99852f9fc6540f507ce938564e6fa Mon Sep 17 00:00:00 2001 From: Philippe Page Date: Tue, 21 Jan 2025 10:04:24 -0500 Subject: [PATCH 2/2] Bump version to v0.0.16 --- packages/python/pyproject.toml | 2 +- packages/python/src/mainframe_orchestra/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/python/pyproject.toml b/packages/python/pyproject.toml index c6a8ba9..e879d62 100755 --- a/packages/python/pyproject.toml +++ b/packages/python/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "mainframe-orchestra" -version = "0.0.15" +version = "0.0.16" description = "Mainframe-Orchestra is a lightweight, open-source agentic framework for building LLM based pipelines and self-orchestrating multi-agent teams" authors = [ "Mainframe Computer Inc. ", diff --git a/packages/python/src/mainframe_orchestra/__init__.py b/packages/python/src/mainframe_orchestra/__init__.py index 633db4e..17eca19 100755 --- a/packages/python/src/mainframe_orchestra/__init__.py +++ b/packages/python/src/mainframe_orchestra/__init__.py @@ -3,7 +3,7 @@ """ # Copyright 2024 Mainframe-Orchestra Contributors. Licensed under Apache License 2.0. -__version__ = "0.0.15" +__version__ = "0.0.16" from .task import Task, configure_logging, LogColors, default_logger from .agent import Agent