From 5936664a1671fdda2381b82d29892e794676ca0e Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 19 Dec 2023 15:00:52 +0000 Subject: [PATCH] fix(ollama.py): raise async errors --- litellm/utils.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 45d5d02f0dc..90678f45c58 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1572,11 +1572,14 @@ def client(original_function): def post_call_processing(original_response, model): try: - call_type = original_function.__name__ - 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) + if original_response is None: + pass + else: + call_type = original_function.__name__ + 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: raise e