scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

fix install on 3.8

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-27 17:09:16 -07:00
parent b3892b871d
commit 37ed201c50
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/cohere/rerank.py
View file

@ -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,

2
litellm/llms/togetherai/rerank.py
View file

@ -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,