fix: circualr imports

Author: asafgardin

ai21/models/chat/chat_completion_response.py

@@ -2,9 +2,9 @@
 from typing import Optional, List
 
 from ai21.models.ai21_base_model_mixin import AI21BaseModelMixin
-from ai21.models.chat import ChatMessage
 from ai21.models.logprobs import Logprobs
 from ai21.models.usage_info import UsageInfo
+from .chat_message import ChatMessage
 
 __all__ = ["ChatCompletionResponse", "ChatCompletionResponseChoice"]
 

ai21/models/chat/chat_message.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 
-from ai21.models import RoleType
 from ai21.models.ai21_base_model_mixin import AI21BaseModelMixin
+from .role_type import RoleType
 
 __all__ = ["ChatMessage"]