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

updates to litedebugger logging

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-25 16:20:57 -07:00
parent 0f7245d153
commit c423db29f3
3 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/integrations/litedebugger.py
View file

@ -62,7 +62,7 @@ class LiteDebugger:
): ):
try: try:
litellm_data_obj = { litellm_data_obj = {
"status": "succeeded", "status": "success",
"additional_details": {"original_response": original_response}, "additional_details": {"original_response": original_response},
"litellm_call_id": litellm_call_id, "litellm_call_id": litellm_call_id,
"user_email": self.user_email, "user_email": self.user_email,

1
litellm/main.py
View file

@ -143,6 +143,7 @@ def completion(
custom_llm_provider=custom_llm_provider, custom_llm_provider=custom_llm_provider,
custom_api_base=custom_api_base, custom_api_base=custom_api_base,
litellm_call_id=litellm_call_id, litellm_call_id=litellm_call_id,
model_alias_map=litellm.model_alias_map
) )
logging = Logging( logging = Logging(
model=model, model=model,

2
litellm/utils.py
View file

@ -552,6 +552,7 @@ def get_litellm_params(
custom_llm_provider=None, custom_llm_provider=None,
custom_api_base=None, custom_api_base=None,
litellm_call_id=None, litellm_call_id=None,
model_alias_map=None
): ):
litellm_params = { litellm_params = {
"return_async": return_async, "return_async": return_async,
@ -562,6 +563,7 @@ def get_litellm_params(
"custom_llm_provider": custom_llm_provider, "custom_llm_provider": custom_llm_provider,
"custom_api_base": custom_api_base, "custom_api_base": custom_api_base,
"litellm_call_id": litellm_call_id, "litellm_call_id": litellm_call_id,
"model_alias_map": model_alias_map
} }
return litellm_params return litellm_params