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

13
litellm/utils.py
View file

@ -1572,11 +1572,14 @@ def client(original_function):
def post_call_processing(original_response, model): def post_call_processing(original_response, model):
try: try:
call_type = original_function.__name__ if original_response is None:
if call_type == CallTypes.completion.value or call_type == CallTypes.acompletion.value: pass
model_response = original_response['choices'][0]['message']['content'] else:
### POST-CALL RULES ### call_type = original_function.__name__
rules_obj.post_call_rules(input=model_response, model=model) if call_type == CallTypes.completion.value or call_type == CallTypes.acompletion.value:
model_response = original_response['choices'][0]['message']['content']
### POST-CALL RULES ###
rules_obj.post_call_rules(input=model_response, model=model)
except Exception as e: except Exception as e:
raise e raise e