snapshot

Author: avelanarius

summarize.go

@@ -0,0 +1,80 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	tea "github.com/charmbracelet/bubbletea"
+	openai "github.com/openai/openai-go"
+	"github.com/openai/openai-go/option"
+)
+
+// SummaryCommand represents a command's description and its captured output
+// used for generating an LLM summary.
+type SummaryCommand struct {
+	Description string
+	Output      string
+}
+
+// Summarizer encapsulates LLM client configuration used for summarization.
+type Summarizer struct {
+	client openai.Client
+	model  string
+}
+
+// NewSummarizer constructs a Summarizer with sensible defaults.
+func NewSummarizer() *Summarizer {
+	return &Summarizer{
+		client: openai.NewClient(option.WithBaseURL("https://openrouter.ai/api/v1")),
+		model:  "openai/gpt-oss-120b:nitro",
+	}
+}
+
+// Summarize generates a summary given a system prompt and a list of command
+// descriptions paired with their outputs. The systemPrompt is passed as a
+// system message, and the concatenated command outputs are passed as a user
+// message.
+func (s *Summarizer) Summarize(systemPrompt string, commands []SummaryCommand) (string, error) {
+	ctx := context.Background()
+
+	var b strings.Builder
+	for i, c := range commands {
+		if strings.TrimSpace(c.Output) == "" {
+			continue
+		}
+		b.WriteString(fmt.Sprintf("Command %d: %s\n", i+1, c.Description))
+		b.WriteString(c.Output)
+		b.WriteString("\n\n")
+	}
+	userContent := b.String()
+
+	resp, err := s.client.Chat.Completions.New(ctx, openai.ChatCompletionNewParams{
+		Model: s.model,
+		Messages: []openai.ChatCompletionMessageParamUnion{
+			openai.SystemMessage(systemPrompt),
+			openai.UserMessage(userContent),
+		},
+	})
+	if err != nil {
+		return "", err
+	}
+	if len(resp.Choices) == 0 {
+		return "", fmt.Errorf("no choices from LLM")
+	}
+	return resp.Choices[0].Message.Content, nil
+}
+
+// summarizeCmd wraps the Summarizer.Summarize call into a Bubble Tea command
+// that returns an llmMsg for the UI state machine.
+func summarizeCmd(s *Summarizer, systemPrompt string, commands []SummaryCommand) tea.Cmd {
+	return func() tea.Msg {
+		summary, err := s.Summarize(systemPrompt, commands)
+		if err != nil {
+			return llmMsg{err: err}
+		}
+		return llmMsg{summary: summary}
+	}
+}
+
+

ui.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"context"
 	"fmt"
 	"math"
 	"strings"
@@ -12,8 +11,6 @@ import (
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/glamour"
 	"github.com/charmbracelet/lipgloss"
-	openai "github.com/openai/openai-go"
-	"github.com/openai/openai-go/option"
 )
 
 var (
@@ -80,6 +77,8 @@ type model struct {
 	summaryErr  error
 
 	done bool
+
+	summarizer *Summarizer
 }
 
 // NewModel constructs a model initialised with all diagnostic commands in a
@@ -102,6 +101,7 @@ func NewModel(tb *Toolbox) *model {
 			return s
 		}(),
 		startTime: time.Now(),
+		summarizer: NewSummarizer(),
 	}
 }
 
@@ -118,46 +118,7 @@ func (m *model) Init() tea.Cmd {
 	return tea.Batch(cmds...)
 }
 
-// summarizeCmd calls OpenAI to generate summary.
-func summarizeCmd(outputs []string, errs []error) tea.Cmd {
-	// Prepare input text
-	var b strings.Builder
-	for i, out := range outputs {
-		if out == "" && errs[i] == nil {
-			continue
-		}
-		b.WriteString(fmt.Sprintf("Command %d:\n", i+1))
-		if out != "" {
-			b.WriteString(out)
-			b.WriteString("\n")
-		}
-		if errs[i] != nil {
-			b.WriteString(fmt.Sprintf("ERROR: %v\n", errs[i]))
-		}
-		b.WriteString("\n")
-	}
-	inputText := b.String()
-
-	return func() tea.Msg {
-		ctx := context.Background()
-		client := openai.NewClient(option.WithBaseURL("https://openrouter.ai/api/v1"))
-		content := "Please analyze the following Linux system diagnostic output. Focus on identifying any performance issues, errors, or notable system characteristics. Be concise but comprehensive. Start with 1-sentence overview and the key finding. Prefer bullet points. Your full output should be maximum 2-3 paragraphs (that includes bullet points - don't overdo them).\n\nDiagnostic output:\n\n" + inputText
-		resp, err := client.Chat.Completions.New(ctx, openai.ChatCompletionNewParams{
-			// Model: "openai/gpt-4.1",
-			Model: "openai/gpt-oss-120b:nitro",
-			Messages: []openai.ChatCompletionMessageParamUnion{
-				openai.UserMessage(content),
-			},
-		})
-		if err != nil {
-			return llmMsg{err: err}
-		}
-		if len(resp.Choices) == 0 {
-			return llmMsg{err: fmt.Errorf("no choices from LLM")}
-		}
-		return llmMsg{summary: resp.Choices[0].Message.Content}
-	}
-}
+// summarizeCmd moved to summarize.go
 
 // downloadToolboxCmd runs the toolbox download in a goroutine.
 func downloadToolboxCmd(tb *Toolbox) tea.Cmd {
@@ -224,7 +185,15 @@ func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		if allDone {
 			if !m.summarizing && m.summary == "" {
 				m.summarizing = true
-				return m, summarizeCmd(m.outputs, m.errors)
+				var sc []SummaryCommand
+				for i := range m.commands {
+					sc = append(sc, SummaryCommand{
+						Description: m.commands[i].Display,
+						Output:      m.outputs[i],
+					})
+				}
+				systemPrompt := "Please analyze the following Linux system diagnostic output. Focus on identifying any performance issues, errors, or notable system characteristics. Be concise but comprehensive. Start with 1-sentence overview and the key finding. Prefer bullet points. Your full output should be maximum 2-3 paragraphs (that includes bullet points - don't overdo them)."
+				return m, summarizeCmd(m.summarizer, systemPrompt, sc)
 			}
 		}
 		// No follow-up commands here.