tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-07 06:29:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
Download Patch File Download Diff File Expand all files Collapse all files
litellm/litellm_core_utils/litellm_logging.py
+4 -2
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: