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

Change EmbeddingOptions from class into interface

Browse Source
This commit is contained in:
Christian Tzolov
2024-01-28 19:10:08 +01:00
parent da12b56dc7
commit 7c7392e657
4 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/spring-ai-postgresml/src/main/java/org/springframework/ai/postgresml/PostgresMlEmbeddingClient.java
View File

@@ -156,7 +156,7 @@ public class PostgresMlEmbeddingClient extends AbstractEmbeddingClient implement
@Override
public EmbeddingResponse embedForResponse(List<String> texts) {
return this.call(new EmbeddingRequest(texts, new EmbeddingOptions()));
return this.call(new EmbeddingRequest(texts, EmbeddingOptions.EMPTY));
}
@Override

2
models/spring-ai-transformers/src/main/java/org/springframework/ai/transformers/TransformersEmbeddingClient.java
View File

@@ -214,7 +214,7 @@ public class TransformersEmbeddingClient extends AbstractEmbeddingClient impleme
@Override
public List<List<Double>> embed(List<String> texts) {
return this.call(new EmbeddingRequest(texts, new EmbeddingOptions()))
return this.call(new EmbeddingRequest(texts, EmbeddingOptions.EMPTY))
.getResults()
.stream()
.map(e -> e.getOutput())

4
spring-ai-core/src/main/java/org/springframework/ai/embedding/EmbeddingClient.java
View File

@@ -50,7 +50,7 @@ public interface EmbeddingClient extends ModelClient<EmbeddingRequest, Embedding
*/
default List<List<Double>> embed(List<String> texts) {
Assert.notNull(texts, "Texts must not be null");
return this.call(new EmbeddingRequest(texts, new EmbeddingOptions()))
return this.call(new EmbeddingRequest(texts, EmbeddingOptions.EMPTY))
.getResults()
.stream()
.map(Embedding::getOutput)
@@ -64,7 +64,7 @@ public interface EmbeddingClient extends ModelClient<EmbeddingRequest, Embedding
*/
default EmbeddingResponse embedForResponse(List<String> texts) {
Assert.notNull(texts, "Texts must not be null");
return this.call(new EmbeddingRequest(texts, new EmbeddingOptions()));
return this.call(new EmbeddingRequest(texts, EmbeddingOptions.EMPTY));
}
/**

5
spring-ai-core/src/main/java/org/springframework/ai/embedding/EmbeddingOptions.java
View File

@@ -21,6 +21,9 @@ import org.springframework.ai.model.ModelOptions;
/**
* @author Christian Tzolov
*/
public class EmbeddingOptions implements ModelOptions {
public interface EmbeddingOptions extends ModelOptions {
public static EmbeddingOptions EMPTY = new EmbeddingOptions() {
};
}