From 5c9473874c00c8d205788d12a3d35f0cf2136b84 Mon Sep 17 00:00:00 2001
From: Tanaro Laptop <tanaro.schaedler@uni-ulm.de>
Date: Sat, 20 Jan 2024 01:50:27 +0100
Subject: [PATCH] change max_tokens type to int

---
 litellm/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/main.py b/litellm/main.py
index 2fef048a698..407e851f5b6 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -366,7 +366,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,