diff --git a/hertzbeat-ai/pom.xml b/hertzbeat-ai/pom.xml
index 235b2481dce..249c6be8c03 100644
--- a/hertzbeat-ai/pom.xml
+++ b/hertzbeat-ai/pom.xml
@@ -26,7 +26,7 @@
hertzbeat-ai
${hertzbeat.version}
- 1.1.1
+ 2.0.0
25
diff --git a/hertzbeat-ai/src/main/java/org/apache/hertzbeat/ai/config/LlmConfig.java b/hertzbeat-ai/src/main/java/org/apache/hertzbeat/ai/config/LlmConfig.java
index b145344610c..bd56377716c 100644
--- a/hertzbeat-ai/src/main/java/org/apache/hertzbeat/ai/config/LlmConfig.java
+++ b/hertzbeat-ai/src/main/java/org/apache/hertzbeat/ai/config/LlmConfig.java
@@ -27,7 +27,6 @@
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.openai.OpenAiChatModel;
import org.springframework.ai.openai.OpenAiChatOptions;
-import org.springframework.ai.openai.api.OpenAiApi;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ConfigurableApplicationContext;
@@ -101,21 +100,17 @@ private ChatClient createChatClient() {
}
}
- OpenAiApi.Builder builder = new OpenAiApi.Builder();
- builder.baseUrl(modelProviderConfig.getBaseUrl());
- builder.apiKey(modelProviderConfig.getApiKey());
- builder.completionsPath("/chat/completions");
-
- // Create Chat Options
+ // Create Chat Options with baseUrl and apiKey
OpenAiChatOptions openAiChatOptions = OpenAiChatOptions.builder()
+ .baseUrl(modelProviderConfig.getBaseUrl())
+ .apiKey(modelProviderConfig.getApiKey())
.model(modelProviderConfig.getModel())
.temperature(0.3)
.build();
-
+
// Create Chat Model
OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
- .openAiApi(builder.build())
- .defaultOptions(openAiChatOptions)
+ .options(openAiChatOptions)
.build();
// Create and return ChatClient