Merge runtime options for Stability AI image gen

Author: habuma

models/spring-ai-stability-ai/src/main/java/org/springframework/ai/stabilityai/StabilityAiImageModel.java

@@ -88,8 +88,9 @@ public ImageResponse call(ImagePrompt imagePrompt) {
 		// Merge the runtime options passed via the prompt with the default options
 		// configured via the constructor.
 		// Runtime options overwrite StabilityAiImageModel options
-		StabilityAiImageOptions requestImageOptions = mergeOptions(imagePrompt.getOptions(), this.defaultOptions);
-
+		StabilityAiImageOptions runtimeOptions = (StabilityAiImageOptions) imagePrompt.getOptions();
+		StabilityAiImageOptions requestImageOptions = mergeOptions(runtimeOptions, this.defaultOptions);
+		System.err.println("requestImageOptions: " + requestImageOptions);
 		// Copy the org.springframework.ai.model derived ImagePrompt and ImageOptions data
 		// types to the data types used in StabilityAiApi
 		StabilityAiApi.GenerateImageRequest generateImageRequest = getGenerateImageRequest(imagePrompt,
@@ -117,7 +118,8 @@ private ImageResponse convertResponse(StabilityAiApi.GenerateImageResponse gener
 	 * Merge runtime and default {@link ImageOptions} to compute the final options to use
 	 * in the request.
 	 */
-	private StabilityAiImageOptions mergeOptions(ImageOptions runtimeOptions, StabilityAiImageOptions defaultOptions) {
+	private StabilityAiImageOptions mergeOptions(StabilityAiImageOptions runtimeOptions,
+			StabilityAiImageOptions defaultOptions) {
 		if (runtimeOptions == null) {
 			return defaultOptions;
 		}
@@ -134,10 +136,10 @@ private StabilityAiImageOptions mergeOptions(ImageOptions runtimeOptions, Stabil
 			// Handle Stability AI specific image options
 			.withCfgScale(defaultOptions.getCfgScale())
 			.withClipGuidancePreset(defaultOptions.getClipGuidancePreset())
-			.withSampler(defaultOptions.getSampler())
+			.withSampler(runtimeOptions.getSampler())
 			.withSeed(defaultOptions.getSeed())
-			.withSteps(defaultOptions.getSteps())
-			.withStylePreset(defaultOptions.getStylePreset())
+			.withSteps(runtimeOptions.getSteps())
+			.withStylePreset(runtimeOptions.getStylePreset())
 			.build();
 	}
 

models/spring-ai-stability-ai/src/main/java/org/springframework/ai/stabilityai/api/StabilityAiApi.java

@@ -100,7 +100,7 @@ public record GenerateImageRequest(@JsonProperty("text_prompts") List<TextPrompt
 			@JsonProperty("cfg_scale") Float cfgScale, @JsonProperty("clip_guidance_preset") String clipGuidancePreset,
 			@JsonProperty("sampler") String sampler, @JsonProperty("samples") Integer samples,
 			@JsonProperty("seed") Long seed, @JsonProperty("steps") Integer steps,
-			@JsonProperty("style_present") String stylePreset) {
+			@JsonProperty("style_preset") String stylePreset) {
 
 		public static Builder builder() {
 			return new Builder();