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