[OpenAI] Added support for 2024-05-01-preview

.vscode/cspell.json

@@ -390,6 +390,9 @@
     "reimaging",
     "Remediations",
     "requireReleaseDeps",
+    "rerank",
+    "Rerank",
+    "RERANK",
     "resourcemanager",
     "Retriable",
     "revapi",

sdk/openai/azure-ai-openai/CHANGELOG.md

@@ -4,7 +4,7 @@
 
 ### Features Added
 
-- Added support for service API version, `2024-04-01-preview`. 
+- Added support for service API versions, `2024-04-01-preview` and `2024-05-01-preview`. 
 - Note that `AOAI` refers to Azure OpenAI and `OAI` refers to OpenAI.
 
 **Audio**
@@ -17,14 +17,36 @@
 
 **Chat**
 
+- [AOAI] Added new class `OnYourDataContextProperty` to represent the context property in the On Your Data deployment.
+- [AOAI] Added new class `OnYourDataVectorSearchAuthenticationType`, `OnYourDataVectorSearchAuthenticationOptions`, 
+  `OnYourDataVectorSearchApiKeyAuthenticationOptions`, `OnYourDataVectorSearchAccessTokenAuthenticationOptions` for the 
+  vector search authentication options.
+- [AOAI] Added new class `AzureChatExtensionRetrievedDocument` and `AzureChatExtensionRetrieveDocumentFilterReason` to 
+  represent the retrieved document and filter reason in the chat extension.
+- [AOAI] Added a new property `all_retrieved_documents` in `AzureChatExtensionsMessageContext` class to represent all 
+  the retrieved documents in the chat extension.
 - [AOAI] Added a new property `indirectAttack` in `ContentFilterResultDetailsForPrompt` class to represent the indirect attack results.
+- [AOAI] Added new properties `max_search_queries`, `allow_partial_result`, and `include_contexts` in 
+  `AzureCosmosDBChatExtensionParameters`, `AzureMachineLearningIndexChatExtensionParameters`, 
+  `AzureSearchChatExtensionParameters`, `ElasticsearchChatExtensionParameters`, `PineconeChatExtensionParameters` class
+  to support additional search queries.
+
+**Embeddings**
+
+- [AOAI] Added a new property `dimensions` in `OnYourDataDeploymentNameVectorizationSource` class to represent the dimensions of the vectorization source.
+
+**Image**
+
+- [AOAI] Added a new property `custom_blocklists` in `ImageGenerationPromptFilterResults` class to represent the prompt filter results.
 
 ### Breaking Changes
 
 - Replaced Jackson Databind annotations with `azure-json` functionality for OpenAI service models.
 - [AOAI] Added a new class `ContentFilterDetailedResults` to represent detailed content filter results, which replaces the
   `customBlocklists` response property type, `List<ContentFilterBlocklistIdResult>` in 
   `ContentFilterResultDetailsForPrompt` and `ContentFilterResultsForChoice` class.
+- [AOAI] Replaced `OnYourDataAuthenticationOptions` with `OnYourDataVectorSearchAuthenticationOptions` in the `OnYourDataEndpointVectorizationSource` class.
+  Currently, `OnYourDataEndpointVectorizationSource` only supports `OnYourDataApiKeyAuthenticationOptions` and `OnYourDataAccessTokenAuthenticationOptions` as authentication options.
 
 ### Bugs Fixed
 

sdk/openai/azure-ai-openai/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "java",
   "TagPrefix": "java/openai/azure-ai-openai",
-  "Tag": "java/openai/azure-ai-openai_589fab4377"
+  "Tag": "java/openai/azure-ai-openai_959f9440df"
 }

sdk/openai/azure-ai-openai/src/main/java/com/azure/ai/openai/OpenAIServiceVersion.java

@@ -43,7 +43,12 @@ public enum OpenAIServiceVersion implements ServiceVersion {
     /**
      * Enum value 2024-04-01-preview.
      */
-    V2024_04_01_PREVIEW("2024-04-01-preview");
+    V2024_04_01_PREVIEW("2024-04-01-preview"),
+
+    /**
+     * Enum value 2024-05-01-preview.
+     */
+    V2024_05_01_PREVIEW("2024-05-01-preview");
 
     private final String version;
 
@@ -65,6 +70,6 @@ public String getVersion() {
      * @return The latest {@link OpenAIServiceVersion}.
      */
     public static OpenAIServiceVersion getLatest() {
-        return V2024_04_01_PREVIEW;
+        return V2024_05_01_PREVIEW;
     }
 }

sdk/openai/azure-ai-openai/src/main/java/com/azure/ai/openai/implementation/OpenAIClientImpl.java

@@ -1090,6 +1090,22 @@ public Response<BinaryData> getCompletionsWithResponse(String deploymentOrModelN
      *                         }
      *                     ]
      *                     intent: String (Optional)
+     *                     all_retrieved_documents (Optional): [
+     *                          (Optional){
+     *                             content: String (Required)
+     *                             title: String (Optional)
+     *                             url: String (Optional)
+     *                             filepath: String (Optional)
+     *                             chunk_id: String (Optional)
+     *                             search_queries (Required): [
+     *                                 String (Required)
+     *                             ]
+     *                             data_source_index: int (Required)
+     *                             original_search_score: Double (Optional)
+     *                             rerank_score: Double (Optional)
+     *                             filter_reason: String(score/rerank) (Optional)
+     *                         }
+     *                     ]
      *                 }
      *             }
      *             logprobs (Required): {
@@ -1328,6 +1344,22 @@ public Mono<Response<BinaryData>> getChatCompletionsWithResponseAsync(String dep
      *                         }
      *                     ]
      *                     intent: String (Optional)
+     *                     all_retrieved_documents (Optional): [
+     *                          (Optional){
+     *                             content: String (Required)
+     *                             title: String (Optional)
+     *                             url: String (Optional)
+     *                             filepath: String (Optional)
+     *                             chunk_id: String (Optional)
+     *                             search_queries (Required): [
+     *                                 String (Required)
+     *                             ]
+     *                             data_source_index: int (Required)
+     *                             original_search_score: Double (Optional)
+     *                             rerank_score: Double (Optional)
+     *                             filter_reason: String(score/rerank) (Optional)
+     *                         }
+     *                     ]
      *                 }
      *             }
      *             logprobs (Required): {
@@ -1515,6 +1547,15 @@ public Response<BinaryData> getChatCompletionsWithResponse(String deploymentOrMo
      *                     detected: boolean (Required)
      *                 }
      *                 jailbreak (Optional): (recursive schema, see jailbreak above)
+     *                 custom_blocklists (Optional): {
+     *                     filtered: boolean (Required)
+     *                     details (Required): [
+     *                          (Required){
+     *                             filtered: boolean (Required)
+     *                             id: String (Required)
+     *                         }
+     *                     ]
+     *                 }
      *             }
      *         }
      *     ]
@@ -1587,6 +1628,15 @@ public Mono<Response<BinaryData>> getImageGenerationsWithResponseAsync(String de
      *                     detected: boolean (Required)
      *                 }
      *                 jailbreak (Optional): (recursive schema, see jailbreak above)
+     *                 custom_blocklists (Optional): {
+     *                     filtered: boolean (Required)
+     *                     details (Required): [
+     *                          (Required){
+     *                             filtered: boolean (Required)
+     *                             id: String (Required)
+     *                         }
+     *                     ]
+     *                 }
      *             }
      *         }
      *     ]

sdk/openai/azure-ai-openai/src/main/java/com/azure/ai/openai/models/AzureChatExtensionRetrieveDocumentFilterReason.java

@@ -0,0 +1,59 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+// Code generated by Microsoft (R) TypeSpec Code Generator.
+package com.azure.ai.openai.models;
+
+import com.azure.core.annotation.Generated;
+import com.azure.core.util.ExpandableStringEnum;
+import java.util.Collection;
+
+/**
+ * The reason for filtering the retrieved document.
+ */
+public final class AzureChatExtensionRetrieveDocumentFilterReason
+    extends ExpandableStringEnum<AzureChatExtensionRetrieveDocumentFilterReason> {
+
+    /**
+     * The document is filtered by original search score threshold defined by `strictness` configure.
+     */
+    @Generated
+    public static final AzureChatExtensionRetrieveDocumentFilterReason SCORE = fromString("score");
+
+    /**
+     * The document is not filtered by original search score threshold, but is filtered by rerank score and
+     * `top_n_documents` configure.
+     */
+    @Generated
+    public static final AzureChatExtensionRetrieveDocumentFilterReason RERANK = fromString("rerank");
+
+    /**
+     * Creates a new instance of AzureChatExtensionRetrieveDocumentFilterReason value.
+     *
+     * @deprecated Use the {@link #fromString(String)} factory method.
+     */
+    @Generated
+    @Deprecated
+    public AzureChatExtensionRetrieveDocumentFilterReason() {
+    }
+
+    /**
+     * Creates or finds a AzureChatExtensionRetrieveDocumentFilterReason from its string representation.
+     *
+     * @param name a name to look for.
+     * @return the corresponding AzureChatExtensionRetrieveDocumentFilterReason.
+     */
+    @Generated
+    public static AzureChatExtensionRetrieveDocumentFilterReason fromString(String name) {
+        return fromString(name, AzureChatExtensionRetrieveDocumentFilterReason.class);
+    }
+
+    /**
+     * Gets known AzureChatExtensionRetrieveDocumentFilterReason values.
+     *
+     * @return known AzureChatExtensionRetrieveDocumentFilterReason values.
+     */
+    @Generated
+    public static Collection<AzureChatExtensionRetrieveDocumentFilterReason> values() {
+        return values(AzureChatExtensionRetrieveDocumentFilterReason.class);
+    }
+}