(fix) test streaming

This commit is contained in:
ishaan-jaff 2024-03-04 08:29:16 -08:00
parent ba460e6838
commit 3e8cecc03b

View file

@ -348,7 +348,7 @@ def test_completion_claude_stream():
},
]
response = completion(
model="claude-instant-1", messages=messages, stream=True, max_tokens=50
model="claude-instant-1.2", messages=messages, stream=True, max_tokens=50
)
complete_response = ""
# Add any assertions here to check the response