# 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