refactor: 🎨 reformat functions with 2+ arguments

Author: benshuk

ai21/clients/common/assistant/messages.py

@@ -11,7 +11,12 @@ class Messages(ABC):
 
     @abstractmethod
     def create(
-        self, thread_id: str, *, role: ThreadMessageRole, content: MessageContentText, **kwargs
+        self,
+        thread_id: str,
+        *,
+        role: ThreadMessageRole,
+        content: MessageContentText,
+        **kwargs,
     ) -> MessageResponse:
         pass
 

ai21/clients/common/assistant/threads.py

@@ -13,7 +13,11 @@ class Threads(ABC):
     messages: Messages
 
     @abstractmethod
-    def create(self, messages: List[Message], **kwargs) -> Thread:
+    def create(
+        self,
+        messages: List[Message],
+        **kwargs,
+    ) -> Thread:
         pass
 
     @abstractmethod

ai21/clients/studio/resources/assistant/studio_thread.py

@@ -17,7 +17,11 @@ def __init__(self, client: AI21HTTPClient):
 
         self.messages = StudioThreadMessage(client)
 
-    def create(self, messages: List[Message], **kwargs) -> Thread:
+    def create(
+        self,
+        messages: List[Message],
+        **kwargs,
+    ) -> Thread:
         body = dict(messages=messages)
 
         return self._post(path=f"/{self._module_name}", body=body, response_cls=Thread)
@@ -32,7 +36,11 @@ def __init__(self, client: AsyncAI21HTTPClient):
 
         self.messages = AsyncStudioThreadMessage(client)
 
-    async def create(self, messages: List[Message], **kwargs) -> Thread:
+    async def create(
+        self,
+        messages: List[Message],
+        **kwargs,
+    ) -> Thread:
         body = dict(messages=messages)
 
         return await self._post(path=f"/{self._module_name}", body=body, response_cls=Thread)

ai21/clients/studio/resources/assistant/studio_thread_message.py

@@ -8,7 +8,12 @@
 
 class StudioThreadMessage(StudioResource, Messages):
     def create(
-        self, thread_id: str, *, role: ThreadMessageRole, content: MessageContentText, **kwargs
+        self,
+        thread_id: str,
+        *,
+        role: ThreadMessageRole,
+        content: MessageContentText,
+        **kwargs,
     ) -> MessageResponse:
         body = dict(
             role=role,
@@ -23,7 +28,12 @@ def list(self, thread_id: str) -> ListMessageResponse:
 
 class AsyncStudioThreadMessage(AsyncStudioResource, Messages):
     async def create(
-        self, thread_id: str, *, role: ThreadMessageRole, content: MessageContentText, **kwargs
+        self,
+        thread_id: str,
+        *,
+        role: ThreadMessageRole,
+        content: MessageContentText,
+        **kwargs,
     ) -> MessageResponse:
         body = dict(
             role=role,