From 3426b905cedd66d59506174efb04d8c53566b53b Mon Sep 17 00:00:00 2001
From: Cesar Garcia <128240629+Chesars@users.noreply.github.com>
Date: Tue, 3 Mar 2026 10:44:47 -0300
Subject: [PATCH] Update
 tests/litellm/proxy/test_batch_x_litellm_model_encoding.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
---
 tests/litellm/proxy/test_batch_x_litellm_model_encoding.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/litellm/proxy/test_batch_x_litellm_model_encoding.py b/tests/litellm/proxy/test_batch_x_litellm_model_encoding.py
index 01d6a8aaacd..062443d9425 100644
--- a/tests/litellm/proxy/test_batch_x_litellm_model_encoding.py
+++ b/tests/litellm/proxy/test_batch_x_litellm_model_encoding.py
@@ -33,8 +33,8 @@ def _make_mock_request(headers: dict) -> MagicMock:
 def _make_batch_response(
     batch_id: str = "batch_abc123",
     input_file_id: str = "file-input456",
-    output_file_id: str = None,
-    error_file_id: str = None,
+    output_file_id: Optional[str] = None,
+    error_file_id: Optional[str] = None,
     status: str = "validating",
 ) -> LiteLLMBatch:
     """Create a mock LiteLLMBatch response from a provider."""