Merge pull request #1009 from julep-ai/x/execution-workflow-tests

fix(agents-api): fix some ``test_execution_workflow`` tests

agents-api/agents_api/activities/task_steps/prompt_step.py

@@ -116,7 +116,7 @@ async def prompt_step(context: StepContext) -> StepOutcome:
     passed_settings: dict = context.current_step.model_dump(
         exclude=excluded_keys, exclude_unset=True
     )
-    passed_settings.update(passed_settings.pop("settings", {}))
+    passed_settings.update(passed_settings.pop("settings", {}) or {})
 
     if not passed_settings.get("tools"):
         passed_settings.pop("tool_choice", None)

agents-api/agents_api/activities/utils.py

@@ -11,6 +11,7 @@
 from collections import deque
 from collections.abc import Callable
 from dataclasses import dataclass
+from functools import reduce
 from threading import Lock as ThreadLock
 from typing import Any, ParamSpec, TypeVar
 
@@ -140,6 +141,7 @@ def safe_extract_json(string: str):
     "sum": sum,
     "tuple": tuple,
     "zip": zip,
+    "reduce": reduce,
     # Safe versions of potentially dangerous functions
     "range": safe_range,
     "load_json": safe_json_loads,

agents-api/agents_api/queries/tasks/get_task.py

@@ -28,7 +28,7 @@
         '[]'::jsonb
     ) as workflows,
     COALESCE(
-        jsonb_agg(tl) FILTER (WHERE tl IS NOT NULL),
+        jsonb_agg(to_jsonb(tl)) FILTER (WHERE tl.tool_id IS NOT NULL),
         '[]'::jsonb
     ) as tools
 FROM

agents-api/agents_api/routers/tasks/create_task_execution.py

@@ -73,6 +73,7 @@ async def start_execution(
     )
 
     execution_input.task = task_to_spec(task)
+    execution_input.task.id = task.id
 
     job_id = uuid7()
 

agents-api/tests/test_execution_workflow.py

@@ -668,6 +668,7 @@ async def _(
                     "arguments": {"test": "_.test"},
                 },
             ],
+            inherit_tools=True,
         ),
         connection_pool=pool,
     )