Merge pull request #1476 from narengogi/chore/nano-banana-improvements

improvements for nano banana

Author: VisargD

src/providers/google-vertex-ai/chatComplete.ts

@@ -599,6 +599,10 @@ export const VertexLlamaChatCompleteConfig: ProviderConfig = {
     param: 'stream',
     default: false,
   },
+  image_config: {
+    param: 'generationConfig',
+    transform: (params: Params) => transformGenerationConfig(params),
+  },
 };
 
 export const GoogleChatCompleteStreamChunkTransform: (

src/providers/google-vertex-ai/transformGenerationConfig.ts

@@ -1,10 +1,9 @@
-import { Params } from '../../types/requestBody';
 import {
   recursivelyDeleteUnsupportedParameters,
   transformGeminiToolParameters,
 } from './utils';
 import { GoogleEmbedParams } from './embed';
-import { EmbedInstancesData } from './types';
+import { EmbedInstancesData, PortkeyGeminiParams } from './types';
 
 export const openaiReasoningEffortToVertexThinkingLevel = (
   reasoningEffort: string
@@ -19,7 +18,7 @@ export const openaiReasoningEffortToVertexThinkingLevel = (
 /**
  * @see https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/gemini#request_body
  */
-export function transformGenerationConfig(params: Params) {
+export function transformGenerationConfig(params: PortkeyGeminiParams) {
   const generationConfig: Record<string, any> = {};
   if (params['temperature'] != null && params['temperature'] != undefined) {
     generationConfig['temperature'] = params['temperature'];
@@ -86,7 +85,16 @@ export function transformGenerationConfig(params: Params) {
       };
     }
   }
-
+  if (params.image_config) {
+    generationConfig['imageConfig'] = {
+      ...(params.image_config.aspect_ratio && {
+        aspectRatio: params.image_config.aspect_ratio,
+      }),
+      ...(params.image_config.image_size && {
+        imageSize: params.image_config.image_size,
+      }),
+    };
+  }
   return generationConfig;
 }
 

src/providers/google-vertex-ai/types.ts

@@ -1,3 +1,4 @@
+import { Params } from '../../types/requestBody';
 import { ChatCompletionResponse, GroundingMetadata } from '../types';
 
 export interface GoogleErrorResponse {
@@ -276,3 +277,9 @@ export enum VERTEX_MODALITY {
   IMAGE = 'IMAGE',
   AUDIO = 'AUDIO',
 }
+export interface PortkeyGeminiParams extends Params {
+  image_config?: {
+    aspect_ratio: string; // '16:9', '4:3', '1:1'
+    image_size: string; // '2K', '4K', '8K'
+  };
+}

src/providers/google/chatComplete.ts

@@ -32,9 +32,12 @@ import {
   generateInvalidProviderResponseError,
   transformFinishReason,
 } from '../utils';
-import { GOOGLE_GENERATE_CONTENT_FINISH_REASON } from './types';
+import {
+  GOOGLE_GENERATE_CONTENT_FINISH_REASON,
+  PortkeyGeminiParams,
+} from './types';
 
-const transformGenerationConfig = (params: Params) => {
+const transformGenerationConfig = (params: PortkeyGeminiParams) => {
   const generationConfig: Record<string, any> = {};
   if (params['temperature'] != null && params['temperature'] != undefined) {
     generationConfig['temperature'] = params['temperature'];
@@ -100,6 +103,16 @@ const transformGenerationConfig = (params: Params) => {
       };
     }
   }
+  if (params.image_config) {
+    generationConfig['imageConfig'] = {
+      ...(params.image_config.aspect_ratio && {
+        aspectRatio: params.image_config.aspect_ratio,
+      }),
+      ...(params.image_config.image_size && {
+        imageSize: params.image_config.image_size,
+      }),
+    };
+  }
   return generationConfig;
 };
 
@@ -464,6 +477,10 @@ export const GoogleChatCompleteConfig: ProviderConfig = {
     param: 'generationConfig',
     transform: (params: Params) => transformGenerationConfig(params),
   },
+  image_config: {
+    param: 'generationConfig',
+    transform: (params: Params) => transformGenerationConfig(params),
+  },
 };
 
 export interface GoogleErrorResponse {

src/providers/google/types.ts

@@ -1,3 +1,5 @@
+import { Params } from '../../types/requestBody';
+
 export enum GOOGLE_GENERATE_CONTENT_FINISH_REASON {
   FINISH_REASON_UNSPECIFIED = 'FINISH_REASON_UNSPECIFIED',
   STOP = 'STOP',
@@ -12,3 +14,9 @@ export enum GOOGLE_GENERATE_CONTENT_FINISH_REASON {
   MALFORMED_FUNCTION_CALL = 'MALFORMED_FUNCTION_CALL',
   IMAGE_SAFETY = 'IMAGE_SAFETY',
 }
+export interface PortkeyGeminiParams extends Params {
+  image_config?: {
+    aspect_ratio: string; // '16:9', '4:3', '1:1'
+    image_size: string; // '2K', '4K', '8K'
+  };
+}