extend sessions and subagents tools with filters and log/send actions

This commit is contained in:
DBT
2026-02-23 13:27:00 +00:00
parent 68afec08d0
commit c86e314d7f
2 changed files with 62 additions and 9 deletions

View File

@@ -36,9 +36,10 @@ func (t *SessionsTool) Parameters() map[string]interface{} {
return map[string]interface{}{ return map[string]interface{}{
"type": "object", "type": "object",
"properties": map[string]interface{}{ "properties": map[string]interface{}{
"action": map[string]interface{}{"type": "string", "description": "list|history"}, "action": map[string]interface{}{"type": "string", "description": "list|history"},
"key": map[string]interface{}{"type": "string", "description": "session key for history"}, "key": map[string]interface{}{"type": "string", "description": "session key for history"},
"limit": map[string]interface{}{"type": "integer", "description": "max items", "default": 20}, "limit": map[string]interface{}{"type": "integer", "description": "max items", "default": 20},
"kinds": map[string]interface{}{"type": "array", "items": map[string]interface{}{"type": "string"}, "description": "optional session kinds filter for list"},
"include_tools": map[string]interface{}{"type": "boolean", "description": "include tool role messages in history", "default": false}, "include_tools": map[string]interface{}{"type": "boolean", "description": "include tool role messages in history", "default": false},
}, },
"required": []string{"action"}, "required": []string{"action"},
@@ -57,17 +58,44 @@ func (t *SessionsTool) Execute(ctx context.Context, args map[string]interface{})
if v, ok := args["include_tools"].(bool); ok { if v, ok := args["include_tools"].(bool); ok {
includeTools = v includeTools = v
} }
kindFilter := map[string]struct{}{}
if rawKinds, ok := args["kinds"].([]interface{}); ok {
for _, it := range rawKinds {
if s, ok := it.(string); ok {
s = strings.ToLower(strings.TrimSpace(s))
if s != "" {
kindFilter[s] = struct{}{}
}
}
}
}
switch action { switch action {
case "list": case "list":
if t.listFn == nil { if t.listFn == nil {
return "sessions list unavailable", nil return "sessions list unavailable", nil
} }
items := t.listFn(limit) items := t.listFn(limit * 3)
if len(items) == 0 { if len(items) == 0 {
return "No sessions.", nil return "No sessions.", nil
} }
if len(kindFilter) > 0 {
filtered := make([]SessionInfo, 0, len(items))
for _, s := range items {
k := strings.ToLower(strings.TrimSpace(s.Kind))
if _, ok := kindFilter[k]; ok {
filtered = append(filtered, s)
}
}
items = filtered
}
if len(items) == 0 {
return "No sessions (after filters).", nil
}
sort.Slice(items, func(i, j int) bool { return items[i].UpdatedAt.After(items[j].UpdatedAt) }) sort.Slice(items, func(i, j int) bool { return items[i].UpdatedAt.After(items[j].UpdatedAt) })
if len(items) > limit {
items = items[:limit]
}
var sb strings.Builder var sb strings.Builder
sb.WriteString("Sessions:\n") sb.WriteString("Sessions:\n")
for _, s := range items { for _, s := range items {

View File

@@ -24,9 +24,9 @@ func (t *SubagentsTool) Parameters() map[string]interface{} {
return map[string]interface{}{ return map[string]interface{}{
"type": "object", "type": "object",
"properties": map[string]interface{}{ "properties": map[string]interface{}{
"action": map[string]interface{}{"type": "string", "description": "list|info|kill|steer"}, "action": map[string]interface{}{"type": "string", "description": "list|info|kill|steer|send|log"},
"id": map[string]interface{}{"type": "string", "description": "subagent id for info/kill/steer"}, "id": map[string]interface{}{"type": "string", "description": "subagent id for info/kill/steer/send/log"},
"message": map[string]interface{}{"type": "string", "description": "steering message for steer action"}, "message": map[string]interface{}{"type": "string", "description": "steering message for steer/send action"},
}, },
"required": []string{"action"}, "required": []string{"action"},
} }
@@ -73,14 +73,39 @@ func (t *SubagentsTool) Execute(ctx context.Context, args map[string]interface{}
return "subagent not found", nil return "subagent not found", nil
} }
return "subagent kill requested", nil return "subagent kill requested", nil
case "steer": case "steer", "send":
if id == "" || message == "" { if id == "" || message == "" {
return "id and message are required for steer", nil return "id and message are required for steer/send", nil
} }
if !t.manager.SteerTask(id, message) { if !t.manager.SteerTask(id, message) {
return "subagent not found", nil return "subagent not found", nil
} }
return "steering message accepted", nil return "steering message accepted", nil
case "log":
if id == "" {
return "id is required for log", nil
}
task, ok := t.manager.GetTask(id)
if !ok {
return "subagent not found", nil
}
var sb strings.Builder
sb.WriteString(fmt.Sprintf("Subagent %s Log\n", task.ID))
sb.WriteString(fmt.Sprintf("Status: %s\n", task.Status))
if len(task.Steering) > 0 {
sb.WriteString("Steering Messages:\n")
for _, m := range task.Steering {
sb.WriteString("- " + m + "\n")
}
}
if strings.TrimSpace(task.Result) != "" {
result := strings.TrimSpace(task.Result)
if len(result) > 500 {
result = result[:500] + "..."
}
sb.WriteString("Result Preview:\n" + result)
}
return strings.TrimSpace(sb.String()), nil
default: default:
return "unsupported action", nil return "unsupported action", nil
} }