Browse Source

AI rule node: correctly call node lifecycle methods

pull/13371/head
Dmytro Skarzhynets 1 year ago
parent
commit
d2c275b761
No known key found for this signature in database GPG Key ID: 2B51652F224037DF
  1. 4
      rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/ai/TbAiNode.java

4
rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/ai/TbAiNode.java

@ -59,6 +59,8 @@ public final class TbAiNode extends TbAbstractExternalNode implements TbNode {
@Override
public void init(TbContext ctx, TbNodeConfiguration configuration) throws TbNodeException {
super.init(ctx);
var config = TbNodeUtils.convert(configuration, TbAiNodeConfiguration.class);
String errorPrefix = "'" + ctx.getSelf().getName() + "' node configuration is invalid: ";
try {
@ -66,6 +68,7 @@ public final class TbAiNode extends TbAbstractExternalNode implements TbNode {
} catch (DataValidationException e) {
throw new TbNodeException(e, true);
}
systemMessage = SystemMessage.from(config.getSystemPrompt());
userPromptTemplate = PromptTemplate.from("""
User-provided task or question: %s
@ -114,6 +117,7 @@ public final class TbAiNode extends TbAbstractExternalNode implements TbNode {
@Override
public void destroy() {
super.destroy();
systemMessage = null;
userPromptTemplate = null;
chatModel = null;

Loading…
Cancel
Save