From 37ed201c5077fb9b69fef47e0ff7bd410e47c208 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 27 Aug 2024 17:09:16 -0700 Subject: [PATCH] fix install on 3.8 --- litellm/llms/cohere/rerank.py | 2 +- litellm/llms/togetherai/rerank.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/llms/cohere/rerank.py b/litellm/llms/cohere/rerank.py index 4ef523e3a1..a2a7476df8 100644 --- a/litellm/llms/cohere/rerank.py +++ b/litellm/llms/cohere/rerank.py @@ -23,7 +23,7 @@ class CohereRerank(BaseLLM): model: str, api_key: str, query: str, - documents: list[Union[str, Dict[str, Any]]], + documents: List[Union[str, Dict[str, Any]]], top_n: Optional[int] = None, rank_fields: Optional[List[str]] = None, return_documents: Optional[bool] = True, diff --git a/litellm/llms/togetherai/rerank.py b/litellm/llms/togetherai/rerank.py index 32a8cdcfdc..5d905071c3 100644 --- a/litellm/llms/togetherai/rerank.py +++ b/litellm/llms/togetherai/rerank.py @@ -23,7 +23,7 @@ class TogetherAIRerank(BaseLLM): model: str, api_key: str, query: str, - documents: list[Union[str, Dict[str, Any]]], + documents: List[Union[str, Dict[str, Any]]], top_n: Optional[int] = None, rank_fields: Optional[List[str]] = None, return_documents: Optional[bool] = True,