scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-22 00:31:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 51

Adding a new catch exception block based on feedback.

Browse source
This commit is contained in:
Josh Bonczkowski 2026-03-13 15:28:20 -04:00
parent aeffbb8320
commit fc6f5fa1b8
2 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -117,6 +117,12 @@ class NewRelicLogger(CustomLogger):
"Integration will be disabled."
)
self.enabled = False
except Exception as e:
verbose_logger.error(
f"Failed to initialize New Relic agent: {e}. "
"Integration will be disabled."
)
self.enabled = False
def _get_newrelic_params(self) -> Dict:
"""

10
tests/test_litellm/integrations/newrelic/test_newrelic.py
View file

@ -126,6 +126,16 @@ class TestNewRelicLoggerInit:
logger = NewRelicLogger()
assert logger.enabled is False
def test_disabled_on_agent_startup_error(self):
with patch.object(
_mock_newrelic_agent,
"register_application",
side_effect=RuntimeError("agent startup failed"),
):
with patch.dict(os.environ, NR_ENV):
logger = NewRelicLogger()
assert logger.enabled is False
def test_record_content_default_true(self):
logger = make_logger()
assert logger.record_content is True