From b5aa441a006233f1adbff99ac7b14a3a32e514fc Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 26 Mar 2025 17:17:56 -0700 Subject: [PATCH] add clear type hints to litellm.messages.create functions --- litellm/messages/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/messages/__init__.py b/litellm/messages/__init__.py index 0a4ea92c2c..b2be67bbd9 100644 --- a/litellm/messages/__init__.py +++ b/litellm/messages/__init__.py @@ -9,6 +9,9 @@ from typing import Dict, List, Optional, Union from litellm.llms.anthropic.experimental_pass_through.handler import ( anthropic_messages as _async_anthropic_messages, ) +from litellm.types.llms.anthropic_messages.anthropic_response import ( + AnthropicMessagesResponse, +) async def acreate( @@ -26,7 +29,7 @@ async def acreate( top_k: Optional[int] = None, top_p: Optional[float] = None, **kwargs -) -> Dict: +) -> AnthropicMessagesResponse: """ Async wrapper for Anthropic's messages API @@ -82,7 +85,7 @@ async def create( top_k: Optional[int] = None, top_p: Optional[float] = None, **kwargs -) -> Dict: +) -> AnthropicMessagesResponse: """ Async wrapper for Anthropic's messages API