Merge pull request #13 from svrc/origin-dev

modifications for recent embeddings changes
0pens0 · Aug 6, 2024 · be072be · be072be
2 parents 7d52f18 + 143feb9
commit be072be
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/conf/manifest.yml b/conf/manifest.yml
@@ -7,6 +7,7 @@ applications:
   routes:
     - route: spring-metal-demo-gitops.vmtanzu.net
       protocol: http2
+
   env:
     JBP_CONFIG_SPRING_AUTO_RECONFIGURATION: '{enabled: false}'
     SPRING_PROFILES_ACTIVE: http2

diff --git a/src/main/java/org/cloudfoundry/samples/music/config/ai/PostgresAiConfiguration.java b/src/main/java/org/cloudfoundry/samples/music/config/ai/PostgresAiConfiguration.java
@@ -35,9 +35,9 @@ public class PostgresAiConfiguration {
 	@Bean
 	public VectorStore vectorStore(JdbcTemplate jcbcTemplate, EmbeddingClient embeddingClient) {
 		return new PgVectorStore(jcbcTemplate, embeddingClient,
-				4096,
+				768,
 				PgVectorStore.PgDistanceType.CosineDistance,
-				false,
+				true,
 				PgVectorStore.PgIndexType.NONE);
 	}
 

diff --git a/src/main/java/org/cloudfoundry/samples/music/config/ai/VectorStoreInitializer.java b/src/main/java/org/cloudfoundry/samples/music/config/ai/VectorStoreInitializer.java
@@ -64,7 +64,7 @@ public void onApplicationEvent(ApplicationReadyEvent event) {
         List<Document> documents = new ArrayList<>();
 
 
-        List<Document> docs = this.vectorStore.similaritySearch("");
+        List<Document> docs = this.vectorStore.similaritySearch("album");
         logger.info("Vector store contains " + docs.size() + " records");
 
         if (docs.size() == 0) {