AI rule node: fix type when configuring temperature for Gemini models

This commit is contained in:
Dmytro Skarzhynets 2025-05-22 16:41:16 +03:00
parent 45fbbf201f
commit 71d1e3640a
No known key found for this signature in database
GPG Key ID: 2B51652F224037DF

View File

@ -23,6 +23,7 @@ import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.thingsboard.rule.engine.api.RuleEngineAiService;
import org.thingsboard.server.common.data.ai.AiSettings;
import org.thingsboard.server.common.data.ai.model.GoogleAiGeminiChatModelConfig;
import org.thingsboard.server.common.data.ai.model.MistralAiChatModelConfig;
import org.thingsboard.server.common.data.ai.model.OpenAiChatModelConfig;
import org.thingsboard.server.common.data.id.AiSettingsId;
@ -74,7 +75,7 @@ class AiServiceImpl implements RuleEngineAiService {
.apiKey(aiSettings.getProviderConfig().getApiKey())
.modelName(aiSettings.getModel());
if (aiSettings.getModelConfig() instanceof OpenAiChatModelConfig config) {
if (aiSettings.getModelConfig() instanceof GoogleAiGeminiChatModelConfig config) {
modelBuilder.temperature(config.getTemperature());
}