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

fix(test_text_completion.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-08 23:03:17 +05:30
parent ff12e023ae
commit fd6ccfca7d
3 changed files with 14 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/utils.py
View file

@ -2267,11 +2267,16 @@ def client(original_function):
and kwargs["complete_response"] == True
):
chunks = []
for idx, chunk in enumerate(result):
async for chunk in result:
chunks.append(chunk)
return litellm.stream_chunk_builder(
chunks, messages=kwargs.get("messages", None)
)
if call_type == CallTypes.acompletion.value:
return litellm.stream_chunk_builder(
chunks, messages=kwargs.get("messages")
)
elif call_type == CallTypes.atext_completion.value:
return litellm.stream_chunk_builder(
chunks, messages=[{"role": "user", "content": kwargs.get("prompt")}]
)
else:
return result