From b3b1498af7d57aa47c8c1f64a8a735e34a9443b7 Mon Sep 17 00:00:00 2001 From: jerryjzhang Date: Sun, 1 Dec 2024 22:53:56 +0800 Subject: [PATCH] [fix][launcher]Change default vector dimension to 512. --- .../config/EmbeddingStoreParameterConfig.java | 6 +++--- .../src/main/resources/application-postgres.yaml | 14 +++++++++++++- 2 files changed, 16 insertions(+), 4 deletions(-) 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 69a6b7ae4..b8ff2227d 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 @@ -38,13 +38,13 @@ public class EmbeddingStoreParameterConfig extends ParameterConfig { new Parameter("s2.embedding.store.timeout", "60", "超时时间(秒)", "", "number", MODULE_NAME); public static final Parameter EMBEDDING_STORE_DIMENSION = - new Parameter("s2.embedding.store.dimension", "", "纬度", "", "number", MODULE_NAME, null, + new Parameter("s2.embedding.store.dimension", "", "向量维度", "", "number", MODULE_NAME, null, getDimensionDependency()); public static final Parameter EMBEDDING_STORE_DATABASE_NAME = new Parameter("s2.embedding.store.databaseName", "", "DatabaseName", "", "string", MODULE_NAME, null, getDatabaseNameDependency()); - public static final Parameter EMBEDDING_STORE_POST = new Parameter("s2.embedding.store.post", + public static final Parameter EMBEDDING_STORE_POST = new Parameter("s2.embedding.store.port", "", "端口", "", "number", MODULE_NAME, null, getPostDependency()); public static final Parameter EMBEDDING_STORE_USER = new Parameter("s2.embedding.store.user", @@ -118,7 +118,7 @@ public class EmbeddingStoreParameterConfig extends ParameterConfig { Lists.newArrayList(EmbeddingStoreType.MILVUS.name(), EmbeddingStoreType.PGVECTOR.name()), ImmutableMap.of(EmbeddingStoreType.MILVUS.name(), "384", - EmbeddingStoreType.PGVECTOR.name(), "768")); + EmbeddingStoreType.PGVECTOR.name(), "512")); } private static List getDatabaseNameDependency() { diff --git a/launchers/standalone/src/main/resources/application-postgres.yaml b/launchers/standalone/src/main/resources/application-postgres.yaml index 20f94226e..a9f4f82f3 100644 --- a/launchers/standalone/src/main/resources/application-postgres.yaml +++ b/launchers/standalone/src/main/resources/application-postgres.yaml @@ -11,4 +11,16 @@ spring: username: postgres password: postgres schema-locations: classpath:db/schema-postgres.sql,classpath:db/schema-postgres-demo.sql - data-locations: classpath:db/data-postgres.sql,classpath:db/data-postgres-demo.sql \ No newline at end of file + data-locations: classpath:db/data-postgres.sql,classpath:db/data-postgres-demo.sql + +#s2: +# embedding: +# store: +# provider: PGVECTOR +# base: +# url: 127.0.0.1 +# port: 5432 +# databaseName: postgres +# user: postgres +# password: postgres +# dimension: 512 \ No newline at end of file