fix anthropic streaming

This commit is contained in:
ishaan-jaff 2023-08-08 16:07:53 -07:00
parent 6ea264d561
commit 4c31ee342d

View file

@ -239,7 +239,7 @@ def completion(
max_tokens_to_sample=max_tokens_to_sample,
**optional_params
)
if optional_params['stream'] == True:
if 'stream' in optional_params and optional_params['stream'] == True:
# don't try to access stream object,
response = CustomStreamWrapper(completion)
return response