scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-26 01:12:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 98

style(mavvrik): address pattern conformance findings 3 and 4

Browse source 
Finding 3 — proxy_server.py: remove inline `import os as _os`
  Every other integration block in _initialize_spend_tracking_background_jobs
  uses the module-level os import already in scope. Changed _os.getenv(...)
  to os.getenv(...) and removed the mid-function import alias.

Finding 4 — custom_logger_registry.py: fix MavvrikLogger import order
  MavvrikLogger was inserted between cloudzero and datadog imports.
  Alphabetically mavvrik sorts after datadog/datadog_llm_obs/datadog_metrics
  and before deepeval. Moved import to the correct alphabetical position.

Co-Authored-By: Claude Sonnet 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Praveen Ghuge 2026-04-26 07:58:21 +05:30
parent fee9c1023c
commit 5af15a2b59
2 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/litellm_core_utils/custom_logger_registry.py
View file

@ -18,11 +18,11 @@ from litellm.integrations.azure_storage.azure_storage import AzureBlobStorageLog
from litellm.integrations.bitbucket import BitBucketPromptManager
from litellm.integrations.braintrust_logging import BraintrustLogger
from litellm.integrations.cloudzero.cloudzero import CloudZeroLogger
from litellm.integrations.mavvrik import Logger as MavvrikLogger
from litellm.integrations.datadog.datadog import DataDogLogger
from litellm.integrations.datadog.datadog_llm_obs import DataDogLLMObsLogger
from litellm.integrations.datadog.datadog_metrics import DatadogMetricsLogger
from litellm.integrations.deepeval import DeepEvalLogger
from litellm.integrations.mavvrik import Logger as MavvrikLogger
from litellm.integrations.dotprompt import DotpromptManager
from litellm.integrations.focus.focus_logger import FocusLogger
from litellm.integrations.vantage.vantage_logger import VantageLogger

7
litellm/proxy/proxy_server.py
View file

@ -6737,14 +6737,13 @@ class ProxyStartupEvent:
# Skip DB load when credentials come from env vars — avoids raising
# if prisma_client is not yet connected at startup.
data = {} if settings.has_env_vars else await settings.load()
import os as _os
client = MavvrikClient(
api_key=data.get("api_key") or _os.getenv("MAVVRIK_API_KEY", ""),
api_key=data.get("api_key") or os.getenv("MAVVRIK_API_KEY", ""),
api_endpoint=data.get("api_endpoint")
or _os.getenv("MAVVRIK_API_ENDPOINT", ""),
or os.getenv("MAVVRIK_API_ENDPOINT", ""),
connection_id=data.get("connection_id")
or _os.getenv("MAVVRIK_CONNECTION_ID", ""),
or os.getenv("MAVVRIK_CONNECTION_ID", ""),
)
uploader = MavvrikUploader(client=client)
orchestrator = MavvrikOrchestrator(client=client, uploader=uploader)