Merge pull request #28 from pier-digital/f/rewrite-connector-logic

Remove custom connectors logic

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "retrack"
-version = "2.4.0"
+version = "2.5.0"
 description = "A business rules engine"
 authors = ["Gabriel Guarisa <[email protected]>", "Nathalia Trotte <[email protected]>"]
 license = "MIT"

retrack/engine/executor.py

@@ -34,6 +34,8 @@ def __init__(
         self._metadata = metadata
 
         input_nodes = self._components_registry.get_by_kind(NodeKind.INPUT)
+        input_nodes.extend(self._components_registry.get_by_kind(NodeKind.CONNECTOR))
+
         self._input_columns = {
             f"{node.id}@{constants.INPUT_OUTPUT_VALUE_CONNECTOR_NAME}": node.data.name
             for node in input_nodes
@@ -132,8 +134,7 @@ def __run_node(self, node_id: str, execution: Execution):
             node.model_dump(by_alias=True),
             current_node_filter,
             execution=execution,
-            include_payload=node.kind() == NodeKind.CONNECTOR
-            or node.kind() == NodeKind.FLOW,
+            include_payload=node.kind() == NodeKind.FLOW,
         )
         output = node.run(**input_params)
 

retrack/engine/request_manager.py

@@ -49,7 +49,10 @@ def inputs(self, inputs: typing.List[BaseNode]):
                     f"inputs[{i}] must be an InputModel, not {type(inputs[i])}"
                 )
 
-            if inputs[i].kind() != NodeKind.INPUT:
+            if (
+                inputs[i].kind() != NodeKind.INPUT
+                and inputs[i].kind() != NodeKind.CONNECTOR
+            ):
                 raise TypeError(
                     f"inputs[{i}] must be an InputModel, not {type(inputs[i])}"
                 )

retrack/nodes/connectors.py

@@ -1,5 +1,3 @@
-import typing
-
 from retrack.nodes.base import NodeKind
 from retrack.nodes.inputs import Input, InputMetadataModel
 
@@ -14,6 +12,3 @@ class BaseConnector(Input):
 
     def kind(self) -> NodeKind:
         return NodeKind.CONNECTOR
-
-    def generate_input_nodes(self) -> typing.List[Input]:
-        return [Input(**self.model_dump(by_alias=True))]