Merge pull request #234 from Yanick-Salzmann/bug/missing-api-version

Added apiVersion parameter to AzureOpenAiEmbeddingModel and AzureOpen…
quarkiverse · Jan 22, 2024 · 5de41ca · 5de41ca
2 parents b53feaa + 8ef8229
commit 5de41ca
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...me/src/main/java/io/quarkiverse/langchain4j/azure/openai/runtime/AzureOpenAiRecorder.java b/...me/src/main/java/io/quarkiverse/langchain4j/azure/openai/runtime/AzureOpenAiRecorder.java
@@ -50,6 +50,7 @@ public Supplier<?> streamingChatModel(Langchain4jAzureOpenAiConfig runtimeConfig
         var builder = AzureOpenAiStreamingChatModel.builder()
                 .baseUrl(getBaseUrl(runtimeConfig))
                 .apiKey(runtimeConfig.apiKey())
+                .apiVersion(runtimeConfig.apiVersion())
                 .timeout(runtimeConfig.timeout())
                 .logRequests(firstOrDefault(false, chatModelConfig.logRequests(), runtimeConfig.logRequests()))
                 .logResponses(firstOrDefault(false, chatModelConfig.logResponses(), runtimeConfig.logResponses()))
@@ -76,6 +77,7 @@ public Supplier<?> embeddingModel(Langchain4jAzureOpenAiConfig runtimeConfig) {
         var builder = AzureOpenAiEmbeddingModel.builder()
                 .baseUrl(getBaseUrl(runtimeConfig))
                 .apiKey(runtimeConfig.apiKey())
+                .apiVersion(runtimeConfig.apiVersion())
                 .timeout(runtimeConfig.timeout())
                 .maxRetries(runtimeConfig.maxRetries())
                 .logRequests(firstOrDefault(false, embeddingModelConfig.logRequests(), runtimeConfig.logRequests()))