Files
GenAIExamples/tests/agent/sql_agent_openai.yaml
2025-01-06 13:25:47 +08:00

37 lines
1.2 KiB
YAML

# 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