fix: all import

Author: asafgardin

ai21/models/chat/__init__.py

@@ -4,3 +4,5 @@
 from .chat_completion_response import ChatCompletionResponseChoice
 from .chat_message import ChatMessage
 from .role_type import RoleType as RoleType
+
+__all__ = ["ChatCompletionResponse", "ChatCompletionResponseChoice", "ChatMessage", "RoleType"]

ai21/models/chat/chat_completion_response.py

@@ -6,8 +6,6 @@
 from ai21.models.usage_info import UsageInfo
 from .chat_message import ChatMessage
 
-__all__ = ["ChatCompletionResponse", "ChatCompletionResponseChoice"]
-
 
 @dataclass
 class ChatCompletionResponseChoice(AI21BaseModelMixin):

ai21/models/chat/chat_message.py

@@ -3,8 +3,6 @@
 from ai21.models.ai21_base_model_mixin import AI21BaseModelMixin
 from .role_type import RoleType
 
-__all__ = ["ChatMessage"]
-
 
 @dataclass
 class ChatMessage(AI21BaseModelMixin):

ai21/models/chat/role_type.py

@@ -1,7 +1,5 @@
 from enum import Enum
 
-__all__ = ["RoleType"]
-
 
 class RoleType(str, Enum):
     USER = "user"

pyproject.toml

@@ -116,5 +116,3 @@ prerelease = true
 
 [tool.ruff]
 line-length = 120
-[tool.ruff.per-file-ignores]
-"__init__.py" = ["F401"]