Skip to content

Commit 8b39617

Browse files
committed
Ollama Embedding keep_alive Duration format
Signed-off-by: lance <[email protected]>
1 parent 3fc1ed6 commit 8b39617

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

models/spring-ai-ollama/src/main/java/org/springframework/ai/ollama/OllamaEmbeddingModel.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -178,8 +178,8 @@ OllamaApi.EmbeddingsRequest ollamaEmbeddingRequest(EmbeddingRequest embeddingReq
178178
}
179179

180180
return new OllamaApi.EmbeddingsRequest(requestOptions.getModel(), embeddingRequest.getInstructions(),
181-
DurationParser.parse(requestOptions.getKeepAlive()),
182-
OllamaEmbeddingOptions.filterNonSupportedFields(requestOptions.toMap()), requestOptions.getTruncate());
181+
requestOptions.getKeepAlive(), OllamaEmbeddingOptions.filterNonSupportedFields(requestOptions.toMap()),
182+
requestOptions.getTruncate());
183183
}
184184

185185
/**

models/spring-ai-ollama/src/main/java/org/springframework/ai/ollama/api/OllamaApi.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -620,7 +620,7 @@ public Duration getEvalDuration() {
620620
public record EmbeddingsRequest(
621621
@JsonProperty("model") String model,
622622
@JsonProperty("input") List<String> input,
623-
@JsonProperty("keep_alive") Duration keepAlive,
623+
@JsonProperty("keep_alive") String keepAlive,
624624
@JsonProperty("options") Map<String, Object> options,
625625
@JsonProperty("truncate") Boolean truncate) {
626626

0 commit comments

Comments
 (0)