Algis DumbrisandClaude Opus 4.6
2ba0f95666
docgardener: each leaf task now launches a real subprocess via exec.CommandContext and records a full reactive_runs + harness_runs row chain with task_id populated, captured prompt, captured response, exit code, duration, tokens, cost. The Agent Runs page and /runs/:id detail page now show real data for the doc-gardener demo — including "What the model saw" and "What the model said" panels — without needing the coordinator LLM loop. agents store: agentSelectSQL and both scanAgent functions extended to read the feature-018 columns (config_hash, parent_agent_id, spawn_depth, system_prompt, autonomy_tier, tool_scope_json, quarantined_at, quarantine_reason). /api/agents and /api/agents/:name now return these fields end-to-end. Web UI agent detail (web/src/routes/agents/[name]/+page.svelte): adds a Trust & Spawn section (config_hash, autonomy tier, spawn depth, parent agent, tool scope chips) and a full-height System Prompt pre block. Rebuilt internal/web/dist/. Verified in Chrome against a fresh ./start.sh && ./run_task.sh run: - Agent Runs page lists 3 completed runs (docs-scanner, cli-verifier, drift-reporter) with task.claim event and non-zero durations - /runs/1 detail page renders captured prompt + structured #finding output with 12 flags - /agents/docs-scanner shows config_hash=a0b5c6538b2d…, parent=#1, depth=1, tool-scope chips, and the 170-char system prompt - #goal-... channel loads all 12 messages (no "Joining..." hang) Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
389 lines
12 KiB
Go
389 lines
12 KiB
Go
package agents
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// AgentStore defines the storage interface for agent operations.
|
|
type AgentStore interface {
|
|
CreateAgent(ctx context.Context, agent *Agent) error
|
|
GetAgentByName(ctx context.Context, name string) (*Agent, error)
|
|
GetAgentByID(ctx context.Context, id int64) (*Agent, error)
|
|
UpdateAgent(ctx context.Context, agent *Agent) error
|
|
DeactivateAgent(ctx context.Context, name string) error
|
|
ListActiveAgents(ctx context.Context) ([]*Agent, error)
|
|
ListAllActiveAgents(ctx context.Context) ([]*Agent, error)
|
|
ListAgentsByOwner(ctx context.Context, ownerID int64) ([]*Agent, error)
|
|
SearchAgentsByCapability(ctx context.Context, query string) ([]*Agent, error)
|
|
GetHumanAgentByOwner(ctx context.Context, ownerID int64) (*Agent, error)
|
|
|
|
// Reactive trigger methods
|
|
UpdateTriggerConfig(ctx context.Context, name string, mode string, cooldown, budget, maxDepth int) error
|
|
UpdateK8sImage(ctx context.Context, name, image, envJSON, preset string) error
|
|
SetPendingWork(ctx context.Context, name string, pending bool) error
|
|
ListReactiveAgents(ctx context.Context) ([]*Agent, error)
|
|
|
|
// Harness config (migration 019). Pass empty strings to leave a
|
|
// field unchanged; pass a blank placeholder ("-") to explicitly
|
|
// clear it. Returns sql.ErrNoRows if the agent doesn't exist.
|
|
UpdateHarnessConfig(ctx context.Context, name, harnessName, localCommand, harnessConfigJSON string) error
|
|
}
|
|
|
|
// SQLiteAgentStore implements AgentStore using SQLite.
|
|
type SQLiteAgentStore struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
// NewSQLiteAgentStore creates a new SQLite-backed agent store.
|
|
func NewSQLiteAgentStore(db *sql.DB) *SQLiteAgentStore {
|
|
return &SQLiteAgentStore{db: db}
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) CreateAgent(ctx context.Context, agent *Agent) error {
|
|
caps := string(agent.Capabilities)
|
|
if caps == "" {
|
|
caps = "{}"
|
|
}
|
|
|
|
// Default trigger values
|
|
triggerMode := agent.TriggerMode
|
|
if triggerMode == "" {
|
|
triggerMode = TriggerModePassive
|
|
}
|
|
cooldown := agent.CooldownSeconds
|
|
if cooldown == 0 {
|
|
cooldown = 600
|
|
}
|
|
budget := agent.DailyTriggerBudget
|
|
if budget == 0 {
|
|
budget = 8
|
|
}
|
|
maxDepth := agent.MaxTriggerDepth
|
|
if maxDepth == 0 {
|
|
maxDepth = 5
|
|
}
|
|
preset := agent.K8sResourcePreset
|
|
if preset == "" {
|
|
preset = "default"
|
|
}
|
|
|
|
result, err := s.db.ExecContext(ctx,
|
|
`INSERT INTO agents (name, display_name, type, capabilities, owner_id, api_key_hash, status, created_at, updated_at)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)`,
|
|
agent.Name, agent.DisplayName, agent.Type, caps, agent.OwnerID, agent.APIKeyHash, AgentStatusActive,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("insert agent: %w", err)
|
|
}
|
|
id, err := result.LastInsertId()
|
|
if err != nil {
|
|
return fmt.Errorf("get agent id: %w", err)
|
|
}
|
|
agent.ID = id
|
|
agent.Status = AgentStatusActive
|
|
agent.TriggerMode = triggerMode
|
|
agent.CooldownSeconds = cooldown
|
|
agent.DailyTriggerBudget = budget
|
|
agent.MaxTriggerDepth = maxDepth
|
|
agent.K8sResourcePreset = preset
|
|
return nil
|
|
}
|
|
|
|
// UpdateTriggerConfig updates the reactive trigger configuration for an agent.
|
|
func (s *SQLiteAgentStore) UpdateTriggerConfig(ctx context.Context, name string, mode string, cooldown, budget, maxDepth int) error {
|
|
_, err := s.db.ExecContext(ctx,
|
|
`UPDATE agents SET trigger_mode = ?, cooldown_seconds = ?, daily_trigger_budget = ?, max_trigger_depth = ?, updated_at = CURRENT_TIMESTAMP
|
|
WHERE name = ? AND status = 'active'`,
|
|
mode, cooldown, budget, maxDepth, name,
|
|
)
|
|
return err
|
|
}
|
|
|
|
// UpdateK8sImage updates the K8s container image and env config for an agent.
|
|
func (s *SQLiteAgentStore) UpdateK8sImage(ctx context.Context, name, image, envJSON, preset string) error {
|
|
_, err := s.db.ExecContext(ctx,
|
|
`UPDATE agents SET k8s_image = ?, k8s_env_json = ?, k8s_resource_preset = ?, updated_at = CURRENT_TIMESTAMP
|
|
WHERE name = ? AND status = 'active'`,
|
|
image, envJSON, preset, name,
|
|
)
|
|
return err
|
|
}
|
|
|
|
// UpdateHarnessConfig updates any subset of harness_name / local_command /
|
|
// harness_config_json for an agent. Pass empty string to leave a field
|
|
// unchanged; pass "-" to explicitly clear (set to NULL).
|
|
func (s *SQLiteAgentStore) UpdateHarnessConfig(ctx context.Context, name, harnessName, localCommand, harnessConfigJSON string) error {
|
|
sets := []string{}
|
|
args := []any{}
|
|
if harnessName != "" {
|
|
sets = append(sets, "harness_name = ?")
|
|
if harnessName == "-" {
|
|
args = append(args, nil)
|
|
} else {
|
|
args = append(args, harnessName)
|
|
}
|
|
}
|
|
if localCommand != "" {
|
|
sets = append(sets, "local_command = ?")
|
|
if localCommand == "-" {
|
|
args = append(args, nil)
|
|
} else {
|
|
args = append(args, localCommand)
|
|
}
|
|
}
|
|
if harnessConfigJSON != "" {
|
|
sets = append(sets, "harness_config_json = ?")
|
|
if harnessConfigJSON == "-" {
|
|
args = append(args, nil)
|
|
} else {
|
|
args = append(args, harnessConfigJSON)
|
|
}
|
|
}
|
|
if len(sets) == 0 {
|
|
return nil
|
|
}
|
|
sets = append(sets, "updated_at = CURRENT_TIMESTAMP")
|
|
args = append(args, name)
|
|
query := "UPDATE agents SET " + strings.Join(sets, ", ") + " WHERE name = ? AND status = 'active'"
|
|
|
|
result, err := s.db.ExecContext(ctx, query, args...)
|
|
if err != nil {
|
|
return fmt.Errorf("update harness config: %w", err)
|
|
}
|
|
n, err := result.RowsAffected()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if n == 0 {
|
|
return sql.ErrNoRows
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// SetPendingWork sets the pending_work flag for an agent.
|
|
func (s *SQLiteAgentStore) SetPendingWork(ctx context.Context, name string, pending bool) error {
|
|
val := 0
|
|
if pending {
|
|
val = 1
|
|
}
|
|
_, err := s.db.ExecContext(ctx,
|
|
`UPDATE agents SET pending_work = ? WHERE name = ? AND status = 'active'`,
|
|
val, name,
|
|
)
|
|
return err
|
|
}
|
|
|
|
// ListReactiveAgents returns all active agents with trigger_mode='reactive'.
|
|
func (s *SQLiteAgentStore) ListReactiveAgents(ctx context.Context) ([]*Agent, error) {
|
|
rows, err := s.db.QueryContext(ctx,
|
|
agentSelectSQL()+` WHERE status = 'active' AND trigger_mode = 'reactive' ORDER BY name`,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
return s.scanAgents(rows)
|
|
}
|
|
|
|
// agentSelectSQL returns the base SELECT clause for agent queries.
|
|
func agentSelectSQL() string {
|
|
return `SELECT id, name, display_name, type, capabilities, owner_id, api_key_hash, status, created_at, updated_at,
|
|
trigger_mode, cooldown_seconds, daily_trigger_budget, max_trigger_depth, k8s_image, k8s_env_json, k8s_resource_preset, pending_work,
|
|
harness_name, local_command, harness_config_json,
|
|
config_hash, parent_agent_id, spawn_depth, system_prompt, autonomy_tier, tool_scope_json, quarantined_at, quarantine_reason
|
|
FROM agents`
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) GetAgentByName(ctx context.Context, name string) (*Agent, error) {
|
|
return s.scanAgent(s.db.QueryRowContext(ctx,
|
|
agentSelectSQL()+` WHERE name = ? AND status = 'active'`, name,
|
|
))
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) GetAgentByID(ctx context.Context, id int64) (*Agent, error) {
|
|
return s.scanAgent(s.db.QueryRowContext(ctx,
|
|
agentSelectSQL()+` WHERE id = ? AND status = 'active'`, id,
|
|
))
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) UpdateAgent(ctx context.Context, agent *Agent) error {
|
|
caps := string(agent.Capabilities)
|
|
if caps == "" {
|
|
caps = "{}"
|
|
}
|
|
|
|
_, err := s.db.ExecContext(ctx,
|
|
`UPDATE agents SET display_name = ?, capabilities = ?, api_key_hash = ?, updated_at = CURRENT_TIMESTAMP
|
|
WHERE name = ? AND status = 'active'`,
|
|
agent.DisplayName, caps, agent.APIKeyHash, agent.Name,
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) DeactivateAgent(ctx context.Context, name string) error {
|
|
result, err := s.db.ExecContext(ctx,
|
|
`UPDATE agents SET status = 'inactive', updated_at = CURRENT_TIMESTAMP
|
|
WHERE name = ? AND status = 'active'`,
|
|
name,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("deactivate agent: %w", err)
|
|
}
|
|
rowsAffected, err := result.RowsAffected()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if rowsAffected == 0 {
|
|
return fmt.Errorf("agent not found: %s", name)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) ListActiveAgents(ctx context.Context) ([]*Agent, error) {
|
|
rows, err := s.db.QueryContext(ctx,
|
|
agentSelectSQL()+` WHERE status = 'active' ORDER BY name`,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
return s.scanAgents(rows)
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) ListAllActiveAgents(ctx context.Context) ([]*Agent, error) {
|
|
rows, err := s.db.QueryContext(ctx,
|
|
agentSelectSQL()+` WHERE status = 'active' AND type != 'human' ORDER BY name`,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
return s.scanAgents(rows)
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) ListAgentsByOwner(ctx context.Context, ownerID int64) ([]*Agent, error) {
|
|
rows, err := s.db.QueryContext(ctx,
|
|
agentSelectSQL()+` WHERE owner_id = ? AND status = 'active' ORDER BY name`,
|
|
ownerID,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
return s.scanAgents(rows)
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) SearchAgentsByCapability(ctx context.Context, query string) ([]*Agent, error) {
|
|
// Simple LIKE search on the capabilities JSON field
|
|
rows, err := s.db.QueryContext(ctx,
|
|
agentSelectSQL()+` WHERE status = 'active' AND capabilities LIKE ? ORDER BY name`,
|
|
"%"+query+"%",
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
return s.scanAgents(rows)
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) GetHumanAgentByOwner(ctx context.Context, ownerID int64) (*Agent, error) {
|
|
return s.scanAgent(s.db.QueryRowContext(ctx,
|
|
agentSelectSQL()+` WHERE owner_id = ? AND type = 'human' AND status = 'active' LIMIT 1`, ownerID,
|
|
))
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) scanAgent(row *sql.Row) (*Agent, error) {
|
|
var agent Agent
|
|
var caps string
|
|
var k8sImage, k8sEnvJSON sql.NullString
|
|
var harnessName, localCommand, harnessConfigJSON sql.NullString
|
|
var pendingWork int
|
|
var parentAgentID sql.NullInt64
|
|
var quarantinedAt sql.NullTime
|
|
var quarantineReason sql.NullString
|
|
err := row.Scan(
|
|
&agent.ID, &agent.Name, &agent.DisplayName, &agent.Type,
|
|
&caps, &agent.OwnerID, &agent.APIKeyHash, &agent.Status,
|
|
&agent.CreatedAt, &agent.UpdatedAt,
|
|
&agent.TriggerMode, &agent.CooldownSeconds, &agent.DailyTriggerBudget,
|
|
&agent.MaxTriggerDepth, &k8sImage, &k8sEnvJSON,
|
|
&agent.K8sResourcePreset, &pendingWork,
|
|
&harnessName, &localCommand, &harnessConfigJSON,
|
|
&agent.ConfigHash, &parentAgentID, &agent.SpawnDepth, &agent.SystemPrompt,
|
|
&agent.AutonomyTier, &agent.ToolScopeJSON, &quarantinedAt, &quarantineReason,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
agent.Capabilities = json.RawMessage(caps)
|
|
agent.K8sImage = k8sImage.String
|
|
agent.K8sEnvJSON = k8sEnvJSON.String
|
|
agent.PendingWork = pendingWork != 0
|
|
agent.HarnessName = harnessName.String
|
|
agent.LocalCommand = localCommand.String
|
|
agent.HarnessConfigJSON = harnessConfigJSON.String
|
|
if parentAgentID.Valid {
|
|
id := parentAgentID.Int64
|
|
agent.ParentAgentID = &id
|
|
}
|
|
if quarantinedAt.Valid {
|
|
t := quarantinedAt.Time
|
|
agent.QuarantinedAt = &t
|
|
}
|
|
agent.QuarantineReason = quarantineReason.String
|
|
return &agent, nil
|
|
}
|
|
|
|
func (s *SQLiteAgentStore) scanAgents(rows *sql.Rows) ([]*Agent, error) {
|
|
var agents []*Agent
|
|
for rows.Next() {
|
|
var agent Agent
|
|
var caps string
|
|
var k8sImage, k8sEnvJSON sql.NullString
|
|
var harnessName, localCommand, harnessConfigJSON sql.NullString
|
|
var pendingWork int
|
|
var parentAgentID sql.NullInt64
|
|
var quarantinedAt sql.NullTime
|
|
var quarantineReason sql.NullString
|
|
err := rows.Scan(
|
|
&agent.ID, &agent.Name, &agent.DisplayName, &agent.Type,
|
|
&caps, &agent.OwnerID, &agent.APIKeyHash, &agent.Status,
|
|
&agent.CreatedAt, &agent.UpdatedAt,
|
|
&agent.TriggerMode, &agent.CooldownSeconds, &agent.DailyTriggerBudget,
|
|
&agent.MaxTriggerDepth, &k8sImage, &k8sEnvJSON,
|
|
&agent.K8sResourcePreset, &pendingWork,
|
|
&harnessName, &localCommand, &harnessConfigJSON,
|
|
&agent.ConfigHash, &parentAgentID, &agent.SpawnDepth, &agent.SystemPrompt,
|
|
&agent.AutonomyTier, &agent.ToolScopeJSON, &quarantinedAt, &quarantineReason,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
agent.Capabilities = json.RawMessage(caps)
|
|
agent.K8sImage = k8sImage.String
|
|
agent.K8sEnvJSON = k8sEnvJSON.String
|
|
agent.PendingWork = pendingWork != 0
|
|
agent.HarnessName = harnessName.String
|
|
agent.LocalCommand = localCommand.String
|
|
agent.HarnessConfigJSON = harnessConfigJSON.String
|
|
if parentAgentID.Valid {
|
|
id := parentAgentID.Int64
|
|
agent.ParentAgentID = &id
|
|
}
|
|
if quarantinedAt.Valid {
|
|
t := quarantinedAt.Time
|
|
agent.QuarantinedAt = &t
|
|
}
|
|
agent.QuarantineReason = quarantineReason.String
|
|
agents = append(agents, &agent)
|
|
}
|
|
if agents == nil {
|
|
agents = []*Agent{}
|
|
}
|
|
return agents, rows.Err()
|
|
}
|