From d1735725adb0718ddabe0818033618df751a06db Mon Sep 17 00:00:00 2001 From: Mark Pollack Date: Mon, 25 Nov 2024 16:33:58 -0500 Subject: [PATCH] Enable easier debugging of Azure Image and Audio Transcription tests --- .../AzureOpenAiAudioTranscriptionModelIT.java | 18 ++++++++++-------- .../openai/image/AzureOpenAiImageModelIT.java | 11 ++++++++--- 2 files changed, 18 insertions(+), 11 deletions(-) diff --git a/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/AzureOpenAiAudioTranscriptionModelIT.java b/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/AzureOpenAiAudioTranscriptionModelIT.java index 18ecd8b73..c04a1ac3e 100644 --- a/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/AzureOpenAiAudioTranscriptionModelIT.java +++ b/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/AzureOpenAiAudioTranscriptionModelIT.java @@ -36,7 +36,7 @@ import org.springframework.core.io.Resource; import static org.assertj.core.api.Assertions.assertThat; /** - * NOTE - Use deployment name "whisper + * NOTE - Use deployment name "whisper" * * @author Piotr Olaszewski */ @@ -87,13 +87,15 @@ class AzureOpenAiAudioTranscriptionModelIT { @Bean public OpenAIClient openAIClient() { - return new OpenAIClientBuilder() - .credential(new AzureKeyCredential(System.getenv("AZURE_OPENAI_TRANSCRIPTION_API_KEY"))) - .endpoint(System.getenv("AZURE_OPENAI_TRANSCRIPTION_ENDPOINT")) - // new - // AzureKeyCredential("2FsJHcxEoyIMWlhT882lUYx6D1ovQeTHNyQjHKsBYtX9Pf9TynrAJQQJ99AKACfhMk5XJ3w3AAAAACOGDOWA")) - // .endpoint("https://mpoll-m3ot7gc7-swedencentral.cognitiveservices.azure.com/") - .serviceVersion(OpenAIServiceVersion.V2024_02_15_PREVIEW) + String apiKey = System.getenv("AZURE_OPENAI_TRANSCRIPTION_API_KEY"); + String endpoint = System.getenv("AZURE_OPENAI_TRANSCRIPTION_ENDPOINT"); + + // System.out.println("API Key: " + apiKey); + // System.out.println("Endpoint: " + endpoint); + + return new OpenAIClientBuilder().credential(new AzureKeyCredential(apiKey)) + .endpoint(endpoint) + // .serviceVersion(OpenAIServiceVersion.V2024_02_15_PREVIEW) .buildClient(); } diff --git a/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/image/AzureOpenAiImageModelIT.java b/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/image/AzureOpenAiImageModelIT.java index f548866a9..325962ff2 100644 --- a/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/image/AzureOpenAiImageModelIT.java +++ b/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/image/AzureOpenAiImageModelIT.java @@ -87,9 +87,14 @@ public class AzureOpenAiImageModelIT { @Bean public OpenAIClient openAIClient() { - return new OpenAIClientBuilder() - .credential(new AzureKeyCredential(System.getenv("AZURE_OPENAI_IMAGE_API_KEY"))) - .endpoint(System.getenv("AZURE_OPENAI_IMAGE_ENDPOINT")) + String apiKey = System.getenv("AZURE_OPENAI_IMAGE_API_KEY"); + String endpoint = System.getenv("AZURE_OPENAI_IMAGE_ENDPOINT"); + + // System.out.println("API Key: " + apiKey); + // System.out.println("Endpoint: " + endpoint); + + return new OpenAIClientBuilder().credential(new AzureKeyCredential(apiKey)) + .endpoint(endpoint) .buildClient(); }