(improvement)(chat) Upgrade and optimize the embedding metastore. (#1198)

common/src/main/java/dev/langchain4j/inmemory/spring/EmbeddingStoreProperties.java

@@ -0,0 +1,11 @@
+package dev.langchain4j.inmemory.spring;
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+class EmbeddingStoreProperties {
+
+    private String filePath;
+}

common/src/main/java/dev/langchain4j/inmemory/spring/InMemoryAutoConfig.java

@@ -0,0 +1,21 @@
+package dev.langchain4j.inmemory.spring;
+
+
+import static dev.langchain4j.inmemory.spring.Properties.PREFIX;
+
+import dev.langchain4j.store.embedding.EmbeddingStoreFactory;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+@EnableConfigurationProperties(Properties.class)
+public class InMemoryAutoConfig {
+
+    @Bean
+    @ConditionalOnProperty(PREFIX + ".embedding-store.file-path")
+    EmbeddingStoreFactory milvusChatModel(Properties properties) {
+        return new InMemoryEmbeddingStoreFactory(properties);
+    }
+}

common/src/main/java/dev/langchain4j/inmemory/spring/InMemoryEmbeddingStoreFactory.java

@@ -0,0 +1,37 @@
+package dev.langchain4j.inmemory.spring;
+
+import dev.langchain4j.store.embedding.EmbeddingQuery;
+import dev.langchain4j.store.embedding.EmbeddingStore;
+import dev.langchain4j.store.embedding.EmbeddingStoreFactory;
+import dev.langchain4j.store.embedding.inmemory.InMemoryEmbeddingStore;
+import java.util.Map;
+import java.util.Objects;
+import java.util.concurrent.ConcurrentHashMap;
+import lombok.extern.slf4j.Slf4j;
+
+@Slf4j
+public class InMemoryEmbeddingStoreFactory implements EmbeddingStoreFactory {
+
+    private static Map<String, InMemoryEmbeddingStore<EmbeddingQuery>> collectionNameToStore =
+            new ConcurrentHashMap<>();
+    private Properties properties;
+
+
+    public InMemoryEmbeddingStoreFactory(Properties properties) {
+        this.properties = properties;
+    }
+
+    @Override
+    public synchronized EmbeddingStore create(String collectionName) {
+        InMemoryEmbeddingStore<EmbeddingQuery> embeddingStore = collectionNameToStore.get(collectionName);
+        if (Objects.nonNull(embeddingStore)) {
+            return embeddingStore;
+        }
+        if (Objects.isNull(embeddingStore)) {
+            embeddingStore = new InMemoryEmbeddingStore();
+            collectionNameToStore.putIfAbsent(collectionName, embeddingStore);
+        }
+        return embeddingStore;
+
+    }
+}

common/src/main/java/dev/langchain4j/inmemory/spring/Properties.java

@@ -0,0 +1,17 @@
+package dev.langchain4j.inmemory.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.in-memory";
+
+    @NestedConfigurationProperty
+    EmbeddingStoreProperties embeddingStore;
+}