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));