Summary

This PR refactors chat session management to ensure only one chat session runs at a time and improves tool execution event handling.

Changes

1. Session Management

• Added cleanup in StartChatSession() to end existing sessions before starting new ones
• Prevents multiple concurrent chat sessions which could cause race conditions

2. Event Enhancement

• Extended ToolExecutionCompletedEvent with RequestID and Timestamp fields
• Provides better tracking and correlation of tool execution events

3. Bug Fixes

• Removed duplicate session cleanup in HandleComputerUseResumedEvent
• Changed auto-save from goroutines to synchronous execution to prevent race conditions
• Removed excessive debug logging that was cluttering output

4. Code Cleanup

• Simplified tool preview rendering logic
• Removed unnecessary debug logging statements
• Improved code organization and readability

5. Testing

• Added comprehensive unit tests for tool execution completion events in agent_test_additions.go

Technical Details

Key Changes:

• internal/services/agent.go: Added session cleanup before starting new chat sessions
• internal/domain/ui_events.go: Enhanced ToolExecutionCompletedEvent struct
• internal/services/conversation.go: Fixed auto-save concurrency by removing goroutines
• internal/handlers/chat_handler.go: Removed duplicate session cleanup
• internal/services/agent_test_additions.go: New test file for event handling

Files Modified:

• 11 files changed, 308 insertions(+), 96 deletions(-)

Impact

• Backward Compatible: Yes, maintains existing API and behavior
• Performance: Improved by reducing unnecessary logging and preventing race conditions
• Reliability: Better session lifecycle management prevents potential deadlocks
• Observability: Enhanced event tracking with request IDs and timestamps

Testing

• All existing tests pass
• New unit tests added for tool execution events
• Manual testing confirms single session enforcement works correctly