ea4all-gradio-mcp-hackathon-runaway

ea4all/ea4all_mcp.py

@@ -162,7 +162,11 @@ async def run_reference_architecture_agentic_system(business_query: str) -> Asyn
             config=config
         ) #astream not loading the graph
         vision_target = response['vision_target']
-        architecture_runway = response['architecture_runway']
+        if isinstance(response['architecture_runway'], str):
+            architecture_runway = response['architecture_runway']
+        else:
+            architecture_runway = None
+
         yield [vision_target, architecture_runway]
 
 async def run_pmo_agentic_system(question:str) -> AsyncGenerator[list, None]:

ea4all/src/ea4all_gra/togaf_task3/graph.py

@@ -177,7 +177,7 @@ def generate_architecture_runway(state: Task3State, config: RunnableConfig):
         "architecture_runway": architecture_runway.content
     }
 
-def save_diagram(state: Task3State, config: RunnableConfig, fmt=["svg","png"]): 
+def save_diagram(state: Task3State, config: RunnableConfig, fmt=["svg","png"]) -> dict | Exception: 
     configuration = AgentConfiguration.from_runnable_config(config)
 
     #_config = config.get('configurable')
@@ -203,8 +203,9 @@ def save_diagram(state: Task3State, config: RunnableConfig, fmt=["svg","png"]):
             view=False,
             ).replace('\\', '/')
     except Exception as e:
-        response=f"Error: Agent couldn't parse the diagram at this time! {e} \n {output_img} \n {input_img}"
-
+        #response=f"Error: Agent couldn't parse the diagram at this time! {e} \n {output_img} \n {input_img}"
+        return e
+    
     return{"architecture_runway": response}
 
 def generate_reference_architecture(state: Task3State):