From a0ef1179c4edd6fffdcb7dd5e0892ae064f5d16b Mon Sep 17 00:00:00 2001
From: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
Date: Fri, 20 Dec 2024 20:38:01 +0000
Subject: [PATCH] Fix PgVectorStoreWithChatMemoryAdvisorIT

 - Use the PgVectorStoreBuilder to create the instance of PgVectorStore
---
 .../pgvector/PgVectorStoreWithChatMemoryAdvisorIT.java       | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/vector-stores/spring-ai-pgvector-store/src/test/java/org/springframework/ai/vectorstore/pgvector/PgVectorStoreWithChatMemoryAdvisorIT.java b/vector-stores/spring-ai-pgvector-store/src/test/java/org/springframework/ai/vectorstore/pgvector/PgVectorStoreWithChatMemoryAdvisorIT.java
index 5ecf7c354..5fadc8599 100644
--- a/vector-stores/spring-ai-pgvector-store/src/test/java/org/springframework/ai/vectorstore/pgvector/PgVectorStoreWithChatMemoryAdvisorIT.java
+++ b/vector-stores/spring-ai-pgvector-store/src/test/java/org/springframework/ai/vectorstore/pgvector/PgVectorStoreWithChatMemoryAdvisorIT.java
@@ -84,9 +84,10 @@ class PgVectorStoreWithChatMemoryAdvisorIT {
 
 	private static PgVectorStore createPgVectorStoreUsingTestcontainer(EmbeddingModel embeddingModel) throws Exception {
 		JdbcTemplate jdbcTemplate = createJdbcTemplateWithConnectionToTestcontainer();
-		PgVectorStore vectorStore = new PgVectorStore.Builder(jdbcTemplate, embeddingModel).withDimensions(3) // match
+		PgVectorStore vectorStore = PgVectorStore.builder(jdbcTemplate, embeddingModel)
+			.dimensions(3) // match
 			// embeddings
-			.withInitializeSchema(true)
+			.initializeSchema(true)
 			.build();
 		initStore(vectorStore);
 		return vectorStore;