Usage sample of BYOD with ChatCompletions (Azure#36511)

* sample of completions with BYOD

* add comments with links to info

Author: brandom-msft

sdk/openai/azure-ai-openai/src/samples/java/com/azure/ai/openai/usage/ChatCompletionsWithYourData.java

@@ -0,0 +1,82 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.ai.openai.usage;
+
+import com.azure.ai.openai.OpenAIClient;
+import com.azure.ai.openai.OpenAIClientBuilder;
+import com.azure.ai.openai.models.AzureChatExtensionConfiguration;
+import com.azure.ai.openai.models.AzureChatExtensionType;
+import com.azure.ai.openai.models.AzureCognitiveSearchChatExtensionConfiguration;
+import com.azure.ai.openai.models.ChatChoice;
+import com.azure.ai.openai.models.ChatCompletions;
+import com.azure.ai.openai.models.ChatCompletionsOptions;
+import com.azure.ai.openai.models.ChatMessage;
+import com.azure.ai.openai.models.ChatRole;
+import com.azure.core.credential.AzureKeyCredential;
+import com.azure.core.util.BinaryData;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+/**
+ * Sample demonstrates the "Azure OpenAI on your data" feature. Prerequisites and guides
+ * for this feature can be found at:
+ * https://learn.microsoft.com/azure/ai-services/openai/use-your-data-quickstart?tabs=command-line&pivots=programming-language-studio
+ */
+public class ChatCompletionsWithYourData {
+    /**
+     * Runs the sample and demonstrates configuration of Azure Cognitive Search as a data source.
+     *
+     * @param args Unused. Arguments to the program.
+     */
+    public static void main(String[] args){
+        String azureOpenaiKey = "{azure-open-ai-key}";
+        String endpoint = "{azure-open-ai-endpoint}";
+        String deploymentOrModelId = "{azure-open-ai-deployment-model-id}";
+
+        OpenAIClient client = new OpenAIClientBuilder()
+            .endpoint(endpoint)
+            .credential(new AzureKeyCredential(azureOpenaiKey))
+            .buildClient();
+
+        List<ChatMessage> chatMessages = new ArrayList<>();
+        chatMessages.add(new ChatMessage(ChatRole.USER, "How many of our customers are using the latest version of our SDK?"));
+
+        ChatCompletionsOptions chatCompletionsOptions = new ChatCompletionsOptions(chatMessages);
+
+        // These configuration values come from your Azure Cognitive Search resource. Using Azure
+        // Cognitive Search as a data source is briefly described in the "Azure OpenAI on
+        // your data" quickstart, linked above. A more detailed guide can be found here:
+        // https://learn.microsoft.com/azure/search/search-get-started-portal
+
+        // Your Azure Cognitive Search endpoint, admin key, and index name
+        String azureSearchEndpoint = "{azure-cognitive-search-endpoint}";
+        String azureSearchAdminKey = "{azure-cognitive-search-key}";
+        String azureSearchIndexName = "{azure-cognitive-search-index-name}";
+
+        AzureCognitiveSearchChatExtensionConfiguration cognitiveSearchConfiguration =
+            new AzureCognitiveSearchChatExtensionConfiguration(
+                azureSearchEndpoint,
+                azureSearchAdminKey,
+                azureSearchIndexName
+            );
+
+        AzureChatExtensionConfiguration extensionConfiguration =
+            new AzureChatExtensionConfiguration(
+                AzureChatExtensionType.AZURE_COGNITIVE_SEARCH,
+                BinaryData.fromObject(cognitiveSearchConfiguration));
+
+        chatCompletionsOptions.setDataSources(Arrays.asList(extensionConfiguration));
+
+        ChatCompletions chatCompletions = client.getChatCompletions(deploymentOrModelId, chatCompletionsOptions);
+
+        for (ChatChoice choice : chatCompletions.getChoices()) {
+            ChatMessage message = choice.getMessage();
+            System.out.printf("Index: %d, Chat Role: %s.%n", choice.getIndex(), message.getRole());
+            System.out.println("Message:");
+            System.out.println(message.getContent());
+        }
+    }
+}