dimitar
aee63bf321
Add the AI layer that turns scan results into suggested next actions. Provider model (internal/ai): - Provider interface with Name() and Suggest(ctx, ScanResult) returning []Suggestion (repo_path, action, message, command, priority 0-2) - NewProvider factory resolves the configured provider and enforces that cloud providers have their API key in the configured env var; Ollama needs no key - OpenAIProvider: Chat Completions with response_format json_object - OllamaProvider: local /api/chat with format:json for structured output - AnthropicProvider: Messages API with system prompt and key/version headers; all three fall back to provider-appropriate defaults for base_url and model - Shared client: 60s timeout, 4 MiB response cap, JSON encode/decode, HTTP error surfaces the upstream status and body Prompt design (BuildPrompt): - Renders the full repository status table (name/status/branch/ahead/ behind/changes) plus strict output rules: exact suggestion schema, allowed actions, smallest-safe-step guidance, no invented repositories, return [] when healthy Parsing (parseSuggestions): - Tolerates ```json fences and surrounding prose, clamps priorities to 0-2, caps the result, and rejects replies without a JSON array Suggestion display (presenter.Suggestions): - "AI SUGGESTIONS" table (repository/action/priority/message/command) rendered below the scan table with priority color-coded (red/yellow/ green); empty results say all repositories are healthy Guarded execution (--ai-execute, experimental): - RunConfirmed executes a suggestion's command inside its repository only after explicit per-command y/N confirmation, and only for actions on an allowlist (commit/push/pull/stash/checkout) so LLM output can never run arbitrary shell commands; cancellation aborts remaining suggestions CLI wiring: - scan: AI block after the table when --ai is set and format is not json (JSON streams stay machine-readable); failures degrade to warnings - watch: AI is queried only on the first frame and when something changed since the previous frame, to avoid hammering the provider every interval - New --ai-execute flag and provider validation in config (openai/ollama/ anthropic), included in the config dump Testing: - httptest-based provider tests verifying request shape (model, auth headers, path), response parsing, API error bodies, HTTP failures, and cancellation - Parse tests: plain/fenced/prose replies, empty arrays, garbage, truncated JSON, priority clamping - Execution tests: unsafe actions and empty commands never run, declined confirmations are skipped, confirmed commands execute in the repo dir, failing commands surface errors - Presenter suggestion table and empty-state tests Verified: go build, go vet, go test -race, gofmt clean; end-to-end smoke test against a local fake Ollama server (request shape confirmed, table + suggestions rendered) and the missing-API-key warning path.
70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
package ai
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"gitea.oblak.solutions/dimitar/gitFlow/pkg/status"
|
|
)
|
|
|
|
// BuildPrompt renders a scan result as an LLM instruction: a repository
|
|
// status table plus strict output rules for the suggestion schema.
|
|
func BuildPrompt(result status.ScanResult) string {
|
|
var b strings.Builder
|
|
fmt.Fprintf(&b, "You are gitflow, a Git repository health assistant. "+
|
|
"A scan of %q at %s found %d repositories.\n\n",
|
|
result.ParentDir, result.ScannedAt.Format(time.RFC3339), len(result.Repos))
|
|
|
|
b.WriteString("Repository status table:\n")
|
|
b.WriteString("| name | status | branch | ahead | behind | changes |\n")
|
|
b.WriteString("|---|---|---|---|---|---|\n")
|
|
for _, r := range result.Repos {
|
|
fmt.Fprintf(&b, "| %s | %s | %s | %d | %d | %s |\n",
|
|
r.Name, r.Status, branchLabel(r), r.AheadBy, r.BehindBy, changeSummary(r))
|
|
}
|
|
|
|
attention := result.NeedsAttention()
|
|
fmt.Fprintf(&b, "\n%d of %d repositories need attention.\n\n", len(attention), len(result.Repos))
|
|
|
|
b.WriteString(`Return ONLY a JSON array of suggestions. Each suggestion has exactly these fields:
|
|
- "repo_path": absolute path of the repository
|
|
- "action": one of "commit", "push", "pull", "stash", "create_pr", "cleanup", "inspect"
|
|
- "message": one short sentence explaining the next step
|
|
- "command": a concrete git command to run, or "" if none is safe
|
|
- "priority": 0 (low), 1 (medium), or 2 (high)
|
|
|
|
Rules:
|
|
- Only suggest actions for repositories that need attention.
|
|
- Prefer the smallest safe step; never suggest destructive commands.
|
|
- Do not invent repositories that are not in the table.
|
|
- If nothing needs attention, return [].
|
|
`)
|
|
return b.String()
|
|
}
|
|
|
|
func branchLabel(r status.RepoInfo) string {
|
|
if r.Branch == "" {
|
|
return "(none)"
|
|
}
|
|
return r.Branch
|
|
}
|
|
|
|
// changeSummary renders staged/modified/untracked counts for the prompt.
|
|
func changeSummary(r status.RepoInfo) string {
|
|
parts := make([]string, 0, 3)
|
|
if n := len(r.StagedFiles); n > 0 {
|
|
parts = append(parts, fmt.Sprintf("%d staged", n))
|
|
}
|
|
if n := len(r.ModifiedFiles); n > 0 {
|
|
parts = append(parts, fmt.Sprintf("%d modified", n))
|
|
}
|
|
if n := len(r.UntrackedFiles); n > 0 {
|
|
parts = append(parts, fmt.Sprintf("%d untracked", n))
|
|
}
|
|
if len(parts) == 0 {
|
|
return "none"
|
|
}
|
|
return strings.Join(parts, ", ")
|
|
}
|