fix e2e tests

Author: Jonas-Isr

sample-code/spring-app/src/main/java/com/sap/ai/sdk/app/services/OrchestrationService.java

@@ -443,12 +443,14 @@ public record Translation(
   @Nonnull
   public OrchestrationChatResponse responseFormatJsonSchema(
       @Nonnull final String word, @Nonnull final Class<?> targetType) {
+    final var configWithGpt4 =
+        new OrchestrationModuleConfig().withLlmConfig(GPT_4O_MINI);
     val schema =
         ResponseJsonSchema.fromType(targetType)
             .withDescription("Output schema for language translation.")
             .withStrict(true);
     val configWithResponseSchema =
-        config.withTemplateConfig(TemplateConfig.create().withJsonSchemaResponse(schema));
+        configWithGpt4.withTemplateConfig(TemplateConfig.create().withJsonSchemaResponse(schema));
 
     val prompt =
         new OrchestrationPrompt(
@@ -564,8 +566,10 @@ public OrchestrationChatResponse templateFromPromptRegistryByScenario(
   @Nonnull
   public OrchestrationChatResponse localPromptTemplate(@Nonnull final String promptTemplate)
       throws IOException {
+    final var configWithGpt4 =
+        new OrchestrationModuleConfig().withLlmConfig(GPT_4O_MINI);
     val template = TemplateConfig.create().fromYaml(promptTemplate);
-    val configWithTemplate = template != null ? config.withTemplateConfig(template) : config;
+    val configWithTemplate = template != null ? configWithGpt4.withTemplateConfig(template) : configWithGpt4;
 
     val inputParams = Map.of("language", "German");
     val prompt = new OrchestrationPrompt(inputParams);

sample-code/spring-app/src/test/java/com/sap/ai/sdk/app/controllers/ScenarioTest.java

@@ -8,7 +8,10 @@
 import com.sap.ai.sdk.orchestration.OrchestrationAiModel;
 import java.lang.reflect.Field;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.Optional;
+import java.util.Set;
+
 import lombok.SneakyThrows;
 import org.assertj.core.api.SoftAssertions;
 import org.junit.jupiter.api.DisplayName;
@@ -95,19 +98,27 @@ void orchestrationAiModelAvailability() {
       }
     }
 
+    // Set of internal-only models
+    var internalOnlyModels = Set.of("abap-codestral");
+
     // Assert that the declared Orchestration models match the expected list
-    assertThat(declaredOrchestrationModelList.keySet())
+    var declaredAndInternalOnlyModels = new HashSet<>(declaredOrchestrationModelList.keySet());
+    declaredAndInternalOnlyModels.addAll(internalOnlyModels);
+
+    assertThat(declaredAndInternalOnlyModels)
         .containsAll(availableOrchestrationModels.keySet());
 
     SoftAssertions softly = new SoftAssertions();
     for (var model : availableOrchestrationModels.entrySet()) {
       Boolean declaredDeprecated = declaredOrchestrationModelList.get(model.getKey());
-      softly
-          .assertThat(declaredDeprecated)
-          .withFailMessage(
-              "%s is deprecated:%s on AI Core but deprecated:%s in AI SDK",
-              model.getKey(), model.getValue(), declaredDeprecated)
-          .isEqualTo(model.getValue());
+      if (!internalOnlyModels.contains(model.getKey())) {
+        softly
+            .assertThat(declaredDeprecated)
+            .withFailMessage(
+                "%s is deprecated:%s on AI Core but deprecated:%s in AI SDK",
+                model.getKey(), model.getValue(), declaredDeprecated)
+            .isEqualTo(model.getValue());
+      }
     }
     softly.assertAll();
   }