add max_tokens

2023-09-17 10:27:32 +02:00 · 2023-09-17 10:27:32 +02:00 · 71afdbc0e5
commit 71afdbc0e5
parent 43a18b9528
1 changed files with 2 additions and 1 deletions
--- 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: