Merge pull request #386 from rumpl/feat-command-palette

Add a command palette to the TUI

Author: rumpl

pkg/app/app.go

@@ -22,6 +22,7 @@ type App struct {
 	firstMessage     *string
 	events           chan tea.Msg
 	throttleDuration time.Duration
+	cancel           context.CancelFunc
 }
 
 func New(title, agentFilename string, rt runtime.Runtime, agents *team.Team, sess *session.Session, firstMessage *string) *App {
@@ -50,7 +51,8 @@ func (a *App) Title() string {
 }
 
 // Run one agent loop
-func (a *App) Run(ctx context.Context, message string) {
+func (a *App) Run(ctx context.Context, cancel context.CancelFunc, message string) {
+	a.cancel = cancel
 	go func() {
 		// Special shell command
 		if strings.HasPrefix(message, "!") {
@@ -62,6 +64,9 @@ func (a *App) Run(ctx context.Context, message string) {
 		// User message
 		a.session.AddMessage(session.UserMessage(a.agentFilename, message))
 		for event := range a.runtime.RunStream(ctx, a.session) {
+			if ctx.Err() != nil {
+				return
+			}
 			a.events <- event
 		}
 	}()
@@ -89,6 +94,24 @@ func (a *App) Resume(confirmationType string) {
 	}
 }
 
+func (a *App) NewSession() {
+	if a.cancel != nil {
+		a.cancel()
+		a.cancel = nil
+	}
+	a.session = session.New()
+}
+
+func (a *App) CompactSession() {
+	if a.runtime != nil && a.session != nil {
+		events := make(chan runtime.Event, 100)
+		a.runtime.Summarize(context.Background(), a.session, events)
+		for event := range events {
+			a.events <- event
+		}
+	}
+}
+
 // ResumeStartOAuth resumes the runtime with OAuth authorization confirmation
 func (a *App) ResumeStartOAuth(confirmation bool) {
 	if a.runtime != nil {

pkg/tui/components/messages/messages.go

@@ -35,7 +35,6 @@ type Model interface {
 	AddOrUpdateToolCall(agentName string, toolCall tools.ToolCall, toolDef tools.Tool, status types.ToolStatus) tea.Cmd
 	AddToolResult(msg *runtime.ToolCallResponseEvent, status types.ToolStatus) tea.Cmd
 	AppendToLastMessage(agentName string, messageType types.MessageType, content string) tea.Cmd
-	ClearMessages()
 	ScrollToBottom() tea.Cmd
 	AddShellOutputMessage(content string) tea.Cmd
 	AddSystemMessage(content string) tea.Cmd
@@ -73,7 +72,7 @@ type model struct {
 // New creates a new message list component
 func New(a *app.App) Model {
 	return &model{
-		width:         80,
+		width:         120,
 		height:        24,
 		app:           a,
 		renderedItems: make(map[int]renderedItem),
@@ -603,16 +602,6 @@ func (m *model) AppendToLastMessage(agentName string, messageType types.MessageT
 	}
 }
 
-// ClearMessages clears all messages
-func (m *model) ClearMessages() {
-	m.messages = nil
-	m.views = nil
-	m.scrollOffset = 0
-	m.rendered = ""
-	m.totalHeight = 0
-	m.renderedItems = make(map[int]renderedItem)
-}
-
 // ScrollToBottom scrolls to the bottom of the chat
 func (m *model) ScrollToBottom() tea.Cmd {
 	return func() tea.Msg {