From 020cbba4ddc4c336b4fd6a39de146aa1392f1e2d Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Sat, 19 Sep 2026 12:39:44 -0700
Subject: [PATCH] refactor(batches): annotate the stored file row so its model
 import is a real use

---
 litellm/repositories/managed_file_content_repository.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/litellm/repositories/managed_file_content_repository.py b/litellm/repositories/managed_file_content_repository.py
index 8810269279c..c55d0060080 100644
--- a/litellm/repositories/managed_file_content_repository.py
+++ b/litellm/repositories/managed_file_content_repository.py
@@ -3,7 +3,7 @@ from typing import TYPE_CHECKING, Final
 from litellm.repositories.table_repositories import PrismaTableRepository
 
 if TYPE_CHECKING:
-    from prisma import models as prisma_models  # noqa: F401  # used by the quoted base-class subscript
+    from prisma import models as prisma_models
 
 
 class ManagedFileContentRepository(PrismaTableRepository["prisma_models.LiteLLM_ManagedFileContentTable"]):
@@ -18,7 +18,9 @@ class ManagedFileContentRepository(PrismaTableRepository["prisma_models.LiteLLM_
         return row.id
 
     async def load(self, row_id: str) -> bytes | None:
-        row: Final = await self.table.find_unique(where={"id": row_id})  # mutable-ok: prisma filters are plain dicts
+        row: Final[prisma_models.LiteLLM_ManagedFileContentTable | None] = await self.table.find_unique(
+            where={"id": row_id}  # mutable-ok: prisma filters are plain dicts
+        )
         return None if row is None else row.content.decode()
 
     async def delete(self, row_id: str) -> None: