refactor: extract readSSEResponse helper, un-export GENERATION_STATUS_LABELS

Author: Shubhangi-Microsoft

content-gen/src/app/frontend/src/api/index.ts

@@ -18,6 +18,27 @@ function normalizeUserId(userId?: string): string {
   return userId || 'anonymous';
 }
 
+/**
+ * Validate an SSE response, extract its body reader, and yield parsed events.
+ * Consolidates the duplicated response → reader → parseSSEStream pipeline
+ * used by streamChat and streamRegenerateImage.
+ */
+async function* readSSEResponse(
+  response: Response,
+  context: string,
+): AsyncGenerator<AgentResponse> {
+  if (!response.ok) {
+    throw new Error(`${context}: ${response.statusText}`);
+  }
+
+  const reader = response.body?.getReader();
+  if (!reader) {
+    throw new Error('No response body');
+  }
+
+  yield* parseSSEStream(reader);
+}
+
 /**
  * Get application configuration including feature flags
  */
@@ -94,16 +115,7 @@ export async function* streamChat(
     }),
   });
 
-  if (!response.ok) {
-    throw new Error(`Chat request failed: ${response.statusText}`);
-  }
-
-  const reader = response.body?.getReader();
-  if (!reader) {
-    throw new Error('No response body');
-  }
-
-  yield* parseSSEStream(reader);
+  yield* readSSEResponse(response, 'Chat request failed');
 }
 
 /**
@@ -219,14 +231,5 @@ export async function* streamRegenerateImage(
     }),
   });
 
-  if (!response.ok) {
-    throw new Error(`Regeneration request failed: ${response.statusText}`);
-  }
-
-  const reader = response.body?.getReader();
-  if (!reader) {
-    throw new Error('No response body');
-  }
-
-  yield* parseSSEStream(reader);
+  yield* readSSEResponse(response, 'Regeneration request failed');
 }

content-gen/src/app/frontend/src/store/appSlice.ts

@@ -32,7 +32,7 @@ export enum GenerationStatus {
 }
 
 /** Display strings shown in the UI for each status. */
-export const GENERATION_STATUS_LABELS: Record<GenerationStatus, string> = {
+const GENERATION_STATUS_LABELS: Record<GenerationStatus, string> = {
   [GenerationStatus.IDLE]: '',
   [GenerationStatus.UPDATING_BRIEF]: 'Updating creative brief...',
   [GenerationStatus.PROCESSING_QUESTION]: 'Processing your question...',