mirror of
https://github.com/sipeed/picoclaw.git
synced 2026-06-12 18:08:54 +00:00
refactor(agent): source discovery identity from AGENT.md frontmatter
This commit is contained in:
@@ -51,7 +51,7 @@ func (r *ToolRegistry) SetAllowlist(names []string) {
|
||||
|
||||
allowlist := make(map[string]struct{}, len(names))
|
||||
for _, name := range names {
|
||||
trimmed := strings.TrimSpace(name)
|
||||
trimmed := strings.ToLower(strings.TrimSpace(name))
|
||||
if trimmed == "" {
|
||||
continue
|
||||
}
|
||||
@@ -172,7 +172,7 @@ func (r *ToolRegistry) toolAllowedLocked(name string) bool {
|
||||
if r.allowlist == nil {
|
||||
return true
|
||||
}
|
||||
_, ok := r.allowlist[name]
|
||||
_, ok := r.allowlist[strings.ToLower(strings.TrimSpace(name))]
|
||||
return ok
|
||||
}
|
||||
|
||||
|
||||
@@ -101,7 +101,7 @@ func TestToolRegistry_RegisterAndGet(t *testing.T) {
|
||||
|
||||
func TestToolRegistry_AllowlistFiltersRegistrations(t *testing.T) {
|
||||
r := NewToolRegistry()
|
||||
r.SetAllowlist([]string{"allowed_tool"})
|
||||
r.SetAllowlist([]string{"Allowed_Tool"})
|
||||
|
||||
r.Register(newMockTool("allowed_tool", "allowed"))
|
||||
r.Register(newMockTool("blocked_tool", "blocked"))
|
||||
|
||||
Reference in New Issue
Block a user