version: '3.8'

services:
  openai-agent:
    build: .
    environment:
      API_BASE_URL: https://api.openai.com/v1
      MODEL_NAME: gpt-4
      API_KEY: ${OPENAI_API_KEY}
      TEMPERATURE: 0.7
      MAX_TOKENS: 2000
    command: ["--task-id", "bug_detection_easy_1"]

  gemini-agent:
    build: .
    environment:
      API_BASE_URL: https://generativelanguage.googleapis.com/v1beta/openai
      MODEL_NAME: gemini-2.0-flash
      API_KEY: ${GEMINI_API_KEY}
    command: ["--task-id", "memory_leak_medium_1"]

  local-agent:
    build: .
    network_mode: host
    environment:
      API_BASE_URL: http://localhost:11434/v1
      MODEL_NAME: llama2
      API_KEY: not-needed
    command: ["--task-id", "security_hard_1"]

  groq-agent:
    build: .
    environment:
      API_BASE_URL: https://api.groq.com/openai/v1
      MODEL_NAME: llama2-70b-4096
      API_KEY: ${GROQ_API_KEY}
    command: ["--task-id", "bug_detection_easy_1"]