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

Fix mypy and code quality issues

Browse source
This commit is contained in:
Sameer Kankute 2026-01-27 16:49:39 +05:30
parent 154ad179af
commit f30742fe6e
2 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/integrations/datadog/datadog_cost_management.py
View file

@ -2,7 +2,7 @@ import asyncio
import os
import time
from datetime import datetime
from typing import Dict, List, Optional
from typing import Dict, List, Optional, Tuple
from litellm._logging import verbose_logger
from litellm.integrations.custom_batch_logger import CustomBatchLogger
@ -93,7 +93,7 @@ class DatadogCostManagementLogger(CustomBatchLogger):
Aggregates costs by Provider, Model, and Date.
Returns a list of DatadogFOCUSCostEntry.
"""
aggregator: Dict[str, DatadogFOCUSCostEntry] = {}
aggregator: Dict[Tuple[str, str, str, Tuple[Tuple[str, str], ...]], DatadogFOCUSCostEntry] = {}
for log in logs:
try:
@ -171,8 +171,10 @@ class DatadogCostManagementLogger(CustomBatchLogger):
tags["user"] = str(metadata["user_api_key_alias"])
if "user_api_key_team_alias" in metadata:
tags["team"] = str(metadata["user_api_key_team_alias"])
if "model_group" in metadata:
tags["model_group"] = str(metadata["model_group"])
# model_group is not in StandardLoggingMetadata TypedDict, so we need to access it via dict.get()
model_group = metadata.get("model_group") # type: ignore[misc]
if model_group:
tags["model_group"] = str(model_group)
return tags

2
litellm/llms/bedrock/chat/converse_transformation.py
View file

@ -1501,7 +1501,7 @@ class AmazonConverseConfig(BaseConfig):
return content_str, tools, reasoningContentBlocks, citationsContentBlocks
def _transform_response(
def _transform_response( # noqa: PLR0915
self,
model: str,
response: httpx.Response,