Merge branch 'main' into spring-ai🍃tools

Author: CharlesDuboisSAP

foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/JacksonMixins.java

@@ -0,0 +1,25 @@
+package com.sap.ai.sdk.foundationmodels.openai;
+
+import com.fasterxml.jackson.annotation.JsonTypeInfo;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.sap.ai.sdk.foundationmodels.openai.generated.model.CreateChatCompletionStreamResponse;
+import lombok.AccessLevel;
+import lombok.NoArgsConstructor;
+
+/**
+ * This class contains Jackson Mixins for customizing the serialization and deserialization behavior
+ * of certain classes in the OpenAI SDK.
+ */
+@NoArgsConstructor(access = AccessLevel.PRIVATE)
+final class JacksonMixins {
+
+  /**
+   * Mixin interface to customize the deserialization of CreateChatCompletionStreamResponse.
+   *
+   * <p>Disables type information inclusion and specifies the concrete class to use for
+   * deserialization.
+   */
+  @JsonTypeInfo(use = JsonTypeInfo.Id.NONE)
+  @JsonDeserialize(as = CreateChatCompletionStreamResponse.class)
+  interface DefaultChatCompletionCreate200ResponseMixIn {}
+}

foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/OpenAiAssistantMessage.java

@@ -0,0 +1,37 @@
+package com.sap.ai.sdk.foundationmodels.openai;
+
+import com.google.common.annotations.Beta;
+import com.sap.ai.sdk.foundationmodels.openai.generated.model.ChatCompletionRequestAssistantMessage;
+import com.sap.ai.sdk.foundationmodels.openai.generated.model.ChatCompletionRequestAssistantMessageContent;
+import javax.annotation.Nonnull;
+import lombok.Value;
+import lombok.experimental.Accessors;
+
+/**
+ * Represents a chat message as 'assistant' to OpenAI service.
+ *
+ * @since 1.4.0
+ */
+@Beta
+@Value
+@Accessors(fluent = true)
+class OpenAiAssistantMessage implements OpenAiMessage {
+
+  /** The role of the message. */
+  @Nonnull String role = "assistant";
+
+  /** The content of the message. */
+  @Nonnull String content;
+
+  /**
+   * Converts the message to a serializable object.
+   *
+   * @return the corresponding {@code ChatCompletionRequestAssistantMessage} object.
+   */
+  @Nonnull
+  ChatCompletionRequestAssistantMessage createChatCompletionRequestMessage() {
+    return new ChatCompletionRequestAssistantMessage()
+        .role(ChatCompletionRequestAssistantMessage.RoleEnum.fromValue(role()))
+        .content(ChatCompletionRequestAssistantMessageContent.create(content));
+  }
+}

foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/OpenAiChatCompletionDelta.java

@@ -0,0 +1,69 @@
+package com.sap.ai.sdk.foundationmodels.openai;
+
+import static com.sap.ai.sdk.foundationmodels.openai.OpenAiUtils.getOpenAiObjectMapper;
+import static lombok.AccessLevel.PACKAGE;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.google.common.annotations.Beta;
+import com.sap.ai.sdk.core.common.StreamedDelta;
+import com.sap.ai.sdk.foundationmodels.openai.generated.model.CompletionUsage;
+import com.sap.ai.sdk.foundationmodels.openai.generated.model.CreateChatCompletionStreamResponse;
+import java.util.Map;
+import java.util.Objects;
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import lombok.EqualsAndHashCode;
+import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+import lombok.ToString;
+
+/**
+ * Represents an OpenAI chat completion output delta for streaming.
+ *
+ * @since 1.4.0
+ */
+@Beta
+@RequiredArgsConstructor(onConstructor_ = @JsonCreator, access = PACKAGE)
+@Getter
+@ToString
+@EqualsAndHashCode
+public class OpenAiChatCompletionDelta implements StreamedDelta {
+  /** The original response from the chat completion stream. */
+  @Nonnull private final CreateChatCompletionStreamResponse originalResponse;
+
+  @Nonnull
+  @Override
+  public String getDeltaContent() {
+    final var choices = getOriginalResponse().getChoices();
+    if (!choices.isEmpty() && choices.get(0).getIndex() == 0) {
+      final var message = choices.get(0).getDelta();
+      return Objects.requireNonNullElse(message.getContent(), "");
+    }
+    return "";
+  }
+
+  @Nullable
+  @Override
+  public String getFinishReason() {
+    final var choices = getOriginalResponse().getChoices();
+    if (!choices.isEmpty()) {
+      final var finishReason = choices.get(0).getFinishReason();
+      return finishReason != null ? finishReason.getValue() : null;
+    }
+    return null;
+  }
+
+  /**
+   * Retrieves the completion usage from the response, or null if it is not available.
+   *
+   * @return The completion usage or null.
+   */
+  @Nullable
+  public CompletionUsage getCompletionUsage() {
+    if (getOriginalResponse().getCustomFieldNames().contains("usage")
+        && getOriginalResponse().getCustomField("usage") instanceof Map<?, ?> usage) {
+      return getOpenAiObjectMapper().convertValue(usage, CompletionUsage.class);
+    }
+    return null;
+  }
+}