92 lines
2.4 KiB
Go
92 lines
2.4 KiB
Go
// internal/skills/review/handlers.go
|
|
package review
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/mathiasbq/supervisor/internal/brain"
|
|
iexec "github.com/mathiasbq/supervisor/internal/exec"
|
|
"github.com/mathiasbq/supervisor/internal/session"
|
|
)
|
|
|
|
type reviewArgs struct {
|
|
ProjectRoot string `json:"project_root"`
|
|
Files []string `json:"files"`
|
|
Context string `json:"context"`
|
|
Model string `json:"model"`
|
|
SessionID string `json:"session_id"`
|
|
}
|
|
|
|
// Handle dispatches the MCP tool call to the appropriate handler.
|
|
func (s *Skill) Handle(ctx context.Context, tool string, args json.RawMessage) (json.RawMessage, error) {
|
|
if tool != "review" {
|
|
return nil, fmt.Errorf("unknown tool: %s", tool)
|
|
}
|
|
var a reviewArgs
|
|
if err := json.Unmarshal(args, &a); err != nil {
|
|
return nil, fmt.Errorf("parse args: %w", err)
|
|
}
|
|
if a.ProjectRoot == "" {
|
|
return nil, fmt.Errorf("project_root is required")
|
|
}
|
|
if len(a.Files) == 0 {
|
|
return nil, fmt.Errorf("files is required")
|
|
}
|
|
|
|
model := a.Model
|
|
if model == "" {
|
|
model = s.cfg.DefaultModel
|
|
}
|
|
|
|
brainCtx, _ := brain.Query(ctx, s.cfg.IngestBaseURL, strings.Join(a.Files, " ")+" "+a.Context, 3)
|
|
|
|
task := fmt.Sprintf(
|
|
"phase: review\nproject_root: %s\nfiles: %s\ncontext: %s\nmodel: %s",
|
|
a.ProjectRoot, strings.Join(a.Files, ", "), a.Context, model,
|
|
)
|
|
task = session.PrependHistory(s.cfg.SessionsDir, a.SessionID, "review", task)
|
|
if brainCtx != "" {
|
|
task = brainCtx + "\n---\n\n" + task
|
|
}
|
|
|
|
if s.cfg.ExecutorFn == nil {
|
|
return nil, fmt.Errorf("no executor configured")
|
|
}
|
|
t0 := time.Now()
|
|
result, err := s.cfg.ExecutorFn(ctx, iexec.Request{
|
|
SkillPrompt: s.cfg.SkillPrompt,
|
|
TaskPrompt: task,
|
|
Model: model,
|
|
Tools: "Read,Bash",
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if a.SessionID != "" && s.cfg.SessionsDir != "" {
|
|
_ = session.Append(s.cfg.SessionsDir, a.SessionID, session.Entry{
|
|
SessionID: a.SessionID,
|
|
Timestamp: time.Now(),
|
|
Skill: "review",
|
|
Phase: "review",
|
|
ProjectRoot: a.ProjectRoot,
|
|
Attempts: session.AttemptsFrom(result.Attempts),
|
|
FinalStatus: result.Status,
|
|
FilePath: result.FilePath,
|
|
ModelUsed: result.ModelUsed,
|
|
DurationMs: time.Since(t0).Milliseconds(),
|
|
Message: result.Message,
|
|
})
|
|
}
|
|
|
|
b, err := json.Marshal(result)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("marshal result: %w", err)
|
|
}
|
|
return b, nil
|
|
}
|