Skip to content

Commit 80021f9

Browse files
committed
Formatting
1 parent a0c8c89 commit 80021f9

File tree

2 files changed

+34
-10
lines changed

2 files changed

+34
-10
lines changed

packages/vertexai/src/models/imagen-model.test.ts

Lines changed: 32 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -100,10 +100,22 @@ describe('ImagenModel', () => {
100100
false,
101101
match((value: string) => {
102102
return (
103-
value.includes(JSON.stringify(imagenModel.modelConfig.addWatermark)) &&
104-
value.includes(JSON.stringify(imagenModel.modelConfig.safetySettings?.safetyFilterLevel)) &&
105-
value.includes(JSON.stringify(imagenModel.modelConfig.safetySettings?.personFilterLevel)) &&
106-
value.includes(JSON.stringify(imagenModel.modelConfig.imageFormat?.mimeType))
103+
value.includes(
104+
JSON.stringify(imagenModel.modelConfig.addWatermark)
105+
) &&
106+
value.includes(
107+
JSON.stringify(
108+
imagenModel.modelConfig.safetySettings?.safetyFilterLevel
109+
)
110+
) &&
111+
value.includes(
112+
JSON.stringify(
113+
imagenModel.modelConfig.safetySettings?.personFilterLevel
114+
)
115+
) &&
116+
value.includes(
117+
JSON.stringify(imagenModel.modelConfig.imageFormat?.mimeType)
118+
)
107119
);
108120
}),
109121
undefined
@@ -176,10 +188,22 @@ describe('ImagenModel', () => {
176188
false,
177189
match((value: string) => {
178190
return (
179-
value.includes(JSON.stringify(imagenModel.modelConfig.addWatermark)) &&
180-
value.includes(JSON.stringify(imagenModel.modelConfig.safetySettings?.safetyFilterLevel)) &&
181-
value.includes(JSON.stringify(imagenModel.modelConfig.safetySettings?.personFilterLevel)) &&
182-
value.includes(JSON.stringify(imagenModel.modelConfig.imageFormat?.mimeType)) &&
191+
value.includes(
192+
JSON.stringify(imagenModel.modelConfig.addWatermark)
193+
) &&
194+
value.includes(
195+
JSON.stringify(
196+
imagenModel.modelConfig.safetySettings?.safetyFilterLevel
197+
)
198+
) &&
199+
value.includes(
200+
JSON.stringify(
201+
imagenModel.modelConfig.safetySettings?.personFilterLevel
202+
)
203+
) &&
204+
value.includes(
205+
JSON.stringify(imagenModel.modelConfig.imageFormat?.mimeType)
206+
) &&
183207
value.includes(`"sampleCount":${requestConfig.numberOfImages}`) &&
184208
value.includes(`"aspectRatio":"${requestConfig.aspectRatio}"`) &&
185209
value.includes(`"negativePrompt":"${requestConfig.negativePrompt}"`)

packages/vertexai/src/types/imagen/internal.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,8 +113,8 @@ export interface PredictRequestBody {
113113
/**
114114
* Contains all possible REST API paramaters.
115115
* This is the union of model-level (`ImagenModelParams`),
116-
* request-level (`ImagenGenerationConfig`) configurations, prompt,
117-
* and gcsURI (for GCS generation only).
116+
* request-level (`ImagenGenerationConfig`) configurations, prompt,
117+
* and gcsURI (for GCS generation only).
118118
*
119119
* @internal
120120
*/

0 commit comments

Comments
 (0)