formatting

Author: dlarocque

packages/vertexai/src/methods/count-tokens.ts

@@ -22,7 +22,7 @@ import {
 } from '../types';
 import { Task, makeRequest } from '../requests/request';
 import { ApiSettings } from '../types/internal';
-import * as GoogleAIMapper from '../googleAIMappers'; // FIXME: (code smell) Is there a better way to namespace this?
+import * as GoogleAIMapper from '../googleAIMappers';
 import { BackendType } from '../public-types';
 
 export async function countTokens(

packages/vertexai/src/public-types.ts

@@ -21,16 +21,16 @@ export * from './types';
 
 /**
  * An instance of the Vertex AI in Firebase SDK.
- * 
+ *
  * For more information, refer to the documentation for the new {@link GenAI}.
- * 
+ *
  * @public
  */
 export type VertexAI = GenAI;
 
 /**
  * Options when initializing the Vertex AI in Firebase SDK.
- * 
+ *
  * @public
  */
 export interface VertexAIOptions {

packages/vertexai/src/requests/stream-reader.ts

@@ -89,14 +89,17 @@ async function* generateResponseSequence(
       break;
     }
 
-    const enhancedResponse =
-      apiSettings.backend.backendType === BackendType.GOOGLE_AI
-        ? createEnhancedContentResponse(
-            GoogleAIMapper.mapGenerateContentResponse(
-              value as GoogleAIGenerateContentResponse
-            )
-          )
-        : createEnhancedContentResponse(value);
+    let enhancedResponse: EnhancedGenerateContentResponse;
+    if (apiSettings.backend.backendType === BackendType.GOOGLE_AI) {
+      enhancedResponse = createEnhancedContentResponse(
+        GoogleAIMapper.mapGenerateContentResponse(
+          value as GoogleAIGenerateContentResponse
+        )
+      );
+    } else {
+      enhancedResponse = createEnhancedContentResponse(value);
+    }
+
     yield enhancedResponse;
   }
 }