# Copyright (C) 2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0

services:
  agent:
    image: ${agent_image}
    container_name: test-comps-agent-endpoint
    volumes:
      - ${TOOLSET_PATH}:/home/user/tools/ # tools
      - ${WORKDIR}/GenAIComps/comps:/home/user/comps # code
      - ${WORKDIR}/TAG-Bench/:/home/user/TAG-Bench # SQL database and hints_file
    ports:
      - "9095:9095"
    ipc: host
    environment:
      ip_address: ${ip_address}
      strategy: sql_agent
      db_name: ${db_name}
      db_path: ${db_path}
      use_hints: false #true
      hints_file: /home/user/TAG-Bench/${db_name}_hints.csv
      recursion_limit: ${recursion_limit}
      llm_engine: openai
      OPENAI_API_KEY: ${OPENAI_API_KEY}
      model: "gpt-4o-mini-2024-07-18"
      temperature: 0
      max_new_tokens: ${max_new_tokens}
      stream: false
      tools: /home/user/tools/sql_agent_tools.yaml # /home/user/tools/custom_tools.yaml #
      require_human_feedback: false
      no_proxy: ${no_proxy}
      http_proxy: ${http_proxy}
      https_proxy: ${https_proxy}
      port: 9095
      GOOGLE_CSE_ID: ${GOOGLE_CSE_ID} #delete
      GOOGLE_API_KEY: ${GOOGLE_API_KEY} # delete
