Minimal integration of new features

- convenience api does not support MultiChatMessage

Author: rpanackal

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

@@ -1,6 +1,6 @@
 package com.sap.ai.sdk.orchestration;
 
-import com.sap.ai.sdk.orchestration.model.ChatMessage;
+import com.sap.ai.sdk.orchestration.model.ChatMessagesInner;
 import com.sap.ai.sdk.orchestration.model.CompletionPostRequest;
 import com.sap.ai.sdk.orchestration.model.ModuleConfigs;
 import com.sap.ai.sdk.orchestration.model.OrchestrationConfig;
@@ -32,7 +32,10 @@ static CompletionPostRequest toCompletionPostRequest(
             OrchestrationConfig.create().moduleConfigurations(toModuleConfigs(configCopy)))
         .inputParams(prompt.getTemplateParameters())
         .messagesHistory(
-            prompt.getMessagesHistory().stream().map(Message::createChatMessage).toList());
+            prompt.getMessagesHistory().stream()
+                .map(Message::createChatMessage)
+                .map(ChatMessagesInner.class::cast)
+                .toList());
   }
 
   @Nonnull
@@ -45,7 +48,7 @@ static TemplatingModuleConfig toTemplateModuleConfig(
      * In this case, the request will fail, since the templating module will try to resolve the parameter.
      * To be fixed with https://github.tools.sap/AI/llm-orchestration/issues/662
      */
-    val messages = template instanceof Template t ? t.getTemplate() : List.<ChatMessage>of();
+    val messages = template instanceof Template t ? t.getTemplate() : List.<ChatMessagesInner>of();
     val messagesWithPrompt = new ArrayList<>(messages);
     messagesWithPrompt.addAll(
         prompt.getMessages().stream().map(Message::createChatMessage).toList());

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

@@ -3,6 +3,7 @@
 import static lombok.AccessLevel.PACKAGE;
 
 import com.sap.ai.sdk.orchestration.model.ChatMessage;
+import com.sap.ai.sdk.orchestration.model.ChatMessagesInner;
 import com.sap.ai.sdk.orchestration.model.CompletionPostResponse;
 import com.sap.ai.sdk.orchestration.model.LLMChoice;
 import com.sap.ai.sdk.orchestration.model.LLMModuleResultSynchronous;
@@ -50,23 +51,30 @@ public TokenUsage getTokenUsage() {
   /**
    * Get all messages. This can be used for subsequent prompts as a message history.
    *
+   * @throws UnsupportedOperationException if the MultiChatMessage type message in chat.
    * @return A list of all messages.
    */
   @Nonnull
-  public List<Message> getAllMessages() {
+  public List<Message> getAllMessages() throws UnsupportedOperationException{
     final var messages = new ArrayList<Message>();
 
-    for (final ChatMessage chatMessage : originalResponse.getModuleResults().getTemplating()) {
-      final var message =
-          switch (chatMessage.getRole()) {
-            case "user" -> new UserMessage(chatMessage.getContent());
-            case "assistant" -> new AssistantMessage(chatMessage.getContent());
-            case "system" -> new SystemMessage(chatMessage.getContent());
-            default -> throw new IllegalStateException("Unexpected role: " + chatMessage.getRole());
-          };
-      messages.add(message);
+    for (final ChatMessagesInner chatMessage : originalResponse.getModuleResults().getTemplating()) {
+      if (chatMessage instanceof ChatMessage simpleMsg) {
+        final var message =
+            switch (simpleMsg.getRole()) {
+              case "user" -> new UserMessage(simpleMsg.getContent());
+              case "assistant" -> new AssistantMessage(simpleMsg.getContent());
+              case "system" -> new SystemMessage(simpleMsg.getContent());
+              default -> throw new IllegalStateException("Unexpected role: " + simpleMsg.getRole());
+            };
+        messages.add(message);
+      }
+      else {
+        throw new UnsupportedOperationException("Currently MultiChatMessage type not supported by convenience API");
+      }
     }
-
+    // TODO: Review { bypassing throwing OrchestrationClientException and
+    //       assumes AssistantMessage is always only type ChatMessage }
     messages.add(new AssistantMessage(getChoice().getMessage().getContent()));
     return messages;
   }

orchestration/src/test/java/com/sap/ai/sdk/orchestration/OrchestrationUnitTest.java

@@ -128,7 +128,7 @@ void testGrounding() {
     assertThat(groundingData.get("grounding_result").toString())
         .startsWith("Joule is the AI copilot that truly understands your business.");
     assertThat(result.getModuleResults().getGrounding().getMessage()).isEqualTo("grounding result");
-    assertThat(result.getModuleResults().getTemplating().get(0).getContent())
+    assertThat(((ChatMessage)result.getModuleResults().getTemplating().get(0)).getContent())
         .startsWith(
             "What does Joule do? Use the following information as additional context: Joule is the AI copilot that truly understands your business.");
     assertThat(llmChoice.getMessage().getContent())
@@ -574,10 +574,12 @@ void streamChatCompletionDeltas() throws IOException {
         final var message0 = (Map<String, Object>) choices0.getCustomField("delta");
         assertThat(message0.get("role")).isEqualTo("");
         assertThat(message0.get("content")).isEqualTo("");
-        List<ChatMessage> templating = deltaList.get(0).getModuleResults().getTemplating();
+        final var templating = deltaList.get(0).getModuleResults().getTemplating();
         assertThat(templating).hasSize(1);
-        assertThat(templating.get(0).getRole()).isEqualTo("user");
-        assertThat(templating.get(0).getContent())
+        
+        final var templateItem = (ChatMessage) templating.get(0);
+        assertThat(templateItem.getRole()).isEqualTo("user");
+        assertThat(templateItem.getContent())
             .isEqualTo("Hello world! Why is this phrase so famous?");
 
         assertThat(result1.getSystemFingerprint()).isEqualTo("fp_808245b034");