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

fix(ollama.py): raise async errors

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-19 15:00:52 +00:00
parent 9eb487efb3
commit 5936664a16
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/utils.py
View file

@ -1572,6 +1572,9 @@ def client(original_function):
def post_call_processing(original_response, model): def post_call_processing(original_response, model):
try: try:
if original_response is None:
pass
else:
call_type = original_function.__name__ call_type = original_function.__name__
if call_type == CallTypes.completion.value or call_type == CallTypes.acompletion.value: if call_type == CallTypes.completion.value or call_type == CallTypes.acompletion.value:
model_response = original_response['choices'][0]['message']['content'] model_response = original_response['choices'][0]['message']['content']