diff --git a/hertzbeat-ai/src/main/java/org/apache/hertzbeat/ai/service/impl/ChatClientProviderServiceImpl.java b/hertzbeat-ai/src/main/java/org/apache/hertzbeat/ai/service/impl/ChatClientProviderServiceImpl.java
index e744d3e0177..dec9bfc90c8 100644
--- a/hertzbeat-ai/src/main/java/org/apache/hertzbeat/ai/service/impl/ChatClientProviderServiceImpl.java
+++ b/hertzbeat-ai/src/main/java/org/apache/hertzbeat/ai/service/impl/ChatClientProviderServiceImpl.java
@@ -126,7 +126,7 @@ public Flux<String> streamChat(ChatRequestContext context) {
             return chatClient.prompt()
                 .messages(messages)
                 .system(systemPrompt)
-                .toolCallbacks(toolCallbackProvider)
+                .tools(toolCallbackProvider)
                 .stream()
                 .content()
                 .doOnComplete(() -> log.info("Streaming completed for conversation: {}", context.getConversationId()))