Support Azure OpenAI and AI Search

langchain4j-azure-spring-boot-starter/pom.xml

@@ -0,0 +1,73 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xmlns="http://maven.apache.org/POM/4.0.0"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>dev.langchain4j</groupId>
+        <artifactId>langchain4j-spring</artifactId>
+        <version>0.29.0-SNAPSHOT</version>
+        <relativePath>../pom.xml</relativePath>
+    </parent>
+
+    <artifactId>langchain4j-azure-spring-boot-starter</artifactId>
+    <name>LangChain4j Spring Boot starter for Azure</name>
+    <packaging>jar</packaging>
+
+    <dependencies>
+
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-azure-open-ai</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-azure-ai-search</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-autoconfigure-processor</artifactId>
+            <optional>true</optional>
+        </dependency>
+
+        <!-- should be listed before spring-boot-configuration-processor -->
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <scope>provided</scope>
+        </dependency>
+
+        <!-- needed to generate automatic metadata about available config properties -->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-configuration-processor</artifactId>
+            <optional>true</optional>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-test</artifactId>
+            <scope>test</scope>
+        </dependency>
+
+    </dependencies>
+
+    <licenses>
+        <license>
+            <name>Apache-2.0</name>
+            <url>https://www.apache.org/licenses/LICENSE-2.0.txt</url>
+            <distribution>repo</distribution>
+            <comments>A business-friendly OSS license</comments>
+        </license>
+    </licenses>
+
+</project>

langchain4j-azure-spring-boot-starter/src/main/java/dev/langchain4j/azure/aisearch/spring/AutoConfig.java

@@ -0,0 +1,25 @@
+package dev.langchain4j.azure.aisearch.spring;
+
+import dev.langchain4j.store.embedding.azure.search.AzureAiSearchEmbeddingStore;
+import org.springframework.boot.autoconfigure.AutoConfiguration;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+
+import static dev.langchain4j.azure.aisearch.spring.Properties.PREFIX;
+
+@AutoConfiguration
+@EnableConfigurationProperties(Properties.class)
+public class AutoConfig {
+
+    @Bean
+    @ConditionalOnProperty(PREFIX + ".api-key")
+    public AzureAiSearchEmbeddingStore azureAiSearchEmbeddingStore(Properties properties) {
+        return AzureAiSearchEmbeddingStore.builder()
+                .endpoint(properties.getEndpoint())
+                .apiKey(properties.getApiKey())
+                .dimensions(properties.getDimensions())
+                .setupIndex(properties.isSetupIndex())
+                .build();
+    }
+}

langchain4j-azure-spring-boot-starter/src/main/java/dev/langchain4j/azure/aisearch/spring/Properties.java

@@ -0,0 +1,18 @@
+package dev.langchain4j.azure.aisearch.spring;
+
+import lombok.Getter;
+import lombok.Setter;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+
+@Getter
+@Setter
+@ConfigurationProperties(prefix = Properties.PREFIX)
+public class Properties {
+
+    static final String PREFIX = "langchain4j.azure.ai-search";
+
+    String endpoint;
+    String apiKey;
+    int dimensions;
+    boolean setupIndex;
+}

langchain4j-azure-spring-boot-starter/src/main/java/dev/langchain4j/azure/openai/spring/AutoConfig.java

@@ -0,0 +1,95 @@
+package dev.langchain4j.azure.openai.spring;
+
+import com.azure.core.http.ProxyOptions;
+import com.azure.core.util.Configuration;
+import dev.langchain4j.model.azure.AzureOpenAiChatModel;
+import dev.langchain4j.model.azure.AzureOpenAiEmbeddingModel;
+import dev.langchain4j.model.azure.AzureOpenAiImageModel;
+import dev.langchain4j.model.azure.AzureOpenAiStreamingChatModel;
+import org.springframework.boot.autoconfigure.AutoConfiguration;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+
+import static dev.langchain4j.azure.openai.spring.Properties.PREFIX;
+
+@AutoConfiguration
+@EnableConfigurationProperties(Properties.class)
+public class AutoConfig {
+
+    @Bean
+    @ConditionalOnProperty(PREFIX + ".chat-model.api-key")
+    AzureOpenAiChatModel openAiChatModel(Properties properties) {
+        ChatModelProperties chatModelProperties = properties.getChatModel();
+        return AzureOpenAiChatModel.builder()
+                .endpoint(chatModelProperties.getEndpoint())
+                .apiKey(chatModelProperties.getApiKey())
+                .deploymentName(chatModelProperties.getDeploymentName())
+                .temperature(chatModelProperties.getTemperature())
+                .topP(chatModelProperties.getTopP())
+                .stop(chatModelProperties.getStop())
+                .maxTokens(chatModelProperties.getMaxTokens())
+                .presencePenalty(chatModelProperties.getPresencePenalty())
+                .frequencyPenalty(chatModelProperties.getFrequencyPenalty())
+                .timeout(chatModelProperties.getTimeout())
+                .maxRetries(chatModelProperties.getMaxRetries())
+                .proxyOptions(ProxyOptions.fromConfiguration(Configuration.getGlobalConfiguration()))
+                .logRequestsAndResponses(chatModelProperties.getLogRequestsAndResponses() != null && chatModelProperties.getLogRequestsAndResponses())
+                .build();
+    }
+
+    @Bean
+    @ConditionalOnProperty(PREFIX + ".streaming-chat-model.api-key")
+    AzureOpenAiStreamingChatModel openAiStreamingChatModel(Properties properties) {
+        ChatModelProperties chatModelProperties = properties.getStreamingChatModel();
+        return AzureOpenAiStreamingChatModel.builder()
+                .endpoint(chatModelProperties.getEndpoint())
+                .apiKey(chatModelProperties.getApiKey())
+                .deploymentName(chatModelProperties.getDeploymentName())
+                .temperature(chatModelProperties.getTemperature())
+                .topP(chatModelProperties.getTopP())
+                .stop(chatModelProperties.getStop())
+                .maxTokens(chatModelProperties.getMaxTokens())
+                .presencePenalty(chatModelProperties.getPresencePenalty())
+                .frequencyPenalty(chatModelProperties.getFrequencyPenalty())
+                .timeout(chatModelProperties.getTimeout())
+                .proxyOptions(ProxyOptions.fromConfiguration(Configuration.getGlobalConfiguration()))
+                .logRequestsAndResponses(chatModelProperties.getLogRequestsAndResponses() != null && chatModelProperties.getLogRequestsAndResponses())
+                .build();
+    }
+
+    @Bean
+    @ConditionalOnProperty(PREFIX + ".embedding-model.api-key")
+    AzureOpenAiEmbeddingModel openAiEmbeddingModel(Properties properties) {
+        EmbeddingModelProperties embeddingModelProperties = properties.getEmbeddingModel();
+        return AzureOpenAiEmbeddingModel.builder()
+                .endpoint(embeddingModelProperties.getEndpoint())
+                .apiKey(embeddingModelProperties.getApiKey())
+                .deploymentName(embeddingModelProperties.getDeploymentName())
+                .timeout(embeddingModelProperties.getTimeout())
+                .maxRetries(embeddingModelProperties.getMaxRetries())
+                .proxyOptions(ProxyOptions.fromConfiguration(Configuration.getGlobalConfiguration()))
+                .logRequestsAndResponses(embeddingModelProperties.getLogRequestsAndResponses() != null && embeddingModelProperties.getLogRequestsAndResponses())
+                .build();
+    }
+
+    @Bean
+    @ConditionalOnProperty(PREFIX + ".image-model.api-key")
+    AzureOpenAiImageModel openAiImageModel(Properties properties) {
+        ImageModelProperties imageModelProperties = properties.getImageModel();
+        return AzureOpenAiImageModel.builder()
+                .endpoint(imageModelProperties.getEndpoint())
+                .apiKey(imageModelProperties.getApiKey())
+                .deploymentName(imageModelProperties.getDeploymentName())
+                .size(imageModelProperties.getSize())
+                .quality(imageModelProperties.getQuality())
+                .style(imageModelProperties.getStyle())
+                .user(imageModelProperties.getUser())
+                .responseFormat(imageModelProperties.getResponseFormat())
+                .timeout(imageModelProperties.getTimeout())
+                .maxRetries(imageModelProperties.getMaxRetries())
+                .proxyOptions(ProxyOptions.fromConfiguration(Configuration.getGlobalConfiguration()))
+                .logRequestsAndResponses(imageModelProperties.getLogRequestsAndResponses() != null && imageModelProperties.getLogRequestsAndResponses())
+                .build();
+    }
+}

langchain4j-azure-spring-boot-starter/src/main/java/dev/langchain4j/azure/openai/spring/ChatModelProperties.java

@@ -0,0 +1,31 @@
+package dev.langchain4j.azure.openai.spring;
+
+import lombok.Getter;
+import lombok.Setter;
+
+import java.time.Duration;
+import java.util.List;
+import java.util.Map;
+
+@Getter
+@Setter
+class ChatModelProperties {
+
+    String endpoint;
+    String apiKey;
+    String organizationId;
+    String deploymentName;
+    Double temperature;
+    Double topP;
+    List<String> stop;
+    Integer maxTokens;
+    Double presencePenalty;
+    Double frequencyPenalty;
+    Map<String, Integer> logitBias;
+    String responseFormat;
+    Integer seed;
+    String user;
+    Duration timeout;
+    Integer maxRetries;
+    Boolean logRequestsAndResponses;
+}

langchain4j-azure-spring-boot-starter/src/main/java/dev/langchain4j/azure/openai/spring/EmbeddingModelProperties.java

@@ -0,0 +1,21 @@
+package dev.langchain4j.azure.openai.spring;
+
+import lombok.Getter;
+import lombok.Setter;
+
+import java.time.Duration;
+
+@Getter
+@Setter
+class EmbeddingModelProperties {
+
+    String endpoint;
+    String apiKey;
+    String organizationId;
+    String deploymentName;
+    Integer dimensions;
+    String user;
+    Duration timeout;
+    Integer maxRetries;
+    Boolean logRequestsAndResponses;
+}

langchain4j-azure-spring-boot-starter/src/main/java/dev/langchain4j/azure/openai/spring/ImageModelProperties.java

@@ -0,0 +1,27 @@
+package dev.langchain4j.azure.openai.spring;
+
+import lombok.Getter;
+import lombok.Setter;
+
+import java.nio.file.Path;
+import java.time.Duration;
+
+@Getter
+@Setter
+class ImageModelProperties {
+
+    String endpoint;
+    String apiKey;
+    String organizationId;
+    String deploymentName;
+    String size;
+    String quality;
+    String style;
+    String user;
+    String responseFormat;
+    Duration timeout;
+    Integer maxRetries;
+    Boolean logRequestsAndResponses;
+    Boolean withPersisting;
+    Path persistTo;
+}

langchain4j-azure-spring-boot-starter/src/main/java/dev/langchain4j/azure/openai/spring/Properties.java

@@ -0,0 +1,26 @@
+package dev.langchain4j.azure.openai.spring;
+
+import lombok.Getter;
+import lombok.Setter;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.boot.context.properties.NestedConfigurationProperty;
+
+@Getter
+@Setter
+@ConfigurationProperties(prefix = Properties.PREFIX)
+public class Properties {
+
+    static final String PREFIX = "langchain4j.azure.open-ai";
+
+    @NestedConfigurationProperty
+    ChatModelProperties chatModel;
+
+    @NestedConfigurationProperty
+    ChatModelProperties streamingChatModel;
+
+    @NestedConfigurationProperty
+    EmbeddingModelProperties embeddingModel;
+
+    @NestedConfigurationProperty
+    ImageModelProperties imageModel;
+}

langchain4j-azure-spring-boot-starter/src/main/resources/META-INF/spring.factories

@@ -0,0 +1,3 @@
+org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
+dev.langchain4j.azure.openai.spring.AutoConfig,\
+dev.langchain4j.azure.aisearch.spring.AutoConfig

langchain4j-azure-spring-boot-starter/src/main/resources/META-INF/spring/org.springframework.boot.autoconfigure.AutoConfiguration.imports

@@ -0,0 +1,2 @@
+dev.langchain4j.azure.openai.spring.AutoConfig
+dev.langchain4j.azure.aisearch.spring.AutoConfig

langchain4j-azure-spring-boot-starter/src/test/java/dev/langchain4j/azure/aisearch/spring/AutoConfigIT.java

@@ -0,0 +1,41 @@
+package dev.langchain4j.azure.aisearch.spring;
+
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.store.embedding.EmbeddingStore;
+import dev.langchain4j.store.embedding.azure.search.AzureAiSearchEmbeddingStore;
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.autoconfigure.AutoConfigurations;
+import org.springframework.boot.test.context.runner.ApplicationContextRunner;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class AutoConfigIT {
+
+    private static final String AZURE_AISEARCH_API_KEY = System.getenv("AZURE_AISEARCH_API_KEY");
+    private static final String AZURE_AISEARCH_ENDPOINT = System.getenv("AZURE_AISEARCH_ENDPOINT");
+    private static final String AZURE_AISEARCH_DIMENSIONS = System.getenv("AZURE_AISEARCH_DIMENSIONS");
+    private static final String AZURE_AISEARCH_SETUP_INDEX = System.getenv("AZURE_AISEARCH_SETUP_INDEX");
+
+
+    ApplicationContextRunner contextRunner = new ApplicationContextRunner()
+            .withConfiguration(AutoConfigurations.of(AutoConfig.class));
+
+    @Test
+    void should_provide_ai_search_store() {
+        contextRunner
+                .withPropertyValues(
+                        "langchain4j.azure.ai-search.api-key=" + AZURE_AISEARCH_API_KEY,
+                        "langchain4j.azure.ai-search.endpoint=" + AZURE_AISEARCH_ENDPOINT,
+                        "langchain4j.azure.ai-search.dimensions=" + AZURE_AISEARCH_DIMENSIONS,
+                        "langchain4j.azure.ai-search.setup-index=" + AZURE_AISEARCH_SETUP_INDEX
+                )
+                .run(context -> {
+
+                    EmbeddingStore<TextSegment> azureAiSearchEmbeddingStore = context.getBean(AzureAiSearchEmbeddingStore.class);
+                    assertThat(azureAiSearchEmbeddingStore).isInstanceOf(AzureAiSearchEmbeddingStore.class);
+
+                    assertThat(context.getBean(AzureAiSearchEmbeddingStore.class)).isSameAs(azureAiSearchEmbeddingStore);
+                });
+    }
+
+}