diff --git a/litellm/llms/anthropic/experimental_pass_through/messages/handler.py b/litellm/llms/anthropic/experimental_pass_through/messages/handler.py index 54826a38ba..0b809a4460 100644 --- a/litellm/llms/anthropic/experimental_pass_through/messages/handler.py +++ b/litellm/llms/anthropic/experimental_pass_through/messages/handler.py @@ -65,7 +65,7 @@ class AnthropicMessagesHandler: @client async def anthropic_messages( max_tokens: int, - messages: List[Dict[str, Union[str, List[Dict[str, str]]]]], + messages: List[Dict], model: str, metadata: Optional[Dict] = None, stop_sequences: Optional[List[str]] = None, diff --git a/litellm/messages/__init__.py b/litellm/messages/__init__.py index b2be67bbd9..c5cb3f5330 100644 --- a/litellm/messages/__init__.py +++ b/litellm/messages/__init__.py @@ -16,7 +16,7 @@ from litellm.types.llms.anthropic_messages.anthropic_response import ( async def acreate( max_tokens: int, - messages: List[Dict[str, Union[str, List[Dict[str, str]]]]], + messages: List[Dict], model: str, metadata: Optional[Dict] = None, stop_sequences: Optional[List[str]] = None, @@ -72,7 +72,7 @@ async def acreate( async def create( max_tokens: int, - messages: List[Dict[str, Union[str, List[Dict[str, str]]]]], + messages: List[Dict], model: str, metadata: Optional[Dict] = None, stop_sequences: Optional[List[str]] = None,