pass configs and an extension value to make sure vector extensions are created and can be overridden by passing that config

Author: shrikanthkr

embedding-stores/pgvector/deployment/src/main/java/io/quarkiverse/langchain4j/pgvector/deployment/PgVectorEmbeddingStoreProcessor.java

@@ -80,7 +80,7 @@ public void createBean(
                 .setRuntimeInit()
                 .unremovable()
                 .scope(ApplicationScoped.class)
-                .supplier(recorder.pgVectorAgroalPoolInterceptor())
+                .supplier(recorder.pgVectorAgroalPoolInterceptor(config))
                 .qualifiers(datasourceQualifier)
                 .done());
 

embedding-stores/pgvector/runtime/src/main/java/io/quarkiverse/langchain4j/pgvector/PgVectorAgroalPoolInterceptor.java

@@ -1,5 +1,6 @@
 package io.quarkiverse.langchain4j.pgvector;
 
+import io.quarkiverse.langchain4j.pgvector.runtime.PgVectorEmbeddingStoreConfig;
 import java.sql.Connection;
 import java.sql.SQLException;
 import java.sql.Statement;
@@ -14,10 +15,18 @@
  */
 public class PgVectorAgroalPoolInterceptor implements AgroalPoolInterceptor {
 
+    PgVectorEmbeddingStoreConfig config;
+
+    public PgVectorAgroalPoolInterceptor(PgVectorEmbeddingStoreConfig config) {
+        this.config = config;
+    }
+
     @Override
     public void onConnectionCreate(Connection connection) {
         try (Statement statement = connection.createStatement()) {
-            if (ConfigUtils.isProfileActive("dev") || ConfigUtils.isProfileActive("test")) {
+            boolean createExtension =
+                ConfigUtils.isProfileActive("dev") || ConfigUtils.isProfileActive("test") || this.config.registerVectorPGExtension();
+            if (createExtension) {
                 statement.executeUpdate("CREATE EXTENSION IF NOT EXISTS vector");
             }
             PGvector.addVectorType(connection);

embedding-stores/pgvector/runtime/src/main/java/io/quarkiverse/langchain4j/pgvector/runtime/PgVectorEmbeddingStoreConfig.java

@@ -2,6 +2,7 @@
 
 import static io.quarkus.runtime.annotations.ConfigPhase.RUN_TIME;
 
+import io.smallrye.config.WithName;
 import java.util.List;
 import java.util.Optional;
 
@@ -53,6 +54,14 @@ public interface PgVectorEmbeddingStoreConfig {
     @WithDefault("false")
     Boolean dropTableFirst();
 
+    /**
+     * Whether the PG extension should be created on Start.
+     * By Default, if it's dev or test environment the value is overridden to true
+     */
+    @WithDefault("false")
+    @WithName("register-vector-pg-extension")
+    Boolean registerVectorPGExtension();
+
     /**
      * Metadata configuration.
      */

embedding-stores/pgvector/runtime/src/main/java/io/quarkiverse/langchain4j/pgvector/runtime/PgVectorEmbeddingStoreRecorder.java

@@ -35,11 +35,11 @@ public PgVectorEmbeddingStore apply(SyntheticCreationalContext<PgVectorEmbedding
         };
     }
 
-    public Supplier<PgVectorAgroalPoolInterceptor> pgVectorAgroalPoolInterceptor() {
+    public Supplier<PgVectorAgroalPoolInterceptor> pgVectorAgroalPoolInterceptor(PgVectorEmbeddingStoreConfig config) {
         return new Supplier<>() {
             @Override
             public PgVectorAgroalPoolInterceptor get() {
-                return new PgVectorAgroalPoolInterceptor();
+                return new PgVectorAgroalPoolInterceptor(config);
             }
         };
     }