File size: 2,704 Bytes
8d4d62e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
# Example Enhanced GAIA Agent Configuration with Vertex AI Parameters
# Copy this file to config.yaml and update with your specific values
model:
# Choose your preferred model
model_id: vertex_ai/gemini-2.5-pro # Vertex AI Gemini model
#model_id: vertex_ai/gemini-1.5-pro # Alternative Gemini version
#model_id: anthropic/claude-sonnet-4 # Alternative: Anthropic Claude
#model_id: openai/gpt-4 # Alternative: OpenAI GPT-4
temperature: 0.2
max_tokens: 8096
# Vertex AI specific configuration (REQUIRED for vertex_ai models)
vertex_ai:
# REQUIRED: Replace with your actual GCP project ID
vertex_project: "your-gcp-project-id"
# REQUIRED: Choose your preferred region
vertex_location: "us-central1"
# Other popular regions: "us-east1", "europe-west1", "asia-southeast1"
# Authentication: Choose ONE of the following options:
# Option 1: Service account file (recommended for local development)
vertex_credentials: "/path/to/your-service-account.json"
# Option 2: Environment variables (comment out vertex_credentials above)
# Set these in your shell:
# export GOOGLE_APPLICATION_CREDENTIALS="/path/to/service-account.json"
# export VERTEXAI_PROJECT="your-gcp-project-id"
# export VERTEXAI_LOCATION="us-central1"
# Option 3: GCP SDK authentication (comment out vertex_credentials above)
# Run: gcloud auth application-default login
# Safety settings for content filtering
# Adjust thresholds based on your use case:
# BLOCK_NONE, BLOCK_LOW_AND_ABOVE, BLOCK_MEDIUM_AND_ABOVE, BLOCK_ONLY_HIGH
safety_settings:
- category: "HARM_CATEGORY_HARASSMENT"
threshold: "BLOCK_MEDIUM_AND_ABOVE"
- category: "HARM_CATEGORY_HATE_SPEECH"
threshold: "BLOCK_MEDIUM_AND_ABOVE"
- category: "HARM_CATEGORY_SEXUALLY_EXPLICIT"
threshold: "BLOCK_MEDIUM_AND_ABOVE"
- category: "HARM_CATEGORY_DANGEROUS_CONTENT"
threshold: "BLOCK_MEDIUM_AND_ABOVE"
# Optional: Enable grounding with Google Search (experimental)
# This adds real-time web search capabilities to responses
enable_grounding: false
# Agent configuration
agent:
name: GAIA-Agent
description: Agent using LiteLLM with enhanced Vertex AI features
max_steps: 5
verbosity_level: 1
# GAIA dataset settings
gaia:
local_path: "./GAIA"
# Cache settings
cache:
enabled: true
directory: cache
# Setup Instructions:
# 1. Copy this file to config.yaml
# 2. Replace "your-gcp-project-id" with your actual GCP project ID
# 3. Choose and configure one authentication method
# 4. Adjust safety settings as needed
# 5. Set vertex_location to your preferred region
# 6. Test with: python agent.py or python app.py |