ea4all-gradio-agents-mcp-hackathon-tools-apm-upload

ea4all/ea4all_mcp.py

@@ -9,12 +9,9 @@ from langchain_core.runnables import RunnableConfig
 
 from ea4all.src.shared.configuration import BaseConfiguration, APM_MOCK_QNA, PMO_MOCK_QNA
 from ea4all.src.ea4all_gra.configuration import AgentConfiguration as gra
-import ea4all.src.ea4all_vqa.graph as e4v
-import ea4all.src.ea4all_gra.graph as e4t
-import ea4all.src.shared.utils as e4u
-from ea4all.src.graph import super_graph
 from ea4all.src.ea4all_apm.graph import apm_graph
 from ea4all.src.ea4all_vqa.graph import diagram_graph
+from ea4all.src.ea4all_togaf.graph import togaf_graph
 from ea4all.src.ea4all_indexer.graph import indexer_graph
 
 #from ea4all.src.pmo_crew.crew_runner import run_pmo_crew
@@ -53,7 +50,7 @@ config = RunnableConfig(
         #stream_mode = "messages"
 )
 
-async def call_indexer_apm(state: State, config: RunnableConfig):
+async def call_indexer_apm(config: RunnableConfig):
     response = await indexer_graph.ainvoke(input={"docs":[]}, config=config)
 
 #ea4all-qna-agent-conversational-with-memory
@@ -144,68 +141,19 @@ async def run_reference_architecture_agentic_system(business_query: str) -> Asyn
         response: High-level architecture blueprint and target diagram.
     """
 
-    if len(business_query) < 50:
+    if len(business_query) < 20:
         agent_response = "Please provide a valid Business Requirement content to start!"
         yield([agent_response,gr.Tabs(visible=True), gr.Tabs(selected="id_togaf"),None, None, gr.Tabs(visible=False)])
     else: 
-        plain_text = e4u.markdown_to_plain_text(business_query)
-        agent_response = "Generating Architecture Blueprint ---TOGAF VISION TARGET--- \n\nI am working on your request..."
-        togaf_chain = e4t.togaf_graph
-        final_diagram = ""
-        vision_message = ""
-        try:
-            async for s in togaf_chain.astream_events(
-                {
-                    "messages": [
-                        HumanMessage(
-                            content=plain_text
-                        )
-                    ],
-                    "business_query": business_query,
-                },
-                config=config,
-                version="v2"
-            ):    
-                kind = s["event"]
-                tags = s.get("tags", [])
-                name = s['name']
-
-                if "gra_stream" in tags and name == "stream_vision_target":
-                    if kind == "on_chain_stream":
-                        data = s["data"]
-                        # Accumulate the chunk of data
-                        chunk = data.get('chunk')
-                        if chunk is not None and hasattr(chunk, "content"):
-                            vision_message += chunk.content
-                        time.sleep(e4u.CFG.STREAM_SLEEP)
-                        yield([vision_message,gr.Tabs(visible=True), gr.Tabs(selected="id_togaf"),None, None, gr.Tabs(visible=False)])
-                elif name == "save_diagram" and kind == 'on_chain_end': #MOVED INTO Togaf_Task3
-                    output = s['data'].get('output', {})
-                    final_diagram = output.get('architecture_runway', "")
-                elif ("assess_business_query" in tags or "assess_landscape" in tags) and kind == 'on_chain_start': ##'on_chat_model_stream':
-                    agent_response += f"\n\n`{tags[-1]}:{name}`"
-
-                    yield([agent_response,gr.Tabs(visible=True), gr.Tabs(selected="id_togaf"),None, None, gr.Tabs(visible=False)])
-                
-            if vision_message=="":
-                agent_response = "I cannot generate the Architecture Vision. Please provide a valid Business Requirement content to start!"
-                yield([agent_response,gr.Tabs(visible=True), gr.Tabs(selected="id_togaf"),None, None, gr.Tabs(visible=False)])
-            elif "Error" not in final_diagram:
-                yield([vision_message,gr.Tabs(visible=True), gr.Tabs(selected="id_togaf"),final_diagram, None, gr.Tabs(visible=True)])
-            else:
-                yield([vision_message,gr.Tabs(visible=True), gr.Tabs(selected="id_togaf"),None, final_diagram, gr.Tabs(visible=True)])
-
-        except Exception as e:  
-            yield(
-                [
-                    e.args[-1],
-                    gr.Tabs(visible=True), 
-                    gr.Tabs(selected="id_togaf"),
-                    None,
-                    None,
-                    gr.Tabs(visible=False)
-                ]
-            )
+        inputs = {"messages": [{"role": "user", "content": business_query}]} #user response
+
+        response = await togaf_graph.ainvoke(
+            input=inputs, 
+            config=config
+        ) #astream not loading the graph
+        yield response.content
+
+
 
 async def run_pmo_agentic_system(question:str) -> AsyncGenerator[list, None]:
     """