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

AI rule node: ensure AI settings exist on rule node init

Browse Source
This commit is contained in:
Dmytro Skarzhynets 2025-06-11 18:10:01 +03:00
parent 2e5aab3fd6
commit ac61136ecc
No known key found for this signature in database
GPG Key ID: 2B51652F224037DF
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/ai/TbAiNode.java
View File

@ -95,6 +95,10 @@ public final class TbAiNode extends TbAbstractExternalNode implements TbNode {
); );
timeoutSeconds = config.getTimeoutSeconds(); timeoutSeconds = config.getTimeoutSeconds();
if (!aiSettingsExist(ctx, config.getAiSettingsId())) {
throw new TbNodeException("[" + ctx.getTenantId() + "] AI settings with ID: " + config.getAiSettingsId() + " were not found", true);
}
aiSettingsId = config.getAiSettingsId(); aiSettingsId = config.getAiSettingsId();
} }
@ -105,6 +109,10 @@ public final class TbAiNode extends TbAbstractExternalNode implements TbNode {
return responseFormatType == ResponseFormatType.JSON && jsonSchema != null ? Langchain4jJsonSchemaAdapter.fromJsonNode(jsonSchema) : null; return responseFormatType == ResponseFormatType.JSON && jsonSchema != null ? Langchain4jJsonSchemaAdapter.fromJsonNode(jsonSchema) : null;
} }
private static boolean aiSettingsExist(TbContext ctx, AiSettingsId aiSettingsId) {
return ctx.getAiSettingsService().findAiSettingsByTenantIdAndId(ctx.getTenantId(), aiSettingsId).isPresent();
}
@Override @Override
public void onMsg(TbContext ctx, TbMsg msg) throws TbNodeException { public void onMsg(TbContext ctx, TbMsg msg) throws TbNodeException {
var ackedMsg = ackIfNeeded(ctx, msg); var ackedMsg = ackIfNeeded(ctx, msg);