AI rule node: cleanup AiProvider enum

This commit is contained in:
Dmytro Skarzhynets 2025-07-02 20:39:17 +03:00
parent cdcb80ef76
commit bc87f99935
No known key found for this signature in database
GPG Key ID: 2B51652F224037DF

View File

@ -15,34 +15,15 @@
*/ */
package org.thingsboard.server.common.data.ai.provider; 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 { public enum AiProvider {
OPENAI(OpenAiChatModel.class), OPENAI,
AZURE_OPENAI(AzureOpenAiChatModel.class), AZURE_OPENAI,
GOOGLE_AI_GEMINI(GoogleAiGeminiChatModel.class), GOOGLE_AI_GEMINI,
GOOGLE_VERTEX_AI_GEMINI(GoogleVertexAiGeminiChatModel.class), GOOGLE_VERTEX_AI_GEMINI,
MISTRAL_AI(MistralAiChatModel.class), MISTRAL_AI,
ANTHROPIC(AnthropicChatModel.class), ANTHROPIC,
AMAZON_BEDROCK(AmazonBedrockChatModel.class), AMAZON_BEDROCK,
GITHUB_MODELS(GitHubModelsChatModel.class); GITHUB_MODELS
private final Class<?> defaultModelClass;
AiProvider(Class<?> defaultModelClass) {
this.defaultModelClass = defaultModelClass;
}
public Class<?> getDefaultModelClass() {
return defaultModelClass;
}
} }