scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

get_custom_logger_for_prompt_management

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-19 17:38:24 -07:00
parent 805542005a
commit a49bd460e0
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/litellm_core_utils/litellm_logging.py
View file

@ -447,7 +447,9 @@ class Logging(LiteLLMLoggingBaseClass):
self.messages = messages
return model, messages, non_default_params
def get_custom_logger_for_prompt_management(self, model: str) -> CustomLogger:
def get_custom_logger_for_prompt_management(
self, model: str
) -> Optional[CustomLogger]:
for (
custom_logger_compatible_callback
) in litellm._known_custom_logger_compatible_callbacks:
@ -468,7 +470,7 @@ class Logging(LiteLLMLoggingBaseClass):
)
if len(custom_prompt_management_loggers) > 0:
return custom_prompt_management_loggers[0]
raise ValueError(f"No custom logger found for model: {model}")
return None
def _get_raw_request_body(self, data: Optional[Union[dict, str]]) -> dict:
if data is None: