From d3a86e7b7f09e1b0a1ca3b52f12bb128acdc0a73 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebasti=C3=A1n=20Est=C3=A9vez?= Date: Tue, 2 Apr 2024 00:28:08 -0400 Subject: [PATCH] Fix max_tokens type in main.py --- litellm/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index 1fca06fa5..47876afab 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -179,7 +179,7 @@ async def acompletion( n: Optional[int] = None, stream: Optional[bool] = None, stop=None, - max_tokens: Optional[float] = None, + max_tokens: Optional[int] = None, presence_penalty: Optional[float] = None, frequency_penalty: Optional[float] = None, logit_bias: Optional[dict] = None, @@ -422,7 +422,7 @@ def completion( n: Optional[int] = None, stream: Optional[bool] = None, stop=None, - max_tokens: Optional[float] = None, + max_tokens: Optional[int] = None, presence_penalty: Optional[float] = None, frequency_penalty: Optional[float] = None, logit_bias: Optional[dict] = None,