add max_tokens

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