fix: Fixed formatting issues

Author: codefionn

cmd/scriptschnell/main.go

@@ -830,7 +830,9 @@ func runTUI(cfg *config.Config, providerMgr *provider.Manager) error {
 									SessionID:    sessionID,
 									ResponseChan: make(chan actor.SessionStorageDeleteResponse, 1),
 								}
-								storageRef.Send(deleteMsg)
+								if err := storageRef.Send(deleteMsg); err != nil {
+									return fmt.Errorf("failed to send delete message: %w", err)
+								}
 								response := <-deleteMsg.ResponseChan
 								if response.Err != nil {
 									return fmt.Errorf("failed to delete session: %w", response.Err)

go.mod

@@ -24,10 +24,10 @@ require (
 	github.com/tree-sitter/tree-sitter-go v0.25.0
 	github.com/tree-sitter/tree-sitter-python v0.25.0
 	github.com/tree-sitter/tree-sitter-typescript v0.23.2
-	golang.org/x/crypto v0.45.0
-	golang.org/x/net v0.47.0
-	golang.org/x/term v0.37.0
-	google.golang.org/genai v1.37.0
+	golang.org/x/crypto v0.46.0
+	golang.org/x/net v0.48.0
+	golang.org/x/term v0.38.0
+	google.golang.org/genai v1.38.0
 )
 
 require (
@@ -42,7 +42,7 @@ require (
 	github.com/charmbracelet/colorprofile v0.3.3 // indirect
 	github.com/charmbracelet/x/ansi v0.11.2 // indirect
 	github.com/charmbracelet/x/cellbuf v0.0.14 // indirect
-	github.com/charmbracelet/x/exp/slice v0.0.0-20251201173703-9f73bfd934ff // indirect
+	github.com/charmbracelet/x/exp/slice v0.0.0-20251210182518-b3d4d1ed2373 // indirect
 	github.com/charmbracelet/x/term v0.2.2 // indirect
 	github.com/clipperhouse/displaywidth v0.6.1 // indirect
 	github.com/clipperhouse/stringish v0.1.1 // indirect
@@ -52,7 +52,7 @@ require (
 	github.com/felixge/httpsnoop v1.0.4 // indirect
 	github.com/go-logr/logr v1.4.3 // indirect
 	github.com/go-logr/stdr v1.2.2 // indirect
-	github.com/go-openapi/jsonpointer v0.22.3 // indirect
+	github.com/go-openapi/jsonpointer v0.22.4 // indirect
 	github.com/go-openapi/swag/jsonname v0.25.4 // indirect
 	github.com/google/go-cmp v0.7.0 // indirect
 	github.com/google/s2a-go v0.1.9 // indirect
@@ -87,12 +87,12 @@ require (
 	github.com/yuin/goldmark v1.7.13 // indirect
 	github.com/yuin/goldmark-emoji v1.0.6 // indirect
 	go.opentelemetry.io/auto/sdk v1.2.1 // indirect
-	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.63.0 // indirect
-	go.opentelemetry.io/otel v1.38.0 // indirect
-	go.opentelemetry.io/otel/metric v1.38.0 // indirect
-	go.opentelemetry.io/otel/trace v1.38.0 // indirect
-	golang.org/x/sys v0.38.0 // indirect
-	golang.org/x/text v0.31.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.64.0 // indirect
+	go.opentelemetry.io/otel v1.39.0 // indirect
+	go.opentelemetry.io/otel/metric v1.39.0 // indirect
+	go.opentelemetry.io/otel/trace v1.39.0 // indirect
+	golang.org/x/sys v0.39.0 // indirect
+	golang.org/x/text v0.32.0 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217 // indirect
 	google.golang.org/grpc v1.77.0 // indirect
 	google.golang.org/protobuf v1.36.10 // indirect

go.sum

@@ -46,6 +46,8 @@ github.com/charmbracelet/x/exp/golden v0.0.0-20241011142426-46044092ad91 h1:payR
 github.com/charmbracelet/x/exp/golden v0.0.0-20241011142426-46044092ad91/go.mod h1:wDlXFlCrmJ8J+swcL/MnGUuYnqgQdW9rhSD61oNMb6U=
 github.com/charmbracelet/x/exp/slice v0.0.0-20251201173703-9f73bfd934ff h1:Uwr+/JS+qnRcO/++xjYEDtW7x+P5E4+4cBiOHTt2Xfk=
 github.com/charmbracelet/x/exp/slice v0.0.0-20251201173703-9f73bfd934ff/go.mod h1:vqEfX6xzqW1pKKZUUiFOKg0OQ7bCh54Q2vR/tserrRA=
+github.com/charmbracelet/x/exp/slice v0.0.0-20251210182518-b3d4d1ed2373 h1:2NaW38gkQs5W7HFoFvCrD0VZnSLzD4126YKiAlMr7nU=
+github.com/charmbracelet/x/exp/slice v0.0.0-20251210182518-b3d4d1ed2373/go.mod h1:vqEfX6xzqW1pKKZUUiFOKg0OQ7bCh54Q2vR/tserrRA=
 github.com/charmbracelet/x/term v0.2.2 h1:xVRT/S2ZcKdhhOuSP4t5cLi5o+JxklsoEObBSgfgZRk=
 github.com/charmbracelet/x/term v0.2.2/go.mod h1:kF8CY5RddLWrsgVwpw4kAa6TESp6EB5y3uxGLeCqzAI=
 github.com/clipperhouse/displaywidth v0.6.1 h1:/zMlAezfDzT2xy6acHBzwIfyu2ic0hgkT83UX5EY2gY=
@@ -77,6 +79,8 @@ github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
 github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=
 github.com/go-openapi/jsonpointer v0.22.3 h1:dKMwfV4fmt6Ah90zloTbUKWMD+0he+12XYAsPotrkn8=
 github.com/go-openapi/jsonpointer v0.22.3/go.mod h1:0lBbqeRsQ5lIanv3LHZBrmRGHLHcQoOXQnf88fHlGWo=
+github.com/go-openapi/jsonpointer v0.22.4 h1:dZtK82WlNpVLDW2jlA1YCiVJFVqkED1MegOUy9kR5T4=
+github.com/go-openapi/jsonpointer v0.22.4/go.mod h1:elX9+UgznpFhgBuaMQ7iu4lvvX1nvNsesQ3oxmYTw80=
 github.com/go-openapi/swag/jsonname v0.25.4 h1:bZH0+MsS03MbnwBXYhuTttMOqk+5KcQ9869Vye1bNHI=
 github.com/go-openapi/swag/jsonname v0.25.4/go.mod h1:GPVEk9CWVhNvWhZgrnvRA6utbAltopbKwDu8mXNUMag=
 github.com/go-openapi/testify/v2 v2.0.2 h1:X999g3jeLcoY8qctY/c/Z8iBHTbwLz7R2WXd6Ub6wls=
@@ -222,37 +226,60 @@ go.opentelemetry.io/auto/sdk v1.2.1 h1:jXsnJ4Lmnqd11kwkBV2LgLoFMZKizbCi5fNZ/ipaZ
 go.opentelemetry.io/auto/sdk v1.2.1/go.mod h1:KRTj+aOaElaLi+wW1kO/DZRXwkF4C5xPbEe3ZiIhN7Y=
 go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.63.0 h1:RbKq8BG0FI8OiXhBfcRtqqHcZcka+gU3cskNuf05R18=
 go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.63.0/go.mod h1:h06DGIukJOevXaj/xrNjhi/2098RZzcLTbc0jDAUbsg=
+go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.64.0 h1:ssfIgGNANqpVFCndZvcuyKbl0g+UAVcbBcqGkG28H0Y=
+go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.64.0/go.mod h1:GQ/474YrbE4Jx8gZ4q5I4hrhUzM6UPzyrqJYV2AqPoQ=
 go.opentelemetry.io/otel v1.38.0 h1:RkfdswUDRimDg0m2Az18RKOsnI8UDzppJAtj01/Ymk8=
 go.opentelemetry.io/otel v1.38.0/go.mod h1:zcmtmQ1+YmQM9wrNsTGV/q/uyusom3P8RxwExxkZhjM=
+go.opentelemetry.io/otel v1.39.0 h1:8yPrr/S0ND9QEfTfdP9V+SiwT4E0G7Y5MO7p85nis48=
+go.opentelemetry.io/otel v1.39.0/go.mod h1:kLlFTywNWrFyEdH0oj2xK0bFYZtHRYUdv1NklR/tgc8=
 go.opentelemetry.io/otel/metric v1.38.0 h1:Kl6lzIYGAh5M159u9NgiRkmoMKjvbsKtYRwgfrA6WpA=
 go.opentelemetry.io/otel/metric v1.38.0/go.mod h1:kB5n/QoRM8YwmUahxvI3bO34eVtQf2i4utNVLr9gEmI=
+go.opentelemetry.io/otel/metric v1.39.0 h1:d1UzonvEZriVfpNKEVmHXbdf909uGTOQjA0HF0Ls5Q0=
+go.opentelemetry.io/otel/metric v1.39.0/go.mod h1:jrZSWL33sD7bBxg1xjrqyDjnuzTUB0x1nBERXd7Ftcs=
 go.opentelemetry.io/otel/sdk v1.38.0 h1:l48sr5YbNf2hpCUj/FoGhW9yDkl+Ma+LrVl8qaM5b+E=
 go.opentelemetry.io/otel/sdk v1.38.0/go.mod h1:ghmNdGlVemJI3+ZB5iDEuk4bWA3GkTpW+DOoZMYBVVg=
+go.opentelemetry.io/otel/sdk v1.39.0 h1:nMLYcjVsvdui1B/4FRkwjzoRVsMK8uL/cj0OyhKzt18=
 go.opentelemetry.io/otel/sdk/metric v1.38.0 h1:aSH66iL0aZqo//xXzQLYozmWrXxyFkBJ6qT5wthqPoM=
 go.opentelemetry.io/otel/sdk/metric v1.38.0/go.mod h1:dg9PBnW9XdQ1Hd6ZnRz689CbtrUp0wMMs9iPcgT9EZA=
+go.opentelemetry.io/otel/sdk/metric v1.39.0 h1:cXMVVFVgsIf2YL6QkRF4Urbr/aMInf+2WKg+sEJTtB8=
 go.opentelemetry.io/otel/trace v1.38.0 h1:Fxk5bKrDZJUH+AMyyIXGcFAPah0oRcT+LuNtJrmcNLE=
 go.opentelemetry.io/otel/trace v1.38.0/go.mod h1:j1P9ivuFsTceSWe1oY+EeW3sc+Pp42sO++GHkg4wwhs=
+go.opentelemetry.io/otel/trace v1.39.0 h1:2d2vfpEDmCJ5zVYz7ijaJdOF59xLomrvj7bjt6/qCJI=
+go.opentelemetry.io/otel/trace v1.39.0/go.mod h1:88w4/PnZSazkGzz/w84VHpQafiU4EtqqlVdxWy+rNOA=
 golang.org/x/crypto v0.45.0 h1:jMBrvKuj23MTlT0bQEOBcAE0mjg8mK9RXFhRH6nyF3Q=
 golang.org/x/crypto v0.45.0/go.mod h1:XTGrrkGJve7CYK7J8PEww4aY7gM3qMCElcJQ8n8JdX4=
+golang.org/x/crypto v0.46.0 h1:cKRW/pmt1pKAfetfu+RCEvjvZkA9RimPbh7bhFjGVBU=
+golang.org/x/crypto v0.46.0/go.mod h1:Evb/oLKmMraqjZ2iQTwDwvCtJkczlDuTmdJXoZVzqU0=
 golang.org/x/exp v0.0.0-20231006140011-7918f672742d h1:jtJma62tbqLibJ5sFQz8bKtEM8rJBtfilJ2qTU199MI=
 golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQzzkH36rKxoZW1tw7ZJpeKx+hdo=
 golang.org/x/net v0.47.0 h1:Mx+4dIFzqraBXUugkia1OOvlD6LemFo1ALMHjrXDOhY=
 golang.org/x/net v0.47.0/go.mod h1:/jNxtkgq5yWUGYkaZGqo27cfGZ1c5Nen03aYrrKpVRU=
+golang.org/x/net v0.48.0 h1:zyQRTTrjc33Lhh0fBgT/H3oZq9WuvRR5gPC70xpDiQU=
+golang.org/x/net v0.48.0/go.mod h1:+ndRgGjkh8FGtu1w1FGbEC31if4VrNVMuKTgcAAnQRY=
 golang.org/x/sync v0.18.0 h1:kr88TuHDroi+UVf+0hZnirlk8o8T+4MrK6mr60WkH/I=
 golang.org/x/sync v0.18.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
+golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
 golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.38.0 h1:3yZWxaJjBmCWXqhN1qh02AkOnCQ1poK6oF+a7xWL6Gc=
 golang.org/x/sys v0.38.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
+golang.org/x/sys v0.39.0 h1:CvCKL8MeisomCi6qNZ+wbb0DN9E5AATixKsvNtMoMFk=
+golang.org/x/sys v0.39.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
 golang.org/x/term v0.37.0 h1:8EGAD0qCmHYZg6J17DvsMy9/wJ7/D/4pV/wfnld5lTU=
 golang.org/x/term v0.37.0/go.mod h1:5pB4lxRNYYVZuTLmy8oR2BH8dflOR+IbTYFD8fi3254=
+golang.org/x/term v0.38.0 h1:PQ5pkm/rLO6HnxFR7N2lJHOZX6Kez5Y1gDSJla6jo7Q=
+golang.org/x/term v0.38.0/go.mod h1:bSEAKrOT1W+VSu9TSCMtoGEOUcKxOKgl3LE5QEF/xVg=
 golang.org/x/text v0.31.0 h1:aC8ghyu4JhP8VojJ2lEHBnochRno1sgL6nEi9WGFGMM=
 golang.org/x/text v0.31.0/go.mod h1:tKRAlv61yKIjGGHX/4tP1LTbc13YSec1pxVEWXzfoeM=
+golang.org/x/text v0.32.0 h1:ZD01bjUt1FQ9WJ0ClOL5vxgxOI/sVCNgX1YtKwcY0mU=
+golang.org/x/text v0.32.0/go.mod h1:o/rUWzghvpD5TXrTIBuJU77MTaN0ljMWE47kxGJQ7jY=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk=
 gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E=
 google.golang.org/genai v1.37.0 h1:dgp71k1wQ+/+APdZrN3LFgAGnVnr5IdTF1Oj0Dg+BQc=
 google.golang.org/genai v1.37.0/go.mod h1:A3kkl0nyBjyFlNjgxIwKq70julKbIxpSxqKO5gw/gmk=
+google.golang.org/genai v1.38.0 h1:aE+kIjkmV9/gX5HjEv7ZQkw1sAAMM+9tW4a/RUhKdFk=
+google.golang.org/genai v1.38.0/go.mod h1:A3kkl0nyBjyFlNjgxIwKq70julKbIxpSxqKO5gw/gmk=
 google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217 h1:gRkg/vSppuSQoDjxyiGfN4Upv/h/DQmIR10ZU8dh4Ww=
 google.golang.org/genproto/googleapis/rpc v0.0.0-20251202230838-ff82c1b0f217/go.mod h1:7i2o+ce6H/6BluujYR+kqX3GKH+dChPTQU19wjRPiGk=
 google.golang.org/grpc v1.77.0 h1:wVVY6/8cGA6vvffn+wWK5ToddbgdU3d8MNENr4evgXM=

internal/config/config.go

@@ -84,9 +84,9 @@ type MCPOpenAIConfig struct {
 
 // AutoSaveConfig holds configuration for automatic session saving
 type AutoSaveConfig struct {
-	Enabled           bool `json:"enabled"`
+	Enabled             bool `json:"enabled"`
 	SaveIntervalSeconds int  `json:"save_interval_seconds"`
-	MaxConcurrentSaves int  `json:"max_concurrent_saves"`
+	MaxConcurrentSaves  int  `json:"max_concurrent_saves"`
 }
 
 // Config represents application configuration
@@ -111,7 +111,7 @@ type Config struct {
 	PromptCacheTTL     string                        `json:"prompt_cache_ttl,omitempty"`    // Cache TTL: "5m" or "1h" (default: "1h", Anthropic only)
 	ContextDirectories map[string][]string           `json:"context_directories,omitempty"` // Workspace-specific context directories (map of workspace path -> directories)
 	OpenTabs           map[string]*WorkspaceTabState `json:"open_tabs,omitempty"`           // Workspace-specific open tabs state (map of workspace path -> tab state)
-	AutoSave           AutoSaveConfig                `json:"auto_save,omitempty"`          // Session auto-save configuration
+	AutoSave           AutoSaveConfig                `json:"auto_save,omitempty"`           // Session auto-save configuration
 
 	authMu          sync.RWMutex `json:"-"` // Protects AuthorizedDomains and AuthorizedCommands for concurrent access
 	secretsPassword string       `json:"-"`
@@ -200,9 +200,9 @@ func DefaultConfig() *Config {
 		PromptCacheTTL:     "1h",                      // Default to 1 hour for longer sessions
 		ContextDirectories: make(map[string][]string), // No context directories by default
 		AutoSave: AutoSaveConfig{
-			Enabled:              true, // Enable by default
-			SaveIntervalSeconds:  5,    // Save every 5 seconds
-			MaxConcurrentSaves: 1,    // Only one save operation at a time
+			Enabled:             true, // Enable by default
+			SaveIntervalSeconds: 5,    // Save every 5 seconds
+			MaxConcurrentSaves:  1,    // Only one save operation at a time
 		},
 	}
 }
@@ -459,8 +459,27 @@ func (c *Config) decryptSensitiveFields(password string) error {
 }
 
 func (c *Config) marshalWithEncryptedSecrets() ([]byte, error) {
-	copyCfg := *c
-	copyCfg.Search = c.Search
+	copyCfg := Config{
+		WorkingDir:         c.WorkingDir,
+		CacheTTL:           c.CacheTTL,
+		MaxCacheEntries:    c.MaxCacheEntries,
+		DefaultTimeout:     c.DefaultTimeout,
+		TempDir:            c.TempDir,
+		Temperature:        c.Temperature,
+		MaxTokens:          c.MaxTokens,
+		ProviderConfigPath: c.ProviderConfigPath,
+		DisableAnimations:  c.DisableAnimations,
+		LogLevel:           c.LogLevel,
+		LogPath:            c.LogPath,
+		AuthorizedDomains:  c.AuthorizedDomains,
+		AuthorizedCommands: c.AuthorizedCommands,
+		Search:             c.Search,
+		MCP:                c.MCP,
+		Secrets:            c.Secrets,
+		EnablePromptCache:  c.EnablePromptCache,
+		AutoSave:           c.AutoSave,
+		secretsPassword:    c.secretsPassword,
+	}
 
 	var err error
 	copyCfg.Search.Exa.APIKey, err = encryptField(copyCfg.Search.Exa.APIKey, c.secretsPassword)

internal/llm/openai_compatible_client_test.go

@@ -39,7 +39,10 @@ func TestOpenAICompatibleClient_UsageData(t *testing.T) {
 		}
 
 		w.Header().Set("Content-Type", "application/json")
-		json.NewEncoder(w).Encode(response)
+		if err := json.NewEncoder(w).Encode(response); err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
 	}))
 	defer server.Close()
 
@@ -110,7 +113,10 @@ func TestOpenAICompatibleClient_UsageData_NoUsage(t *testing.T) {
 		}
 
 		w.Header().Set("Content-Type", "application/json")
-		json.NewEncoder(w).Encode(response)
+		if err := json.NewEncoder(w).Encode(response); err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
 	}))
 	defer server.Close()
 
@@ -142,4 +148,4 @@ func TestOpenAICompatibleClient_UsageData_NoUsage(t *testing.T) {
 	if resp.Usage != nil {
 		t.Errorf("Expected usage data to be nil, got %v", resp.Usage)
 	}
-}
+}

internal/orchestrator/orchestrator.go

@@ -1009,12 +1009,32 @@ func (o *Orchestrator) TestMCPServer(serverName string) error {
 		return fmt.Errorf("mcp server not found: %s", serverName)
 	}
 
-	tmpCfg := *o.config
-	tmpCfg.MCP.Servers = map[string]*config.MCPServerConfig{
-		serverName: server,
+	tmpCfg := &config.Config{
+		WorkingDir:         o.config.WorkingDir,
+		CacheTTL:           o.config.CacheTTL,
+		MaxCacheEntries:    o.config.MaxCacheEntries,
+		DefaultTimeout:     o.config.DefaultTimeout,
+		TempDir:            o.config.TempDir,
+		Temperature:        o.config.Temperature,
+		MaxTokens:          o.config.MaxTokens,
+		ProviderConfigPath: o.config.ProviderConfigPath,
+		DisableAnimations:  o.config.DisableAnimations,
+		LogLevel:           o.config.LogLevel,
+		LogPath:            o.config.LogPath,
+		AuthorizedDomains:  o.config.AuthorizedDomains,
+		AuthorizedCommands: o.config.AuthorizedCommands,
+		Search:             o.config.Search,
+		MCP: config.MCPConfig{
+			Servers: map[string]*config.MCPServerConfig{
+				serverName: server,
+			},
+		},
+		Secrets:           o.config.Secrets,
+		EnablePromptCache: o.config.EnablePromptCache,
+		AutoSave:          o.config.AutoSave,
 	}
 
-	manager := mcp.NewManager(&tmpCfg, o.workingDir, o.providerMgr)
+	manager := mcp.NewManager(tmpCfg, o.workingDir, o.providerMgr)
 	tools, errs := manager.BuildTools()
 	if len(errs) > 0 {
 		messages := make([]string, 0, len(errs))

internal/orchestrator/tool_validation_test.go

@@ -114,4 +114,4 @@ func TestToolCallValidationErrorNil(t *testing.T) {
 	if err.Error() != "" {
 		t.Errorf("Expected empty string for nil error, got: %s", err.Error())
 	}
-}
+}

internal/session/session.go

@@ -32,7 +32,7 @@ type Message struct {
 // Session manages a conversation session
 type Session struct {
 	ID                  string
-	Title               string          // Auto-generated title for the session
+	Title               string // Auto-generated title for the session
 	WorkingDir          string
 	Messages            []*Message
 	FilesRead           map[string]string // path -> content

internal/session/session_test.go

@@ -50,20 +50,20 @@ func TestCompactWithSummaryMismatch(t *testing.T) {
 func TestSaveEmptySession(t *testing.T) {
 	// Create a temporary directory for testing
 	tempDir := t.TempDir()
-	
+
 	// Create session storage
 	storage, err := NewSessionStorage()
 	if err != nil {
 		t.Fatalf("Failed to create session storage: %v", err)
 	}
-	
+
 	// Test 1: Try to save an empty session
 	emptySession := NewSession("test-empty", tempDir)
 	err = storage.SaveSession(emptySession, "Empty Session")
 	if err != nil {
 		t.Errorf("Unexpected error saving empty session: %v", err)
 	}
-	
+
 	// Verify no session was saved
 	sessions, err := storage.ListSessions(tempDir)
 	if err != nil {
@@ -72,18 +72,18 @@ func TestSaveEmptySession(t *testing.T) {
 	if len(sessions) != 0 {
 		t.Errorf("Expected 0 sessions after saving empty session, got %d", len(sessions))
 	}
-	
+
 	// Test 2: Save a session with messages
 	sessionWithMessages := NewSession("test-with-messages", tempDir)
 	sessionWithMessages.AddMessage(&Message{
-			Role:    "user",
-			Content: "Hello, world!",
-		})
+		Role:    "user",
+		Content: "Hello, world!",
+	})
 	err = storage.SaveSession(sessionWithMessages, "Session with Messages")
 	if err != nil {
 		t.Errorf("Unexpected error saving session with messages: %v", err)
 	}
-	
+
 	// Verify session was saved
 	sessions, err = storage.ListSessions(tempDir)
 	if err != nil {
@@ -101,26 +101,26 @@ func TestSaveSessionStorageDirectoryOverride(t *testing.T) {
 	// This test validates the behavior when we need to test with a specific directory
 	// In practice, the actual tests above work with the system's session storage
 	tempDir := t.TempDir()
-	
+
 	// Create a session
 	s := NewSession("test", tempDir)
-	
+
 	// Initially it should be dirty
 	if !s.IsDirty() {
 		t.Error("New session should be dirty")
 	}
-	
+
 	// Add a message
 	s.AddMessage(&Message{Role: "user", Content: "test"})
-	
+
 	// Should still be dirty
 	if !s.IsDirty() {
 		t.Error("Session with new message should be dirty")
 	}
-	
+
 	// Mark as saved
 	s.MarkSaved(s.UpdatedAt)
-	
+
 	// Should no longer be dirty
 	if s.IsDirty() {
 		t.Error("Session should not be dirty after being marked as saved")