- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 | - - - - - - - - - - - - - - - - - - - - - - -1x - - - - - - - - - - - - - -1x -71x -71x -71x -71x - -71x - - - - - -70x - - - - - - -10x - - - - - - -4x - - - - - - -42x -42x - - - - - - - - - - - - - - -36x -1x - - - - - - -35x -4x -4x - - - -31x -31x -31x - - -31x - -31x -31x - -20x - -19x - - - - -19x -19x - -1x - - -20x - - -11x - -6x -6x -6x - - -5x -5x -5x - -31x -31x - - - - - - - -31x - - -31x -31x -31x - - - -31x - - - - - -31x - - - - - - -31x -31x -31x - - -31x -31x - - -31x - -52x -6x - - - -46x -25x - - -25x - - - - -25x -25x - - - -46x -20x -20x - - - - -20x - - - - - - - - - - - - - -50x -6x - - - - - - - -2x -2x -2x -2x - - - - - - -2x -1x - - - - - - - - - - - -10x -10x -10x - - - - - - -166x - - - - - - -42x -42x - - - | /**
- * InlineEditService - Business logic for inline AI text editing
- *
- * Responsibilities:
- * - State machine management (INPUT -> LOADING -> RESULT)
- * - LLM streaming integration with cancellation support
- * - Concurrent request blocking
- *
- * Uses the same LLMService infrastructure as the chat system.
- */
-
-import type { LLMService } from './llm/core/LLMService';
-import type {
- InlineEditState,
- InlineEditRequest,
- InlineEditResult,
- InlineEditCallbacks
-} from '../ui/inline-edit/types';
-
-/**
- * System prompt for inline editing operations
- * Instructs the LLM to only return the edited text, no explanations
- */
-const INLINE_EDIT_SYSTEM_PROMPT = `You are a precise text editor. Your task is to modify the given text according to the user's instructions.
-
-Rules:
-1. Return ONLY the modified text - no explanations, no markdown code blocks, no preamble
-2. Preserve the original formatting style (markdown, indentation, etc.) unless instructed otherwise
-3. If the instruction is unclear, make your best interpretation
-4. If the instruction cannot be applied, return the original text unchanged
-
-You will receive:
-- The selected text to edit
-- An instruction for how to modify it
-
-Respond with only the edited text.`;
-
-export class InlineEditService {
- private state: InlineEditState = { phase: 'input', selectedText: '' };
- private abortController: AbortController | null = null;
- private isActive = false;
- private callbacks: InlineEditCallbacks = {};
-
- constructor(private llmService: LLMService) {}
-
- /**
- * Set callbacks for state changes and events
- */
- setCallbacks(callbacks: InlineEditCallbacks): void {
- this.callbacks = callbacks;
- }
-
- /**
- * Get current state
- */
- getState(): InlineEditState {
- return this.state;
- }
-
- /**
- * Check if a generation is currently active
- */
- isGenerating(): boolean {
- return this.isActive;
- }
-
- /**
- * Initialize with selected text (transition to INPUT state)
- */
- initialize(selectedText: string): void {
- this.state = { phase: 'input', selectedText };
- this.notifyStateChange();
- }
-
- /**
- * Generate edited text from instruction
- *
- * State transitions:
- * INPUT -> LOADING -> RESULT (success)
- * INPUT -> LOADING -> ERROR (failure)
- *
- * @param request - The edit request parameters
- * @returns Promise resolving to the edit result
- */
- async generate(request: InlineEditRequest): Promise<InlineEditResult> {
- // Block concurrent requests
- if (this.isActive) {
- return {
- success: false,
- error: 'A generation is already in progress. Please wait or cancel first.'
- };
- }
-
- // Validate instruction
- if (!request.instruction || request.instruction.trim().length === 0) {
- this.transitionToError('Please enter an instruction for how to edit the text.');
- return { success: false, error: 'Empty instruction' };
- }
-
- // Transition to loading state
- this.isActive = true;
- this.state = { phase: 'loading', progress: 'Connecting...', streamedText: '' };
- this.notifyStateChange();
-
- // Create abort controller for cancellation
- this.abortController = new AbortController();
-
- try {
- const result = await this.executeGeneration(request);
-
- if (result.success && result.editedText) {
- // Transition to result state
- this.state = {
- phase: 'result',
- original: request.selectedText,
- edited: result.editedText
- };
- this.notifyStateChange();
- this.callbacks.onComplete?.(result);
- } else {
- this.transitionToError(result.error || 'Unknown error occurred', request.instruction);
- }
-
- return result;
- } catch (error) {
- // Handle abort specifically
- if (error instanceof DOMException && error.name === 'AbortError') {
- // User cancelled - return to input state
- this.state = { phase: 'input', selectedText: request.selectedText };
- this.notifyStateChange();
- return { success: false, error: 'Cancelled by user' };
- }
-
- const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
- this.transitionToError(errorMessage, request.instruction);
- return { success: false, error: errorMessage };
- } finally {
- this.isActive = false;
- this.abortController = null;
- }
- }
-
- /**
- * Execute the LLM generation with streaming
- */
- private async executeGeneration(request: InlineEditRequest): Promise<InlineEditResult> {
- const { selectedText, instruction, modelConfig, context } = request;
-
- // Build user prompt with context
- let userPrompt = `TEXT TO EDIT:\n${selectedText}\n\nINSTRUCTION: ${instruction}`;
- if (context?.fileName) {
- userPrompt = `[File: ${context.fileName}]\n\n${userPrompt}`;
- }
-
- // Build messages array for LLM
- const messages = [
- { role: 'system', content: INLINE_EDIT_SYSTEM_PROMPT },
- { role: 'user', content: userPrompt }
- ];
-
- // Stream options
- const options = {
- provider: modelConfig.provider,
- model: modelConfig.model,
- temperature: 0.3, // Lower temperature for more predictable edits
- abortSignal: this.abortController?.signal
- };
-
- let accumulatedText = '';
- let inputTokens = 0;
- let outputTokens = 0;
-
- // Update progress state
- this.state = { phase: 'loading', progress: 'Generating...', streamedText: '' };
- this.notifyStateChange();
-
- // Stream the response
- for await (const chunk of this.llmService.generateResponseStream(messages, options)) {
- // Check for abort
- if (this.abortController?.signal.aborted) {
- throw new DOMException('Generation aborted by user', 'AbortError');
- }
-
- // Accumulate text
- if (chunk.chunk) {
- accumulatedText += chunk.chunk;
-
- // Update state with streamed text
- this.state = {
- phase: 'loading',
- progress: 'Generating...',
- streamedText: accumulatedText
- };
- this.notifyStateChange();
- this.callbacks.onStreamChunk?.(chunk.chunk);
- }
-
- // Capture usage on completion
- if (chunk.complete && chunk.usage) {
- inputTokens = chunk.usage.promptTokens || 0;
- outputTokens = chunk.usage.completionTokens || 0;
- }
- }
-
- // Return result
- return {
- success: true,
- editedText: accumulatedText.trim(),
- tokenUsage: {
- input: inputTokens,
- output: outputTokens
- }
- };
- }
-
- /**
- * Cancel current generation
- */
- cancel(): void {
- if (this.abortController && this.isActive) {
- this.abortController.abort();
- }
- }
-
- /**
- * Reset to input state (for retry from result)
- */
- reset(selectedText: string, preserveInstruction?: string): void {
- this.cancel();
- this.isActive = false;
- this.state = { phase: 'input', selectedText };
- this.notifyStateChange();
- }
-
- /**
- * Update the edited text (user editing in result state)
- */
- updateEditedText(newText: string): void {
- if (this.state.phase === 'result') {
- this.state = {
- ...this.state,
- edited: newText
- };
- // Don't notify - this is just tracking local edits
- }
- }
-
- /**
- * Transition to error state
- */
- private transitionToError(message: string, lastInstruction?: string): void {
- this.state = { phase: 'error', message, lastInstruction };
- this.notifyStateChange();
- this.callbacks.onError?.(message);
- }
-
- /**
- * Notify callbacks of state change
- */
- private notifyStateChange(): void {
- this.callbacks.onStateChange?.(this.state);
- }
-
- /**
- * Clean up resources
- */
- dispose(): void {
- this.cancel();
- this.callbacks = {};
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| services | -
-
- |
- 100% | -75/75 | -93.1% | -27/29 | -100% | -13/13 | -100% | -75/75 | -
| ui/chat/components | -
-
- |
- 51.83% | -113/218 | -32.14% | -18/56 | -42% | -21/50 | -53.62% | -111/207 | -
| ui/chat/services | -
-
- |
- 70.85% | -141/199 | -72.5% | -58/80 | -57.57% | -19/33 | -72.82% | -134/184 | -
| ui/chat/utils | -
-
- |
- 100% | -38/38 | -100% | -17/17 | -100% | -7/7 | -100% | -34/34 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 | - - - - - - - - - - - - - - - - - - - - - - -1x - - - - - - - - - - - - - -1x -73x -73x -73x -73x - -73x - - - - - -72x - - - - - - -11x - - - - - - -5x - - - - - - -43x -43x - - - - - - - - - - - - - - -37x -1x - - - - - - -36x -4x -4x - - - -32x -32x -32x - - -32x - -32x -32x - -20x - -19x - - - - -19x -19x - -1x - - -20x - - -12x - -6x -6x -6x - - -6x -6x -6x - -32x -32x - - - - - - - -32x - - -32x -32x -32x - - - -32x - - - - - -32x - - - - - - -32x -32x -32x - - -32x -32x - - -32x - -53x -6x - - - -47x -26x - - -26x - - - - -26x -26x - - - -47x -20x -20x - - - - -20x - - - - - - - - - - - - - -51x -6x - - - - - - - -2x -2x -2x -2x - - - - - - -2x -1x - - - - - - - - - - - -11x -11x -11x - - - - - - -171x - - - - - - -43x -43x - - - | /**
- * InlineEditService - Business logic for inline AI text editing
- *
- * Responsibilities:
- * - State machine management (INPUT -> LOADING -> RESULT)
- * - LLM streaming integration with cancellation support
- * - Concurrent request blocking
- *
- * Uses the same LLMService infrastructure as the chat system.
- */
-
-import type { LLMService } from './llm/core/LLMService';
-import type {
- InlineEditState,
- InlineEditRequest,
- InlineEditResult,
- InlineEditCallbacks
-} from '../ui/inline-edit/types';
-
-/**
- * System prompt for inline editing operations
- * Instructs the LLM to only return the edited text, no explanations
- */
-const INLINE_EDIT_SYSTEM_PROMPT = `You are a precise text editor. Your task is to modify the given text according to the user's instructions.
-
-Rules:
-1. Return ONLY the modified text - no explanations, no markdown code blocks, no preamble
-2. Preserve the original formatting style (markdown, indentation, etc.) unless instructed otherwise
-3. If the instruction is unclear, make your best interpretation
-4. If the instruction cannot be applied, return the original text unchanged
-
-You will receive:
-- The selected text to edit
-- An instruction for how to modify it
-
-Respond with only the edited text.`;
-
-export class InlineEditService {
- private state: InlineEditState = { phase: 'input', selectedText: '' };
- private abortController: AbortController | null = null;
- private isActive = false;
- private callbacks: InlineEditCallbacks = {};
-
- constructor(private llmService: LLMService) {}
-
- /**
- * Set callbacks for state changes and events
- */
- setCallbacks(callbacks: InlineEditCallbacks): void {
- this.callbacks = callbacks;
- }
-
- /**
- * Get current state
- */
- getState(): InlineEditState {
- return this.state;
- }
-
- /**
- * Check if a generation is currently active
- */
- isGenerating(): boolean {
- return this.isActive;
- }
-
- /**
- * Initialize with selected text (transition to INPUT state)
- */
- initialize(selectedText: string): void {
- this.state = { phase: 'input', selectedText };
- this.notifyStateChange();
- }
-
- /**
- * Generate edited text from instruction
- *
- * State transitions:
- * INPUT -> LOADING -> RESULT (success)
- * INPUT -> LOADING -> ERROR (failure)
- *
- * @param request - The edit request parameters
- * @returns Promise resolving to the edit result
- */
- async generate(request: InlineEditRequest): Promise<InlineEditResult> {
- // Block concurrent requests
- if (this.isActive) {
- return {
- success: false,
- error: 'A generation is already in progress. Please wait or cancel first.'
- };
- }
-
- // Validate instruction
- if (!request.instruction || request.instruction.trim().length === 0) {
- this.transitionToError('Please enter an instruction for how to edit the text.');
- return { success: false, error: 'Empty instruction' };
- }
-
- // Transition to loading state
- this.isActive = true;
- this.state = { phase: 'loading', progress: 'Connecting...', streamedText: '' };
- this.notifyStateChange();
-
- // Create abort controller for cancellation
- this.abortController = new AbortController();
-
- try {
- const result = await this.executeGeneration(request);
-
- if (result.success && result.editedText) {
- // Transition to result state
- this.state = {
- phase: 'result',
- original: request.selectedText,
- edited: result.editedText
- };
- this.notifyStateChange();
- this.callbacks.onComplete?.(result);
- } else {
- this.transitionToError(result.error || 'Unknown error occurred', request.instruction);
- }
-
- return result;
- } catch (error) {
- // Handle abort specifically
- if (error instanceof DOMException && error.name === 'AbortError') {
- // User cancelled - return to input state
- this.state = { phase: 'input', selectedText: request.selectedText };
- this.notifyStateChange();
- return { success: false, error: 'Cancelled by user' };
- }
-
- const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
- this.transitionToError(errorMessage, request.instruction);
- return { success: false, error: errorMessage };
- } finally {
- this.isActive = false;
- this.abortController = null;
- }
- }
-
- /**
- * Execute the LLM generation with streaming
- */
- private async executeGeneration(request: InlineEditRequest): Promise<InlineEditResult> {
- const { selectedText, instruction, modelConfig, context } = request;
-
- // Build user prompt with context
- let userPrompt = `TEXT TO EDIT:\n${selectedText}\n\nINSTRUCTION: ${instruction}`;
- if (context?.fileName) {
- userPrompt = `[File: ${context.fileName}]\n\n${userPrompt}`;
- }
-
- // Build messages array for LLM
- const messages = [
- { role: 'system', content: INLINE_EDIT_SYSTEM_PROMPT },
- { role: 'user', content: userPrompt }
- ];
-
- // Stream options
- const options = {
- provider: modelConfig.provider,
- model: modelConfig.model,
- temperature: 0.3, // Lower temperature for more predictable edits
- abortSignal: this.abortController?.signal
- };
-
- let accumulatedText = '';
- let inputTokens = 0;
- let outputTokens = 0;
-
- // Update progress state
- this.state = { phase: 'loading', progress: 'Generating...', streamedText: '' };
- this.notifyStateChange();
-
- // Stream the response
- for await (const chunk of this.llmService.generateResponseStream(messages, options)) {
- // Check for abort
- if (this.abortController?.signal.aborted) {
- throw new DOMException('Generation aborted by user', 'AbortError');
- }
-
- // Accumulate text
- if (chunk.chunk) {
- accumulatedText += chunk.chunk;
-
- // Update state with streamed text
- this.state = {
- phase: 'loading',
- progress: 'Generating...',
- streamedText: accumulatedText
- };
- this.notifyStateChange();
- this.callbacks.onStreamChunk?.(chunk.chunk);
- }
-
- // Capture usage on completion
- if (chunk.complete && chunk.usage) {
- inputTokens = chunk.usage.promptTokens || 0;
- outputTokens = chunk.usage.completionTokens || 0;
- }
- }
-
- // Return result
- return {
- success: true,
- editedText: accumulatedText.trim(),
- tokenUsage: {
- input: inputTokens,
- output: outputTokens
- }
- };
- }
-
- /**
- * Cancel current generation
- */
- cancel(): void {
- if (this.abortController && this.isActive) {
- this.abortController.abort();
- }
- }
-
- /**
- * Reset to input state (for retry from result)
- */
- reset(selectedText: string): void {
- this.cancel();
- this.isActive = false;
- this.state = { phase: 'input', selectedText };
- this.notifyStateChange();
- }
-
- /**
- * Update the edited text (user editing in result state)
- */
- updateEditedText(newText: string): void {
- if (this.state.phase === 'result') {
- this.state = {
- ...this.state,
- edited: newText
- };
- // Don't notify - this is just tracking local edits
- }
- }
-
- /**
- * Transition to error state
- */
- private transitionToError(message: string, lastInstruction?: string): void {
- this.state = { phase: 'error', message, lastInstruction };
- this.notifyStateChange();
- this.callbacks.onError?.(message);
- }
-
- /**
- * Notify callbacks of state change
- */
- private notifyStateChange(): void {
- this.callbacks.onStateChange?.(this.state);
- }
-
- /**
- * Clean up resources
- */
- dispose(): void {
- this.cancel();
- this.callbacks = {};
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| InlineEditService.ts | -
-
- |
- 100% | -75/75 | -93.1% | -27/29 | -100% | -13/13 | -100% | -75/75 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 | - - - - - - - -1x - - - - - - -1x - - - - -9x - - - -9x -9x - -9x -9x -9x - - - - - - -9x - -9x - - - - - - -9x - - -9x -9x - -9x - - - - - - -9x - - -9x -9x -9x -9x - - - - - - -5x -5x - - - - - - -5x -2x -2x - - -3x -3x - - -3x -3x - - -3x - - - - - - -3x -3x - -3x -3x - - -3x -3x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -12x - - - - - - -5x -4x - - - - - - -3x -3x - - - - - - -11x -11x - - - - - - -4x - -2x -2x - -2x - - - - - - - - - - - - - - - - - - - - -2x -2x - - | /**
- * MessageBranchNavigator - UI component for navigating between message branches
- *
- * Shows mini "< X/Y >" navigation for individual messages that have multiple branches
- * Only displays when message has branches (human or subagent)
- */
-
-import { ConversationMessage } from '../../../types/chat/ChatTypes';
-import { setIcon, Component } from 'obsidian';
-
-export interface MessageBranchNavigatorEvents {
- onAlternativeChanged: (messageId: string, alternativeIndex: number) => void;
- onError: (message: string) => void;
-}
-
-export class MessageBranchNavigator {
- private container: HTMLElement;
- private branchIndicator!: HTMLElement;
- private prevButton!: HTMLButtonElement;
- private nextButton!: HTMLButtonElement;
- private currentMessage: ConversationMessage | null = null;
-
- constructor(
- container: HTMLElement,
- private events: MessageBranchNavigatorEvents,
- private component?: Component
- ) {
- this.container = container;
- this.createBranchNavigator();
- this.hide(); // Hidden by default
- }
-
- /**
- * Create the mini branch navigation UI
- */
- private createBranchNavigator(): void {
- this.container.addClass('message-branch-navigator');
-
- this.prevButton = this.container.createEl('button', {
- cls: 'message-action-btn message-branch-prev clickable-icon',
- attr: {
- 'aria-label': 'Previous alternative',
- 'title': 'Go to previous alternative response'
- }
- });
- setIcon(this.prevButton, 'chevron-left');
-
- // Branch indicator (shows current/total like "2/4")
- this.branchIndicator = this.container.createDiv('message-branch-indicator');
- this.branchIndicator.textContent = '1/1';
-
- this.nextButton = this.container.createEl('button', {
- cls: 'message-action-btn message-branch-next clickable-icon',
- attr: {
- 'aria-label': 'Next alternative',
- 'title': 'Go to next alternative response'
- }
- });
- setIcon(this.nextButton, 'chevron-right');
-
- // Event listeners
- const prevHandler = () => this.handlePreviousAlternative();
- const nextHandler = () => this.handleNextAlternative();
- this.component!.registerDomEvent(this.prevButton, 'click', prevHandler);
- this.component!.registerDomEvent(this.nextButton, 'click', nextHandler);
- }
-
- /**
- * Update the navigator for a message
- */
- updateMessage(message: ConversationMessage): void {
- this.currentMessage = message;
- this.updateDisplay();
- }
-
- /**
- * Update the display based on current message
- */
- private updateDisplay(): void {
- if (!this.currentMessage || !this.hasAlternatives()) {
- this.hide();
- return;
- }
-
- const alternativeCount = this.getAlternativeCount();
- const currentIndex = this.currentMessage.activeAlternativeIndex || 0;
-
- // Show and update the indicator (1-based display)
- this.show();
- this.branchIndicator.textContent = `${currentIndex + 1}/${alternativeCount}`;
-
- // Update button states
- this.updateButtonStates(currentIndex, alternativeCount);
- }
-
- /**
- * Update navigation button states
- */
- private updateButtonStates(currentIndex: number, totalCount: number): void {
- const isFirst = currentIndex === 0;
- const isLast = currentIndex === totalCount - 1;
-
- this.prevButton.disabled = isFirst;
- this.nextButton.disabled = isLast;
-
- // Update visual states
- this.prevButton.toggleClass('disabled', isFirst);
- this.nextButton.toggleClass('disabled', isLast);
- }
-
- /**
- * Handle previous alternative navigation
- */
- private async handlePreviousAlternative(): Promise<void> {
- Iif (!this.currentMessage) return;
-
- const currentIndex = this.currentMessage.activeAlternativeIndex || 0;
- Iif (currentIndex <= 0) return;
-
- const newIndex = currentIndex - 1;
- this.events.onAlternativeChanged(this.currentMessage.id, newIndex);
- this.updateDisplay();
- }
-
- /**
- * Handle next alternative navigation
- */
- private async handleNextAlternative(): Promise<void> {
- Iif (!this.currentMessage) return;
-
- const currentIndex = this.currentMessage.activeAlternativeIndex || 0;
- const totalCount = this.getAlternativeCount();
- Iif (currentIndex >= totalCount - 1) return;
-
- const newIndex = currentIndex + 1;
- this.events.onAlternativeChanged(this.currentMessage.id, newIndex);
- this.updateDisplay();
- }
-
- /**
- * Check if current message has branches
- */
- private hasAlternatives(): boolean {
- return !!(this.currentMessage?.branches && this.currentMessage.branches.length > 0);
- }
-
- /**
- * Get total branch count (including the original message)
- */
- private getAlternativeCount(): number {
- if (!this.hasAlternatives()) return 1;
- return (this.currentMessage!.branches!.length) + 1; // +1 for original message
- }
-
- /**
- * Show the navigator
- */
- private show(): void {
- this.container.removeClass('message-branch-navigator-hidden');
- this.container.addClass('message-branch-navigator-visible');
- }
-
- /**
- * Hide the navigator
- */
- private hide(): void {
- this.container.removeClass('message-branch-navigator-visible');
- this.container.addClass('message-branch-navigator-hidden');
- }
-
- /**
- * Get current alternative information for external use
- */
- getCurrentAlternativeInfo(): { current: number; total: number; hasAlternatives: boolean } | null {
- if (!this.currentMessage) return null;
-
- const currentIndex = this.currentMessage.activeAlternativeIndex || 0;
- const totalCount = this.getAlternativeCount();
-
- return {
- current: currentIndex + 1, // 1-based for display
- total: totalCount,
- hasAlternatives: this.hasAlternatives()
- };
- }
-
- /**
- * Check if navigator is currently visible
- */
- isVisible(): boolean {
- return this.container.hasClass('message-branch-navigator-visible');
- }
-
- /**
- * Clean up resources.
- * Note: Event listeners registered via component.registerDomEvent() are
- * automatically cleaned up when the Obsidian Component unloads, so no
- * manual removeEventListener calls are needed here.
- */
- destroy(): void {
- this.container.empty();
- this.currentMessage = null;
- }
-} |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 -285 -286 -287 -288 -289 -290 -291 -292 -293 -294 -295 -296 -297 -298 -299 -300 -301 -302 -303 -304 -305 -306 -307 -308 -309 -310 -311 -312 -313 -314 -315 -316 -317 -318 -319 -320 -321 -322 -323 -324 -325 -326 -327 -328 -329 -330 -331 -332 -333 -334 -335 -336 -337 -338 -339 -340 -341 -342 -343 -344 -345 -346 -347 -348 -349 -350 -351 -352 -353 -354 -355 -356 -357 -358 -359 -360 -361 -362 -363 -364 -365 -366 -367 -368 -369 -370 -371 -372 -373 -374 -375 -376 -377 -378 -379 -380 -381 -382 -383 -384 -385 -386 -387 -388 -389 -390 -391 -392 -393 -394 -395 -396 -397 -398 -399 -400 -401 -402 -403 -404 -405 -406 -407 | - - - - - - -1x - -1x - -1x -6x -6x -6x - - -6x -6x -6x -6x -6x -6x -6x -6x - -6x - - - - - - - - - -7x -7x -7x - - -7x -7x -7x -7x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -6x -6x - -6x -6x - -6x -6x - - -6x - - - - - - - - - - - -13x -2x - -13x - -13x -13x - -13x -6x -6x - - - -7x - - -7x -13x -13x - - -7x - - - - - - - -13x - - - - - - - -13x - - - - - - - - - - -13x - -13x - - - -13x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -4x - - - - - - -1x -1x - -1x -1x - - -1x - -1x -1x -1x - - - - - - - - - - - - - - - - - - - - -14x -14x -14x - - - - - - - - - - - - - - - - - - - - - - - - - -1x -2x - -1x -1x - - - | /**
- * MessageDisplay - Main chat message display area
- *
- * Shows conversation messages with user/AI bubbles and tool execution displays
- */
-
-import { ConversationData, ConversationMessage } from '../../../types/chat/ChatTypes';
-import { MessageBubble } from './MessageBubble';
-import { BranchManager } from '../services/BranchManager';
-import { App, setIcon, ButtonComponent } from 'obsidian';
-
-export class MessageDisplay {
- private conversation: ConversationData | null = null;
- private currentConversationId: string | null = null;
- private messageBubbles: Map<string, MessageBubble> = new Map();
-
- constructor(
- private container: HTMLElement,
- private app: App,
- private branchManager: BranchManager,
- private onRetryMessage?: (messageId: string) => void,
- private onEditMessage?: (messageId: string, newContent: string) => void,
- private onToolEvent?: (messageId: string, event: 'detected' | 'updated' | 'started' | 'completed', data: any) => void,
- private onMessageAlternativeChanged?: (messageId: string, alternativeIndex: number) => void,
- private onViewBranch?: (branchId: string) => void
- ) {
- this.render();
- }
-
- /**
- * Set conversation to display.
- * Uses incremental reconciliation when updating the same conversation (preserves
- * live progressive tool accordions, branch navigator state, and avoids flicker).
- * Falls back to full render when switching to a different conversation.
- */
- setConversation(conversation: ConversationData): void {
- const previousConversationId = this.currentConversationId;
- this.conversation = conversation;
- this.currentConversationId = conversation.id;
-
- // Full render for conversation switches or first load
- if (previousConversationId !== conversation.id) {
- this.render();
- this.scrollToBottom();
- return;
- }
-
- // Incremental reconciliation for same conversation updates
- this.reconcile(conversation);
- this.scrollToBottom();
- }
-
- /**
- * Incrementally reconcile the displayed messages with the new conversation data.
- * Reuses existing MessageBubble instances for messages that still exist,
- * removes stale ones, and creates new ones -- preserving live UI state.
- */
- private reconcile(conversation: ConversationData): void {
- const messagesContainer = this.container.querySelector('.messages-container');
- Iif (!messagesContainer) {
- // No messages container yet (e.g., was showing welcome) -- fall back to full render
- this.render();
- return;
- }
-
- const newMessages = conversation.messages;
- const newMessageIds = new Set(newMessages.map(m => m.id));
-
- // 1. Remove stale bubbles (messages no longer in conversation)
- for (const [id, bubble] of this.messageBubbles) {
- Iif (!newMessageIds.has(id)) {
- const element = bubble.getElement();
- Iif (element) {
- element.remove();
- }
- bubble.cleanup();
- this.messageBubbles.delete(id);
- }
- }
-
- // 2. Walk new messages in order: update existing, create new, ensure DOM order
- let previousElement: Element | null = null;
- for (const message of newMessages) {
- const existingBubble = this.messageBubbles.get(message.id);
-
- if (existingBubble) {
- // Update the existing bubble in place
- existingBubble.updateWithNewMessage(message);
- const element = existingBubble.getElement();
-
- // Ensure DOM order: element should follow previousElement
- Iif (element) {
- const expectedNext: Element | null = previousElement ? previousElement.nextElementSibling : messagesContainer.firstElementChild;
- Iif (element !== expectedNext) {
- if (previousElement) {
- previousElement.after(element);
- } else {
- messagesContainer.prepend(element);
- }
- }
- previousElement = element;
- }
- } else {
- // Create a new bubble for this message
- const bubbleEl = this.createMessageBubble(message);
-
- // Insert at the correct position
- if (previousElement) {
- previousElement.after(bubbleEl);
- } else {
- messagesContainer.prepend(bubbleEl);
- }
- previousElement = bubbleEl;
- }
- }
- }
-
- /**
- * Add a user message immediately (for optimistic updates)
- */
- addUserMessage(content: string): void {
- const message: ConversationMessage = {
- id: `temp_${Date.now()}`,
- role: 'user',
- content,
- timestamp: Date.now(),
- conversationId: this.conversation?.id || 'unknown'
- };
-
- const bubble = this.createMessageBubble(message);
- const messagesContainer = this.container.querySelector('.messages-container');
- Iif (messagesContainer) {
- messagesContainer.appendChild(bubble);
- }
- this.scrollToBottom();
- }
-
- /**
- * Add a message immediately using the actual message object (prevents duplicate message creation)
- */
- addMessage(message: ConversationMessage): void {
- const bubble = this.createMessageBubble(message);
- this.container.querySelector('.messages-container')?.appendChild(bubble);
- this.scrollToBottom();
- }
-
- /**
- * Add an AI message immediately (for streaming setup)
- */
- addAIMessage(message: ConversationMessage): void {
- const bubble = this.createMessageBubble(message);
- this.container.querySelector('.messages-container')?.appendChild(bubble);
- this.scrollToBottom();
- }
-
- /**
- * Update a specific message content for final display (streaming handled by StreamingController)
- */
- updateMessageContent(messageId: string, content: string): void {
- const messageBubble = this.messageBubbles.get(messageId);
- Iif (messageBubble) {
- messageBubble.updateContent(content);
- }
- }
-
- /**
- * Update a specific message with new data (including tool calls) without full re-render
- */
- updateMessage(messageId: string, updatedMessage: ConversationMessage): void {
- Iif (!this.conversation) {
- return;
- }
-
- // Update the message in conversation data
- const messageIndex = this.conversation.messages.findIndex(msg => msg.id === messageId);
- Iif (messageIndex !== -1) {
- this.conversation.messages[messageIndex] = updatedMessage;
- }
-
- // Update the bubble in place
- const messageBubble = this.messageBubbles.get(messageId);
- Iif (messageBubble) {
- messageBubble.updateWithNewMessage(updatedMessage);
- }
- }
-
- /**
- * Escape HTML for safe display
- */
- private escapeHtml(text: string): string {
- const div = document.createElement('div');
- div.textContent = text;
- return div.innerHTML;
- }
-
-
- /**
- * Show welcome state
- */
- showWelcome(): void {
- this.container.empty();
- this.container.addClass('message-display');
-
- const welcome = this.container.createDiv('chat-welcome');
- const welcomeContent = welcome.createDiv('chat-welcome-content');
-
- const welcomeIcon = welcomeContent.createDiv('chat-welcome-icon');
- setIcon(welcomeIcon, 'message-circle');
-
- // Use Obsidian's ButtonComponent
- new ButtonComponent(welcomeContent)
- .setButtonText('New conversation')
- .setIcon('plus')
- .setClass('chat-welcome-button');
- }
-
- /**
- * Full render - destroys all existing bubbles and rebuilds from scratch.
- * Used for conversation switches and initial load.
- */
- private render(): void {
- // Cleanup all existing bubbles before clearing the DOM
- for (const bubble of this.messageBubbles.values()) {
- bubble.cleanup();
- }
- this.messageBubbles.clear();
-
- this.container.empty();
- this.container.addClass('message-display');
-
- if (!this.conversation) {
- this.showWelcome();
- return;
- }
-
- // Create scrollable messages container
- const messagesContainer = this.container.createDiv('messages-container');
-
- // Render all messages (no branch filtering needed for message-level alternatives)
- this.conversation.messages.forEach((message) => {
- const messageEl = this.createMessageBubble(message);
- messagesContainer.appendChild(messageEl);
- });
-
- this.scrollToBottom();
- }
-
- /**
- * Create a message bubble element
- */
- private createMessageBubble(message: ConversationMessage): HTMLElement {
- // Render using the currently active alternative content/tool calls so branch selection persists across re-renders
- const displayMessage = this.branchManager
- ? {
- ...message,
- content: this.branchManager.getActiveMessageContent(message),
- toolCalls: this.branchManager.getActiveMessageToolCalls(message)
- }
- : message;
-
- const bubble = new MessageBubble(
- displayMessage,
- this.app,
- (messageId: string) => this.onCopyMessage(messageId),
- (messageId: string) => this.handleRetryMessage(messageId),
- (messageId: string, newContent: string) => this.handleEditMessage(messageId, newContent),
- this.onToolEvent,
- this.onMessageAlternativeChanged ? (messageId: string, alternativeIndex: number) => this.handleMessageAlternativeChanged(messageId, alternativeIndex) : undefined,
- this.onViewBranch
- );
-
- this.messageBubbles.set(message.id, bubble);
-
- const bubbleEl = bubble.createElement();
-
- // Tool accordion is now rendered inside MessageBubble's content area
-
- return bubbleEl;
- }
-
- /**
- * Handle copy message action
- */
- private onCopyMessage(messageId: string): void {
- const message = this.findMessage(messageId);
- Iif (message) {
- navigator.clipboard.writeText(message.content).then(() => {
- // Message copied to clipboard
- }).catch(err => {
- // Failed to copy message
- });
- }
- }
-
- /**
- * Handle retry message action
- */
- private handleRetryMessage(messageId: string): void {
- Iif (this.onRetryMessage) {
- this.onRetryMessage(messageId);
- }
- }
-
- /**
- * Handle edit message action
- */
- private handleEditMessage(messageId: string, newContent: string): void {
- Iif (this.onEditMessage) {
- this.onEditMessage(messageId, newContent);
- }
- }
-
- /**
- * Handle message alternative changed action
- */
- private handleMessageAlternativeChanged(messageId: string, alternativeIndex: number): void {
- Iif (this.onMessageAlternativeChanged) {
- this.onMessageAlternativeChanged(messageId, alternativeIndex);
- }
- }
-
- /**
- * Find message by ID
- */
- private findMessage(messageId: string): ConversationMessage | undefined {
- return this.conversation?.messages.find(msg => msg.id === messageId);
- }
-
- /**
- * Find MessageBubble by messageId for tool events
- */
- findMessageBubble(messageId: string): MessageBubble | undefined {
- return this.messageBubbles.get(messageId);
- }
-
- /**
- * Update MessageBubble with new message ID (for handling temporary -> real ID updates)
- */
- updateMessageId(oldId: string, newId: string, updatedMessage: ConversationMessage): void {
- const messageBubble = this.messageBubbles.get(oldId);
- if (messageBubble) {
- // Re-key the bubble in the Map under the new ID
- this.messageBubbles.delete(oldId);
- this.messageBubbles.set(newId, messageBubble);
-
- // Update the MessageBubble's message reference and DOM attribute
- messageBubble.updateWithNewMessage(updatedMessage);
-
- const element = messageBubble.getElement();
- if (element) {
- element.setAttribute('data-message-id', newId);
- }
- }
- }
-
- /**
- * Check if any message bubbles have progressive tool accordions
- */
- hasProgressiveToolAccordions(): boolean {
- for (const bubble of this.messageBubbles.values()) {
- Iif (bubble.getProgressiveToolAccordions().size > 0) {
- return true;
- }
- }
- return false;
- }
-
- /**
- * Scroll to bottom of messages
- */
- private scrollToBottom(): void {
- const messagesContainer = this.container.querySelector('.messages-container');
- if (messagesContainer) {
- messagesContainer.scrollTop = messagesContainer.scrollHeight;
- }
- }
-
- /**
- * Get current scroll position
- */
- getScrollPosition(): number {
- const messagesContainer = this.container.querySelector('.messages-container');
- return messagesContainer?.scrollTop ?? 0;
- }
-
- /**
- * Set scroll position
- */
- setScrollPosition(position: number): void {
- const messagesContainer = this.container.querySelector('.messages-container');
- Iif (messagesContainer) {
- messagesContainer.scrollTop = position;
- }
- }
-
- /**
- * Cleanup resources
- */
- cleanup(): void {
- for (const bubble of this.messageBubbles.values()) {
- bubble.cleanup();
- }
- this.messageBubbles.clear();
- this.currentConversationId = null;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| MessageBranchNavigator.ts | -
-
- |
- 71.42% | -50/70 | -57.89% | -11/19 | -68.75% | -11/16 | -77.41% | -48/62 | -
| MessageDisplay.ts | -
-
- |
- 42.56% | -63/148 | -18.91% | -7/37 | -29.41% | -10/34 | -43.44% | -63/145 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 -285 -286 -287 -288 -289 -290 -291 -292 -293 -294 -295 -296 -297 -298 -299 -300 -301 -302 -303 -304 -305 -306 -307 -308 -309 -310 -311 -312 -313 -314 -315 -316 -317 -318 -319 -320 -321 -322 -323 -324 -325 -326 -327 -328 -329 -330 -331 -332 -333 -334 -335 -336 -337 -338 -339 -340 -341 -342 -343 -344 -345 -346 -347 -348 -349 -350 -351 -352 -353 -354 -355 -356 -357 -358 | - - - - - - - - - - - - - - - - - - - - - -1x - -26x -26x - - - - - - - - - - - -4x - -8x -4x -1x -1x - - -3x - - -3x -3x - - - -3x -3x - -3x - - - -3x - - - - - - - - - - -3x - - -3x - - -3x - - - -3x - -3x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -1x -2x -1x - - - -1x -1x - -1x - - - -1x -1x - - - - - - - - - - - - - - -3x -1x - - -4x -2x -1x - - -1x -1x -1x - - - - - - - - - -15x - - -15x -6x - - -9x -9x -8x - - -1x - - - - - - -3x -3x -2x -1x - - - - -1x - -1x - - - - - - -4x -4x -2x -1x - - - - -1x - -2x - - - - - - -2x -2x -2x -1x - - -1x - - - - - - - - - - - - - - -2x -2x -2x - -2x - -2x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -2x -2x - - - - - - -2x -2x -2x - - - | /**
- * BranchManager - Handles message-level branching operations
- *
- * Manages creating and switching between branches for individual messages.
- * Works with the unified branch model where both human alternatives and
- * subagent branches share the same data structure.
- *
- * Human branches: inheritContext=true (includes parent context)
- * Subagent branches: inheritContext=false (fresh start)
- */
-
-// import { ConversationRepository } from '../../../database/services/chat/ConversationRepository';
-type ConversationRepository = any;
-import { ConversationData, ConversationMessage } from '../../../types/chat/ChatTypes';
-import type { ConversationBranch, HumanBranchMetadata } from '../../../types/branch/BranchTypes';
-
-export interface BranchManagerEvents {
- onBranchCreated: (messageId: string, branchId: string) => void;
- onBranchSwitched: (messageId: string, branchId: string) => void;
- onError: (message: string) => void;
-}
-
-export class BranchManager {
- constructor(
- private conversationRepo: ConversationRepository,
- private events: BranchManagerEvents
- ) {}
-
- /**
- * Create a human branch (alternative response) for a specific message
- */
- async createHumanBranch(
- conversation: ConversationData,
- messageId: string,
- alternativeResponse: ConversationMessage,
- description?: string
- ): Promise<string | null> {
- try {
- // Find the message in the conversation
- const messageIndex = conversation.messages.findIndex((msg) => msg.id === messageId);
- if (messageIndex === -1) {
- console.error('[BranchManager] Message not found:', messageId);
- return null;
- }
-
- const message = conversation.messages[messageIndex];
-
- // Initialize branches array if it doesn't exist
- if (!message.branches) {
- message.branches = [];
- }
-
- // Create the new branch
- const now = Date.now();
- const branchId = `branch-${now}-${Math.random().toString(36).substring(2, 9)}`;
-
- const metadata: HumanBranchMetadata = {
- description: description || `Alternative response ${message.branches.length + 1}`,
- };
-
- const newBranch: ConversationBranch = {
- id: branchId,
- type: 'human',
- inheritContext: true,
- messages: [alternativeResponse],
- created: now,
- updated: now,
- metadata,
- };
-
- // Add the new branch
- message.branches.push(newBranch);
-
- // Set the new branch as active
- message.activeAlternativeIndex = message.branches.length; // 1-based (0 = original)
-
- // Save the updated conversation to repository
- await this.conversationRepo.updateConversation(conversation.id, {
- messages: conversation.messages,
- });
-
- this.events.onBranchCreated(messageId, branchId);
-
- return branchId;
- } catch (error) {
- console.error('[BranchManager] Failed to create branch:', error);
- this.events.onError('Failed to create alternative response');
- return null;
- }
- }
-
- /**
- * Switch to a specific branch by ID
- */
- async switchToBranch(
- conversation: ConversationData,
- messageId: string,
- branchId: string
- ): Promise<boolean> {
- try {
- // Find the message in the conversation
- const messageIndex = conversation.messages.findIndex((msg) => msg.id === messageId);
- Iif (messageIndex === -1) {
- console.error('[BranchManager] Message not found:', messageId);
- return false;
- }
-
- const message = conversation.messages[messageIndex];
-
- // Find the branch index
- Iif (!message.branches) {
- console.error('[BranchManager] No branches on message:', messageId);
- return false;
- }
-
- const branchIndex = message.branches.findIndex((b) => b.id === branchId);
- Iif (branchIndex === -1) {
- console.error('[BranchManager] Branch not found:', branchId);
- return false;
- }
-
- // Update the active alternative index
- // activeAlternativeIndex: 0 = original, 1+ = branch index + 1
- message.activeAlternativeIndex = branchIndex + 1;
-
- // Save the updated conversation to repository
- await this.conversationRepo.updateConversation(conversation.id, {
- messages: conversation.messages,
- });
-
- this.events.onBranchSwitched(messageId, branchId);
-
- return true;
- } catch (error) {
- console.error('[BranchManager] Failed to switch branch:', error);
- this.events.onError('Failed to switch to branch');
- return false;
- }
- }
-
- /**
- * Switch to original message (no branch)
- */
- async switchToOriginal(
- conversation: ConversationData,
- messageId: string
- ): Promise<boolean> {
- try {
- const messageIndex = conversation.messages.findIndex((msg) => msg.id === messageId);
- Iif (messageIndex === -1) {
- return false;
- }
-
- const message = conversation.messages[messageIndex];
- message.activeAlternativeIndex = 0;
-
- await this.conversationRepo.updateConversation(conversation.id, {
- messages: conversation.messages,
- });
-
- this.events.onBranchSwitched(messageId, 'original');
- return true;
- } catch (error) {
- console.error('[BranchManager] Failed to switch to original:', error);
- return false;
- }
- }
-
- /**
- * Switch to a branch by index (0 = original, 1+ = branch index)
- */
- async switchToBranchByIndex(
- conversation: ConversationData,
- messageId: string,
- index: number
- ): Promise<boolean> {
- if (index === 0) {
- return this.switchToOriginal(conversation, messageId);
- }
-
- const message = conversation.messages.find((msg) => msg.id === messageId);
- if (!message?.branches) {
- return false;
- }
-
- const branchIndex = index - 1;
- if (branchIndex < 0 || branchIndex >= message.branches.length) {
- return false;
- }
-
- return this.switchToBranch(conversation, messageId, message.branches[branchIndex].id);
- }
-
- /**
- * Get the currently active branch for a message
- */
- getActiveBranch(message: ConversationMessage): ConversationBranch | null {
- const activeIndex = message.activeAlternativeIndex || 0;
-
- // Index 0 is the original message
- if (activeIndex === 0 || !message.branches) {
- return null;
- }
-
- const branchIndex = activeIndex - 1;
- if (branchIndex >= 0 && branchIndex < message.branches.length) {
- return message.branches[branchIndex];
- }
-
- return null;
- }
-
- /**
- * Get the currently active message content (original or from branch)
- */
- getActiveMessageContent(message: ConversationMessage): string {
- const branch = this.getActiveBranch(message);
- if (branch) {
- if (branch.messages.length > 0) {
- return branch.messages[branch.messages.length - 1].content;
- }
- // Active branch exists but has no messages yet (e.g. still loading).
- // Return empty string rather than falling through to the original message
- // content, which would be misleading since the user selected this branch.
- return '';
- }
- return message.content;
- }
-
- /**
- * Get the currently active message tool calls
- */
- getActiveMessageToolCalls(message: ConversationMessage): any[] | undefined {
- const branch = this.getActiveBranch(message);
- if (branch) {
- if (branch.messages.length > 0) {
- return branch.messages[branch.messages.length - 1].toolCalls;
- }
- // Active branch exists but has no messages -- return undefined (no data)
- // rather than the original message's tool calls which belong to a
- // different alternative.
- return undefined;
- }
- return message.toolCalls;
- }
-
- /**
- * Get the currently active message reasoning
- */
- getActiveMessageReasoning(message: ConversationMessage): string | undefined {
- const branch = this.getActiveBranch(message);
- if (branch) {
- if (branch.messages.length > 0) {
- return branch.messages[branch.messages.length - 1].reasoning;
- }
- // Active branch exists but has no messages -- return undefined
- return undefined;
- }
- return message.reasoning;
- }
-
- /**
- * Get branch navigation info for a message
- */
- getBranchInfo(message: ConversationMessage): {
- current: number;
- total: number;
- hasBranches: boolean;
- activeBranchId?: string;
- activeBranchType?: 'human' | 'subagent';
- } {
- const activeIndex = message.activeAlternativeIndex || 0;
- const branchCount = message.branches?.length || 0;
- const total = branchCount + 1; // +1 for original
-
- const branch = this.getActiveBranch(message);
-
- return {
- current: activeIndex + 1, // 1-based for display
- total,
- hasBranches: branchCount > 0,
- activeBranchId: branch?.id,
- activeBranchType: branch?.type,
- };
- }
-
- /**
- * Check if a message has any branches
- */
- hasBranches(message: ConversationMessage): boolean {
- return (message.branches?.length || 0) > 0;
- }
-
- /**
- * Get all branches for a message
- */
- getBranches(message: ConversationMessage): ConversationBranch[] {
- return message.branches || [];
- }
-
- /**
- * Get branch by ID from a message
- */
- getBranchById(message: ConversationMessage, branchId: string): ConversationBranch | null {
- Iif (!message.branches) {
- return null;
- }
- return message.branches.find((b) => b.id === branchId) || null;
- }
-
- /**
- * Check if any branch on a message is a subagent branch
- */
- hasSubagentBranches(message: ConversationMessage): boolean {
- Iif (!message.branches) {
- return false;
- }
- return message.branches.some((b) => b.type === 'subagent');
- }
-
- /**
- * Get only subagent branches for a message
- */
- getSubagentBranches(message: ConversationMessage): ConversationBranch[] {
- Iif (!message.branches) {
- return [];
- }
- return message.branches.filter((b) => b.type === 'subagent');
- }
-
- /**
- * Get only human branches for a message
- */
- getHumanBranches(message: ConversationMessage): ConversationBranch[] {
- Iif (!message.branches) {
- return [];
- }
- return message.branches.filter((b) => b.type === 'human');
- }
-
- /**
- * Navigate to previous branch/original
- */
- getPreviousIndex(message: ConversationMessage): number | null {
- const currentIndex = message.activeAlternativeIndex || 0;
- return currentIndex > 0 ? currentIndex - 1 : null;
- }
-
- /**
- * Navigate to next branch
- */
- getNextIndex(message: ConversationMessage): number | null {
- const currentIndex = message.activeAlternativeIndex || 0;
- const total = (message.branches?.length || 0) + 1;
- return currentIndex < total - 1 ? currentIndex + 1 : null;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 | - - - - - - - - - - - - - - - - - - - - - - - -1x - - - - - - - - - - - - - - - - - - - -1x -22x -22x - - -22x - - -22x -22x -22x -22x -22x - - - - - - - - - - - - - - - - - - - - -23x -1x - - -42x -22x - -21x -21x - - -20x -19x -19x - - -19x - -19x -19x - - -19x -19x -19x - -19x - - - - - - - - - - -19x - - - - - - - - -19x -19x - - - - - - - - - - -19x -19x -19x -19x -19x - - - -19x - - -19x - - -19x -19x - - -19x - - - - -19x - - - - - - - - - - - -15x - - -15x - - -4x - - -2x -2x -2x - -2x - -1x -1x -1x - -1x -1x - - -1x -1x -1x - -1x - - - -2x - -2x - - -19x -19x -19x -19x - - - - - - - -1x - - - - - - - - - - -1x - - - | /**
- * Location: /src/ui/chat/services/MessageAlternativeService.ts
- *
- * Purpose: Handles creation of alternative AI responses for message branching
- * Extracted from MessageManager.ts to follow Single Responsibility Principle
- *
- * Flow:
- * 1. Save the original content as a branch (preserving old response)
- * 2. Clear the current message content and set loading state
- * 3. Fire UI update so the user sees a cleared message with loading indicator
- * 4. Stream the new response directly into the live conversation message
- * 5. On complete: update message, fire final events
- * 6. On abort: keep partial content (original is safe in the branch)
- * 7. Branch arrows allow navigation between original (branch) and new (current)
- *
- * Used by: MessageManager for retry and alternative response generation
- * Dependencies: ChatService, BranchManager, MessageStreamHandler
- */
-
-import { ChatService } from '../../../services/chat/ChatService';
-import { ConversationData, ConversationMessage } from '../../../types/chat/ChatTypes';
-import { BranchManager } from './BranchManager';
-import { MessageStreamHandler } from './MessageStreamHandler';
-import { AbortHandler } from '../utils/AbortHandler';
-import { filterCompletedToolCalls } from '../utils/toolCallUtils';
-
-export interface MessageAlternativeServiceEvents {
- onStreamingUpdate: (messageId: string, content: string, isComplete: boolean, isIncremental?: boolean) => void;
- onConversationUpdated: (conversation: ConversationData) => void;
- onToolCallsDetected: (messageId: string, toolCalls: any[]) => void;
- onLoadingStateChanged: (isLoading: boolean) => void;
- onError: (message: string) => void;
-}
-
-/**
- * Service for creating alternative AI responses when retrying messages.
- *
- * Clear-and-restream flow:
- * 1. Save original content into a branch (preserves old response)
- * 2. Clear message content and stream new response fresh
- * 3. On success: message has new content, branch has old content
- * 4. On abort: keep partial new content (original safe in branch)
- * 5. Branch arrows navigate between new (current) and old (branch)
- */
-export class MessageAlternativeService {
- private currentAbortController: AbortController | null = null;
- private currentStreamingMessageId: string | null = null;
-
- /** Guard against concurrent retries on the same message */
- private retryInProgress: Set<string> = new Set();
-
- constructor(
- private chatService: ChatService,
- private branchManager: BranchManager,
- private streamHandler: MessageStreamHandler,
- private abortHandler: AbortHandler,
- private events: MessageAlternativeServiceEvents
- ) {}
-
- /**
- * Create an alternative response for an AI message.
- *
- * Saves the original content as a branch, clears the message,
- * and streams a fresh response directly into the conversation.
- */
- async createAlternativeResponse(
- conversation: ConversationData,
- aiMessageId: string,
- options?: {
- provider?: string;
- model?: string;
- systemPrompt?: string;
- workspaceId?: string;
- sessionId?: string;
- }
- ): Promise<void> {
- // Concurrent retry guard: if a retry is already in progress for this message, bail
- if (this.retryInProgress.has(aiMessageId)) {
- return;
- }
-
- const aiMessageIndex = conversation.messages.findIndex(msg => msg.id === aiMessageId);
- if (aiMessageIndex === -1) return;
-
- const aiMessage = conversation.messages[aiMessageIndex];
- if (!aiMessage || aiMessage.role !== 'assistant') return;
-
- // Must have a preceding user message to retry against
- if (aiMessageIndex === 0) return;
- const userMessage = conversation.messages[aiMessageIndex - 1];
- Iif (!userMessage || userMessage.role !== 'user') return;
-
- // Mark retry as in progress
- this.retryInProgress.add(aiMessageId);
-
- try {
- this.events.onLoadingStateChanged(true);
-
- // 1. Save original content as a branch FIRST (preserves old response)
- const originalContent = aiMessage.content;
- const originalToolCalls = aiMessage.toolCalls ? [...aiMessage.toolCalls] : undefined;
- const originalReasoning = aiMessage.reasoning;
-
- const branchMessage: ConversationMessage = {
- id: `alt_${Date.now()}_${Math.random().toString(36).substring(2, 10)}`,
- role: 'assistant',
- content: originalContent,
- timestamp: aiMessage.timestamp,
- conversationId: conversation.id,
- state: aiMessage.state || 'complete',
- toolCalls: originalToolCalls,
- reasoning: originalReasoning
- };
-
- const branchId = await this.branchManager.createHumanBranch(
- conversation,
- aiMessageId,
- branchMessage
- );
-
- // 1b. Collect and move continuation messages (e.g. tool-call follow-ups)
- // that follow the retried AI message into the branch so they don't
- // linger as stale content after the new response streams in.
- const continuationMessages = conversation.messages.splice(aiMessageIndex + 1);
- Iif (continuationMessages.length > 0 && branchId) {
- const targetBranch = aiMessage.branches?.find(b => b.id === branchId);
- Iif (targetBranch) {
- targetBranch.messages.push(...continuationMessages);
- targetBranch.updated = Date.now();
- // Persist the branch with its continuation messages
- await this.chatService.updateConversation(conversation);
- }
- }
-
- // 2. Clear the current message for fresh streaming
- aiMessage.content = '';
- aiMessage.toolCalls = undefined;
- aiMessage.reasoning = undefined;
- aiMessage.isLoading = true;
- aiMessage.state = 'draft';
-
- // Set activeAlternativeIndex to 0 so the UI shows the current message
- // (the original content is now in the branch, navigable via branch arrows)
- aiMessage.activeAlternativeIndex = 0;
-
- // 3. Fire UI update so the user sees the cleared message with loading state
- this.events.onConversationUpdated(conversation);
-
- // 4. Create abort controller for this retry
- this.currentAbortController = new AbortController();
- this.currentStreamingMessageId = aiMessageId;
-
- // 5. Get user message content for the LLM request
- const userMessageContent = userMessage.content;
-
- // 6. Stream new response directly into the live conversation
- // The stream handler mutates conversation.messages[aiMessageIndex] in-place,
- // fires onStreamingUpdate events for live UI updates, and handles tool calls.
- await this.streamHandler.streamResponse(
- conversation,
- userMessageContent,
- aiMessageId,
- {
- ...options,
- excludeFromMessageId: aiMessageId,
- abortSignal: this.currentAbortController.signal
- }
- );
-
- // 7. After streaming completes, save the updated conversation
- await this.chatService.updateConversation(conversation);
-
- // 8. Fire final UI update
- this.events.onConversationUpdated(conversation);
-
- } catch (error) {
- if (error instanceof Error && error.name === 'AbortError') {
- // On abort: keep whatever partial content was streamed.
- // The original response is safe in the branch.
- const abortedMessage = conversation.messages[aiMessageIndex];
- if (abortedMessage) {
- const hasContent = abortedMessage.content && abortedMessage.content.trim();
-
- if (hasContent) {
- // Keep partial content, clean up incomplete tool calls
- abortedMessage.toolCalls = filterCompletedToolCalls(abortedMessage.toolCalls);
- abortedMessage.isLoading = false;
- abortedMessage.state = 'aborted';
-
- await this.chatService.updateConversation(conversation);
- this.events.onStreamingUpdate(aiMessageId, abortedMessage.content, true, false);
- } else {
- // No content streamed yet - mark as aborted with empty content
- abortedMessage.isLoading = false;
- abortedMessage.state = 'aborted';
- abortedMessage.content = '';
-
- await this.chatService.updateConversation(conversation);
- }
- }
-
- this.events.onConversationUpdated(conversation);
- } else {
- this.events.onError('Failed to generate alternative response');
- }
- } finally {
- this.retryInProgress.delete(aiMessageId);
- this.currentAbortController = null;
- this.currentStreamingMessageId = null;
- this.events.onLoadingStateChanged(false);
- }
- }
-
- /**
- * Cancel current alternative generation
- */
- cancel(): void {
- Iif (this.currentAbortController && this.currentStreamingMessageId) {
- this.currentAbortController.abort();
- this.currentAbortController = null;
- this.currentStreamingMessageId = null;
- }
- }
-
- /**
- * Check if currently generating an alternative
- */
- isGenerating(): boolean {
- return this.currentAbortController !== null;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| BranchManager.ts | -
-
- |
- 60.33% | -73/121 | -70.58% | -36/51 | -51.85% | -14/27 | -62.5% | -70/112 | -
| MessageAlternativeService.ts | -
-
- |
- 87.17% | -68/78 | -75.86% | -22/29 | -83.33% | -5/6 | -88.88% | -64/72 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 | - - - - - - - - - - - -1x - - - - - - - - - - -1x - -15x -15x - - - - - - - - - - - - - - -11x - -20x -10x - -9x -9x - - -9x -1x -1x - - - -8x - -5x -5x -5x - - -5x - - -5x - - -5x - - -3x -3x - - -3x - - -3x - - -3x - - - - - - - -7x - - - - - - - - - - - - -2x -1x -1x - -1x - - - | /**
- * Location: /src/ui/chat/utils/AbortHandler.ts
- *
- * Purpose: Unified abort handling utility for AI message generation
- * Extracted from MessageManager.ts to eliminate DRY violations (4+ repeated abort patterns)
- *
- * Used by: MessageManager, MessageAlternativeService for handling abort scenarios
- * Dependencies: ChatService
- */
-
-import { ConversationData, ConversationMessage } from '../../../types/chat/ChatTypes';
-import { ChatService } from '../../../services/chat/ChatService';
-import { filterCompletedToolCalls } from './toolCallUtils';
-
-export interface AbortHandlerEvents {
- onStreamingUpdate: (messageId: string, content: string, isComplete: boolean, isIncremental?: boolean) => void;
- onConversationUpdated: (conversation: ConversationData) => void;
-}
-
-/**
- * Handles abort scenarios for AI message generation
- * Consolidates repeated abort handling logic throughout MessageManager
- */
-export class AbortHandler {
- constructor(
- private chatService: ChatService,
- private events: AbortHandlerEvents
- ) {}
-
- /**
- * Handle abort for an AI message being generated
- *
- * @param conversation - The conversation containing the message
- * @param aiMessageId - ID of the AI message being generated
- * @param customHandler - Optional custom handler for specific abort scenarios
- */
- async handleAbort(
- conversation: ConversationData,
- aiMessageId: string | null,
- customHandler?: (hasContent: boolean, aiMessage: ConversationMessage) => Promise<void>
- ): Promise<void> {
- if (!aiMessageId) return;
-
- const aiMessageIndex = conversation.messages.findIndex(msg => msg.id === aiMessageId);
- if (aiMessageIndex < 0) return;
-
- const aiMessage = conversation.messages[aiMessageIndex];
- const hasContent = aiMessage.content && aiMessage.content.trim();
-
- // Use custom handler if provided
- if (customHandler) {
- await customHandler(!!hasContent, aiMessage);
- return;
- }
-
- // Default abort handling
- if (hasContent) {
- // Keep partial response - clean up incomplete tool calls
- aiMessage.toolCalls = filterCompletedToolCalls(aiMessage.toolCalls);
- aiMessage.isLoading = false;
- aiMessage.state = 'aborted'; // Mark as aborted (will be included in context)
-
- // Save conversation with cleaned partial message
- await this.chatService.updateConversation(conversation);
-
- // Finalize streaming with partial content (stops animation, renders final content)
- this.events.onStreamingUpdate(aiMessageId, aiMessage.content, true, false);
-
- // Update UI to show final partial message
- this.events.onConversationUpdated(conversation);
- } else {
- // No content generated - mark as invalid and delete
- aiMessage.state = 'invalid'; // Mark as invalid (will be filtered from context)
- aiMessage.isLoading = false;
-
- // Delete the empty message entirely
- conversation.messages.splice(aiMessageIndex, 1);
-
- // Save conversation without the empty message
- await this.chatService.updateConversation(conversation);
-
- // Update UI to remove the empty message bubble
- this.events.onConversationUpdated(conversation);
- }
- }
-
- /**
- * Check if an error is an abort error
- */
- isAbortError(error: unknown): boolean {
- return error instanceof Error && error.name === 'AbortError';
- }
-
- /**
- * Handle abort with error checking
- * Convenience method that checks if error is abort before handling
- */
- async handleIfAbortError(
- error: unknown,
- conversation: ConversationData,
- aiMessageId: string | null,
- customHandler?: (hasContent: boolean, aiMessage: ConversationMessage) => Promise<void>
- ): Promise<boolean> {
- if (this.isAbortError(error)) {
- await this.handleAbort(conversation, aiMessageId, customHandler);
- return true;
- }
- return false;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| AbortHandler.ts | -
-
- |
- 100% | -32/32 | -100% | -10/10 | -100% | -5/5 | -100% | -29/29 | -
| toolCallUtils.ts | -
-
- |
- 100% | -6/6 | -100% | -7/7 | -100% | -2/2 | -100% | -5/5 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 | - - - - - - - - - - - - - - -3x -20x -3x - - -37x - -17x - - | /**
- * Location: /src/ui/chat/utils/toolCallUtils.ts
- *
- * Purpose: Utility functions for tool call filtering and inspection
- * Used by: AbortHandler, MessageAlternativeService
- */
-
-import { ToolCall } from '../../../types/chat/ChatTypes';
-
-/**
- * Filter tool calls to keep only those that have completed execution.
- * A tool call is considered complete if it has a result or a success flag set.
- *
- * Returns undefined if no completed tool calls remain (or input is empty/undefined).
- */
-export function filterCompletedToolCalls(toolCalls?: ToolCall[]): ToolCall[] | undefined {
- if (!toolCalls || toolCalls.length === 0) {
- return undefined;
- }
-
- const completed = toolCalls.filter(tc => tc.result !== undefined || tc.success !== undefined);
-
- return completed.length > 0 ? completed : undefined;
-}
- |