format and fix types

Author: MHHukiewitz

src/aleph/sdk/client.py

@@ -60,7 +60,7 @@
     MessageNotFoundError,
     MultipleMessagesError,
 )
-from .models import MessagesResponse, PostsResponse, Post
+from .models import MessagesResponse, Post, PostsResponse
 from .utils import check_unix_socket_valid, get_message_type_value
 
 logger = logging.getLogger(__name__)
@@ -722,7 +722,7 @@ async def get_messages(
         pagination: int = 200,
         page: int = 1,
         message_type: Optional[MessageType] = None,
-        message_types: Optional[List[MessageType]] = None,
+        message_types: Optional[Iterable[MessageType]] = None,
         content_types: Optional[Iterable[str]] = None,
         content_keys: Optional[Iterable[str]] = None,
         refs: Optional[Iterable[str]] = None,

src/aleph/sdk/models.py

@@ -1,6 +1,6 @@
-from typing import List, Optional, Any, Dict, Union
+from typing import Any, Dict, List, Optional, Union
 
-from aleph_message.models import AlephMessage, BaseMessage, ItemHash, ChainRef
+from aleph_message.models import AlephMessage, BaseMessage, ChainRef, ItemHash
 from pydantic import BaseModel, Field
 
 
@@ -23,16 +23,25 @@ class Post(BaseMessage):
     A post is a type of message that can be updated. Over the get_posts API
     we get the latest version of a post.
     """
+
     hash: ItemHash = Field(description="Hash of the content (sha256 by default)")
-    original_item_hash: ItemHash = Field(description="Hash of the original content (sha256 by default)")
+    original_item_hash: ItemHash = Field(
+        description="Hash of the original content (sha256 by default)"
+    )
     original_signature: Optional[str] = Field(
         description="Cryptographic signature of the original message by the sender"
     )
-    original_type: str = Field(description="The original, user-generated 'content-type' of the POST message")
-    content: Dict[str, Any] = Field(description="The content.content of the POST message")
+    original_type: str = Field(
+        description="The original, user-generated 'content-type' of the POST message"
+    )
+    content: Dict[str, Any] = Field(
+        description="The content.content of the POST message"
+    )
     type: str = Field(description="The content.type of the POST message")
     address: str = Field(description="The address of the sender of the POST message")
-    ref: Optional[Union[str, ChainRef]] = Field(description="Other message referenced by this one")
+    ref: Optional[Union[str, ChainRef]] = Field(
+        description="Other message referenced by this one"
+    )
 
 
 class PostsResponse(PaginationResponse):

tests/unit/test_asynchronous_get.py

@@ -3,7 +3,7 @@
 from unittest.mock import AsyncMock
 
 import pytest
-from aleph_message.models import MessagesResponse, MessageType
+from aleph_message.models import MessagesResponse
 
 from aleph.sdk.client import AlephClient
 from aleph.sdk.conf import settings