diff --git a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/pom.xml b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/pom.xml
index 522c75d8b..6be03393b 100644
--- a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/pom.xml
+++ b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/pom.xml
@@ -96,6 +96,12 @@
ollama
test
+
+
+ com.fasterxml.jackson.module
+ jackson-module-kotlin
+ test
+
diff --git a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaChatAutoConfigurationIT.java b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaChatAutoConfigurationIT.java
index 65e388a59..9ecca5a29 100644
--- a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaChatAutoConfigurationIT.java
+++ b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaChatAutoConfigurationIT.java
@@ -114,8 +114,8 @@ public class OllamaChatAutoConfigurationIT extends BaseOllamaIT {
@Test
void chatActivation() {
- this.contextRunner.withPropertyValues("spring.ai.ollama.chat.enabled=false").run(context -> {
- assertThat(context.getBeansOfType(OllamaChatProperties.class)).isNotEmpty();
+ this.contextRunner.withPropertyValues("spring.ai.model.chat=none").run(context -> {
+ assertThat(context.getBeansOfType(OllamaChatProperties.class)).isEmpty();
assertThat(context.getBeansOfType(OllamaChatModel.class)).isEmpty();
});
@@ -124,7 +124,7 @@ public class OllamaChatAutoConfigurationIT extends BaseOllamaIT {
assertThat(context.getBeansOfType(OllamaChatModel.class)).isNotEmpty();
});
- this.contextRunner.withPropertyValues("spring.ai.ollama.chat.enabled=true").run(context -> {
+ this.contextRunner.withPropertyValues("spring.ai.model.chat=ollama").run(context -> {
assertThat(context.getBeansOfType(OllamaChatProperties.class)).isNotEmpty();
assertThat(context.getBeansOfType(OllamaChatModel.class)).isNotEmpty();
});
diff --git a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaEmbeddingAutoConfigurationIT.java b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaEmbeddingAutoConfigurationIT.java
index 7bf54314f..d01c5205a 100644
--- a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaEmbeddingAutoConfigurationIT.java
+++ b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaEmbeddingAutoConfigurationIT.java
@@ -84,8 +84,8 @@ public class OllamaEmbeddingAutoConfigurationIT extends BaseOllamaIT {
@Test
void embeddingActivation() {
- this.contextRunner.withPropertyValues("spring.ai.ollama.embedding.enabled=false").run(context -> {
- assertThat(context.getBeansOfType(OllamaEmbeddingProperties.class)).isNotEmpty();
+ this.contextRunner.withPropertyValues("spring.ai.model.embedding=none").run(context -> {
+ assertThat(context.getBeansOfType(OllamaEmbeddingProperties.class)).isEmpty();
assertThat(context.getBeansOfType(OllamaEmbeddingModel.class)).isEmpty();
});
diff --git a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/kotlin/org/springframework/ai/model/ollama/autoconfigure/tool/FunctionCallbackContextKotlinIT.kt b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/kotlin/org/springframework/ai/model/ollama/autoconfigure/tool/FunctionCallbackContextKotlinIT.kt
index 27182cbd9..1b7d10077 100644
--- a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/kotlin/org/springframework/ai/model/ollama/autoconfigure/tool/FunctionCallbackContextKotlinIT.kt
+++ b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/kotlin/org/springframework/ai/model/ollama/autoconfigure/tool/FunctionCallbackContextKotlinIT.kt
@@ -20,12 +20,11 @@ import org.assertj.core.api.Assertions.assertThat
import org.junit.jupiter.api.BeforeAll
import org.junit.jupiter.api.Test
import org.slf4j.LoggerFactory
-
-import org.springframework.ai.model.ollama.autoconfigure.BaseOllamaIT
-import org.springframework.ai.model.ollama.autoconfigure.OllamaChatAutoConfiguration
import org.springframework.ai.chat.messages.UserMessage
import org.springframework.ai.chat.prompt.Prompt
-import org.springframework.ai.model.function.FunctionCallingOptions
+import org.springframework.ai.model.ollama.autoconfigure.BaseOllamaIT
+import org.springframework.ai.model.ollama.autoconfigure.OllamaChatAutoConfiguration
+import org.springframework.ai.model.tool.ToolCallingChatOptions
import org.springframework.ai.ollama.OllamaChatModel
import org.springframework.ai.ollama.api.OllamaOptions
import org.springframework.boot.autoconfigure.AutoConfigurations
@@ -87,8 +86,8 @@ class FunctionCallbackResolverKotlinIT : BaseOllamaIT() {
val userMessage = UserMessage(
"What are the weather conditions in San Francisco, Tokyo, and Paris? Find the temperature in Celsius for each of the three locations.")
- val functionOptions = FunctionCallingOptions.builder()
- .function("weatherInfo")
+ val functionOptions = ToolCallingChatOptions.builder()
+ .toolNames("weatherInfo")
.build()
val response = chatModel.call(Prompt(listOf(userMessage), functionOptions));