From 5db2a0c8850385b9e56b17bcd8053bab4fac0b59 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Fri, 18 Sep 2026 16:05:42 -0700
Subject: [PATCH] test(proxy): type the sqlstate test parameters

---
 tests/test_litellm/proxy/db/test_exception_handler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_litellm/proxy/db/test_exception_handler.py b/tests/test_litellm/proxy/db/test_exception_handler.py
index 26ac1ea65ad..f7cc5e3ed83 100644
--- a/tests/test_litellm/proxy/db/test_exception_handler.py
+++ b/tests/test_litellm/proxy/db/test_exception_handler.py
@@ -681,7 +681,7 @@ def test_is_deadlock_error_excludes_non_deadlocks(error):
         (httpx.ReadTimeout("no reply"), None),
     ],
 )
-def test_postgres_sqlstate_reads_the_code_prisma_attached_to_the_failed_statement(error, sqlstate):
+def test_postgres_sqlstate_reads_the_code_prisma_attached_to_the_failed_statement(error: Exception, sqlstate: str | None):
     """Only a prisma data error carrying Postgres's own error code yields a SQLSTATE; a
     codeless or malformed payload, an engine-level error, and a transport error yield None."""
     assert PrismaDBExceptionHandler.postgres_sqlstate(error) == sqlstate