phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix anthropic streaming

Browse source
This commit is contained in:
ishaan-jaff 2023-08-08 16:07:53 -07:00
parent 36a6ac9b08
commit 5040d08f79
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

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