Merge pull request #1503 from krissetto/editable-session-titles

Editable session titles

Author: dgageot

cmd/root/new.go

@@ -12,6 +12,7 @@ import (
 	"github.com/docker/cagent/pkg/creator"
 	"github.com/docker/cagent/pkg/runtime"
 	"github.com/docker/cagent/pkg/session"
+	"github.com/docker/cagent/pkg/sessiontitle"
 	"github.com/docker/cagent/pkg/telemetry"
 	"github.com/docker/cagent/pkg/tui"
 )
@@ -73,6 +74,13 @@ func (f *newFlags) runNewCommand(cmd *cobra.Command, args []string) error {
 }
 
 func runTUI(ctx context.Context, rt runtime.Runtime, sess *session.Session, opts ...app.Opt) error {
+	// For local runtime, create and pass a title generator.
+	if pr, ok := rt.(*runtime.PersistentRuntime); ok {
+		if model := pr.CurrentAgent().Model(); model != nil {
+			opts = append(opts, app.WithTitleGenerator(sessiontitle.New(model)))
+		}
+	}
+
 	a := app.New(ctx, rt, sess, opts...)
 	m := tui.New(ctx, a)
 

docs/USAGE.md

@@ -172,6 +172,7 @@ During TUI sessions, you can use special slash commands. Type `/` to see all ava
 | `/star`     | Toggle star on current session                                      |
 | `/theme`    | Change the color theme (see [Theming](#theming))                    |
 | `/think`    | Toggle thinking/reasoning mode                                      |
+| `/title`    | Set or regenerate session title (usage: /title [new title])         |
 | `/yolo`     | Toggle automatic approval of tool calls                             |
 
 #### Runtime Model Switching
@@ -279,6 +280,30 @@ Themes can customize colors in three sections: `colors`, `chroma` (syntax highli
 
 See the [built-in themes on GitHub](https://github.com/docker/cagent/tree/main/pkg/tui/styles/themes) for complete examples.
 
+#### Session Title Editing
+
+You can customize session titles to make them more meaningful and easier to find later. By default, cagent automatically generates titles based on your first message, but you can override or regenerate them at any time.
+
+**Using the `/title` command:**
+
+```
+/title                     # Regenerate title using AI (based on at most the last 2 user messages)
+/title My Custom Title     # Set a specific title
+```
+
+**Using the sidebar:**
+
+In the TUI, you can click on the pencil icon (✎) next to the session title in the sidebar to edit it inline:
+
+1. Click the pencil icon next to the title
+2. Type your new title
+3. Press Enter to save, or Escape to cancel
+
+**Notes:**
+- Manually set titles are preserved and won't be overwritten by auto-generation
+- Title changes are persisted immediately to the session
+- Works with both local and remote runtimes
+
 ## 🔧 Configuration Reference
 
 ### Agent Properties

e2e/runtime_test.go

@@ -32,7 +32,7 @@ func TestRuntime_OpenAI_Basic(t *testing.T) {
 
 	response := sess.GetLastAssistantMessageContent()
 	assert.Equal(t, "2 + 2 equals 4.", response)
-	assert.Equal(t, "Simple Math: Addition of 2 and 2", sess.Title)
+	// Title generation is now handled by pkg/app or pkg/server, not the runtime
 }
 
 func TestRuntime_Mistral_Basic(t *testing.T) {
@@ -55,5 +55,5 @@ func TestRuntime_Mistral_Basic(t *testing.T) {
 
 	response := sess.GetLastAssistantMessageContent()
 	assert.Equal(t, "The sum of 2 + 2 is 4.", response)
-	assert.Equal(t, "Math Basics: Simple Addition", sess.Title)
+	// Title generation is now handled by pkg/app or pkg/server, not the runtime
 }

e2e/testdata/cassettes/TestRuntime_Mistral_Basic.yaml

@@ -58,7 +58,7 @@ interactions:
         proto_minor: 1
         content_length: 0
         host: api.mistral.ai
-        body: '{"messages":[{"content":"You are a helpful AI assistant that generates concise, descriptive titles for conversations. You will be given a conversation history and asked to create a single-line title that captures the main topic. Never use newlines or line breaks in your response.","role":"system"},{"content":"Based on the following message a user sent to an AI assistant, generate a short, descriptive title (maximum 50 characters) that captures the main topic or purpose of the conversation. Return ONLY the title text on a single line, nothing else. Do not include any newlines, explanations, or formatting.\n\nUser message: What''s 2+2?\n\n","role":"user"}],"model":"mistral-small","stream_options":{"include_usage":true},"stream":true}'
+        body: '{"messages":[{"content":"You are a helpful AI assistant that generates concise, descriptive titles for conversations. You will be given up to 2 recent user messages and asked to create a single-line title that captures the main topic. Never use newlines or line breaks in your response.","role":"system"},{"content":"Based on the following recent user messages from a conversation with an AI assistant, generate a short, descriptive title (maximum 50 characters) that captures the main topic or purpose of the conversation. Return ONLY the title text on a single line, nothing else. Do not include any newlines, explanations, or formatting.\n\nRecent user messages:\n1. What''s 2+2?\n\n\n","role":"user"}],"model":"mistral-small","stream_options":{"include_usage":true},"stream":true}'
         url: https://api.mistral.ai/v1/chat/completions
         method: POST
       response:

e2e/testdata/cassettes/TestRuntime_OpenAI_Basic.yaml

@@ -52,7 +52,7 @@ interactions:
         proto_minor: 1
         content_length: 0
         host: api.openai.com
-        body: '{"messages":[{"content":"You are a helpful AI assistant that generates concise, descriptive titles for conversations. You will be given a conversation history and asked to create a single-line title that captures the main topic. Never use newlines or line breaks in your response.","role":"system"},{"content":"Based on the following message a user sent to an AI assistant, generate a short, descriptive title (maximum 50 characters) that captures the main topic or purpose of the conversation. Return ONLY the title text on a single line, nothing else. Do not include any newlines, explanations, or formatting.\n\nUser message: What''s 2+2?\n\n","role":"user"}],"model":"gpt-3.5-turbo","stream_options":{"include_usage":true},"stream":true}'
+        body: '{"messages":[{"content":"You are a helpful AI assistant that generates concise, descriptive titles for conversations. You will be given up to 2 recent user messages and asked to create a single-line title that captures the main topic. Never use newlines or line breaks in your response.","role":"system"},{"content":"Based on the following recent user messages from a conversation with an AI assistant, generate a short, descriptive title (maximum 50 characters) that captures the main topic or purpose of the conversation. Return ONLY the title text on a single line, nothing else. Do not include any newlines, explanations, or formatting.\n\nRecent user messages:\n1. What''s 2+2?\n\n\n","role":"user"}],"model":"gpt-3.5-turbo","stream_options":{"include_usage":true},"stream":true}'
         url: https://api.openai.com/v1/chat/completions
         method: POST
       response: