diff --git a/literalai/api/__init__.py b/literalai/api/__init__.py
index fd2413d..480a185 100644
--- a/literalai/api/__init__.py
+++ b/literalai/api/__init__.py
@@ -227,7 +227,8 @@ def raise_error(error):
                 json = response.json()
             except ValueError as e:
                 raise_error(
-                    f"Failed to parse JSON response: {e}, content: {response.content!r}"
+                    f"""Failed to parse JSON response: {
+                        e}, content: {response.content!r}"""
                 )
 
             if json.get("errors"):
@@ -238,7 +239,8 @@ def raise_error(error):
                     for key, value in json["data"].items():
                         if value and value.get("ok") is False:
                             raise_error(
-                                f"Failed to {description}: {value.get('message')}"
+                                f"""Failed to {description}: {
+                                    value.get('message')}"""
                             )
 
             return json
@@ -277,7 +279,8 @@ def make_rest_call(self, subpath: str, body: Dict[str, Any]) -> Dict:
                 return response.json()
             except ValueError as e:
                 raise ValueError(
-                    f"Failed to parse JSON response: {e}, content: {response.content!r}"
+                    f"""Failed to parse JSON response: {
+                        e}, content: {response.content!r}"""
                 )
 
     def gql_helper(
@@ -695,14 +698,15 @@ def upload_file(
         fields: Dict = request_dict.get("fields", {})
         object_key: Optional[str] = fields.get("key")
         upload_type: Literal["raw", "multipart"] = cast(
-            Literal["raw", "multipart"], request_dict.get("uploadType", "multipart")
+            Literal["raw", "multipart"], request_dict.get(
+                "uploadType", "multipart")
         )
         signed_url: Optional[str] = json_res.get("signedUrl")
 
         # Prepare form data
         form_data = (
             {}
-        )  # type: Dict[str, Union[Tuple[Union[str, None], Any], Tuple[Union[str, None], Any, Any]]]
+        ) # type: Dict[str, Union[Tuple[Union[str, None], Any], Tuple[Union[str, None], Any, Any]]]
         for field_name, field_value in fields.items():
             form_data[field_name] = (None, field_value)
 
@@ -772,7 +776,8 @@ def create_attachment(
             if active_steps := active_steps_var.get([]):
                 step_id = active_steps[-1].id
             else:
-                raise Exception("No step_id provided and no active step found.")
+                raise Exception(
+                    "No step_id provided and no active step found.")
 
         (
             query,
@@ -794,7 +799,8 @@ def create_attachment(
         )
 
         if content:
-            uploaded = self.upload_file(content=content, thread_id=thread_id, mime=mime)
+            uploaded = self.upload_file(
+                content=content, thread_id=thread_id, mime=mime)
 
             if uploaded["object_key"] is None or uploaded["url"] is None:
                 raise Exception("Failed to upload file")
@@ -1002,7 +1008,7 @@ def send_steps(self, steps: List[Union[StepDict, "Step"]]):
         Sends a list of steps to be processed.
 
         Args:
-            steps (List[Union[StepDict, "Step"]]): A list of steps or step dictionaries to send.
+            steps (List[Union[StepDict, Step]]): A list of steps or step dictionaries to send.
 
         Returns:
             The result of the GraphQL helper function for sending steps.
@@ -1448,7 +1454,8 @@ def raise_error(error):
                 json = response.json()
             except ValueError as e:
                 raise_error(
-                    f"Failed to parse JSON response: {e}, content: {response.content!r}"
+                    f"""Failed to parse JSON response: {
+                        e}, content: {response.content!r}"""
                 )
 
             if json.get("errors"):
@@ -1459,7 +1466,8 @@ def raise_error(error):
                     for key, value in json["data"].items():
                         if value and value.get("ok") is False:
                             raise_error(
-                                f"Failed to {description}: {value.get('message')}"
+                                f"""Failed to {description}: {
+                                    value.get('message')}"""
                             )
 
             return json
@@ -1498,7 +1506,8 @@ async def make_rest_call(self, subpath: str, body: Dict[str, Any]) -> Dict:
                 return response.json()
             except ValueError as e:
                 raise ValueError(
-                    f"Failed to parse JSON response: {e}, content: {response.content!r}"
+                    f"""Failed to parse JSON response: {
+                        e}, content: {response.content!r}"""
                 )
 
     async def gql_helper(
@@ -1943,14 +1952,15 @@ async def upload_file(
         fields: Dict = request_dict.get("fields", {})
         object_key: Optional[str] = fields.get("key")
         upload_type: Literal["raw", "multipart"] = cast(
-            Literal["raw", "multipart"], request_dict.get("uploadType", "multipart")
+            Literal["raw", "multipart"], request_dict.get(
+                "uploadType", "multipart")
         )
         signed_url: Optional[str] = json_res.get("signedUrl")
 
         # Prepare form data
         form_data = (
-            {}
-        )  # type: Dict[str, Union[Tuple[Union[str, None], Any], Tuple[Union[str, None], Any, Any]]]
+            {}  
+        ) # type: Dict[str, Union[Tuple[Union[str, None], Any], Tuple[Union[str, None], Any, Any]]]
         for field_name, field_value in fields.items():
             form_data[field_name] = (None, field_value)
 
@@ -2231,7 +2241,7 @@ async def send_steps(self, steps: List[Union[StepDict, "Step"]]):
         Asynchronously sends a list of steps to be processed.
 
         Args:
-            steps (List[Union[StepDict, "Step"]]): A list of steps or step dictionaries to send.
+            steps (List[Union[StepDict, Step]]): A list of steps or step dictionaries to send.
 
         Returns:
             The result of the GraphQL helper function for sending steps.