diff --git a/docs/my-website/docs/completion/input.md b/docs/my-website/docs/completion/input.md index 676e4d232..445fd072b 100644 --- a/docs/my-website/docs/completion/input.md +++ b/docs/my-website/docs/completion/input.md @@ -73,7 +73,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, diff --git a/litellm/main.py b/litellm/main.py index 407e851f5..fb7574b06 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -143,7 +143,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, @@ -1884,7 +1884,7 @@ def batch_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, diff --git a/litellm/types/completion.py b/litellm/types/completion.py index 5eac90575..3ce441880 100644 --- a/litellm/types/completion.py +++ b/litellm/types/completion.py @@ -12,7 +12,7 @@ class CompletionRequest(BaseModel): n: Optional[int] = None stream: Optional[bool] = None stop: Optional[dict] = 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