432ff7577d
The cross_encoder for Gemini already supported passing in a custom client.
I replicated the same input pattern to embedder and llm_client.
The value is, you can support custom API endpoints and other options like below:
cross_encoder=GeminiRerankerClient(
client=genai.Client(
api_key=os.environ.get('GOOGLE_GENAI_API_KEY'),
http_options=types.HttpOptions(api_version='v1alpha')),
config=LLMConfig(
model="gemini-2.5-flash-lite-preview-06-17"
)
))
|
||
|---|---|---|
| .. | ||
| __init__.py | ||
| azure_openai.py | ||
| client.py | ||
| gemini.py | ||
| openai.py | ||
| voyage.py | ||