Skip to content

Commit a0ef117

Browse files
committed
Fix PgVectorStoreWithChatMemoryAdvisorIT
- Use the PgVectorStoreBuilder to create the instance of PgVectorStore
1 parent 1e68862 commit a0ef117

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vector-stores/spring-ai-pgvector-store/src/test/java/org/springframework/ai/vectorstore/pgvector/PgVectorStoreWithChatMemoryAdvisorIT.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,10 @@ private static void initStore(PgVectorStore store) throws Exception {
8484

8585
private static PgVectorStore createPgVectorStoreUsingTestcontainer(EmbeddingModel embeddingModel) throws Exception {
8686
JdbcTemplate jdbcTemplate = createJdbcTemplateWithConnectionToTestcontainer();
87-
PgVectorStore vectorStore = new PgVectorStore.Builder(jdbcTemplate, embeddingModel).withDimensions(3) // match
87+
PgVectorStore vectorStore = PgVectorStore.builder(jdbcTemplate, embeddingModel)
88+
.dimensions(3) // match
8889
// embeddings
89-
.withInitializeSchema(true)
90+
.initializeSchema(true)
9091
.build();
9192
initStore(vectorStore);
9293
return vectorStore;

0 commit comments

Comments
 (0)