phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix test custom callback router

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-28 16:54:28 -07:00
parent a765bae2b6
commit c417ab7a77
1 changed files with 21 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

21
litellm/llms/azure.py
View file

@ -717,11 +717,32 @@ class AzureChatCompletion(BaseLLM):
model_response_object=model_response, model_response_object=model_response,
) )
except AzureOpenAIError as e: except AzureOpenAIError as e:
## LOGGING
logging_obj.post_call(
input=data["messages"],
api_key=api_key,
additional_args={"complete_input_dict": data},
original_response=str(e),
)
exception_mapping_worked = True exception_mapping_worked = True
raise e raise e
except asyncio.CancelledError as e: except asyncio.CancelledError as e:
## LOGGING
logging_obj.post_call(
input=data["messages"],
api_key=api_key,
additional_args={"complete_input_dict": data},
original_response=str(e),
)
raise AzureOpenAIError(status_code=500, message=str(e)) raise AzureOpenAIError(status_code=500, message=str(e))
except Exception as e: except Exception as e:
## LOGGING
logging_obj.post_call(
input=data["messages"],
api_key=api_key,
additional_args={"complete_input_dict": data},
original_response=str(e),
)
if hasattr(e, "status_code"): if hasattr(e, "status_code"):
raise e raise e
else: else: