This is an automated email from the ASF dual-hosted git repository. xtsong pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/flink-agents.git
commit 1e581a18ad16dccfa847f9d1755a2f2d2250e282 Author: WenjinXie <[email protected]> AuthorDate: Fri Oct 31 11:20:28 2025 +0800 [hotfix] Adapt ollama example after supporting explicit tool call handling in OllamaChatModel. --- .../flink/agents/integration/test/AgentWithOllama.java | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/e2e-test/integration-test/src/main/java/org/apache/flink/agents/integration/test/AgentWithOllama.java b/e2e-test/integration-test/src/main/java/org/apache/flink/agents/integration/test/AgentWithOllama.java index 00c4144..d651db4 100644 --- a/e2e-test/integration-test/src/main/java/org/apache/flink/agents/integration/test/AgentWithOllama.java +++ b/e2e-test/integration-test/src/main/java/org/apache/flink/agents/integration/test/AgentWithOllama.java @@ -30,6 +30,8 @@ import org.apache.flink.agents.api.chat.messages.ChatMessage; import org.apache.flink.agents.api.chat.messages.MessageRole; import org.apache.flink.agents.api.chat.model.BaseChatModelSetup; import org.apache.flink.agents.api.context.RunnerContext; +import org.apache.flink.agents.api.event.ChatRequestEvent; +import org.apache.flink.agents.api.event.ChatResponseEvent; import org.apache.flink.agents.api.resource.ResourceDescriptor; import org.apache.flink.agents.api.resource.ResourceType; import org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelConnection; @@ -126,10 +128,15 @@ public class AgentWithOllama extends Agent { public static void process(InputEvent event, RunnerContext ctx) throws Exception { BaseChatModelSetup chatModel = (BaseChatModelSetup) ctx.getResource("ollamaChatModel", ResourceType.CHAT_MODEL); - ChatMessage response = - chatModel.chat( + ctx.sendEvent( + new ChatRequestEvent( + "ollamaChatModel", Collections.singletonList( - new ChatMessage(MessageRole.USER, (String) event.getInput()))); - ctx.sendEvent(new OutputEvent(response.getContent())); + new ChatMessage(MessageRole.USER, (String) event.getInput())))); + } + + @Action(listenEvents = {ChatResponseEvent.class}) + public static void processChatResponse(ChatResponseEvent event, RunnerContext ctx) { + ctx.sendEvent(new OutputEvent(event.getResponse().getContent())); } }
