Merge branch 'master' into aarushikansal/open-1574-deploy-backend-to-k8s

rnd/autogpt_server/autogpt_server/blocks/if_block.py

@@ -0,0 +1,101 @@
+from enum import Enum
+from typing import Any
+
+from autogpt_server.data.block import Block, BlockOutput, BlockSchema
+from autogpt_server.data.model import SchemaField
+
+
+class ComparisonOperator(Enum):
+    EQUAL = "=="
+    NOT_EQUAL = "!="
+    GREATER_THAN = ">"
+    LESS_THAN = "<"
+    GREATER_THAN_OR_EQUAL = ">="
+    LESS_THAN_OR_EQUAL = "<="
+
+
+class ConditionBlock(Block):
+    class Input(BlockSchema):
+        value1: Any = SchemaField(
+            description="Enter the first value for comparison",
+            placeholder="For example: 10 or 'hello' or True",
+        )
+        operator: ComparisonOperator = SchemaField(
+            description="Choose the comparison operator",
+            placeholder="Select an operator",
+        )
+        value2: Any = SchemaField(
+            description="Enter the second value for comparison",
+            placeholder="For example: 20 or 'world' or False",
+        )
+        yes_value: Any = SchemaField(
+            description="(Optional) Value to output if the condition is true. If not provided, value1 will be used.",
+            placeholder="Leave empty to use value1, or enter a specific value",
+            default=None,
+        )
+        no_value: Any = SchemaField(
+            description="(Optional) Value to output if the condition is false. If not provided, value1 will be used.",
+            placeholder="Leave empty to use value1, or enter a specific value",
+            default=None,
+        )
+
+    class Output(BlockSchema):
+        result: bool = SchemaField(
+            description="The result of the condition evaluation (True or False)"
+        )
+        yes_output: Any = SchemaField(
+            description="The output value if the condition is true"
+        )
+        no_output: Any = SchemaField(
+            description="The output value if the condition is false"
+        )
+
+    def __init__(self):
+        super().__init__(
+            id="715696a0-e1da-45c8-b209-c2fa9c3b0be6",
+            input_schema=ConditionBlock.Input,
+            output_schema=ConditionBlock.Output,
+            description="Handles conditional logic based on comparison operators",
+            test_input={
+                "value1": 10,
+                "operator": ComparisonOperator.GREATER_THAN.value,
+                "value2": 5,
+                "yes_value": "Greater",
+                "no_value": "Not greater",
+            },
+            test_output=[
+                ("result", True),
+                ("yes_output", "Greater"),
+            ],
+        )
+
+    def run(self, input_data: Input) -> BlockOutput:
+        value1 = input_data.value1
+        operator = input_data.operator
+        value2 = input_data.value2
+        yes_value = input_data.yes_value if input_data.yes_value is not None else value1
+        no_value = input_data.no_value if input_data.no_value is not None else value1
+
+        comparison_funcs = {
+            ComparisonOperator.EQUAL: lambda a, b: a == b,
+            ComparisonOperator.NOT_EQUAL: lambda a, b: a != b,
+            ComparisonOperator.GREATER_THAN: lambda a, b: a > b,
+            ComparisonOperator.LESS_THAN: lambda a, b: a < b,
+            ComparisonOperator.GREATER_THAN_OR_EQUAL: lambda a, b: a >= b,
+            ComparisonOperator.LESS_THAN_OR_EQUAL: lambda a, b: a <= b,
+        }
+
+        try:
+            result = comparison_funcs[operator](value1, value2)
+
+            yield "result", result
+
+            if result:
+                yield "yes_output", yes_value
+            else:
+                yield "no_output", no_value
+
+        except Exception:
+            yield "result", None
+            yield "yes_output", None
+            yield "no_output", None