dendi-wang/thingsboard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

AI rule node: cleanup AiProvider enum

Browse Source
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
1 changed files with 8 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
common/data/src/main/java/org/thingsboard/server/common/data/ai/provider/AiProvider.java
View File

@ -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
}