topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 12:37:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[fix][common]Fix embedding store sys configs.

Browse Source
This commit is contained in:
jerryjzhang
2024-12-02 08:50:31 +08:00
parent 7de3662259
commit 27c6b8ecb1
3 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
common/src/main/java/com/tencent/supersonic/common/config/EmbeddingStoreParameterConfig.java
View File

@@ -100,7 +100,9 @@ public class EmbeddingStoreParameterConfig extends ParameterConfig {
} }
private static List<Parameter.Dependency> getApiKeyDependency() { private static List<Parameter.Dependency> getApiKeyDependency() {
return Lists.newArrayList(); return getDependency(EMBEDDING_STORE_PROVIDER.getName(),
Lists.newArrayList(EmbeddingStoreType.MILVUS.name()),
ImmutableMap.of(EmbeddingStoreType.MILVUS.name(), DEMO));
} }
private static List<Parameter.Dependency> getPathDependency() { private static List<Parameter.Dependency> getPathDependency() {

2
common/src/main/java/dev/langchain4j/pgvector/spring/PgvectorEmbeddingStoreFactory.java
View File

@@ -26,7 +26,7 @@ public class PgvectorEmbeddingStoreFactory extends BaseEmbeddingStoreFactory {
embeddingStore.setPort(storeConfig.getPost()); embeddingStore.setPort(storeConfig.getPost());
embeddingStore.setDatabase(storeConfig.getDatabaseName()); embeddingStore.setDatabase(storeConfig.getDatabaseName());
embeddingStore.setUser(storeConfig.getUser()); embeddingStore.setUser(storeConfig.getUser());
embeddingStore.setPassword(storeConfig.getApiKey()); embeddingStore.setPassword(storeConfig.getPassword());
return embeddingStore; return embeddingStore;
} }

1
headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/dataobject/QueryStatDO.java
View File

@@ -28,6 +28,7 @@ public class QueryStatDO {
private String queryStructCmd; private String queryStructCmd;
@TableField("struct_cmd_md5") @TableField("struct_cmd_md5")
private String queryStructCmdMd5; private String queryStructCmdMd5;
@TableField("\"sql\"")
private String sql; private String sql;
private String sqlMd5; private String sqlMd5;
private String queryEngine; private String queryEngine;