removed warnings

Author: CharlesDuboisSAP

orchestration/src/main/java/com/sap/ai/sdk/orchestration/spring/OrchestrationChatModel.java

@@ -46,7 +46,7 @@ public ChatResponse call(@Nonnull final Prompt prompt) {
 
       val orchestrationPrompt = toOrchestrationPrompt(prompt);
       val response = client.chatCompletion(orchestrationPrompt, options.getConfig());
-      return OrchestrationChatResponse.fromOrchestrationResponse(response.getOriginalResponse());
+      return new OrchestrationSpringChatResponse(response);
     }
     throw new IllegalArgumentException(
         "Please add OrchestrationChatOptions to the Prompt: new Prompt(\"message\", new OrchestrationChatOptions(config))");

…on/spring/OrchestrationChatResponse.java …ing/OrchestrationSpringChatResponse.javaorchestration/src/main/java/com/sap/ai/sdk/orchestration/spring/OrchestrationChatResponse.java renamed to orchestration/src/main/java/com/sap/ai/sdk/orchestration/spring/OrchestrationSpringChatResponse.java orchestration/src/main/java/com/sap/ai/sdk/orchestration/spring/OrchestrationChatResponse.java renamed to orchestration/src/main/java/com/sap/ai/sdk/orchestration/spring/OrchestrationSpringChatResponse.java

@@ -1,7 +1,6 @@
 package com.sap.ai.sdk.orchestration.spring;
 
-import com.google.common.annotations.Beta;
-import com.sap.ai.sdk.orchestration.model.CompletionPostResponse;
+import com.sap.ai.sdk.orchestration.OrchestrationChatResponse;
 import com.sap.ai.sdk.orchestration.model.LLMChoice;
 import com.sap.ai.sdk.orchestration.model.LLMModuleResultSynchronous;
 import com.sap.ai.sdk.orchestration.model.TokenUsage;
@@ -18,34 +17,26 @@
 import org.springframework.ai.chat.model.ChatResponse;
 import org.springframework.ai.chat.model.Generation;
 
-/**
- * Response from the orchestration service in a Spring AI {@link ChatResponse}.
- *
- * @since 1.2.0
- */
-@Beta
+/** Response from the orchestration service in a Spring AI {@link ChatResponse}. */
 @Value
 @EqualsAndHashCode(callSuper = true)
-public class OrchestrationChatResponse extends ChatResponse {
+class OrchestrationSpringChatResponse extends ChatResponse {
 
-  private OrchestrationChatResponse(
-      @Nonnull final List<Generation> generations, @Nonnull final ChatResponseMetadata metadata) {
-    super(generations, metadata);
-  }
+  OrchestrationChatResponse response;
 
-  @Nonnull
-  static OrchestrationChatResponse fromOrchestrationResponse(
-      @Nonnull final CompletionPostResponse response) {
-    val result = (LLMModuleResultSynchronous) response.getOrchestrationResult();
-    val generations = toGenerations(result);
-    val metadata = toChatResponseMetadata(result);
-    return new OrchestrationChatResponse(generations, metadata);
+  OrchestrationSpringChatResponse(@Nonnull final OrchestrationChatResponse response) {
+    super(
+        toGenerations(
+            (LLMModuleResultSynchronous) response.getOriginalResponse().getOrchestrationResult()),
+        toChatResponseMetadata(
+            (LLMModuleResultSynchronous) response.getOriginalResponse().getOrchestrationResult()));
+    this.response = response;
   }
 
   @Nonnull
   static List<Generation> toGenerations(@Nonnull final LLMModuleResultSynchronous result) {
     return result.getChoices().stream()
-        .map(OrchestrationChatResponse::toAssistantMessage)
+        .map(OrchestrationSpringChatResponse::toAssistantMessage)
         .map(Generation::new)
         .toList();
   }

orchestration/src/test/java/com/sap/ai/sdk/orchestration/spring/OrchestrationChatResponseTest.java

@@ -20,7 +20,7 @@ void testToAssistantMessage() {
             .message(ChatMessage.create().role("assistant").content("Hello, world!"))
             .finishReason("stop");
 
-    AssistantMessage message = OrchestrationChatResponse.toAssistantMessage(choice);
+    AssistantMessage message = OrchestrationSpringChatResponse.toAssistantMessage(choice);
 
     assertThat(message.getContent()).isEqualTo("Hello, world!");
     assertThat(message.getMetadata()).containsEntry("finish_reason", "stop");
@@ -38,7 +38,7 @@ void testToChatResponseMetadata() {
             .choices(List.of())
             .usage(TokenUsage.create().completionTokens(20).promptTokens(10).totalTokens(30));
 
-    var metadata = OrchestrationChatResponse.toChatResponseMetadata(moduleResult);
+    var metadata = OrchestrationSpringChatResponse.toChatResponseMetadata(moduleResult);
 
     assertThat(metadata.getId()).isEqualTo("test-id");
     assertThat(metadata.getModel()).isEqualTo("test-model");