Issue 32637 ai azure vendor

dotCMS/pom.xml

@@ -86,6 +86,12 @@
  <version>${langchain4j.version}</version>
  </dependency>
 
+ <dependency>
+ <groupId>dev.langchain4j</groupId>
+ <artifactId>langchain4j-azure-open-ai</artifactId>
+ <version>${langchain4j.version}</version>
+ </dependency>
+
  <!-- https://mvnrepository.com/artifact/dev.langchain4j/langchain4j-embeddings-all-minilm-l6-v2 -->
  <dependency>
  <groupId>dev.langchain4j</groupId>

dotCMS/src/main/java/com/dotcms/ai/api/provider/VendorModelProviderFactory.java

@@ -1,6 +1,7 @@
 package com.dotcms.ai.api.provider;
 
 import com.dotcms.ai.api.provider.anthropic.AnthropicVendorModelProviderImpl;
+import com.dotcms.ai.api.provider.azure.AzureVendorModelProviderImpl;
 import com.dotcms.ai.api.provider.openai.OpenAiVendorModelProviderImpl;
 import com.dotcms.ai.config.AiModelConfig;
 import dev.langchain4j.model.chat.ChatModel;
@@ -22,7 +23,8 @@ public class VendorModelProviderFactory {
  private final Map<String, VendorModelProvider> providers = new ConcurrentHashMap<>();
 
  public VendorModelProviderFactory() {
- this(List.of(new OpenAiVendorModelProviderImpl(), new AnthropicVendorModelProviderImpl()));
+ this(List.of(new OpenAiVendorModelProviderImpl(), new AnthropicVendorModelProviderImpl(),
+ new AzureVendorModelProviderImpl()));
  }
 
  public VendorModelProviderFactory(final List<VendorModelProvider> providerList) {

dotCMS/src/main/java/com/dotcms/ai/api/provider/azure/AzureChatModelProviderImpl.java

@@ -0,0 +1,68 @@
+package com.dotcms.ai.api.provider.azure;
+
+import com.dotcms.ai.api.provider.ChatModelProvider;
+import com.dotcms.ai.config.AiModel;
+import com.dotcms.ai.config.AiModelConfig;
+import com.dotcms.util.ConversionUtils;
+import dev.langchain4j.model.azure.AzureOpenAiChatModel;
+import dev.langchain4j.model.azure.AzureOpenAiStreamingChatModel;
+import dev.langchain4j.model.chat.ChatModel;
+import dev.langchain4j.model.chat.StreamingChatModel;
+import dev.langchain4j.model.openai.OpenAiStreamingChatModel;
+
+import java.time.Duration;
+
+/**
+ * Azure OpenAI Chat Model Provider Implementation
+ * Supports chat models on Azure OpenAI Service.
+ */
+public class AzureChatModelProviderImpl implements ChatModelProvider {
+
+ @Override
+ public ChatModel create(final AiModelConfig config) {
+
+ final String apiKey = config.get(AiModelConfig.API_KEY);
+ final String endpoint = config.getOrDefault(AiModelConfig.API_URL, "");
+ final String deployment = config.getOrDefault("deploymentName", "");
+ final double temperature = ConversionUtils.toDouble(
+ config.getOrDefault(AiModelConfig.TEMPERATURE, "0.3"),
+ 0.3
+ );
+ final long timeoutMs = ConversionUtils.toLong(
+ config.getOrDefault(AiModelConfig.TIMEOUT_MS, "30000"),
+ 30000L
+ );
+
+ return AzureOpenAiChatModel.builder()
+ .apiKey(apiKey)
+ .endpoint(endpoint)
+ .deploymentName(deployment)
+ .temperature(temperature)
+ .timeout(Duration.ofMillis(timeoutMs))
+ .build();
+ }
+
+ @Override
+ public StreamingChatModel createStreaming(final AiModelConfig config) {
+
+ final String apiKey = config.get(AiModelConfig.API_KEY);
+ final String endpoint = config.getOrDefault(AiModelConfig.API_URL, "");
+ final String deployment = config.getOrDefault("deploymentName", "");
+ final double temperature = ConversionUtils.toDouble(
+ config.getOrDefault(AiModelConfig.TEMPERATURE, "0.3"),
+ 0.3
+ );
+ final long timeoutMs = ConversionUtils.toLong(
+ config.getOrDefault(AiModelConfig.TIMEOUT_MS, "30000"),
+ 30000L
+ );
+
+ return AzureOpenAiStreamingChatModel.builder()
+ .apiKey(apiKey)
+ .endpoint(endpoint)
+ .deploymentName(deployment)
+ .temperature(temperature)
+ .timeout(Duration.ofMillis(timeoutMs))
+ .build();
+ }
+}

dotCMS/src/main/java/com/dotcms/ai/api/provider/azure/AzureEmbeddingModelProviderImpl.java

@@ -0,0 +1,35 @@
+package com.dotcms.ai.api.provider.azure;
+
+import com.dotcms.ai.api.provider.EmbeddingModelProvider;
+import com.dotcms.ai.config.AiModelConfig;
+import com.dotcms.util.ConversionUtils;
+import dev.langchain4j.model.azure.AzureOpenAiEmbeddingModel;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+
+import java.time.Duration;
+
+/**
+ * Azure OpenAI Embedding Model Provider Implementation
+ * Supports embedding models on Azure OpenAI Service.
+ */
+public class AzureEmbeddingModelProviderImpl implements EmbeddingModelProvider {
+
+ @Override
+ public EmbeddingModel createEmbedding(final AiModelConfig config) {
+
+ final String apiKey = config.get(AiModelConfig.API_KEY);
+ final String endpoint = config.getOrDefault(AiModelConfig.API_URL, "");
+ final String deployment = config.getOrDefault("deploymentName", "");
+ final long timeoutMs = ConversionUtils.toLong(
+ config.getOrDefault(AiModelConfig.TIMEOUT_MS, "30000"),
+ 20000L
+ );
+
+ return AzureOpenAiEmbeddingModel.builder()
+ .apiKey(apiKey)
+ .endpoint(endpoint)
+ .deploymentName(deployment)
+ .timeout(Duration.ofMillis(timeoutMs))
+ .build();
+ }
+}

dotCMS/src/main/java/com/dotcms/ai/api/provider/azure/AzureVendorModelProviderImpl.java

@@ -0,0 +1,34 @@
+package com.dotcms.ai.api.provider.azure;
+
+import com.dotcms.ai.api.provider.VendorModelProvider;
+import com.dotcms.ai.config.AiModelConfig;
+import com.dotcms.ai.config.AiVendor;
+import dev.langchain4j.model.chat.ChatModel;
+import dev.langchain4j.model.chat.StreamingChatModel;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+
+public class AzureVendorModelProviderImpl implements VendorModelProvider {
+
+ private final AzureChatModelProviderImpl azureChatModelProvider = new AzureChatModelProviderImpl();
+ private final AzureEmbeddingModelProviderImpl embeddingProvider = new AzureEmbeddingModelProviderImpl();
+
+ @Override
+ public String getVendorName() {
+ return AiVendor.AZURE_OPEN_AI.getVendorName();
+ }
+
+ @Override
+ public ChatModel create(final AiModelConfig config) {
+ return azureChatModelProvider.create(config);
+ }
+
+ @Override
+ public StreamingChatModel createStreaming(final AiModelConfig config) {
+ return azureChatModelProvider.createStreaming(config);
+ }
+
+ @Override
+ public EmbeddingModel createEmbedding(final AiModelConfig config) {
+ return embeddingProvider.createEmbedding(config);
+ }
+}

dotCMS/src/main/java/com/dotcms/ai/config/AiVendor.java

@@ -7,7 +7,8 @@
 public enum AiVendor {
 
  OPEN_AI("openai"),
- ANTHROPIC("anthropic");
+ ANTHROPIC("anthropic"),
+ AZURE_OPEN_AI("azureopenai");
 
  private final String vendorName;