Enable easier debugging of Azure Image and Audio Transcription tests

spring-projects · Nov 25, 2024 · d173572 · d173572
1 parent d030b82
commit d173572
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 11 deletions.
diff --git a/...c/test/java/org/springframework/ai/azure/openai/AzureOpenAiAudioTranscriptionModelIT.java b/...c/test/java/org/springframework/ai/azure/openai/AzureOpenAiAudioTranscriptionModelIT.java
@@ -36,7 +36,7 @@
 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 @@ public static class TestConfiguration {
 
 		@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/...enai/src/test/java/org/springframework/ai/azure/openai/image/AzureOpenAiImageModelIT.java b/...enai/src/test/java/org/springframework/ai/azure/openai/image/AzureOpenAiImageModelIT.java
@@ -87,9 +87,14 @@ public static class TestConfiguration {
 
 		@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();
 		}