forked from phoenix/litellm-mirror
23 lines
559 B
Python
23 lines
559 B
Python
import json
|
|
import os
|
|
import sys
|
|
from datetime import datetime
|
|
from unittest.mock import AsyncMock
|
|
|
|
sys.path.insert(
|
|
0, os.path.abspath("../..")
|
|
) # Adds the parent directory to the system path
|
|
|
|
|
|
from base_rerank_unit_tests import BaseLLMRerankTest
|
|
import litellm
|
|
|
|
|
|
class TestJinaAI(BaseLLMRerankTest):
|
|
def get_custom_llm_provider(self) -> litellm.LlmProviders:
|
|
return litellm.LlmProviders.JINA_AI
|
|
|
|
def get_base_rerank_call_args(self) -> dict:
|
|
return {
|
|
"model": "jina_ai/jina-reranker-v2-base-multilingual",
|
|
}
|