diff --git a/agents-api/agents_api/clients/temporal.py b/agents-api/agents_api/clients/temporal.py
index c812991f1..4bb25cbc9 100644
--- a/agents-api/agents_api/clients/temporal.py
+++ b/agents-api/agents_api/clients/temporal.py
@@ -1,11 +1,11 @@
 from datetime import timedelta
 from uuid import UUID
 
-from ..common.retry_policies import DEFAULT_RETRY_POLICY
 from temporalio.client import Client, TLSConfig
 
 from ..autogen.openapi_model import TransitionTarget
 from ..common.protocol.tasks import ExecutionInput
+from ..common.retry_policies import DEFAULT_RETRY_POLICY
 from ..env import (
     temporal_client_cert,
     temporal_namespace,
@@ -55,7 +55,7 @@ async def run_task_execution_workflow(
         task_queue=temporal_task_queue,
         id=str(job_id),
         run_timeout=timedelta(days=31),
-        retry_policy=DEFAULT_RETRY_POLICY
+        retry_policy=DEFAULT_RETRY_POLICY,
         # TODO: Should add search_attributes for queryability
     )
 
diff --git a/agents-api/agents_api/common/retry_policies.py b/agents-api/agents_api/common/retry_policies.py
index ce31201e6..3bd51e69c 100644
--- a/agents-api/agents_api/common/retry_policies.py
+++ b/agents-api/agents_api/common/retry_policies.py
@@ -1,4 +1,5 @@
 from datetime import timedelta
+
 from temporalio.common import RetryPolicy
 
 DEFAULT_RETRY_POLICY = RetryPolicy(
diff --git a/agents-api/agents_api/routers/docs/create_doc.py b/agents-api/agents_api/routers/docs/create_doc.py
index aa2ba719a..16754dbe3 100644
--- a/agents-api/agents_api/routers/docs/create_doc.py
+++ b/agents-api/agents_api/routers/docs/create_doc.py
@@ -1,7 +1,6 @@
 from typing import Annotated
 from uuid import UUID, uuid4
 
-from ...common.retry_policies import DEFAULT_RETRY_POLICY
 from fastapi import BackgroundTasks, Depends
 from starlette.status import HTTP_201_CREATED
 from temporalio.client import Client as TemporalClient
@@ -9,6 +8,7 @@
 from ...activities.types import EmbedDocsPayload
 from ...autogen.openapi_model import CreateDocRequest, ResourceCreatedResponse
 from ...clients import temporal
+from ...common.retry_policies import DEFAULT_RETRY_POLICY
 from ...dependencies.developer_id import get_developer_id
 from ...env import temporal_task_queue, testing
 from ...models.docs.create_doc import create_doc as create_doc_query
@@ -42,7 +42,7 @@ async def run_embed_docs_task(
         embed_payload,
         task_queue=temporal_task_queue,
         id=str(job_id),
-        retry_policy=DEFAULT_RETRY_POLICY
+        retry_policy=DEFAULT_RETRY_POLICY,
     )
 
     # TODO: Remove this conditional once we have a way to run workflows in
diff --git a/agents-api/agents_api/workflows/demo.py b/agents-api/agents_api/workflows/demo.py
index e5725065e..0599a4392 100644
--- a/agents-api/agents_api/workflows/demo.py
+++ b/agents-api/agents_api/workflows/demo.py
@@ -16,5 +16,5 @@ async def run(self, a: int, b: int) -> int:
             demo_activity,
             args=[a, b],
             start_to_close_timeout=timedelta(seconds=30),
-            retry_policy=DEFAULT_RETRY_POLICY
+            retry_policy=DEFAULT_RETRY_POLICY,
         )
diff --git a/agents-api/agents_api/workflows/embed_docs.py b/agents-api/agents_api/workflows/embed_docs.py
index e04ebd0f5..83eefe907 100644
--- a/agents-api/agents_api/workflows/embed_docs.py
+++ b/agents-api/agents_api/workflows/embed_docs.py
@@ -19,5 +19,5 @@ async def run(self, embed_payload: EmbedDocsPayload) -> None:
             embed_docs,
             embed_payload,
             schedule_to_close_timeout=timedelta(seconds=600),
-            retry_policy=DEFAULT_RETRY_POLICY
+            retry_policy=DEFAULT_RETRY_POLICY,
         )
diff --git a/agents-api/agents_api/workflows/mem_rating.py b/agents-api/agents_api/workflows/mem_rating.py
index ffcc4bb93..0a87fd787 100644
--- a/agents-api/agents_api/workflows/mem_rating.py
+++ b/agents-api/agents_api/workflows/mem_rating.py
@@ -18,5 +18,5 @@ async def run(self, memory: str) -> None:
             mem_rating,
             memory,
             schedule_to_close_timeout=timedelta(seconds=600),
-            retry_policy=DEFAULT_RETRY_POLICY
+            retry_policy=DEFAULT_RETRY_POLICY,
         )
diff --git a/agents-api/agents_api/workflows/summarization.py b/agents-api/agents_api/workflows/summarization.py
index e117930a9..96ce4c460 100644
--- a/agents-api/agents_api/workflows/summarization.py
+++ b/agents-api/agents_api/workflows/summarization.py
@@ -18,5 +18,5 @@ async def run(self, session_id: str) -> None:
             summarization,
             session_id,
             schedule_to_close_timeout=timedelta(seconds=600),
-            retry_policy=DEFAULT_RETRY_POLICY
+            retry_policy=DEFAULT_RETRY_POLICY,
         )
diff --git a/agents-api/agents_api/workflows/task_execution/__init__.py b/agents-api/agents_api/workflows/task_execution/__init__.py
index ea0797b11..9b99f82e2 100644
--- a/agents-api/agents_api/workflows/task_execution/__init__.py
+++ b/agents-api/agents_api/workflows/task_execution/__init__.py
@@ -4,11 +4,12 @@
 from datetime import timedelta
 from typing import Any
 
-from ...common.retry_policies import DEFAULT_RETRY_POLICY
 from pydantic import RootModel
 from temporalio import workflow
 from temporalio.exceptions import ApplicationError
 
+from ...common.retry_policies import DEFAULT_RETRY_POLICY
+
 # Import necessary modules and types
 with workflow.unsafe.imports_passed_through():
     from ...activities import task_steps
@@ -387,7 +388,7 @@ async def run(
                     task_steps.raise_complete_async,
                     args=[context, output],
                     schedule_to_close_timeout=timedelta(days=31),
-                    retry_policy=DEFAULT_RETRY_POLICY
+                    retry_policy=DEFAULT_RETRY_POLICY,
                 )
 
                 state = PartialTransition(type="resume", output=result)
@@ -420,7 +421,7 @@ async def run(
                     task_steps.raise_complete_async,
                     args=[context, tool_calls_input],
                     schedule_to_close_timeout=timedelta(days=31),
-                    retry_policy=DEFAULT_RETRY_POLICY
+                    retry_policy=DEFAULT_RETRY_POLICY,
                 )
 
                 # Feed the tool call results back to the model
@@ -432,7 +433,7 @@ async def run(
                     schedule_to_close_timeout=timedelta(
                         seconds=30 if debug or testing else 600
                     ),
-                    retry_policy=DEFAULT_RETRY_POLICY
+                    retry_policy=DEFAULT_RETRY_POLICY,
                 )
                 state = PartialTransition(output=new_response.output, type="resume")
 
@@ -476,7 +477,7 @@ async def run(
                     task_steps.raise_complete_async,
                     args=[context, tool_call],
                     schedule_to_close_timeout=timedelta(days=31),
-                    retry_policy=DEFAULT_RETRY_POLICY
+                    retry_policy=DEFAULT_RETRY_POLICY,
                 )
 
                 state = PartialTransition(output=tool_call_response, type="resume")
@@ -507,7 +508,7 @@ async def run(
                     schedule_to_close_timeout=timedelta(
                         seconds=30 if debug or testing else 600
                     ),
-                    retry_policy=DEFAULT_RETRY_POLICY
+                    retry_policy=DEFAULT_RETRY_POLICY,
                 )
 
                 state = PartialTransition(output=tool_call_response)
diff --git a/agents-api/agents_api/workflows/task_execution/helpers.py b/agents-api/agents_api/workflows/task_execution/helpers.py
index fb3e104d1..04449db58 100644
--- a/agents-api/agents_api/workflows/task_execution/helpers.py
+++ b/agents-api/agents_api/workflows/task_execution/helpers.py
@@ -4,7 +4,9 @@
 
 from temporalio import workflow
 from temporalio.exceptions import ApplicationError
+
 from ...common.retry_policies import DEFAULT_RETRY_POLICY
+
 with workflow.unsafe.imports_passed_through():
     from ...activities import task_steps
     from ...autogen.openapi_model import (
@@ -33,7 +35,7 @@ async def continue_as_child(
             previous_inputs,
             user_state,
         ],
-        retry_policy=DEFAULT_RETRY_POLICY
+        retry_policy=DEFAULT_RETRY_POLICY,
     )
 
 
@@ -170,7 +172,7 @@ async def execute_map_reduce_step(
             task_steps.base_evaluate,
             args=[reduce, {"results": result, "_": output}],
             schedule_to_close_timeout=timedelta(seconds=30),
-            retry_policy=DEFAULT_RETRY_POLICY
+            retry_policy=DEFAULT_RETRY_POLICY,
         )
 
     return result
@@ -246,7 +248,7 @@ async def execute_map_reduce_step_parallel(
                     extra_lambda_strs,
                 ],
                 schedule_to_close_timeout=timedelta(seconds=30),
-                retry_policy=DEFAULT_RETRY_POLICY
+                retry_policy=DEFAULT_RETRY_POLICY,
             )
 
         except BaseException as e:
diff --git a/agents-api/agents_api/workflows/task_execution/transition.py b/agents-api/agents_api/workflows/task_execution/transition.py
index 90e3bdccc..035322dad 100644
--- a/agents-api/agents_api/workflows/task_execution/transition.py
+++ b/agents-api/agents_api/workflows/task_execution/transition.py
@@ -1,7 +1,6 @@
 from datetime import timedelta
 
 from temporalio import workflow
-from ...common.retry_policies import DEFAULT_RETRY_POLICY
 from temporalio.exceptions import ApplicationError
 
 from ...activities import task_steps
@@ -11,6 +10,7 @@
     TransitionTarget,
 )
 from ...common.protocol.tasks import PartialTransition, StepContext
+from ...common.retry_policies import DEFAULT_RETRY_POLICY
 
 
 async def transition(
@@ -45,7 +45,7 @@ async def transition(
             task_steps.transition_step,
             args=[context, transition_request],
             schedule_to_close_timeout=timedelta(seconds=30),
-            retry_policy=DEFAULT_RETRY_POLICY
+            retry_policy=DEFAULT_RETRY_POLICY,
         )
 
     except Exception as e:
diff --git a/agents-api/agents_api/workflows/truncation.py b/agents-api/agents_api/workflows/truncation.py
index 2a84b2c5f..d12a186b9 100644
--- a/agents-api/agents_api/workflows/truncation.py
+++ b/agents-api/agents_api/workflows/truncation.py
@@ -18,5 +18,5 @@ async def run(self, session_id: str, token_count_threshold: int) -> None:
             truncation,
             args=[session_id, token_count_threshold],
             schedule_to_close_timeout=timedelta(seconds=600),
-            retry_policy=DEFAULT_RETRY_POLICY
+            retry_policy=DEFAULT_RETRY_POLICY,
         )
diff --git a/agents-api/notebooks/03-summarise.ipynb b/agents-api/notebooks/03-summarise.ipynb
index a934fd1b9..98e6f5e0a 100644
--- a/agents-api/notebooks/03-summarise.ipynb
+++ b/agents-api/notebooks/03-summarise.ipynb
@@ -766,7 +766,9 @@
     "    messages.append(user(start_message))\n",
     "\n",
     "    print(\"Starting chatml generation\")\n",
-    "    trim_result = generate(messages, model=\"gpt-4-turbo\", temperature=0.1, stop=[\"</ct\"])\n",
+    "    trim_result = generate(\n",
+    "        messages, model=\"gpt-4-turbo\", temperature=0.1, stop=[\"</ct\"]\n",
+    "    )\n",
     "    print(\"End chatml generation\")\n",
     "    messages.append(trim_result)\n",
     "\n",
diff --git a/agents-api/notebooks/RecSum-experiments.ipynb b/agents-api/notebooks/RecSum-experiments.ipynb
index 43e6cfe33..1315533dc 100644
--- a/agents-api/notebooks/RecSum-experiments.ipynb
+++ b/agents-api/notebooks/RecSum-experiments.ipynb
@@ -19,6 +19,7 @@
    ],
    "source": [
     "from dotenv import load_dotenv\n",
+    "\n",
     "load_dotenv(\"../../.env\")"
    ]
   },
@@ -56,7 +57,9 @@
     "}\n",
     "\n",
     "user = lambda content, name=None: make_chatml(role=\"user\", content=content, name=name)\n",
-    "assistant = lambda content, name=None: make_chatml(role=\"assistant\", content=content, name=name)\n",
+    "assistant = lambda content, name=None: make_chatml(\n",
+    "    role=\"assistant\", content=content, name=name\n",
+    ")\n",
     "system = lambda content, name=None: make_chatml(content, name=name)\n",
     "thought = lambda content: system(content, name=\"thought\")\n",
     "information = lambda content: system(content, name=\"information\")\n",
@@ -76,7 +79,13 @@
     "    # ((chat_session, 0): (chat_summaries, 1)),\n",
     "]\n",
     "\n",
-    "generate = lambda history, model=\"gpt-3.5-turbo\", **kwargs: client.chat.completions.create(messages=history, model=model, **kwargs).choices[0].message.__dict__"
+    "generate = (\n",
+    "    lambda history, model=\"gpt-3.5-turbo\", **kwargs: client.chat.completions.create(\n",
+    "        messages=history, model=model, **kwargs\n",
+    "    )\n",
+    "    .choices[0]\n",
+    "    .message.__dict__\n",
+    ")"
    ]
   },
   {
@@ -88,13 +97,14 @@
    "source": [
     "from pprint import pprint\n",
     "\n",
+    "\n",
     "def chat():\n",
-    "    while (user_input := input('You: ')) != \"bye\":\n",
+    "    while (user_input := input(\"You: \")) != \"bye\":\n",
     "        chat_session.append(user(user_input))\n",
-    "        \n",
+    "\n",
     "        result = generate(chat_session)\n",
     "        print(f\"GPT: {result['content']}\")\n",
-    "        \n",
+    "\n",
     "        chat_session.append(make_chatml(**result))"
    ]
   },
@@ -106,23 +116,19 @@
    "outputs": [],
    "source": [
     "import json\n",
+    "\n",
     "example_chat = [\n",
     "    {\n",
     "        \"role\": \"system\",\n",
     "        \"content\": \"just have a casual chat with a friend friend\",\n",
     "        \"index\": 0,\n",
     "    },\n",
-    "    {\n",
-    "        \"role\": \"user\",\n",
-    "        \"content\": \"yo\",\n",
-    "        \"index\": 1,\n",
-    "        \"name\": \"Camille\"\n",
-    "    },\n",
+    "    {\"role\": \"user\", \"content\": \"yo\", \"index\": 1, \"name\": \"Camille\"},\n",
     "    {\n",
     "        \"role\": \"assistant\",\n",
     "        \"content\": \"Hey, how's it going?\",\n",
     "        \"index\": 2,\n",
-    "        \"name\": \"JaneBot\"\n",
+    "        \"name\": \"JaneBot\",\n",
     "    },\n",
     "    {\n",
     "        \"role\": \"user\",\n",
@@ -221,7 +227,7 @@
     "    },\n",
     "    {\n",
     "        \"role\": \"assistant\",\n",
-    "        \"content\": \"Based on the themes of starting anew, embracing failure, and the journey from zero to one, I would guess that the title of this story is \\\"Back to One.\\\" It seems to capture the essence of the protagonist's experience and mindset throughout the story. How did I do?\",\n",
+    "        \"content\": 'Based on the themes of starting anew, embracing failure, and the journey from zero to one, I would guess that the title of this story is \"Back to One.\" It seems to capture the essence of the protagonist\\'s experience and mindset throughout the story. How did I do?',\n",
     "        \"index\": 22,\n",
     "    },\n",
     "    {\n",
@@ -283,7 +289,7 @@
     "        \"role\": \"assistant\",\n",
     "        \"content\": \"From my interactions with you so far, I can tell that you have a great sense of humor and enjoy playful banter. You seem open to random and fun conversations, and you're not afraid to keep things interesting with unexpected twists and turns.\\n\\nYour creativity shines through in the way you approach different topics, and you have a knack for surprising me with your responses. You're also quite adept at changing the direction of the conversation while maintaining a light and friendly tone.\\n\\nOverall, chatting with you has been enjoyable and entertaining, and I appreciate the lighthearted and casual vibe you bring to our interactions. I look forward to more fun conversations with you in the future!\",\n",
     "        \"index\": 34,\n",
-    "    }\n",
+    "    },\n",
     "]"
    ]
   },
@@ -316,17 +322,23 @@
     "2. JaneBot (The assistant): Engages in lighthearted conversation and tries to guess user's thoughts.\n",
     "3. Elon Musk: Polarizing tech and space industry figure.\n",
     "4. Bananas: User has a tattoo of one on their arm.\"\"\"),\n",
-    "    summary(\"Event: Camille says hi to JaneBot and they reply that they are working and trying to stay busy.\"),\n",
-    "    summary(\"Event: They discuss Elon Musk and agree that he can be a polarizing personality.\"),\n",
+    "    summary(\n",
+    "        \"Event: Camille says hi to JaneBot and they reply that they are working and trying to stay busy.\"\n",
+    "    ),\n",
+    "    summary(\n",
+    "        \"Event: They discuss Elon Musk and agree that he can be a polarizing personality.\"\n",
+    "    ),\n",
     "    *example_chat[9:17],\n",
     "    summary(\"Event: Camille appreciates JaneBot's sense of humor.\"),\n",
     "    *example_chat[21:23],\n",
-    "    summary('Event: Camille asks JaneBot to play \"What number I am thinking of?\" game but she keeps saying \"no\" to all guesses which JaneBot finds really funny.'),\n",
+    "    summary(\n",
+    "        'Event: Camille asks JaneBot to play \"What number I am thinking of?\" game but she keeps saying \"no\" to all guesses which JaneBot finds really funny.'\n",
+    "    ),\n",
     "    example_chat[33],\n",
     "    summary(\"\"\"Event: JaneBot wrote an essay about Camille. Summary of the essay:\n",
     "1. You have a great sense of humor and enjoy playful, lighthearted banter in conversations.\n",
     "2. Your creativity is evident in how you approach topics in unexpected ways and keep conversations interesting.\n",
-    "3. Chatting with you is enjoyable due to the casual, friendly tone you bring to interactions.\"\"\")\n",
+    "3. Chatting with you is enjoyable due to the casual, friendly tone you bring to interactions.\"\"\"),\n",
     "]"
    ]
   },
@@ -423,7 +435,8 @@
    "source": [
     "# Load test chat from json file\n",
     "import json\n",
-    "with open(\"./test-chat.json\", 'r') as f:\n",
+    "\n",
+    "with open(\"./test-chat.json\", \"r\") as f:\n",
     "    chat_session = json.load(f)"
    ]
   },
@@ -444,19 +457,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "def get_leaf_nodes(chat_session=chat_session, chat_summaries=chat_summaries, session_summary_map=session_summary_map):\n",
+    "def get_leaf_nodes(\n",
+    "    chat_session=chat_session,\n",
+    "    chat_summaries=chat_summaries,\n",
+    "    session_summary_map=session_summary_map,\n",
+    "):\n",
     "    all_messages = [*chat_session, *chat_summaries]\n",
     "\n",
     "    # Collect non-leaf nodes\n",
-    "    non_leaf_nodes = [\n",
-    "        source[idx]\n",
-    "        for (source, idx), _ in session_summary_map\n",
-    "    ]\n",
-    "    \n",
+    "    non_leaf_nodes = [source[idx] for (source, idx), _ in session_summary_map]\n",
+    "\n",
     "    # Remove non-leaf nodes\n",
-    "    remaining = [\n",
-    "        msg for msg in all_messages if msg not in non_leaf_nodes\n",
-    "    ]\n",
+    "    remaining = [msg for msg in all_messages if msg not in non_leaf_nodes]\n",
     "\n",
     "    return remaining"
    ]
@@ -501,16 +513,17 @@
     "from pprint import pprint\n",
     "import time\n",
     "\n",
+    "\n",
     "def compactify(chat_session=chat_session):\n",
     "    start = time.time()\n",
     "    pprint(start)\n",
-    "    \n",
+    "\n",
     "    _sys_msg, *history_to_compact = get_leaf_nodes(chat_session=chat_session)\n",
     "\n",
     "    history_to_compact = [\n",
-    "        {\"index\": i+1, **msg} for i, msg in enumerate(history_to_compact)\n",
+    "        {\"index\": i + 1, **msg} for i, msg in enumerate(history_to_compact)\n",
     "    ]\n",
-    "    \n",
+    "\n",
     "    system_prompt = f\"\"\"\\\n",
     "You are given a session history of a chat between a user and a gpt-4-turbo language model created through the\n",
     "openai api.\n",
@@ -552,8 +565,8 @@
     "\n",
     "    pprint(time.time() - start)\n",
     "    print(\"Starting CoT generation\")\n",
-    "    cot_result = generate(messages, model=\"gpt-4-turbo\", stop=['</ct'], temperature=0.7)\n",
-    "    \n",
+    "    cot_result = generate(messages, model=\"gpt-4-turbo\", stop=[\"</ct\"], temperature=0.7)\n",
+    "\n",
     "    cot_result = make_chatml(**cot_result)\n",
     "    pprint(cot_result)\n",
     "\n",
@@ -569,16 +582,15 @@
     "\"\"\".strip()\n",
     "\n",
     "    messages.append(user(start_message))\n",
-    "    \n",
+    "\n",
     "    pprint(time.time() - start)\n",
     "    print(\"Starting chatml generation\")\n",
     "\n",
-    "    result = generate(messages, model=\"gpt-4-turbo\", temperature=0.1, stop=['</ct'])\n",
+    "    result = generate(messages, model=\"gpt-4-turbo\", temperature=0.1, stop=[\"</ct\"])\n",
     "\n",
     "    print(\"End chatml generation\")\n",
     "    pprint(time.time() - start)\n",
     "\n",
-    "    \n",
     "    return make_chatml(**result)"
    ]
   },
diff --git a/agents-api/tests/test_activities.py b/agents-api/tests/test_activities.py
index 987ba5b01..6f65cd034 100644
--- a/agents-api/tests/test_activities.py
+++ b/agents-api/tests/test_activities.py
@@ -50,7 +50,7 @@ async def _():
             args=[1, 2],
             id=str(uuid4()),
             task_queue=temporal_task_queue,
-            retry_policy=DEFAULT_RETRY_POLICY
+            retry_policy=DEFAULT_RETRY_POLICY,
         )
 
         assert result == 3