From f30742fe6e8e0ae3a753f37631f1c20fd0124f0f Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Tue, 27 Jan 2026 16:49:39 +0530 Subject: [PATCH] Fix mypy and code quality issues --- .../integrations/datadog/datadog_cost_management.py | 10 ++++++---- litellm/llms/bedrock/chat/converse_transformation.py | 2 +- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/litellm/integrations/datadog/datadog_cost_management.py b/litellm/integrations/datadog/datadog_cost_management.py index 9bfb2aba718..2eb94b59dd8 100644 --- a/litellm/integrations/datadog/datadog_cost_management.py +++ b/litellm/integrations/datadog/datadog_cost_management.py @@ -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 diff --git a/litellm/llms/bedrock/chat/converse_transformation.py b/litellm/llms/bedrock/chat/converse_transformation.py index 2ec27a7af08..ec665142073 100644 --- a/litellm/llms/bedrock/chat/converse_transformation.py +++ b/litellm/llms/bedrock/chat/converse_transformation.py @@ -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,