scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Minor fixes for basic observability.

Browse source
This commit is contained in:
Josh Bonczkowski 2025-11-18 15:35:18 -05:00
parent ab8f155ffc
commit 375545e0ca
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/integrations/newrelic/newrelic.py
View file

@ -139,7 +139,7 @@ class NewRelicLogger(CustomLogger):
litellm_params = kwargs.get("litellm_params", {}) or {}
return litellm_params.get("custom_llm_provider", "unknown")
def _get_model_names(self, kwargs: Dict, response_obj: Dict) -> tuple[str, str]:
def _get_model_names(self, kwargs: Dict, response_obj: Dict) -> Tuple[str, str]:
"""
Extract request and response model names.
@ -299,6 +299,7 @@ class NewRelicLogger(CustomLogger):
except Exception as e:
verbose_logger.warning(f"Failed to record New Relic summary event: {e}")
self.handle_callback_failure("newrelic")
def _record_message_events(
self,
@ -334,6 +335,7 @@ class NewRelicLogger(CustomLogger):
except Exception as e:
verbose_logger.warning(f"Failed to record New Relic message events: {e}")
self.handle_callback_failure("newrelic")
def _record_error_metric(self):
"""Record error metric to New Relic."""
@ -345,6 +347,7 @@ class NewRelicLogger(CustomLogger):
except Exception as e:
verbose_logger.warning(f"Failed to record New Relic error metric: {e}")
self.handle_callback_failure("newrelic")
def _process_success(self, kwargs: Dict, response_obj: Dict):
"""
@ -414,6 +417,7 @@ class NewRelicLogger(CustomLogger):
self._process_success(kwargs, response_obj)
except Exception as e:
verbose_logger.warning(f"Error in New Relic log_success_event: {e}")
self.handle_callback_failure("newrelic")
async def async_log_success_event(self, kwargs, response_obj, start_time, end_time):
"""
@ -426,6 +430,7 @@ class NewRelicLogger(CustomLogger):
self._process_success(kwargs, response_obj)
except Exception as e:
verbose_logger.warning(f"Error in New Relic async_log_success_event: {e}")
self.handle_callback_failure("newrelic")
def log_failure_event(self, kwargs, response_obj, start_time, end_time):
"""
@ -441,6 +446,7 @@ class NewRelicLogger(CustomLogger):
except Exception as e:
verbose_logger.warning(f"Error in New Relic log_failure_event: {e}")
self.handle_callback_failure("newrelic")
async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time):
"""
@ -456,3 +462,4 @@ class NewRelicLogger(CustomLogger):
except Exception as e:
verbose_logger.warning(f"Error in New Relic async_log_failure_event: {e}")
self.handle_callback_failure("newrelic")