283 lines
13 KiB
Go
283 lines
13 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
|
|
mujica "mujicaui"
|
|
)
|
|
|
|
const agentCategory = "Agent"
|
|
|
|
var agentPages = []page{
|
|
{ID: "261", Name: "ToolCallCard", ZhName: "工具调用卡片", Category: agentCategory, Demo: toolCallCardDemo, Code: toolCallCardCode,
|
|
Keys: "Tab 到参数、结果折叠按钮，Enter / Space 展开或折叠。"},
|
|
{ID: "262", Name: "ToolCallGroup", ZhName: "工具调用组", Category: agentCategory, Demo: toolCallGroupDemo, Code: toolCallGroupCode,
|
|
Keys: "Tab 到组标题，Enter / Space 展开或折叠。"},
|
|
{ID: "263", Name: "ToolApprovalDialog", ZhName: "工具审批对话框", Category: agentCategory, Demo: toolApprovalDemo, Code: toolApprovalDialogCode,
|
|
Keys: "打开时焦点在拒绝；Enter 只触发聚焦按钮，Tab 到允许才能批准；Escape 拒绝。"},
|
|
{ID: "264", Name: "PermissionPrompt", ZhName: "权限请求", Category: agentCategory, Demo: permissionDemo, Code: permissionPromptCode,
|
|
Keys: "打开时焦点在拒绝；Tab 到始终允许、允许一次；Escape 拒绝。"},
|
|
{ID: "265", Name: "AgentStepList", ZhName: "执行步骤列表", Category: agentCategory, Demo: agentStepListDemo, Code: agentStepListCode,
|
|
Keys: "↑ ↓ 在步骤与重试间移动；Enter 展开详情或重试。"},
|
|
{ID: "266", Name: "AgentPlan", ZhName: "执行计划", Category: agentCategory, Demo: agentPlanDemo, Code: agentPlanCode,
|
|
Keys: "Space 勾选；Tab 到跳过或恢复；在输入框中按 Enter 新增。"},
|
|
{ID: "267", Name: "AgentStatus", ZhName: "Agent 状态", Category: agentCategory, Demo: agentStatusDemo, Code: agentStatusCode,
|
|
Keys: "不获取焦点；屏幕阅读器读出状态文字。"},
|
|
{ID: "268", Name: "AgentProgress", ZhName: "Agent 进度", Category: agentCategory, Demo: agentProgressDemo, Code: agentProgressCode,
|
|
Keys: "不获取焦点；当前动作以状态读出。"},
|
|
{ID: "269", Name: "SubAgentTree", ZhName: "子 Agent 树", Category: agentCategory, Demo: subAgentTreeDemo, Code: subAgentTreeCode,
|
|
Keys: "↑ ↓ 移动；→ 展开，← 折叠；Enter / Space 切换。"},
|
|
{ID: "270", Name: "FileChangeCard", ZhName: "文件变更卡片", Category: agentCategory, Demo: fileChangeCardDemo, Code: fileChangeCardCode,
|
|
Keys: "Tab 到打开差异、拒绝、接受或撤销，Enter / Space 触发。"},
|
|
{ID: "271", Name: "MultiFileDiffReview", ZhName: "多文件差异审阅", Category: agentCategory, Demo: multiFileDiffDemo, Code: multiFileDiffReviewCode,
|
|
Keys: "文件列表中 ↑ ↓ 移动，Enter 选择；Tab 到接受、拒绝、全部接受。"},
|
|
{ID: "272", Name: "CommandExecutionCard", ZhName: "命令执行卡片", Category: agentCategory, Demo: commandExecutionDemo, Code: commandExecutionCardCode,
|
|
Keys: "Tab 到中止，Enter / Space 请求中止；输出区可滚动。"},
|
|
{ID: "273", Name: "ScreenshotStream", ZhName: "Agent 画面流", Category: agentCategory, Demo: screenshotStreamDemo, Code: screenshotStreamCode,
|
|
Keys: "画面区中 ← → 逐帧，Space 暂停或恢复。"},
|
|
{ID: "274", Name: "ArtifactPanel", ZhName: "产物面板", Category: agentCategory, Demo: artifactPanelDemo, Code: artifactPanelCode,
|
|
Keys: "切换组中 ← → 移动、Enter 选择；Tab 到复制、导出。"},
|
|
{ID: "275", Name: "ArtifactVersionSwitcher", ZhName: "产物版本切换", Category: agentCategory, Demo: artifactVersionDemo, Code: artifactVersionSwitcherCode,
|
|
Keys: "焦点在组件内时 ← → 切换版本；Tab 到比较按钮。"},
|
|
{ID: "276", Name: "SandboxStatus", ZhName: "沙箱状态", Category: agentCategory, Demo: sandboxStatusDemo, Code: sandboxStatusCode,
|
|
Keys: "不获取焦点；剩余时间每秒刷新。"},
|
|
{ID: "277", Name: "MCPServerList", ZhName: "MCP 服务器列表", Category: agentCategory, Demo: mcpServerListDemo, Code: mcpServerListCode,
|
|
Keys: "Tab 到各行的启动、停止、重新连接，Enter / Space 触发。"},
|
|
{ID: "278", Name: "MemoryPanel", ZhName: "记忆面板", Category: agentCategory, Demo: memoryPanelDemo, Code: memoryPanelCode,
|
|
Keys: "Tab 到搜索、编辑、删除；删除确认默认聚焦取消，Escape 取消。"},
|
|
{ID: "279", Name: "CheckpointList", ZhName: "检查点列表", Category: agentCategory, Demo: checkpointListDemo, Code: checkpointListCode,
|
|
Keys: "Tab 到回滚，Enter 打开确认；确认框默认聚焦取消。"},
|
|
{ID: "280", Name: "HumanInputRequest", ZhName: "人工输入请求", Category: agentCategory, Demo: humanInputDemo, Code: humanInputRequestCode,
|
|
Keys: "方向键选择选项；输入框中 Enter 发送；Tab 到发送回答。"},
|
|
}
|
|
|
|
// agentStates are the states a demo can pick from.
|
|
var agentStates = []string{"Pending", "Planning", "Running", "Waiting", "Done", "Failed", "Skipped"}
|
|
|
|
// agentStatePick is a select over agentStates, logging changes.
|
|
func agentStatePick(c *ui.Context, log func(string), v *string) mujica.AgentState {
|
|
if ui.Select(c, v, agentStates).Changed() {
|
|
log("state = " + *v)
|
|
}
|
|
for i, s := range agentStates {
|
|
if s == *v {
|
|
return mujica.AgentState(i)
|
|
}
|
|
}
|
|
panic("gallery: unknown state " + *v)
|
|
}
|
|
|
|
type toolCallDemoState struct {
|
|
state string
|
|
failed bool
|
|
still bool
|
|
}
|
|
|
|
func toolCallCardDemo(c *ui.Context, log func(string)) {
|
|
s := ui.Local(c.Root(), "261", func() toolCallDemoState { return toolCallDemoState{state: "Done"} })
|
|
var st mujica.AgentState
|
|
overlayControls(c, func() {
|
|
st = agentStatePick(c, log, &s.state)
|
|
overlayCheck(c, log, &s.failed, "Error")
|
|
overlayCheck(c, log, &s.still, "Still")
|
|
})
|
|
call := mujica.ToolCall{ID: "1", Name: "read_file", Args: `{"path":"court/ledger.md","limit":40}`,
|
|
Result: `{"lines":40,"truncated":false}`, State: st, Duration: 420 * time.Millisecond}
|
|
if s.failed {
|
|
call.Error = "permission denied: court/ledger.md"
|
|
}
|
|
if mujica.ToolCallCard(c, call, mujica.ToolCallCardOptions{Still: s.still}).Changed() {
|
|
log("folded or unfolded a section")
|
|
}
|
|
}
|
|
|
|
func toolCallGroupDemo(c *ui.Context, log func(string)) {
|
|
open := ui.Local(c.Root(), "262", func() bool { return true })
|
|
calls := []mujica.ToolCall{
|
|
{ID: "a", Name: "list_dir", Args: `{"path":"court"}`, Result: `["ledger.md","menu.md"]`, State: mujica.AgentDone, Duration: 300 * time.Millisecond},
|
|
{ID: "b", Name: "read_file", Args: `{"path":"court/wine.md"}`, Error: "not found", State: mujica.AgentFailed, Duration: 1200 * time.Millisecond},
|
|
{ID: "c", Name: "grep", Args: `{"pattern":"Sakiko"}`, Result: `{"matches":3}`, State: mujica.AgentDone, Duration: 500 * time.Millisecond},
|
|
}
|
|
if mujica.ToolCallGroup(c, open, calls, mujica.ToolCallGroupOptions{}).Changed() {
|
|
log(fmt.Sprintf("group open = %v", *open))
|
|
}
|
|
}
|
|
|
|
func toolApprovalDemo(c *ui.Context, log func(string)) {
|
|
open := ui.Local(c.Root(), "263", func() bool { return false })
|
|
if mujica.Button(c, "Run delete_files", mujica.ButtonOptions{}).Clicked() {
|
|
*open = true
|
|
log("asked for approval")
|
|
}
|
|
r := mujica.ToolApprovalDialog(c, open, mujica.ToolApprovalOptions{Tool: "delete_files",
|
|
Args: `{"glob":"build/*.tmp","dry_run":false}`, Reason: "The agent wants to clean temporary files from the build folder."})
|
|
if r.Allowed {
|
|
log("allowed")
|
|
}
|
|
if r.Denied {
|
|
log("denied")
|
|
}
|
|
}
|
|
|
|
type permissionDemoState struct {
|
|
open bool
|
|
grant string
|
|
}
|
|
|
|
func permissionDemo(c *ui.Context, log func(string)) {
|
|
s := ui.Local(c.Root(), "264", func() permissionDemoState { return permissionDemoState{grant: "none"} })
|
|
ui.Row(c).Gap(12).AlignItems(ui.Center).Children(func() {
|
|
if mujica.Button(c, "Ask to write", mujica.ButtonOptions{}).Clicked() {
|
|
s.open = true
|
|
log("asked for permission")
|
|
}
|
|
ui.Text(c, "Grant: "+s.grant)
|
|
})
|
|
r := mujica.PermissionPrompt(c, &s.open, mujica.PermissionPromptOptions{Title: "Write to the ledger?",
|
|
Description: "The agent wants to record tonight's guests.", Scope: "Write access to ~/court/ledger and its subfolders"})
|
|
if r.Choice != mujica.PermissionNone {
|
|
s.grant = [...]string{"none", "once", "always", "denied"}[r.Choice]
|
|
log("choice = " + s.grant)
|
|
}
|
|
}
|
|
|
|
type stepDemoState struct {
|
|
steps []mujica.AgentStep
|
|
disabled bool
|
|
}
|
|
|
|
func agentStepListDemo(c *ui.Context, log func(string)) {
|
|
s := ui.Local(c.Root(), "265", func() stepDemoState {
|
|
return stepDemoState{steps: []mujica.AgentStep{
|
|
{ID: "read", Title: "Read the guest list", State: mujica.AgentDone, Duration: 800 * time.Millisecond, Detail: "42 guests, 3 arrive late."},
|
|
{ID: "seat", Title: "Assign the seats", State: mujica.AgentFailed, Duration: 2 * time.Second, Detail: "Table 4 is over capacity."},
|
|
{ID: "send", Title: "Send the invitations", State: mujica.AgentPending},
|
|
}}
|
|
})
|
|
overlayControls(c, func() { overlayCheck(c, log, &s.disabled, "Disabled") })
|
|
r := mujica.AgentStepList(c, s.steps, mujica.AgentStepListOptions{Disabled: s.disabled})
|
|
if r.Toggled != "" {
|
|
log("toggled " + r.Toggled)
|
|
}
|
|
if r.Retry != "" {
|
|
log("retry " + r.Retry)
|
|
for i := range s.steps {
|
|
if s.steps[i].ID == r.Retry {
|
|
s.steps[i].State, s.steps[i].Detail = mujica.AgentRunning, "Retrying with a larger table."
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
type planDemoState struct {
|
|
items []mujica.PlanItem
|
|
disabled bool
|
|
}
|
|
|
|
func agentPlanDemo(c *ui.Context, log func(string)) {
|
|
s := ui.Local(c.Root(), "266", func() planDemoState {
|
|
return planDemoState{items: []mujica.PlanItem{
|
|
{ID: "a", Text: "Read the guest list", State: mujica.AgentDone},
|
|
{ID: "b", Text: "Assign the seats", State: mujica.AgentRunning},
|
|
{ID: "c", Text: "Order the flowers"},
|
|
}}
|
|
})
|
|
overlayControls(c, func() {
|
|
overlayCheck(c, log, &s.disabled, "Disabled")
|
|
if mujica.Button(c, "Agent finishes a step", mujica.ButtonOptions{Variant: mujica.Secondary}).Clicked() {
|
|
for i := range s.items {
|
|
if s.items[i].State != mujica.AgentDone && s.items[i].State != mujica.AgentSkipped {
|
|
s.items[i].State = mujica.AgentDone
|
|
log("agent checked " + s.items[i].Text)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
})
|
|
if mujica.AgentPlan(c, &s.items, mujica.AgentPlanOptions{Disabled: s.disabled}).Changed() {
|
|
log(fmt.Sprintf("plan changed (%d items)", len(s.items)))
|
|
}
|
|
}
|
|
|
|
type statusDemoState struct {
|
|
state string
|
|
still bool
|
|
}
|
|
|
|
func agentStatusDemo(c *ui.Context, log func(string)) {
|
|
s := ui.Local(c.Root(), "267", func() statusDemoState { return statusDemoState{state: "Running"} })
|
|
var st mujica.AgentState
|
|
overlayControls(c, func() {
|
|
st = agentStatePick(c, log, &s.state)
|
|
overlayCheck(c, log, &s.still, "Still")
|
|
})
|
|
ui.Row(c).Gap(8).Wrap().Children(func() {
|
|
mujica.AgentStatus(c, st, mujica.AgentStatusOptions{Still: s.still})
|
|
for _, v := range []mujica.AgentState{mujica.AgentPlanning, mujica.AgentRunning, mujica.AgentWaiting, mujica.AgentDone, mujica.AgentFailed} {
|
|
mujica.AgentStatus(c, v, mujica.AgentStatusOptions{Still: s.still})
|
|
}
|
|
})
|
|
}
|
|
|
|
type progressDemoState struct {
|
|
value float64
|
|
indeterminate bool
|
|
}
|
|
|
|
func agentProgressDemo(c *ui.Context, log func(string)) {
|
|
s := ui.Local(c.Root(), "268", func() progressDemoState { return progressDemoState{value: 0.4} })
|
|
overlayControls(c, func() {
|
|
if ui.Slider(c, &s.value, 0, 1).Width(200).Changed() {
|
|
log(fmt.Sprintf("value = %.2f", s.value))
|
|
}
|
|
overlayCheck(c, log, &s.indeterminate, "Indeterminate")
|
|
})
|
|
action := "Reading ledger.md"
|
|
if s.value > 0.5 {
|
|
action = "Writing the seating chart"
|
|
}
|
|
mujica.AgentProgress(c, s.value, mujica.AgentProgressOptions{Label: "Preparing the ball", Action: action, Indeterminate: s.indeterminate})
|
|
}
|
|
|
|
func subAgentTreeDemo(c *ui.Context, log func(string)) {
|
|
disabled := ui.Local(c.Root(), "269", func() bool { return false })
|
|
overlayControls(c, func() { overlayCheck(c, log, disabled, "Disabled") })
|
|
agents := []mujica.SubAgent{{ID: "lead", Name: "Steward", Task: "Prepare the ball", State: mujica.AgentRunning, Children: []mujica.SubAgent{
|
|
{ID: "music", Name: "Musician", Task: "Hire the quartet", State: mujica.AgentDone},
|
|
{ID: "kitchen", Name: "Cook", Task: "Plan the menu", State: mujica.AgentWaiting, Children: []mujica.SubAgent{
|
|
{ID: "wine", Name: "Sommelier", Task: "Choose the wine", State: mujica.AgentFailed},
|
|
}},
|
|
}}}
|
|
if mujica.SubAgentTree(c, agents, mujica.SubAgentTreeOptions{Disabled: *disabled}).Changed() {
|
|
log("expanded or collapsed an agent")
|
|
}
|
|
}
|
|
|
|
type fileChangeDemoState struct {
|
|
decision mujica.FileDecision
|
|
disabled bool
|
|
preview float64
|
|
}
|
|
|
|
func fileChangeCardDemo(c *ui.Context, log func(string)) {
|
|
s := ui.Local(c.Root(), "270", func() fileChangeDemoState { return fileChangeDemoState{preview: 8} })
|
|
overlayControls(c, func() {
|
|
overlayCheck(c, log, &s.disabled, "Disabled")
|
|
overlaySlider(c, log, &s.preview, "Preview", "lines", 2, 12, 1)
|
|
})
|
|
r := mujica.FileChangeCard(c, &s.decision, agentDemoChange, mujica.FileChangeCardOptions{Preview: int(s.preview), Disabled: s.disabled})
|
|
if r.Decided {
|
|
log(fmt.Sprintf("decision = %d", s.decision))
|
|
}
|
|
if r.OpenDiff {
|
|
log("open the full diff")
|
|
}
|
|
}
|
|
|
|
var agentDemoChange = mujica.FileChange{Path: "court/ledger.md",
|
|
Old: "# Ledger\nGuests: 40\nWine: claret\nMusic: none\nHall: east wing\n",
|
|
New: "# Ledger\nGuests: 42\nWine: claret\nMusic: quartet\nFlowers: lilies\nHall: east wing\n"}
|