Update GoogleVertexLLMService default model to gemini-2.5-flash

This commit is contained in:
Mark Backman
2025-11-05 11:28:41 -05:00
parent 80d127aaa4
commit c18ff999a5
2 changed files with 4 additions and 2 deletions

View File

@@ -33,6 +33,8 @@ reason")`.
### Changed
- Updated the default model for `GoogleVertexLLMService` to `gemini-2.5-flash`.
- Updated the `GoogleVertexLLMService` to use the `GoogleLLMService` as a base
class instead of the `OpenAILLMService`.

View File

@@ -99,7 +99,7 @@ class GoogleVertexLLMService(GoogleLLMService):
*,
credentials: Optional[str] = None,
credentials_path: Optional[str] = None,
model: str = "google/gemini-2.0-flash-001",
model: str = "gemini-2.5-flash",
location: Optional[str] = None,
project_id: Optional[str] = None,
params: Optional[GoogleLLMService.InputParams] = None,
@@ -114,7 +114,7 @@ class GoogleVertexLLMService(GoogleLLMService):
Args:
credentials: JSON string of service account credentials.
credentials_path: Path to the service account JSON file.
model: Model identifier (e.g., "google/gemini-2.0-flash-001").
model: Model identifier (e.g., "gemini-2.5-flash").
location: GCP region for Vertex AI endpoint (e.g., "us-east4").
project_id: Google Cloud project ID.
params: Input parameters for the model.