use args kwargs for run

NapthaAI · Sep 30, 2024 · 08b7333 · 08b7333
1 parent 6d9f4d1
commit 08b7333
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/babyagi_task_finalizer/run.py b/babyagi_task_finalizer/run.py
@@ -35,9 +35,10 @@ def llm_call(messages, response_model=None):
         )
     return response
 
-def run(inputs: InputSchema, worker_nodes = None, orchestrator_node = None, flow_run = None, cfg: dict = None):
+def run(inputs: InputSchema, *args, **kwargs):
     logger.info(f"Running with inputs {inputs.objective}")
     logger.info(f"Running with inputs {inputs.task}")
+    cfg = kwargs["cfg"]
 
     user_prompt = cfg["inputs"]["user_message_template"].replace("{{task}}", inputs.task).replace("{{objective}}", inputs.objective)