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

AI rule node: use Map.ofEntries() in model type resolver

Browse Source
This commit is contained in:
Dmytro Skarzhynets 2025-06-27 14:56:48 +03:00
parent f2c6fd45d8
commit 00a653c29f
No known key found for this signature in database
GPG Key ID: 2B51652F224037DF
1 changed files with 52 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

112
common/data/src/main/java/org/thingsboard/server/common/data/ai/model/AiModelTypeIdResolver.java
View File

@ -27,75 +27,67 @@ import org.thingsboard.server.common.data.ai.model.chat.GoogleVertexAiGeminiChat
import org.thingsboard.server.common.data.ai.model.chat.MistralAiChatModel; import org.thingsboard.server.common.data.ai.model.chat.MistralAiChatModel;
import org.thingsboard.server.common.data.ai.model.chat.OpenAiChatModel; import org.thingsboard.server.common.data.ai.model.chat.OpenAiChatModel;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map; import java.util.Map;
public final class AiModelTypeIdResolver extends TypeIdResolverBase { public final class AiModelTypeIdResolver extends TypeIdResolverBase {
private static final Map<String, Class<?>> typeIdToModelClass; private static final Map<String, Class<?>> typeIdToModelClass = Map.ofEntries(
// OpenAI models
Map.entry("OPENAI::o4-mini", OpenAiChatModel.class),
// Map.entry("OPENAI::o3-pro", OpenAiChatModel.class); // needs verification with Gov ID :)
// Map.entry("OPENAI::o3", OpenAiChatModel.class); // needs verification with Gov ID :)
Map.entry("OPENAI::o3-mini", OpenAiChatModel.class),
// Map.entry("OPENAI::o1-pro", OpenAiChatModel.class); // LC4j sends requests to v1/chat/completions, but o1-pro is only supported in v1/responses
Map.entry("OPENAI::o1", OpenAiChatModel.class),
Map.entry("OPENAI::gpt-4.1", OpenAiChatModel.class),
Map.entry("OPENAI::gpt-4.1-mini", OpenAiChatModel.class),
Map.entry("OPENAI::gpt-4.1-nano", OpenAiChatModel.class),
Map.entry("OPENAI::gpt-4o", OpenAiChatModel.class),
Map.entry("OPENAI::gpt-4o-mini", OpenAiChatModel.class),
static { // Google AI Gemini models
Map<String, Class<?>> map = new HashMap<>(); Map.entry("GOOGLE_AI_GEMINI::gemini-2.5-pro", GoogleAiGeminiChatModel.class),
Map.entry("GOOGLE_AI_GEMINI::gemini-2.5-flash", GoogleAiGeminiChatModel.class),
Map.entry("GOOGLE_AI_GEMINI::gemini-2.0-flash", GoogleAiGeminiChatModel.class),
Map.entry("GOOGLE_AI_GEMINI::gemini-2.0-flash-lite", GoogleAiGeminiChatModel.class),
Map.entry("GOOGLE_AI_GEMINI::gemini-1.5-pro", GoogleAiGeminiChatModel.class),
Map.entry("GOOGLE_AI_GEMINI::gemini-1.5-flash", GoogleAiGeminiChatModel.class),
Map.entry("GOOGLE_AI_GEMINI::gemini-1.5-flash-8b", GoogleAiGeminiChatModel.class),
// OpenAI models // Google Vertex AI Gemini models
map.put("OPENAI::o4-mini", OpenAiChatModel.class); Map.entry("GOOGLE_VERTEX_AI_GEMINI::gemini-2.5-pro", GoogleVertexAiGeminiChatModel.class),
// map.put("OPENAI::o3-pro", OpenAiChatModel.class); // needs verification with Gov ID :) Map.entry("GOOGLE_VERTEX_AI_GEMINI::gemini-2.5-flash", GoogleVertexAiGeminiChatModel.class),
// map.put("OPENAI::o3", OpenAiChatModel.class); // needs verification with Gov ID :) Map.entry("GOOGLE_VERTEX_AI_GEMINI::gemini-2.0-flash", GoogleVertexAiGeminiChatModel.class),
map.put("OPENAI::o3-mini", OpenAiChatModel.class); Map.entry("GOOGLE_VERTEX_AI_GEMINI::gemini-2.0-flash-lite", GoogleVertexAiGeminiChatModel.class),
// map.put("OPENAI::o1-pro", OpenAiChatModel.class); // LC4j sends requests to v1/chat/completions, but o1-pro is only supported in v1/responses Map.entry("GOOGLE_VERTEX_AI_GEMINI::gemini-1.5-pro", GoogleVertexAiGeminiChatModel.class),
map.put("OPENAI::o1", OpenAiChatModel.class); Map.entry("GOOGLE_VERTEX_AI_GEMINI::gemini-1.5-flash", GoogleVertexAiGeminiChatModel.class),
map.put("OPENAI::gpt-4.1", OpenAiChatModel.class); Map.entry("GOOGLE_VERTEX_AI_GEMINI::gemini-1.5-flash-8b", GoogleVertexAiGeminiChatModel.class),
map.put("OPENAI::gpt-4.1-mini", OpenAiChatModel.class);
map.put("OPENAI::gpt-4.1-nano", OpenAiChatModel.class);
map.put("OPENAI::gpt-4o", OpenAiChatModel.class);
map.put("OPENAI::gpt-4o-mini", OpenAiChatModel.class);
// Google AI Gemini models // Mistral AI models
map.put("GOOGLE_AI_GEMINI::gemini-2.5-pro", GoogleAiGeminiChatModel.class); Map.entry("MISTRAL_AI::magistral-medium-latest", MistralAiChatModel.class),
map.put("GOOGLE_AI_GEMINI::gemini-2.5-flash", GoogleAiGeminiChatModel.class); Map.entry("MISTRAL_AI::magistral-small-latest", MistralAiChatModel.class),
map.put("GOOGLE_AI_GEMINI::gemini-2.0-flash", GoogleAiGeminiChatModel.class); Map.entry("MISTRAL_AI::mistral-large-latest", MistralAiChatModel.class),
map.put("GOOGLE_AI_GEMINI::gemini-2.0-flash-lite", GoogleAiGeminiChatModel.class); Map.entry("MISTRAL_AI::mistral-medium-latest", MistralAiChatModel.class),
map.put("GOOGLE_AI_GEMINI::gemini-1.5-pro", GoogleAiGeminiChatModel.class); Map.entry("MISTRAL_AI::mistral-small-latest", MistralAiChatModel.class),
map.put("GOOGLE_AI_GEMINI::gemini-1.5-flash", GoogleAiGeminiChatModel.class); Map.entry("MISTRAL_AI::pixtral-large-latest", MistralAiChatModel.class),
map.put("GOOGLE_AI_GEMINI::gemini-1.5-flash-8b", GoogleAiGeminiChatModel.class); Map.entry("MISTRAL_AI::ministral-8b-latest", MistralAiChatModel.class),
Map.entry("MISTRAL_AI::ministral-3b-latest", MistralAiChatModel.class),
Map.entry("MISTRAL_AI::open-mistral-nemo", MistralAiChatModel.class),
// Google Vertex AI Gemini models // Anthropic models
map.put("GOOGLE_VERTEX_AI_GEMINI::gemini-2.5-pro", GoogleVertexAiGeminiChatModel.class); Map.entry("ANTHROPIC::claude-opus-4-0", AnthropicChatModel.class),
map.put("GOOGLE_VERTEX_AI_GEMINI::gemini-2.5-flash", GoogleVertexAiGeminiChatModel.class); Map.entry("ANTHROPIC::claude-sonnet-4-0", AnthropicChatModel.class),
map.put("GOOGLE_VERTEX_AI_GEMINI::gemini-2.0-flash", GoogleVertexAiGeminiChatModel.class); Map.entry("ANTHROPIC::claude-3-7-sonnet-latest", AnthropicChatModel.class),
map.put("GOOGLE_VERTEX_AI_GEMINI::gemini-2.0-flash-lite", GoogleVertexAiGeminiChatModel.class); Map.entry("ANTHROPIC::claude-3-5-sonnet-latest", AnthropicChatModel.class),
map.put("GOOGLE_VERTEX_AI_GEMINI::gemini-1.5-pro", GoogleVertexAiGeminiChatModel.class); Map.entry("ANTHROPIC::claude-3-5-haiku-latest", AnthropicChatModel.class),
map.put("GOOGLE_VERTEX_AI_GEMINI::gemini-1.5-flash", GoogleVertexAiGeminiChatModel.class); Map.entry("ANTHROPIC::claude-3-opus-latest", AnthropicChatModel.class),
map.put("GOOGLE_VERTEX_AI_GEMINI::gemini-1.5-flash-8b", GoogleVertexAiGeminiChatModel.class);
// Mistral AI models // Amazon Bedrock models
map.put("MISTRAL_AI::magistral-medium-latest", MistralAiChatModel.class); Map.entry("AMAZON_BEDROCK::amazon.nova-lite-v1:0", AmazonBedrockChatModel.class),
map.put("MISTRAL_AI::magistral-small-latest", MistralAiChatModel.class);
map.put("MISTRAL_AI::mistral-large-latest", MistralAiChatModel.class);
map.put("MISTRAL_AI::mistral-medium-latest", MistralAiChatModel.class);
map.put("MISTRAL_AI::mistral-small-latest", MistralAiChatModel.class);
map.put("MISTRAL_AI::pixtral-large-latest", MistralAiChatModel.class);
map.put("MISTRAL_AI::ministral-8b-latest", MistralAiChatModel.class);
map.put("MISTRAL_AI::ministral-3b-latest", MistralAiChatModel.class);
map.put("MISTRAL_AI::open-mistral-nemo", MistralAiChatModel.class);
// Anthropic models // GitHub Models models
map.put("ANTHROPIC::claude-opus-4-0", AnthropicChatModel.class); Map.entry("GITHUB_MODELS::gpt-4o", GitHubModelsChatModel.class)
map.put("ANTHROPIC::claude-sonnet-4-0", AnthropicChatModel.class); );
map.put("ANTHROPIC::claude-3-7-sonnet-latest", AnthropicChatModel.class);
map.put("ANTHROPIC::claude-3-5-sonnet-latest", AnthropicChatModel.class);
map.put("ANTHROPIC::claude-3-5-haiku-latest", AnthropicChatModel.class);
map.put("ANTHROPIC::claude-3-opus-latest", AnthropicChatModel.class);
// Amazon Bedrock models
map.put("AMAZON_BEDROCK::amazon.nova-lite-v1:0", AmazonBedrockChatModel.class);
// GitHub Models models
map.put("GITHUB_MODELS::gpt-4o", GitHubModelsChatModel.class);
typeIdToModelClass = Collections.unmodifiableMap(map);
}
private JavaType baseType; private JavaType baseType;
@ -117,7 +109,7 @@ public final class AiModelTypeIdResolver extends TypeIdResolverBase {
@Override @Override
public JavaType typeFromId(DatabindContext context, String id) { public JavaType typeFromId(DatabindContext context, String id) {
Class<?> modelClass = typeIdToModelClass.get(id); Class<?> modelClass = typeIdToModelClass.get(id);
if (modelClass == null) { if (modelClass == null) { // TODO: if provider is unknown - throw, if provider is valid but model is unknown - fallback to default model
throw new IllegalArgumentException("Unknown model type ID: " + id); throw new IllegalArgumentException("Unknown model type ID: " + id);
} }
return context.constructSpecializedType(baseType, modelClass); return context.constructSpecializedType(baseType, modelClass);