From f6ef0a24e03efd0ddd3c42c3ee5a3c4a8f1e025d Mon Sep 17 00:00:00 2001 From: kerry Date: Tue, 15 Sep 2026 22:28:23 +0000 Subject: [PATCH] fix(tests): restore rerank response import Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/test_litellm/test_cost_calculator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_litellm/test_cost_calculator.py b/tests/test_litellm/test_cost_calculator.py index 7fedb43ab6e..aa8a1d34bc1 100644 --- a/tests/test_litellm/test_cost_calculator.py +++ b/tests/test_litellm/test_cost_calculator.py @@ -16,6 +16,7 @@ from litellm.litellm_core_utils.litellm_logging import Logging from litellm.llms.base_llm.ocr.transformation import OCRPage, OCRResponse, OCRUsageInfo from litellm.types.llms.base import CachedTokensDetails from litellm.types.llms.openai import OpenAIRealtimeStreamList +from litellm.types.rerank import RerankResponse from litellm.types.utils import ( ModelInfo, ModelResponse,