From 79ddfc8d1bca266e2b39a1d7ee6ba546cfa6a938 Mon Sep 17 00:00:00 2001
From: Devin Conathan <dconathan@gmail.com>
Date: Fri, 15 Aug 2025 15:29:05 -0400
Subject: [PATCH] fix: correct BaseRanker.rank_async annotations to match sync
 rank method

---
 rerankers/models/ranker.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rerankers/models/ranker.py b/rerankers/models/ranker.py
index 6c8f45a..bf7f326 100644
--- a/rerankers/models/ranker.py
+++ b/rerankers/models/ranker.py
@@ -30,8 +30,8 @@ def rank(
     async def rank_async(
         self,
         query: str,
-        docs: List[str],
-        doc_ids: Optional[Union[List[str], str]] = None,
+        docs: Union[str, List[str], Document, List[Document]],
+        doc_ids: Optional[Union[List[str], List[int]]] = None,
     ) -> RankedResults: