Fix execute task

actions.py

@@ -29,7 +29,6 @@ def remove_input(*visibility):
 
 
 def _is_task_row_fully_invisible(row: List[int]) -> bool:
-    print(row)
     for visible in row:
         if bool(visible):
             return False
@@ -60,14 +59,12 @@ def add_task(index, *visibility):
 def remove_task(*visibility):
     visibility = list(visibility)
     n_avail_tasks = len(Task.AVAILABLE_TASKS)
-    print(visibility, n_avail_tasks)
 
     for i in range(MAX_TASKS):
         start_row = i * n_avail_tasks
         is_row_invisible = _is_task_row_fully_invisible(
             visibility[start_row : start_row + n_avail_tasks]
         )
-        print(start_row, start_row + n_avail_tasks, is_row_invisible)
         if is_row_invisible:
             unchanged_up_to = start_row - n_avail_tasks
             return (
@@ -82,25 +79,27 @@ def _clear_error():
     return gr.HighlightedText.update(value=None, visible=False)
 
 
-def execute_task(id_: int, prev_error_value, n_inputs, *vars_in_scope):
+def execute_task(id_: int, prev_error_value, n_task_inputs, *vars_in_scope):
     """
     Params:
         - id_: This will tell us which task to execute.
         - prev_error_value: I carry around whether there is an error in the execution, to be displayed at the end.
-        - n_inputs: How many inputs does this task have?
-        - vars: All variables in scope. This can be a) task inputs, input varaibles or previous task outputs.
+        - n_task_inputs: How many inputs does this task have?
+        - vars_in_scope: All variables in scope. This can be a) input varaibles, b) task inputs or c) previous task outputs.
     """
-    n_inputs = int(n_inputs)
-    task_inputs = vars_in_scope[:n_inputs]
-    input_vars = vars_in_scope[n_inputs:MAX_INPUTS]
-    task_outputs = vars_in_scope[MAX_INPUTS:]
+    n_task_inputs = int(n_task_inputs)
+    task_inputs = vars_in_scope[:n_task_inputs]
+    input_vars = vars_in_scope[n_task_inputs:MAX_INPUTS]
+    task_outputs = vars_in_scope[n_task_inputs + MAX_INPUTS :]
     non_empty_task_inputs = [ti for ti in task_inputs if ti]
 
     # Put all defined variables into a dict, with names (except task inputs)
     vars = {
         f"{Input.VNAME}{i}": input_ for i, input_ in enumerate(input_vars) if input_
     }
-    vars.update({f"{Task.VNAME}{i}": task for i, task in enumerate(task_outputs)})
+    vars.update(
+        {f"{Task.VNAME}{i}": task_output for i, task_output in enumerate(task_outputs)}
+    )
     # Get all variables referenced within the task inputs
     prompt_vars = {v for ti in non_empty_task_inputs for v in re.findall("{(.*?)}", ti)}
 

ai/llm.py

@@ -23,6 +23,7 @@ def call(
     if not temperature:
         temperature = TEMPERATURE
 
+    print(f"Running OpenAI with :: {locals()}")
     return openai.ChatCompletion.create(  # type: ignore
         model=model,
         messages=messages,

components.py

@@ -36,7 +36,6 @@ class Component(ABC):
         self.gr_component = self._render(self._id, self._initial_visibility)
 
     def execute(self, *args):
-        print(f"Executing {self._source} :: {self._id}")
         return self._execute(*args)
 
 
@@ -178,7 +177,7 @@ class Task:
 
     def execute(self, *args):
         inner_task = self.inner_tasks[self.active_task]
-        print(f"Executing {inner_task._source} :: {inner_task._id}")
+        print(f"Executing {inner_task._source}: {inner_task._id}")
         return inner_task.execute(*args)