Merge pull request #30 from mainframecomputer/patch/initial_response

patch initial_response handling in task.py
mainframecomputer · Jan 22, 2025 · c4cfe76 · c4cfe76
2 parents 539603b + 9b9123c
commit c4cfe76
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git 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. <[email protected]>",

diff --git 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

diff --git 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