phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

basic cohere rerank logging

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-06 15:18:16 -07:00
parent 4626c5a365
commit 1852e1cd9a
3 changed files with 32 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/rerank_api/main.py
View file

@ -9,7 +9,7 @@ from litellm.llms.cohere.rerank import CohereRerank
from litellm.llms.togetherai.rerank import TogetherAIRerank
from litellm.secret_managers.main import get_secret
from litellm.types.router import *
from litellm.utils import supports_httpx_timeout
from litellm.utils import client, supports_httpx_timeout
from .types import RerankRequest, RerankResponse
@ -20,6 +20,7 @@ together_rerank = TogetherAIRerank()
#################################################
@client
async def arerank(
model: str,
query: str,
@ -64,6 +65,7 @@ async def arerank(
raise e
@client
def rerank(
model: str,
query: str,