Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Set embeddings on added Document objects for Milvus and PgVector stores.

Browse Source
This commit is contained in:
Francisco Javier Torres
2024-07-03 12:41:02 +02:00
committed by Mark Pollack
parent 6931148377
commit 867bd173e5
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vector-stores/spring-ai-milvus-store/src/main/java/org/springframework/ai/vectorstore/MilvusVectorStore.java
View File

@@ -272,7 +272,7 @@ public class MilvusVectorStore implements VectorStore, InitializingBean {
for (Document document : documents) {
List<Double> embedding = this.embeddingModel.embed(document);
document.setEmbedding(embedding);
docIdArray.add(document.getId());
// Use a (future) DocumentTextLayoutFormatter instance to extract
// the content used to compute the embeddings

4
vector-stores/spring-ai-pgvector-store/src/main/java/org/springframework/ai/vectorstore/PgVectorStore.java
View File

@@ -166,7 +166,9 @@ public class PgVectorStore implements VectorStore, InitializingBean {
var document = documents.get(i);
var content = document.getContent();
var json = toJson(document.getMetadata());
var pGvector = new PGvector(toFloatArray(embeddingModel.embed(document)));
var embedding = embeddingModel.embed(document);
document.setEmbedding(embedding);
var pGvector = new PGvector(toFloatArray(embedding));
StatementCreatorUtils.setParameterValue(ps, 1, SqlTypeValue.TYPE_UNKNOWN,
UUID.fromString(document.getId()));