DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-05-04 04:27:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: expand mcp transports and skill execution

Browse Source
This commit is contained in:
lpf
2026-03-08 11:08:41 +08:00
parent db86b3471d
commit f043de5384
21 changed files with 1447 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
pkg/agent/loop_allowlist_test.go
View File

@@ -19,6 +19,9 @@ func TestEnsureToolAllowedByContext(t *testing.T) {
if err := ensureToolAllowedByContext(restricted, "write_file", map[string]interface{}{}); err == nil {
t.Fatalf("expected disallowed tool to fail")
}
if err := ensureToolAllowedByContext(restricted, "skill_exec", map[string]interface{}{}); err != nil {
t.Fatalf("expected skill_exec to bypass subagent allowlist, got: %v", err)
}
}
func TestEnsureToolAllowedByContextParallelNested(t *testing.T) {
@@ -41,6 +44,15 @@ func TestEnsureToolAllowedByContextParallelNested(t *testing.T) {
if err := ensureToolAllowedByContext(restricted, "parallel", badArgs); err == nil {
t.Fatalf("expected parallel with disallowed nested tool to fail")
}
skillArgs := map[string]interface{}{
"calls": []interface{}{
map[string]interface{}{"tool": "skill_exec", "arguments": map[string]interface{}{"skill": "demo", "script": "scripts/run.sh"}},
},
}
if err := ensureToolAllowedByContext(restricted, "parallel", skillArgs); err != nil {
t.Fatalf("expected parallel with nested skill_exec to pass, got: %v", err)
}
}
func TestEnsureToolAllowedByContext_GroupAllowlist(t *testing.T) {
@@ -52,3 +64,48 @@ func TestEnsureToolAllowedByContext_GroupAllowlist(t *testing.T) {
t.Fatalf("expected files_read group to block write_file")
}
}
func TestFilteredToolDefinitionsForContext(t *testing.T) {
ctx := withToolAllowlistContext(context.Background(), []string{"read_file", "parallel"})
toolDefs := []map[string]interface{}{
{"function": map[string]interface{}{"name": "read_file", "description": "read", "parameters": map[string]interface{}{}}},
{"function": map[string]interface{}{"name": "write_file", "description": "write", "parameters": map[string]interface{}{}}},
{"function": map[string]interface{}{"name": "parallel", "description": "parallel", "parameters": map[string]interface{}{}}},
{"function": map[string]interface{}{"name": "skill_exec", "description": "skill", "parameters": map[string]interface{}{}}},
}
filtered := filterToolDefinitionsByContext(ctx, toolDefs)
got := map[string]bool{}
for _, td := range filtered {
fnRaw, _ := td["function"].(map[string]interface{})
name, _ := fnRaw["name"].(string)
got[name] = true
}
if !got["read_file"] || !got["parallel"] || !got["skill_exec"] {
t.Fatalf("expected filtered tools to include read_file, parallel, and inherited skill_exec, got: %v", got)
}
if got["write_file"] {
t.Fatalf("expected filtered tools to exclude write_file, got: %v", got)
}
}
func TestWithToolRuntimeArgsForSkillExec(t *testing.T) {
mainArgs := withToolRuntimeArgs(context.Background(), "skill_exec", map[string]interface{}{"skill": "demo"})
if mainArgs["caller_agent"] != "main" || mainArgs["caller_scope"] != "main_agent" {
t.Fatalf("expected main agent runtime args, got: %#v", mainArgs)
}
subagentCtx := withMemoryNamespaceContext(context.Background(), "coder")
subArgs := withToolRuntimeArgs(subagentCtx, "skill_exec", map[string]interface{}{"skill": "demo"})
if subArgs["caller_agent"] != "coder" || subArgs["caller_scope"] != "subagent" {
t.Fatalf("expected subagent runtime args, got: %#v", subArgs)
}
}
func TestSubagentToolVisibilityMode(t *testing.T) {
if got := subagentToolVisibilityMode("skill_exec"); got != "inherited" {
t.Fatalf("expected skill_exec inherited, got %q", got)
}
if got := subagentToolVisibilityMode("write_file"); got != "allowlist" {
t.Fatalf("expected write_file allowlist, got %q", got)
}
}