ea4all-gradio-agents-mcp-hackathon-tools-apm-upload
Browse files- ea4all/ea4all_mcp.py +10 -0
ea4all/ea4all_mcp.py
CHANGED
@@ -15,6 +15,7 @@ import ea4all.src.shared.utils as e4u
|
|
15 |
from ea4all.src.graph import super_graph
|
16 |
from ea4all.src.ea4all_apm.graph import apm_graph
|
17 |
from ea4all.src.ea4all_vqa.graph import diagram_graph
|
|
|
18 |
|
19 |
#from ea4all.src.pmo_crew.crew_runner import run_pmo_crew
|
20 |
|
@@ -52,6 +53,9 @@ config = RunnableConfig(
|
|
52 |
#stream_mode = "messages"
|
53 |
)
|
54 |
|
|
|
|
|
|
|
55 |
#ea4all-qna-agent-conversational-with-memory
|
56 |
async def run_qna_agentic_system(question: str) -> AsyncGenerator[list, None]:
|
57 |
"""
|
@@ -415,3 +419,9 @@ with gr.Blocks(title="Your ArchitectGPT",fill_height=True, fill_width=True) as e
|
|
415 |
|
416 |
#Set initial state of apm and llm
|
417 |
ea4all_mcp.load(ea4all_agent_init, outputs=[ea4all_agent_metadata,ea4all_chatbot, ea4all_vqa, pmo_chatbot, confluence_list, confluence_df, apm_df, pmo_df], show_api=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
from ea4all.src.graph import super_graph
|
16 |
from ea4all.src.ea4all_apm.graph import apm_graph
|
17 |
from ea4all.src.ea4all_vqa.graph import diagram_graph
|
18 |
+
from ea4all.src.ea4all_indexer.graph import indexer_graph
|
19 |
|
20 |
#from ea4all.src.pmo_crew.crew_runner import run_pmo_crew
|
21 |
|
|
|
53 |
#stream_mode = "messages"
|
54 |
)
|
55 |
|
56 |
+
async def call_indexer_apm(state: State, config: RunnableConfig):
|
57 |
+
response = await indexer_graph.ainvoke(input={"docs":[]}, config=config)
|
58 |
+
|
59 |
#ea4all-qna-agent-conversational-with-memory
|
60 |
async def run_qna_agentic_system(question: str) -> AsyncGenerator[list, None]:
|
61 |
"""
|
|
|
419 |
|
420 |
#Set initial state of apm and llm
|
421 |
ea4all_mcp.load(ea4all_agent_init, outputs=[ea4all_agent_metadata,ea4all_chatbot, ea4all_vqa, pmo_chatbot, confluence_list, confluence_df, apm_df, pmo_df], show_api=False)
|
422 |
+
ea4all_mcp.then(
|
423 |
+
call_indexer_apm,
|
424 |
+
inputs=None,
|
425 |
+
outputs=None,
|
426 |
+
show_api=False
|
427 |
+
) #call indexer graph to load the APM
|