Skip to content

Commit dfbc75a

Browse files
committed
refactoring
1 parent 9ccb1bb commit dfbc75a

2 files changed

Lines changed: 12 additions & 4 deletions

File tree

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

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,12 @@ export function transformGenerationConfig(params: PortkeyGeminiParams) {
8787
}
8888
if (params.image_config) {
8989
generationConfig['imageConfig'] = {
90-
aspectRatio: params.image_config.aspect_ratio,
91-
imageSize: params.image_config.image_size,
90+
...(params.image_config.aspect_ratio && {
91+
aspectRatio: params.image_config.aspect_ratio,
92+
}),
93+
...(params.image_config.image_size && {
94+
imageSize: params.image_config.image_size,
95+
}),
9296
};
9397
}
9498
return generationConfig;

src/providers/google/chatComplete.ts

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,8 +105,12 @@ const transformGenerationConfig = (params: PortkeyGeminiParams) => {
105105
}
106106
if (params.image_config) {
107107
generationConfig['imageConfig'] = {
108-
aspectRatio: params.image_config.aspect_ratio,
109-
imageSize: params.image_config.image_size,
108+
...(params.image_config.aspect_ratio && {
109+
aspectRatio: params.image_config.aspect_ratio,
110+
}),
111+
...(params.image_config.image_size && {
112+
imageSize: params.image_config.image_size,
113+
}),
110114
};
111115
}
112116
return generationConfig;

0 commit comments

Comments
 (0)