From 71afdbc0e577d37eb9f5d21e27e0ae3fdcdcf443 Mon Sep 17 00:00:00 2001
From: WilliamEspegren <espegren.w@gmail.com>
Date: Sun, 17 Sep 2023 10:27:32 +0200
Subject: [PATCH] add max_tokens

---
 litellm/tests/test_stream_chunk_builder.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/litellm/tests/test_stream_chunk_builder.py b/litellm/tests/test_stream_chunk_builder.py
index fbfc78554a..a0ae18d6cc 100644
--- a/litellm/tests/test_stream_chunk_builder.py
+++ b/litellm/tests/test_stream_chunk_builder.py
@@ -10,7 +10,8 @@ def test_stream_chunk_builder():
     response = completion(
         model="gpt-3.5-turbo",
         messages=messages,
-        stream=True
+        stream=True,
+        max_tokens=10,
     )
 
     for chunk in response: