File size: 4,474 Bytes
43e9024 70a2c7c 43e9024 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 |
import json
from pathlib import Path
import gradio as gr
from gradio_agent_inspector import AgentInspector
import os
def simulate_conversation():
dir_path = Path(os.path.dirname(os.path.realpath(__file__)))
initial_state = {
"state": {},
"events": [],
}
states = []
for i in range(2):
session_value_p = dir_path / "session-sample" / f"value-{i}.json"
with session_value_p.open("r", encoding="utf-8") as f:
session_value = json.load(f)
# attach event trace and graph info to the event
for e in session_value["events"]:
event_trace_p = dir_path / "event-trace" / f"trace-{e['id']}.json"
if event_trace_p.exists():
with event_trace_p.open("r", encoding="utf-8") as trace_f:
event_trace = json.load(trace_f)
if "gcp.vertex.agent.llm_request" in event_trace:
event_trace["gcp.vertex.agent.llm_request"] = json.loads(
event_trace["gcp.vertex.agent.llm_request"]
)
if "gcp.vertex.agent.llm_response" in event_trace:
event_trace["gcp.vertex.agent.llm_response"] = json.loads(
event_trace["gcp.vertex.agent.llm_response"]
)
e["trace"] = event_trace
event_graph_p = dir_path / "event-trace" / f"graph-{e['id']}.json"
if event_graph_p.exists():
with event_graph_p.open("r", encoding="utf-8") as graph_f:
event_graph = json.load(graph_f)
e["graph"] = event_graph
states.append(session_value)
return initial_state, states
def update_conversation_state(state_index, states):
if (state_index + 1) >= len(states):
return states[state_index], state_index
else:
new_index = state_index + 1
return states[new_index], new_index
initial_state, conversation_states = simulate_conversation()
with gr.Blocks() as demo:
gr.Markdown(
"""# 🕵️ Agent Inspector
Debugging agent-based applications can be tricky!
Agent Inspector is a Gradio component designed to make this process easier and more transparent.
Inspired by tools like the [ADK web](https://github.com/google/adk-web) debug panel, Agent Inspector brings similar functionality to the Gradio ecosystem.
Whether you're building, testing, or fine-tuning your agent, Agent Inspector helps you understand what's happening under the hood.
Demo App with ADK :
[](https://huggingface.co/spaces/Agents-MCP-Hackathon/adk-gradio?logs=container)
Github repo :
[](https://github.com/francoislanc/adk-gradio)
"""
)
state_counter = gr.State(-1)
agent_inspector = AgentInspector(json.dumps(initial_state))
with gr.Row():
next_btn = gr.Button(
f"▶️ Next ({0} / {len(conversation_states)})", variant="primary"
)
reset_btn = gr.Button("🔄 Reset", variant="secondary")
def next_state(current_counter):
new_state, new_counter = update_conversation_state(
current_counter, conversation_states
)
json_state = json.dumps(new_state)
next_button_label = f"▶️ Next ({new_counter+1} / {len(conversation_states)})"
return json_state, new_counter, next_button_label
def reset_conversation():
json_state = json.dumps(initial_state)
next_button_label = f"▶️ Next ({0} / {len(conversation_states)})"
return json_state, -1, next_button_label
next_btn.click(
next_state,
inputs=[state_counter],
outputs=[agent_inspector, state_counter, next_btn],
)
reset_btn.click(
reset_conversation, outputs=[agent_inspector, state_counter, next_btn]
)
# examples = gr.Examples(
# examples=[
# s for s in conversation_states
# ],
# inputs=[initial_state],
# )
if __name__ == "__main__":
demo.launch()
|