Fix naming

Author: reakaleek

src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/StreamingAiMessage.tsx

@@ -48,7 +48,14 @@ export const StreamingAiMessage = ({
                 updateAiMessage(message.id, contentRef.current, 'complete')
             }
         },
-        onError: () => {
+        onError: (error) => {
+            console.error('[AI Provider] Error in StreamingAiMessage:', {
+                messageId: message.id,
+                errorMessage: error.message,
+                errorStack: error.stack,
+                errorName: error.name,
+                fullError: error,
+            })
             updateAiMessage(
                 message.id,
                 message.content || 'Error occurred',

src/Elastic.Documentation.Site/Assets/web-components/SearchOrAskAi/AskAi/useAskAi.ts

@@ -61,18 +61,44 @@ export const useAskAi = (props: Props): UseAskAiResponse => {
     const onMessage = useCallback(
         (sseEvent: EventSourceMessage) => {
             try {
-                // Parse and validate the canonical AskAiEvent format
+                // Parse JSON first
                 const rawData = JSON.parse(sseEvent.data)
-                const askAiEvent = AskAiEventSchema.parse(rawData)
 
-                processMessage(askAiEvent)
+                // Use safeParse with reportInput to include input data in validation errors
+                const result = AskAiEventSchema.safeParse(rawData, {
+                    reportInput: true,
+                })
+
+                if (!result.success) {
+                    // Log detailed validation errors with input data
+                    console.error('[AI Provider] Failed to parse SSE event:', {
+                        eventId: sseEvent.id || 'unknown',
+                        eventType: sseEvent.event || 'unknown',
+                        rawEventData: sseEvent.data,
+                        validationErrors: result.error.issues,
+                    })
+                    throw new Error(
+                        `Event validation failed: ${result.error.issues.map((e) => `${e.path.join('.')}: ${e.message}`).join('; ')}`
+                    )
+                }
+
+                processMessage(result.data)
             } catch (error) {
+                // Handle JSON parsing errors or other unexpected errors
+                if (error instanceof Error && error.message.includes('Event validation failed')) {
+                    // Already logged above, just re-throw
+                    throw error
+                }
+
+                // Log JSON parsing or other errors
                 console.error('[AI Provider] Failed to parse SSE event:', {
-                    eventData: sseEvent.data,
-                    error:
-                        error instanceof Error ? error.message : String(error),
+                    eventId: sseEvent.id || 'unknown',
+                    eventType: sseEvent.event || 'unknown',
+                    rawEventData: sseEvent.data,
+                    error: error instanceof Error ? error.message : String(error),
+                    errorStack: error instanceof Error ? error.stack : undefined,
                 })
-                // Re-throw to trigger onError handler
+
                 throw new Error(
                     `Event parsing failed: ${error instanceof Error ? error.message : String(error)}`
                 )
@@ -86,6 +112,12 @@ export const useAskAi = (props: Props): UseAskAiResponse => {
         headers,
         onMessage,
         onError: (error) => {
+            console.error('[AI Provider] Error in useFetchEventSource:', {
+                errorMessage: error.message,
+                errorStack: error.stack,
+                errorName: error.name,
+                fullError: error,
+            })
             setError(error)
             props.onError?.(error)
         },

src/api/Elastic.Documentation.Api.Infrastructure/Adapters/AskAi/StreamTransformerBase.cs

@@ -209,10 +209,10 @@ private async Task ProcessStreamAsync(PipeReader reader, PipeWriter writer, Acti
 						_ = parseActivity?.SetTag("tool.result_summary", toolResult.Result);
 						break;
 					}
-				case AskAiEvent.MessageComplete chunkComplete:
+				case AskAiEvent.MessageComplete messageComplete:
 					{
-						outputMessageParts.Add(new MessagePart("text", chunkComplete.FullContent));
-						Logger.LogInformation("AskAI output message: {OutputMessage}", chunkComplete.FullContent);
+						outputMessageParts.Add(new MessagePart("text", messageComplete.FullContent));
+						Logger.LogInformation("AskAI output message: {OutputMessage}", messageComplete.FullContent);
 						break;
 					}
 				case AskAiEvent.ConversationEnd: