From f09736ceb95bf037f7f00bbc3b8f53c164c6408b Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Mon, 3 Nov 2025 18:02:19 -0800 Subject: [PATCH] fix mypy lint --- litellm/integrations/langfuse/langfuse_otel.py | 12 ++++++------ litellm/llms/azure_ai/ocr/transformation.py | 2 ++ .../deepgram/audio_transcription/transformation.py | 2 +- 3 files changed, 9 insertions(+), 7 deletions(-) diff --git a/litellm/integrations/langfuse/langfuse_otel.py b/litellm/integrations/langfuse/langfuse_otel.py index 0ca430e1334..6992ea17cc8 100644 --- a/litellm/integrations/langfuse/langfuse_otel.py +++ b/litellm/integrations/langfuse/langfuse_otel.py @@ -168,16 +168,16 @@ class LangfuseOtelLogger(OpenTelemetry): output = response_obj.get("output", []) if output: - output_data = [] + output_items_data: list[dict] = [] for item in output: if hasattr(item, "type"): item_type = item.type if item_type == "reasoning" and hasattr(item, "summary"): for summary in item.summary: if hasattr(summary, "text"): - output_data.append({"role": "reasoning_summary", "content": summary.text}) + output_items_data.append({"role": "reasoning_summary", "content": summary.text}) elif item_type == "message": - output_data.append({ + output_items_data.append({ "role": getattr(item, "role", "assistant"), "content": getattr(getattr(item, "content", [{}])[0], "text", "") }) @@ -191,9 +191,9 @@ class LangfuseOtelLogger(OpenTelemetry): "type": "function_call", "arguments": arguments_obj, } - output_data.append(langfuse_tool_call) - if output_data: - safe_set_attribute(span, LangfuseSpanAttributes.OBSERVATION_OUTPUT.value, safe_dumps(output_data)) + output_items_data.append(langfuse_tool_call) + if output_items_data: + safe_set_attribute(span, LangfuseSpanAttributes.OBSERVATION_OUTPUT.value, safe_dumps(output_items_data)) @staticmethod def _set_langfuse_specific_attributes(span: Span, kwargs, response_obj): diff --git a/litellm/llms/azure_ai/ocr/transformation.py b/litellm/llms/azure_ai/ocr/transformation.py index eade2dd765f..24fc9e86134 100644 --- a/litellm/llms/azure_ai/ocr/transformation.py +++ b/litellm/llms/azure_ai/ocr/transformation.py @@ -35,6 +35,7 @@ class AzureAIOCRConfig(MistralOCRConfig): model: str, api_key: Optional[str] = None, api_base: Optional[str] = None, + litellm_params: Optional[dict] = None, **kwargs, ) -> Dict: """ @@ -73,6 +74,7 @@ class AzureAIOCRConfig(MistralOCRConfig): api_base: Optional[str], model: str, optional_params: dict, + litellm_params: Optional[dict] = None, **kwargs, ) -> str: """ diff --git a/litellm/llms/deepgram/audio_transcription/transformation.py b/litellm/llms/deepgram/audio_transcription/transformation.py index fe63ad11bc7..6a540d72778 100644 --- a/litellm/llms/deepgram/audio_transcription/transformation.py +++ b/litellm/llms/deepgram/audio_transcription/transformation.py @@ -150,7 +150,7 @@ class DeepgramAudioTranscriptionConfig(BaseAudioTranscriptionConfig): segments = [] current_speaker = None - current_words = [] + current_words: list[str] = [] for word_obj in words: speaker = word_obj.get("speaker")