add max_tokens

This commit is contained in:
WilliamEspegren 2023-09-17 10:27:32 +02:00
parent 43a18b9528
commit 71afdbc0e5
No known key found for this signature in database
GPG key ID: 30E0CDDF1B5262CD

View file

@ -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: