mirror of
https://github.com/YspCoder/clawgo.git
synced 2026-05-03 10:57:35 +08:00
add subagent index selectors and session active filters
This commit is contained in:
@@ -5,7 +5,9 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"sort"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"clawgo/pkg/providers"
|
"clawgo/pkg/providers"
|
||||||
)
|
)
|
||||||
@@ -101,6 +103,12 @@ func statusCmd() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if recent, err := collectRecentSubagentSessions(sessionsDir, 5); err == nil && len(recent) > 0 {
|
||||||
|
fmt.Println("Recent Subagent Sessions:")
|
||||||
|
for _, key := range recent {
|
||||||
|
fmt.Printf(" - %s\n", key)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -133,3 +141,49 @@ func collectSessionKindCounts(sessionsDir string) (map[string]int, error) {
|
|||||||
}
|
}
|
||||||
return counts, nil
|
return counts, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func collectRecentSubagentSessions(sessionsDir string, limit int) ([]string, error) {
|
||||||
|
entries, err := os.ReadDir(sessionsDir)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
type item struct {
|
||||||
|
key string
|
||||||
|
updated int64
|
||||||
|
}
|
||||||
|
items := make([]item, 0)
|
||||||
|
for _, e := range entries {
|
||||||
|
if e.IsDir() || !strings.HasSuffix(e.Name(), ".meta") {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
metaPath := filepath.Join(sessionsDir, e.Name())
|
||||||
|
data, err := os.ReadFile(metaPath)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
var meta struct {
|
||||||
|
Kind string `json:"kind"`
|
||||||
|
Updated string `json:"updated"`
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal(data, &meta); err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if strings.ToLower(strings.TrimSpace(meta.Kind)) != "subagent" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
t, err := time.Parse(time.RFC3339Nano, meta.Updated)
|
||||||
|
if err != nil {
|
||||||
|
t, _ = time.Parse(time.RFC3339, meta.Updated)
|
||||||
|
}
|
||||||
|
items = append(items, item{key: strings.TrimSuffix(e.Name(), ".meta"), updated: t.UnixMilli()})
|
||||||
|
}
|
||||||
|
sort.Slice(items, func(i, j int) bool { return items[i].updated > items[j].updated })
|
||||||
|
if limit > 0 && len(items) > limit {
|
||||||
|
items = items[:limit]
|
||||||
|
}
|
||||||
|
out := make([]string, 0, len(items))
|
||||||
|
for _, it := range items {
|
||||||
|
out = append(out, it.key)
|
||||||
|
}
|
||||||
|
return out, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -36,11 +36,12 @@ 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"},
|
"active_minutes": map[string]interface{}{"type": "integer", "description": "only sessions updated in recent N minutes (list action)"},
|
||||||
"include_tools": map[string]interface{}{"type": "boolean", "description": "include tool role messages in history", "default": false},
|
"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},
|
||||||
},
|
},
|
||||||
"required": []string{"action"},
|
"required": []string{"action"},
|
||||||
}
|
}
|
||||||
@@ -58,6 +59,10 @@ 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
|
||||||
}
|
}
|
||||||
|
activeMinutes := 0
|
||||||
|
if v, ok := args["active_minutes"].(float64); ok && int(v) > 0 {
|
||||||
|
activeMinutes = int(v)
|
||||||
|
}
|
||||||
kindFilter := map[string]struct{}{}
|
kindFilter := map[string]struct{}{}
|
||||||
if rawKinds, ok := args["kinds"].([]interface{}); ok {
|
if rawKinds, ok := args["kinds"].([]interface{}); ok {
|
||||||
for _, it := range rawKinds {
|
for _, it := range rawKinds {
|
||||||
@@ -89,6 +94,16 @@ func (t *SessionsTool) Execute(ctx context.Context, args map[string]interface{})
|
|||||||
}
|
}
|
||||||
items = filtered
|
items = filtered
|
||||||
}
|
}
|
||||||
|
if activeMinutes > 0 {
|
||||||
|
cutoff := time.Now().Add(-time.Duration(activeMinutes) * time.Minute)
|
||||||
|
filtered := make([]SessionInfo, 0, len(items))
|
||||||
|
for _, s := range items {
|
||||||
|
if s.UpdatedAt.After(cutoff) {
|
||||||
|
filtered = append(filtered, s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
items = filtered
|
||||||
|
}
|
||||||
if len(items) == 0 {
|
if len(items) == 0 {
|
||||||
return "No sessions (after filters).", nil
|
return "No sessions (after filters).", nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,6 +3,8 @@ package tools
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"sort"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -52,40 +54,48 @@ func (t *SubagentsTool) Execute(ctx context.Context, args map[string]interface{}
|
|||||||
}
|
}
|
||||||
var sb strings.Builder
|
var sb strings.Builder
|
||||||
sb.WriteString("Subagents:\n")
|
sb.WriteString("Subagents:\n")
|
||||||
for _, task := range tasks {
|
sort.Slice(tasks, func(i, j int) bool { return tasks[i].Created > tasks[j].Created })
|
||||||
sb.WriteString(fmt.Sprintf("- %s [%s] label=%s\n", task.ID, task.Status, task.Label))
|
for i, task := range tasks {
|
||||||
|
sb.WriteString(fmt.Sprintf("- #%d %s [%s] label=%s\n", i+1, task.ID, task.Status, task.Label))
|
||||||
}
|
}
|
||||||
return strings.TrimSpace(sb.String()), nil
|
return strings.TrimSpace(sb.String()), nil
|
||||||
case "info":
|
case "info":
|
||||||
if id == "" {
|
resolvedID, err := t.resolveTaskID(id)
|
||||||
return "id is required for info", nil
|
if err != nil {
|
||||||
|
return err.Error(), nil
|
||||||
}
|
}
|
||||||
task, ok := t.manager.GetTask(id)
|
task, ok := t.manager.GetTask(resolvedID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return "subagent not found", nil
|
return "subagent not found", nil
|
||||||
}
|
}
|
||||||
return fmt.Sprintf("ID: %s\nStatus: %s\nLabel: %s\nCreated: %d\nUpdated: %d\nSteering Count: %d\nTask: %s\nResult:\n%s", task.ID, task.Status, task.Label, task.Created, task.Updated, len(task.Steering), task.Task, task.Result), nil
|
return fmt.Sprintf("ID: %s\nStatus: %s\nLabel: %s\nCreated: %d\nUpdated: %d\nSteering Count: %d\nTask: %s\nResult:\n%s", task.ID, task.Status, task.Label, task.Created, task.Updated, len(task.Steering), task.Task, task.Result), nil
|
||||||
case "kill":
|
case "kill":
|
||||||
if id == "" {
|
resolvedID, err := t.resolveTaskID(id)
|
||||||
return "id is required for kill", nil
|
if err != nil {
|
||||||
|
return err.Error(), nil
|
||||||
}
|
}
|
||||||
if !t.manager.KillTask(id) {
|
if !t.manager.KillTask(resolvedID) {
|
||||||
return "subagent not found", nil
|
return "subagent not found", nil
|
||||||
}
|
}
|
||||||
return "subagent kill requested", nil
|
return "subagent kill requested", nil
|
||||||
case "steer", "send":
|
case "steer", "send":
|
||||||
if id == "" || message == "" {
|
if message == "" {
|
||||||
return "id and message are required for steer/send", nil
|
return "message is required for steer/send", nil
|
||||||
}
|
}
|
||||||
if !t.manager.SteerTask(id, message) {
|
resolvedID, err := t.resolveTaskID(id)
|
||||||
|
if err != nil {
|
||||||
|
return err.Error(), nil
|
||||||
|
}
|
||||||
|
if !t.manager.SteerTask(resolvedID, message) {
|
||||||
return "subagent not found", nil
|
return "subagent not found", nil
|
||||||
}
|
}
|
||||||
return "steering message accepted", nil
|
return "steering message accepted", nil
|
||||||
case "log":
|
case "log":
|
||||||
if id == "" {
|
resolvedID, err := t.resolveTaskID(id)
|
||||||
return "id is required for log", nil
|
if err != nil {
|
||||||
|
return err.Error(), nil
|
||||||
}
|
}
|
||||||
task, ok := t.manager.GetTask(id)
|
task, ok := t.manager.GetTask(resolvedID)
|
||||||
if !ok {
|
if !ok {
|
||||||
return "subagent not found", nil
|
return "subagent not found", nil
|
||||||
}
|
}
|
||||||
@@ -110,3 +120,26 @@ func (t *SubagentsTool) Execute(ctx context.Context, args map[string]interface{}
|
|||||||
return "unsupported action", nil
|
return "unsupported action", nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (t *SubagentsTool) resolveTaskID(idOrIndex string) (string, error) {
|
||||||
|
idOrIndex = strings.TrimSpace(idOrIndex)
|
||||||
|
if idOrIndex == "" {
|
||||||
|
return "", fmt.Errorf("id is required")
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(idOrIndex, "#") {
|
||||||
|
n, err := strconv.Atoi(strings.TrimPrefix(idOrIndex, "#"))
|
||||||
|
if err != nil || n <= 0 {
|
||||||
|
return "", fmt.Errorf("invalid subagent index")
|
||||||
|
}
|
||||||
|
tasks := t.manager.ListTasks()
|
||||||
|
if len(tasks) == 0 {
|
||||||
|
return "", fmt.Errorf("no subagents")
|
||||||
|
}
|
||||||
|
sort.Slice(tasks, func(i, j int) bool { return tasks[i].Created > tasks[j].Created })
|
||||||
|
if n > len(tasks) {
|
||||||
|
return "", fmt.Errorf("subagent index out of range")
|
||||||
|
}
|
||||||
|
return tasks[n-1].ID, nil
|
||||||
|
}
|
||||||
|
return idOrIndex, nil
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user