ea4all-gradio-agents-mcp-hackathon-tools-refactoring-vqa-safety-status

ea4all/ea4all_mcp.py

@@ -132,7 +132,7 @@ async def run_vqa_agentic_system(question: str, diagram: str, request: gr.Reques
 
                 vqa_image = diagram
                 response = await diagram_graph.ainvoke({"question":msg, "image": vqa_image}, config)
-                chat_memory.append(ChatMessage(role="assistant", content=response['messages'][-1].content))
+                chat_memory.append(ChatMessage(role="assistant", content=response['messages'][-1].content if len(response['messages']) else response['safety_status']['description']))
                 
                 yield chat_memory
 
@@ -160,7 +160,7 @@ async def run_reference_architecture_agentic_system(business_query: str) -> Asyn
             input=inputs, 
             config=config
         ) #astream not loading the graph
-        yield response.content
+        yield response
 
 async def run_pmo_agentic_system(question:str) -> AsyncGenerator[list, None]:
     """

ea4all/src/ea4all_vqa/graph.py

@@ -193,22 +193,13 @@ Given the conversation above, is the image safe to be processed? Does the image
     safe_request = result['isSafe']
     description = result['description']
 
-    if architecture_image and safe_request:
-        print("---ROUTE REQUEST TO DIAGRAM SUPERVISOR---")
-        print(f"Architecture Diagram: {architecture_image} --- isSafe: {safe_request} --- {question} --- {description}")
-        result = True
-    else:
-        print("---ROUTE REQUEST TO FINISH---")
-        print(f"Architecture Diagram: {architecture_image} --- isSafe: {safe_request} -- What is it? {description}")
-        result = False
-    
-    return {'safety_status': result}
+    return {"safety_status": result}
+
+def call_finish(state:OverallState, config:RunnableConfig) -> dict:
 
-def call_finish(state):
     return {
-        "state": state,
         "messages": [],
-        "next": "end",
+        "safety_status": getattr(state, 'safety_status', {}),
     } 
 
 def make_supervisor_node(model: BaseChatModel, members: list[str]) -> RunnableLambda:
@@ -352,7 +343,11 @@ def enter_graph(state:OverallState, config:RunnableConfig) -> Command[Literal['s
     }
 
 async def choose_next(state: OverallState):
-    return "diagram_supervisor" if state.safety_status else "final"
+    """Choose the next node based on the safety status."""
+    isArcihitectureImage = getattr(state, 'safety_status', {}).get('isArchitectureImage', False)
+    isSafe = getattr(state, 'safety_status', {}).get('isSafe', False)
+
+    return "diagram_supervisor" if isArcihitectureImage and isSafe else "final"
 
 def build_vqa_graph():
     model = get_llm_client(BaseConfiguration.supervisor_model, api_base_url="", streaming=BaseConfiguration.streaming)

ea4all/src/ea4all_vqa/state.py

@@ -44,7 +44,7 @@ class OutputState:
     """
 
     messages: Optional[Annotated[Sequence[MessagesState], operator.add]] = None
-    safety_status: Optional[bool] = None
+    safety_status: Optional[dict[str,str]] = None
 
     """Attributes:
     safety_status: safety status of the diagram provided by the user