From 6a67fc37a385e26d24eea60d1792059b050162f7 Mon Sep 17 00:00:00 2001
From: Akihiko Kuroda <akihikokuroda2020@gmail.com>
Date: Mon, 24 Feb 2025 18:55:56 -0500
Subject: [PATCH] wrap workflow.run with asyncio.run

Signed-off-by: Akihiko Kuroda <akihikokuroda2020@gmail.com>
---
 maestro/cli/commands.py    | 3 ++-
 maestro/deployments/api.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/maestro/cli/commands.py b/maestro/cli/commands.py
index c8987dc..a05eb4c 100644
--- a/maestro/cli/commands.py
+++ b/maestro/cli/commands.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import os, yaml, json, jsonschema, traceback
+import asyncio
 
 from openai import OpenAI
 from jsonschema.exceptions import ValidationError, SchemaError
@@ -171,7 +172,7 @@ def __init__(self, args):
     def __run_agents_workflow(self, agents_yaml, workflow_yaml):
         try:
             workflow = Workflow(agents_yaml, workflow_yaml[0])
-            workflow.run()
+            asyncio.run(workflow.run())
         except Exception as e:
             self._check_verbose()
             raise RuntimeError("Unable to run workflow: {message}".format(message=str(e)))
diff --git a/maestro/deployments/api.py b/maestro/deployments/api.py
index 2e0c325..6929c45 100644
--- a/maestro/deployments/api.py
+++ b/maestro/deployments/api.py
@@ -4,6 +4,7 @@
 import json
 import sys
 import io
+import asyncio
 
 import yaml
 from src.workflow import Workflow
@@ -27,7 +28,7 @@ def process_workflow():
 
         output = io.StringIO()
         sys.stdout = output
-        workflow_instance.run()
+        asyncio.run(workflow_instance.run())
         sys.stdout = sys.__stdout__
 
         response = {