diff --git a/agents/creativeAgent.yaml b/agents/creativeAgent.yaml
index a39575f..bc1260e 100644
--- a/agents/creativeAgent.yaml
+++ b/agents/creativeAgent.yaml
@@ -19,7 +19,7 @@ tools:
 
 provider:
   type: gemini
-  model: gemini-3.1-flash-lite-preview
+  model: gemini-2.5-flash
   settings:
     thinking_budget: 0
 
diff --git a/agents/exampleAgent.yaml b/agents/exampleAgent.yaml
index 8a597e8..7eebaa6 100644
--- a/agents/exampleAgent.yaml
+++ b/agents/exampleAgent.yaml
@@ -16,7 +16,7 @@ tools:
 
 provider:
   type: gemini
-  model: gemini-3.1-flash-lite-preview
+  model: gemini-2.5-flash
   settings:
     thinking_budget: 0
 
diff --git a/agents/whatsapp_agent.yaml b/agents/whatsapp_agent.yaml
index 5084722..00711db 100644
--- a/agents/whatsapp_agent.yaml
+++ b/agents/whatsapp_agent.yaml
@@ -24,6 +24,6 @@ system_prompt: |
 
 provider:
   type: gemini
-  model: gemini-2.0-flash
+  model: gemini-2.5-flash
   settings:
     thinking_budget: 0
diff --git a/requirements-api.txt b/requirements-api.txt
index 89eb63f..f1a0c61 100644
--- a/requirements-api.txt
+++ b/requirements-api.txt
@@ -6,7 +6,7 @@ fastapi>=0.115.0
 uvicorn[standard]>=0.30.0
 pydantic>=2.8.0
 pydantic-settings>=2.3.0
-pydantic-ai==1.98.0
+pydantic-ai-slim[google,vertexai]==1.98.0
 google-genai>=1.0.0
 google-cloud-firestore>=2.0.0
 numpy>=1.26.0