From bc87f999354bd3f37640b8ff8f2fce9af80358a3 Mon Sep 17 00:00:00 2001 From: Dmytro Skarzhynets Date: Wed, 2 Jul 2025 20:39:17 +0300 Subject: [PATCH] AI rule node: cleanup AiProvider enum --- .../common/data/ai/provider/AiProvider.java | 35 +++++-------------- 1 file changed, 8 insertions(+), 27 deletions(-) diff --git a/common/data/src/main/java/org/thingsboard/server/common/data/ai/provider/AiProvider.java b/common/data/src/main/java/org/thingsboard/server/common/data/ai/provider/AiProvider.java index f4ae843447..d0a5bd0510 100644 --- a/common/data/src/main/java/org/thingsboard/server/common/data/ai/provider/AiProvider.java +++ b/common/data/src/main/java/org/thingsboard/server/common/data/ai/provider/AiProvider.java @@ -15,34 +15,15 @@ */ package org.thingsboard.server.common.data.ai.provider; -import org.thingsboard.server.common.data.ai.model.chat.AmazonBedrockChatModel; -import org.thingsboard.server.common.data.ai.model.chat.AnthropicChatModel; -import org.thingsboard.server.common.data.ai.model.chat.AzureOpenAiChatModel; -import org.thingsboard.server.common.data.ai.model.chat.GitHubModelsChatModel; -import org.thingsboard.server.common.data.ai.model.chat.GoogleAiGeminiChatModel; -import org.thingsboard.server.common.data.ai.model.chat.GoogleVertexAiGeminiChatModel; -import org.thingsboard.server.common.data.ai.model.chat.MistralAiChatModel; -import org.thingsboard.server.common.data.ai.model.chat.OpenAiChatModel; - public enum AiProvider { - OPENAI(OpenAiChatModel.class), - AZURE_OPENAI(AzureOpenAiChatModel.class), - GOOGLE_AI_GEMINI(GoogleAiGeminiChatModel.class), - GOOGLE_VERTEX_AI_GEMINI(GoogleVertexAiGeminiChatModel.class), - MISTRAL_AI(MistralAiChatModel.class), - ANTHROPIC(AnthropicChatModel.class), - AMAZON_BEDROCK(AmazonBedrockChatModel.class), - GITHUB_MODELS(GitHubModelsChatModel.class); - - private final Class defaultModelClass; - - AiProvider(Class defaultModelClass) { - this.defaultModelClass = defaultModelClass; - } - - public Class getDefaultModelClass() { - return defaultModelClass; - } + OPENAI, + AZURE_OPENAI, + GOOGLE_AI_GEMINI, + GOOGLE_VERTEX_AI_GEMINI, + MISTRAL_AI, + ANTHROPIC, + AMAZON_BEDROCK, + GITHUB_MODELS }