Skip to content

Commit 76a390b

Browse files
committed
dependencies: upgrade Spring AI to 1.0.0-M2
1 parent 8b4ab3d commit 76a390b

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

redis-om-spring/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@
8080
<djl.starter.version>0.26</djl.starter.version>
8181
<djl.version>0.27.0</djl.version>
8282
<junit-bom.version>5.10.2</junit-bom.version>
83-
<spring-ai.version>1.0.0-M1</spring-ai.version>
83+
<spring-ai.version>1.0.0-M2</spring-ai.version>
8484
</properties>
8585

8686
<dependencyManagement>

redis-om-spring/src/main/java/com/redis/om/spring/vectorize/DefaultEmbedder.java

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -163,25 +163,26 @@ private float[] getSentenceEmbeddingAsFloatArrayFor(String text) {
163163
private List<byte[]> getEmbeddingsAsByteArrayFor(List<String> texts, EmbeddingModel model) {
164164
EmbeddingResponse embeddingResponse = model.embedForResponse(texts);
165165
List<Embedding> embeddings = embeddingResponse.getResults();
166-
return embeddings.stream().map(e -> ObjectUtils.doubleListToByteArray(e.getOutput())).toList();
166+
167+
return embeddings.stream().map(e -> ObjectUtils.floatArrayToByteArray(e.getOutput())).toList();
167168
}
168169

169170
private List<float[]> getEmbeddingAsFloatArrayFor(List<String> texts, EmbeddingModel model) {
170171
EmbeddingResponse embeddingResponse = model.embedForResponse(texts);
171172
List<Embedding> embeddings = embeddingResponse.getResults();
172-
return embeddings.stream().map(e -> ObjectUtils.doubleListToFloatArray(e.getOutput())).toList();
173+
return embeddings.stream().map(Embedding::getOutput).toList();
173174
}
174175

175176
private byte[] getEmbeddingsAsByteArrayFor(String text, EmbeddingModel model) {
176177
EmbeddingResponse embeddingResponse = model.embedForResponse(List.of(text));
177178
Embedding embedding = embeddingResponse.getResult();
178-
return ObjectUtils.doubleListToByteArray(embedding.getOutput());
179+
return ObjectUtils.floatArrayToByteArray(embedding.getOutput());
179180
}
180181

181182
private float[] getEmbeddingAsFloatArrayFor(String text, EmbeddingModel model) {
182183
EmbeddingResponse embeddingResponse = model.embedForResponse(List.of(text));
183184
Embedding embedding = embeddingResponse.getResult();
184-
return ObjectUtils.doubleListToFloatArray(embedding.getOutput());
185+
return embedding.getOutput();
185186
}
186187

187188
@Override

redis-om-spring/src/test/java/com/redis/om/spring/annotations/hash/vectorize/VectorizeOllamaHashTest.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,7 @@ void testEmbedderCanVectorizeSentence() {
153153
List<byte[]> embeddings = embedder.getTextEmbeddingsAsBytes(List.of(cat.getText()), HashWithOllamaEmbedding$.TEXT);
154154
assertAll( //
155155
() -> assertThat(embeddings).isNotEmpty(), //
156+
() -> assertThat(embeddings.get(0).length).isEqualTo(catEmbedding.length), //
156157
() -> assertThat(embeddings.get(0)).isEqualTo(catEmbedding)
157158
);
158159
}

0 commit comments

Comments
 (0)