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

fix(utils.py): fix exception raised

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-10 11:28:17 -08:00
parent 18a8bd5543
commit 62290ec5d9
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -2563,7 +2563,7 @@ def handle_failure(exception, traceback_exception, start_time, end_time, args, k
def convert_to_model_response_object(response_object: Optional[dict]=None, model_response_object: Optional[ModelResponse]=None):
try:
if response_object is None or model_response_object is None:
raise OpenAIError(status_code=500, message="Error in response object format")
raise Exception("Error in response object format")
choice_list=[]
for idx, choice in enumerate(response_object["choices"]):
message = Message(content=choice["message"]["content"], role=choice["message"]["role"], function_call=choice["message"].get("function_call", None))
@ -2581,7 +2581,7 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model
model_response_object.model = response_object["model"]
return model_response_object
except:
OpenAIError(status_code=500, message="Invalid response object.")
Exception("Invalid response object.")
# NOTE: DEPRECATING this in favor of using success_handler() in Logging: