AI rule node: cleanup AiProvider enum
This commit is contained in:
parent
cdcb80ef76
commit
bc87f99935
@ -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;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user