From 6ffbb3acc692ca8c2246ae3dcda706df857bc5d4 Mon Sep 17 00:00:00 2001 From: Mark Pollack Date: Sat, 3 May 2025 18:22:05 -0400 Subject: [PATCH] Add simple hello world test to reproduce issue #2957 with qwen/ollama --- .../tool/OllamaFunctionCallbackIT.java | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/tool/OllamaFunctionCallbackIT.java b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/tool/OllamaFunctionCallbackIT.java index cbaf6362b..d24df5fb8 100644 --- a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/tool/OllamaFunctionCallbackIT.java +++ b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/tool/OllamaFunctionCallbackIT.java @@ -25,6 +25,7 @@ import org.slf4j.Logger; import org.slf4j.LoggerFactory; import reactor.core.publisher.Flux; +import org.springframework.ai.chat.client.ChatClient; import org.springframework.ai.chat.messages.AssistantMessage; import org.springframework.ai.chat.messages.UserMessage; import org.springframework.ai.chat.model.ChatResponse; @@ -65,6 +66,24 @@ public class OllamaFunctionCallbackIT extends BaseOllamaIT { initializeOllama(MODEL_NAME); } + /** + * See https://github.com/spring-projects/spring-ai/issues/2957 + */ + @Test + void chatClientHelloWorld() { + this.contextRunner.run(context -> { + + OllamaChatModel chatModel = context.getBean(OllamaChatModel.class); + ChatClient chatClient = ChatClient.builder(chatModel).build(); + + UserMessage userMessage = new UserMessage("What is 2+2"); + + var response = chatClient.prompt(new Prompt(userMessage)).call().content(); + logger.info("Response: " + response); + + }); + } + @Test void functionCallTest() { this.contextRunner.run(context -> {