Browse Source
AI rule node: fix type when configuring temperature for Gemini models
pull/13371/head
Dmytro Skarzhynets
1 year ago
No known key found for this signature in database
GPG Key ID: 2B51652F224037DF
1 changed files with
2 additions and
1 deletions
-
application/src/main/java/org/thingsboard/server/service/ai/AiServiceImpl.java
|
|
|
@ -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()); |
|
|
|
} |
|
|
|
|
|
|
|
|
