diff --git a/common/src/main/java/com/tencent/supersonic/common/config/EmbeddingStoreParameterConfig.java b/common/src/main/java/com/tencent/supersonic/common/config/EmbeddingStoreParameterConfig.java index 126d8d7d6..5591e63cb 100644 --- a/common/src/main/java/com/tencent/supersonic/common/config/EmbeddingStoreParameterConfig.java +++ b/common/src/main/java/com/tencent/supersonic/common/config/EmbeddingStoreParameterConfig.java @@ -100,7 +100,9 @@ public class EmbeddingStoreParameterConfig extends ParameterConfig { } private static List getApiKeyDependency() { - return Lists.newArrayList(); + return getDependency(EMBEDDING_STORE_PROVIDER.getName(), + Lists.newArrayList(EmbeddingStoreType.MILVUS.name()), + ImmutableMap.of(EmbeddingStoreType.MILVUS.name(), DEMO)); } private static List getPathDependency() { diff --git a/common/src/main/java/dev/langchain4j/pgvector/spring/PgvectorEmbeddingStoreFactory.java b/common/src/main/java/dev/langchain4j/pgvector/spring/PgvectorEmbeddingStoreFactory.java index cde961c8c..b840f6df7 100644 --- a/common/src/main/java/dev/langchain4j/pgvector/spring/PgvectorEmbeddingStoreFactory.java +++ b/common/src/main/java/dev/langchain4j/pgvector/spring/PgvectorEmbeddingStoreFactory.java @@ -26,7 +26,7 @@ public class PgvectorEmbeddingStoreFactory extends BaseEmbeddingStoreFactory { embeddingStore.setPort(storeConfig.getPost()); embeddingStore.setDatabase(storeConfig.getDatabaseName()); embeddingStore.setUser(storeConfig.getUser()); - embeddingStore.setPassword(storeConfig.getApiKey()); + embeddingStore.setPassword(storeConfig.getPassword()); return embeddingStore; } diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/dataobject/QueryStatDO.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/dataobject/QueryStatDO.java index 67a7edabf..4235390d7 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/dataobject/QueryStatDO.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/dataobject/QueryStatDO.java @@ -28,6 +28,7 @@ public class QueryStatDO { private String queryStructCmd; @TableField("struct_cmd_md5") private String queryStructCmdMd5; + @TableField("\"sql\"") private String sql; private String sqlMd5; private String queryEngine;