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

[improvement][common] Fix the issue of missing 'ORDER BY' clause generation (#1802)

Browse Source
This commit is contained in:
lexluo09
2024-10-15 10:22:22 +08:00
committed by GitHub
parent 1ef642d0dd
commit d7f301064a
6 changed files with 61 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -32,15 +32,10 @@ public class PgvectorEmbeddingStoreFactory extends BaseEmbeddingStoreFactory {
@Override
public EmbeddingStore<TextSegment> createEmbeddingStore(String collectionName) {
return PgVectorEmbeddingStore.builder()
.host(storeProperties.getHost())
.port(storeProperties.getPort())
.database(storeProperties.getDatabase())
.user(storeProperties.getUser())
.password(storeProperties.getPassword())
.table(collectionName)
.dimension(storeProperties.getDimension())
.build();
return PgVectorEmbeddingStore.builder().host(storeProperties.getHost())
.port(storeProperties.getPort()).database(storeProperties.getDatabase())
.user(storeProperties.getUser()).password(storeProperties.getPassword())
.table(collectionName).dimension(storeProperties.getDimension()).build();
}
}

3
common/src/main/java/dev/langchain4j/pgvector/spring/Properties.java
View File

@@ -12,5 +12,6 @@ public class Properties {
static final String PREFIX = "langchain4j.pgvector";
@NestedConfigurationProperty EmbeddingStoreProperties embeddingStore;
@NestedConfigurationProperty
EmbeddingStoreProperties embeddingStore;
}

6
common/src/main/java/dev/langchain4j/store/embedding/EmbeddingStoreFactoryProvider.java
View File

@@ -35,9 +35,9 @@ public class EmbeddingStoreFactoryProvider {
return factoryMap.computeIfAbsent(embeddingStoreConfig,
storeConfig -> new MilvusEmbeddingStoreFactory(storeConfig));
}
if (EmbeddingStoreType.PGVECTOR.name().equalsIgnoreCase(embeddingStoreConfig.getProvider())) {
return factoryMap.computeIfAbsent(
embeddingStoreConfig,
if (EmbeddingStoreType.PGVECTOR.name()
.equalsIgnoreCase(embeddingStoreConfig.getProvider())) {
return factoryMap.computeIfAbsent(embeddingStoreConfig,
storeConfig -> new PgvectorEmbeddingStoreFactory(storeConfig));
}
if (EmbeddingStoreType.IN_MEMORY.name()