Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion coaching/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ dependencies = [

# Google Gen AI SDK (replaces deprecated google-cloud-aiplatform for generative AI)
# Migration: https://cloud.google.com/vertex-ai/generative-ai/docs/deprecations/genai-vertexai-sdk
"google-genai>=1.56.0",
"google-genai>=1.67.0",
]

[project.optional-dependencies]
Expand Down
2 changes: 1 addition & 1 deletion coaching/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ langchain-anthropic==0.3.9
langchain-aws==0.2.9
langchain-openai==0.2.14
langgraph==0.2.62
google-genai==1.56.0
google-genai==1.67.0

# Redis - Latest version
redis==6.4.0
Expand Down
12 changes: 6 additions & 6 deletions coaching/scripts/test_google_vertex.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,14 +42,14 @@ def check_prerequisites() -> bool:

print(f"\n2. GOOGLE_PROJECT_ID: {project_id or '(will use credentials file)'}")

# Check if google-cloud-aiplatform is installed
# Check if google-genai is installed
try:
import google.cloud.aiplatform # noqa: F401
import google.genai # noqa: F401

print("\n3. google-cloud-aiplatform: ✅ Installed")
print("\n3. google-genai: ✅ Installed")
except ImportError:
print("\n3. google-cloud-aiplatform: ❌ NOT installed")
print(" Run: uv pip install google-cloud-aiplatform>=1.40.0")
print("\n3. google-genai: ❌ NOT installed")
print(" Run: uv pip install google-genai>=1.67.0")
return False

if not creds_path and not project_id:
Expand All @@ -72,7 +72,7 @@ async def test_gemini_model() -> None:
print("=" * 60)

project_id = os.getenv("GOOGLE_PROJECT_ID")
location = os.getenv("GOOGLE_VERTEX_LOCATION", "us-central1")
location = os.getenv("GOOGLE_VERTEX_LOCATION", "global")

print("\nInitializing GoogleVertexLLMProvider...")
print(f" Project ID: {project_id or '(from credentials)'}")
Expand Down
6 changes: 3 additions & 3 deletions coaching/src/core/llm_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -373,12 +373,12 @@ class SupportedModel:
cost_per_1k_tokens=0.00025, # Avg of $0.10/1M input, $0.40/1M output
is_active=True,
),
# Gemini 3 Pro - Latest generation (Preview - December 2025)
# Gemini 3.1 Pro - Latest generation (Preview - February 2026)
"GEMINI_3_PRO": SupportedModel(
code="GEMINI_3_PRO",
provider=LLMProvider.GOOGLE_VERTEX,
model_name="gemini-3-pro-preview",
version="3.0",
model_name="gemini-3.1-pro-preview",
version="3.1",
provider_class="GoogleVertexLLMProvider",
capabilities=[
"chat",
Expand Down
6 changes: 3 additions & 3 deletions coaching/src/infrastructure/llm/google_vertex_provider.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ class GoogleVertexLLMProvider:
Google Vertex AI adapter implementing LLMProviderPort.

This adapter provides Google Vertex AI-backed LLM access using the new
google-genai SDK (v1.56.0+), implementing the provider port interface
google-genai SDK (v1.67.0+), implementing the provider port interface
defined in the domain layer.

Design:
Expand All @@ -72,7 +72,7 @@ class GoogleVertexLLMProvider:
# Supported Vertex AI model IDs
SUPPORTED_MODELS: ClassVar[list[str]] = [
# Gemini 3.x Series (latest - December 2025)
"gemini-3-pro-preview",
"gemini-3.1-pro-preview",
# Gemini 2.5 Series
"gemini-2.5-pro",
"gemini-2.5-flash",
Expand Down Expand Up @@ -141,7 +141,7 @@ async def _get_client(self) -> Any:
except ImportError as e:
raise ImportError(
"Google Gen AI SDK not installed. "
"Install with: pip install google-genai>=1.56.0"
"Install with: pip install google-genai>=1.67.0"
) from e

# Get project_id and credentials from Secrets Manager if not provided
Expand Down
Loading