feat: add base_url option to Gemini LLM class

PiperOrigin-RevId: 866146209
This commit is contained in:
Matthew Chan
2026-02-05 15:43:34 -08:00
committed by Copybara-Service
parent d9e8e9cf32
commit 781f605a1e
+4
View File
@@ -91,6 +91,9 @@ class Gemini(BaseLlm):
model: str = 'gemini-2.5-flash'
base_url: Optional[str] = None
"""The base URL for the AI platform service endpoint."""
speech_config: Optional[types.SpeechConfig] = None
use_interactions_api: bool = False
@@ -305,6 +308,7 @@ class Gemini(BaseLlm):
http_options=types.HttpOptions(
headers=self._tracking_headers(),
retry_options=self.retry_options,
base_url=self.base_url,
)
)