phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix tg ai stream check

Browse source
This commit is contained in:
ishaan-jaff 2023-08-17 10:49:48 -07:00
parent 35fd82fd3f
commit c22e0fe173
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

@ -324,7 +324,7 @@ def completion(
## LOGGING ## LOGGING
logging(model=model, input=prompt, custom_llm_provider=custom_llm_provider, logger_fn=logger_fn) logging(model=model, input=prompt, custom_llm_provider=custom_llm_provider, logger_fn=logger_fn)
if stream == True or optional_params['stream_tokens'] == True: if stream == True:
return together_ai_completion_streaming({ return together_ai_completion_streaming({
"model": model, "model": model,
"prompt": prompt, "prompt": prompt,