Added a new invoke_function: if_else (#15)

SolaceDev · Aug 1, 2024 · a8f72bf · a8f72bf
1 parent d40411a
commit a8f72bf
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/solace_ai_connector/common/invoke_functions.py b/src/solace_ai_connector/common/invoke_functions.py
@@ -25,6 +25,7 @@
 empty_tuple = tuple
 empty_float = lambda: 0.0
 empty_int = lambda: 0
+if_else = lambda x, y, z: y if x else z
 
 
 # A few test functions

diff --git a/tests/test_invoke.py b/tests/test_invoke.py
@@ -302,6 +302,24 @@
                         },
                     },
                 },
+                "if_else_true": {
+                    "invoke": {
+                        "module": "invoke_functions",
+                        "function": "if_else",
+                        "params": {
+                            "positional": [True, 1, 2],
+                        },
+                    },
+                },
+                "if_else_false": {
+                    "invoke": {
+                        "module": "invoke_functions",
+                        "function": "if_else",
+                        "params": {
+                            "positional": [False, 1, 2],
+                        },
+                    },
+                },
             },
         },
         "expected": {
@@ -325,6 +343,8 @@
                 "not_op": False,
                 "in_op": True,
                 "negate": -1,
+                "if_else_true": 1,
+                "if_else_false": 2,
             },
         },
     },