litellm_mypy_fix_batch1: fix type errors in files, opentelemetry, gemini transformation, key management

litellm/files/main.py

@@ -9,7 +9,7 @@
 import contextvars
 import os
 import time
-import uuid
+import uuid as uuid_module
 from functools import partial
 from typing import Any, Coroutine, Dict, Literal, Optional, Union, cast
 
@@ -451,7 +451,7 @@ def file_retrieve(
                         stream=False,
                         call_type="afile_retrieve" if _is_async else "file_retrieve",
                         start_time=time.time(),
-                        litellm_call_id=kwargs.get("litellm_call_id", str(uuid.uuid4())),
+                        litellm_call_id=kwargs.get("litellm_call_id", str(uuid_module.uuid4())),
                         function_id=str(kwargs.get("id") or ""),
                     )
 
@@ -660,7 +660,7 @@ def file_delete(
                         stream=False,
                         call_type="afile_delete" if _is_async else "file_delete",
                         start_time=time.time(),
-                        litellm_call_id=kwargs.get("litellm_call_id", str(uuid.uuid4())),
+                        litellm_call_id=kwargs.get("litellm_call_id", str(uuid_module.uuid4())),
                         function_id=str(kwargs.get("id") or ""),
                     )
 
@@ -793,7 +793,7 @@ def file_list(
                     stream=False,
                     call_type="afile_list" if _is_async else "file_list",
                     start_time=time.time(),
-                    litellm_call_id=kwargs.get("litellm_call_id", str(uuid.uuid4())),
+                    litellm_call_id=kwargs.get("litellm_call_id", str(uuid_module.uuid4())),
                     function_id=str(kwargs.get("id", "")),
                 )
 

litellm/integrations/opentelemetry.py

@@ -1631,7 +1631,7 @@ def set_attributes(  # noqa: PLR0915
                                     )
 
         except Exception as e:
-            self.handle_callback_failure(callback_name= self.callback_name)  
+            self.handle_callback_failure(callback_name=self.callback_name or "opentelemetry")  
             verbose_logger.exception(
                 "OpenTelemetry logging error in set_attributes %s", str(e)
             )

litellm/llms/gemini/files/transformation.py

@@ -301,7 +301,7 @@ def transform_delete_file_request(
         url = f"{api_base}/v1beta/{file_name}"
 
         # Add API key as header (Google AI Studio uses x-goog-api-key header)
-        params = {}
+        params: dict = {}
 
         return url, params
 

litellm/proxy/management_endpoints/key_management_endpoints.py

@@ -1541,13 +1541,14 @@ async def _process_single_key_update(
     )
 
     # Check team member permissions
-    await TeamMemberPermissionChecks.can_team_member_execute_key_management_endpoint(
-        user_api_key_dict=user_api_key_dict,
-        route=KeyManagementRoutes.KEY_UPDATE,
-        prisma_client=prisma_client,
-        existing_key_row=existing_key_row,
-        user_api_key_cache=user_api_key_cache,
-    )
+    if prisma_client is not None:
+        await TeamMemberPermissionChecks.can_team_member_execute_key_management_endpoint(
+            user_api_key_dict=user_api_key_dict,
+            route=KeyManagementRoutes.KEY_UPDATE,
+            prisma_client=prisma_client,
+            existing_key_row=existing_key_row,
+            user_api_key_cache=user_api_key_cache,
+        )
 
     # Create UpdateKeyRequest from BulkUpdateKeyRequestItem
     update_key_request = UpdateKeyRequest(