Add grounding e2e test.

Author: Jonas-Isr

orchestration/src/main/java/com/sap/ai/sdk/orchestration/ConfigToRequestTransformer.java

@@ -70,6 +70,7 @@ static ModuleConfigs toModuleConfigs(@Nonnull final OrchestrationModuleConfig co
 
     Option.of(config.getFilteringConfig()).forEach(moduleConfig::filteringModuleConfig);
     Option.of(config.getMaskingConfig()).forEach(moduleConfig::maskingModuleConfig);
+    Option.of(config.getGroundingConfig()).forEach(moduleConfig::groundingModuleConfig);
 
     return moduleConfig;
   }

orchestration/src/main/java/com/sap/ai/sdk/orchestration/OrchestrationModuleConfig.java

@@ -6,6 +6,7 @@
 import com.sap.ai.sdk.orchestration.model.MaskingModuleConfig;
 import com.sap.ai.sdk.orchestration.model.OutputFilteringConfig;
 import com.sap.ai.sdk.orchestration.model.TemplatingModuleConfig;
+import com.sap.ai.sdk.orchestration.model.GroundingModuleConfig;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Objects;
@@ -83,6 +84,14 @@ public class OrchestrationModuleConfig {
    */
   @Nullable FilteringModuleConfig filteringConfig;
 
+  /**
+   * A grounding configuration to provide additional context to the AI model.
+   *
+   * @link <a href="https://help.sap.com/docs/sap-ai-core/sap-ai-core-service-guide/grounding">SAP
+   *     AI Core: Orchestration - </a>
+   */
+  @Nullable GroundingModuleConfig groundingConfig;
+
   /**
    * Creates a new configuration with the given LLM configuration.
    *

sample-code/spring-app/src/main/java/com/sap/ai/sdk/app/controllers/OrchestrationController.java

@@ -13,6 +13,10 @@
 import com.sap.ai.sdk.orchestration.OrchestrationModuleConfig;
 import com.sap.ai.sdk.orchestration.OrchestrationPrompt;
 import com.sap.ai.sdk.orchestration.model.DPIEntities;
+import com.sap.ai.sdk.orchestration.model.DataRepositoryType;
+import com.sap.ai.sdk.orchestration.model.DocumentGroundingFilter;
+import com.sap.ai.sdk.orchestration.model.GroundingModuleConfig;
+import com.sap.ai.sdk.orchestration.model.GroundingModuleConfigConfig;
 import com.sap.ai.sdk.orchestration.model.Template;
 import java.util.List;
 import java.util.Map;
@@ -207,4 +211,37 @@ OrchestrationChatResponse maskingPseudonymization() {
 
     return client.chatCompletion(prompt, configWithMasking);
   }
+
+  /**
+   * Using grounding to provide additional context to the AI model.
+   *
+   * @link <a href="https://help.sap.com/docs/sap-ai-core/sap-ai-core-service-guide/grounding">SAP
+   *     AI Core: Orchestration - Grounding</a>
+   */
+  @GetMapping("/grounding")
+  @Nonnull
+  OrchestrationChatResponse grounding() {
+    final var message =
+        Message.user(
+            "{{?groundingInput}} Use the following information as additional context: {{?groundingOutput}}");
+    final var prompt =
+        new OrchestrationPrompt(
+            Map.of("groundingInput", "Give the number of letters in the word 'world'."), message);
+
+    final var filterInner =
+        DocumentGroundingFilter.create().id("someID").dataRepositoryType(DataRepositoryType.VECTOR);
+    final var groundingConfigConfig =
+        GroundingModuleConfigConfig.create()
+            .inputParams(List.of("groundingInput"))
+            .outputParam("groundingOutput");
+    groundingConfigConfig.setFilters(List.of(filterInner));
+    final var groundingConfig =
+        GroundingModuleConfig.create()
+            .type(GroundingModuleConfig.TypeEnum.DOCUMENT_GROUNDING_SERVICE)
+            .config(groundingConfigConfig);
+        final var configWithGrounding = config.withGroundingConfig(groundingConfig);
+
+        return client.chatCompletion(prompt, configWithGrounding);
+//    return client.chatCompletion(prompt, config);
+  }
 }

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

@@ -154,8 +154,10 @@ void testMaskingPseudonymization() {
   @Test
   @DisabledIfSystemProperty(named = "aicore.landscape", matches = "production")
   void testGrounding() {
-    // Placeholder for grounding test
     assertThat(System.getProperty("aicore.landscape")).isNotEqualTo("production");
+    var response = controller.grounding();
+    assertThat(response).isNotNull();
+    assertThat(response.getOriginalResponse().getModuleResults().getGrounding()).isNotNull();
   }
 
   @Test