| # 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 |