From 018a6a90ad18a82f681bfa93f9edc1b59f85843e Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 22 Sep 2026 08:27:40 +0000
Subject: [PATCH] refactor(vercel_ai_gateway): type the embedding error headers
 dict

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 litellm/llms/vercel_ai_gateway/embedding/transformation.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/litellm/llms/vercel_ai_gateway/embedding/transformation.py b/litellm/llms/vercel_ai_gateway/embedding/transformation.py
index bb07a945f7e..8a672bf8991 100644
--- a/litellm/llms/vercel_ai_gateway/embedding/transformation.py
+++ b/litellm/llms/vercel_ai_gateway/embedding/transformation.py
@@ -161,7 +161,9 @@ class VercelAIGatewayEmbeddingConfig(BaseEmbeddingConfig):
                 optional_params[param] = value
         return optional_params
 
-    def get_error_class(self, error_message: str, status_code: int, headers: dict | httpx.Headers) -> BaseLLMException:
+    def get_error_class(
+        self, error_message: str, status_code: int, headers: dict[str, str] | httpx.Headers
+    ) -> BaseLLMException:
         """
         Get the error class for Vercel AI Gateway errors.
         """