From 7f6c5282911485bb812c7ddd547117e714d4d0cc Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Tue, 18 Feb 2025 14:45:11 +0800
Subject: [PATCH] feat: The discriminator supports variables (#2311)

---
 .../flow/step_node/condition_node/impl/base_condition_node.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/apps/application/flow/step_node/condition_node/impl/base_condition_node.py b/apps/application/flow/step_node/condition_node/impl/base_condition_node.py
index 2c5260f89..0c12e8b1a 100644
--- a/apps/application/flow/step_node/condition_node/impl/base_condition_node.py
+++ b/apps/application/flow/step_node/condition_node/impl/base_condition_node.py
@@ -36,6 +36,10 @@ class BaseConditionNode(IConditionNode):
         return all(condition_list) if condition == 'and' else any(condition_list)
 
     def assertion(self, field_list: List[str], compare: str, value):
+        try:
+            value = self.workflow_manage.generate_prompt(value)
+        except Exception as e:
+            pass
         field_value = self.workflow_manage.get_reference_field(field_list[0], field_list[1:])
         for compare_handler in compare_handle_list:
             if compare_handler.support(field_list[0], field_list[1:], field_value, compare, value):