decodingtrust-agent-sdk 0.1.0__py3-none-any.whl

dt_arena/config/env.yaml

@@ -0,0 +1,515 @@
+# Environment (Docker Compose) configurations
+# Defines docker environments, their port mappings, and instance limits
+
+# Default max instances for environments not explicitly listed
+default_max_instances: null  # null means unlimited
+
+environments:
+  customer_service:
+    docker_compose: "dt_arena/envs/customer_service/docker-compose.yml"
+    max_instances: 4
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${CUSTOMER_SERVICE_RESET_PORT}/reset"
+        method: POST
+    ports:
+      CUSTOMER_SERVICE_DB_PORT:
+        default: 5444
+        container_port: 5432
+      CUSTOMER_SERVICE_API_PORT:
+        default: 8080
+        container_port: 8080
+      CUSTOMER_SERVICE_RESET_PORT:
+        default: 8081
+        container_port: 8081
+
+  gmail:
+    docker_compose: "dt_arena/envs/gmail/docker-compose-hub.yml"
+    # max_instances: 5  # Uncomment to limit parallel gmail instances
+    reset_endpoints:
+      auth:
+        url: "http://127.0.0.1:${GMAIL_AUTH_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      GMAIL_SMTP_PORT:
+        default: 1025
+        container_port: 1025
+      GMAIL_AUTH_PORT:
+        default: 8030
+        container_port: 8030
+      GMAIL_PROXY_PORT:
+        default: 8031
+        container_port: 8031
+      GMAIL_UI_PORT:
+        default: 8025
+        container_port: 8025
+      GMAIL_FRONTEND_PORT:
+        default: 8050
+        container_port: 80
+
+  slack:
+    docker_compose: "dt_arena/envs/slack/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${SLACK_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      SLACK_API_PORT:
+        default: 8034
+        container_port: 8034
+      SLACK_UI_PORT:
+        default: 8028
+        container_port: 80
+
+  paypal:
+    docker_compose: "dt_arena/envs/paypal/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${PAYPAL_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      PAYPAL_PG_PORT:
+        default: 5454
+        container_port: 5454
+      PAYPAL_API_PORT:
+        default: 8035
+        container_port: 8035
+      PAYPAL_UI_PORT:
+        default: 8029
+        container_port: 80
+
+  calendar:
+    docker_compose: "dt_arena/envs/calendar/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${CALENDAR_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      CALENDAR_API_PORT:
+        default: 8032
+        container_port: 8032
+      CALENDAR_UI_PORT:
+        default: 8026
+        container_port: 80
+
+  zoom:
+    docker_compose: "dt_arena/envs/zoom/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${ZOOM_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      ZOOM_API_PORT:
+        default: 8033
+        container_port: 8033
+      ZOOM_UI_PORT:
+        default: 8027
+        container_port: 80
+
+  google-form:
+    docker_compose: "dt_arena/envs/google-form/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${GOOGLE_FORM_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      GOOGLE_FORM_API_PORT:
+        default: 8054
+        container_port: 8054
+      GOOGLE_FORM_UI_PORT:
+        default: 8055
+        container_port: 80
+
+  googlesheets:
+    docker_compose: "dt_arena/envs/googlesheets/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${GOOGLESHEETS_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      GOOGLESHEETS_PG_PORT:
+        default: 5476
+        container_port: 5432
+      GOOGLESHEETS_API_PORT:
+        default: 8056
+        container_port: 8056
+      GOOGLESHEETS_FRONTEND_PORT:
+        default: 8057
+        container_port: 80
+
+  googledrive:
+    docker_compose: "dt_arena/envs/googledrive/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${GOOGLEDRIVE_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      GOOGLEDRIVE_PG_PORT:
+        default: 5477
+        container_port: 5432
+      GOOGLEDRIVE_API_PORT:
+        default: 8058
+        container_port: 8058
+      GOOGLEDRIVE_FRONTEND_PORT:
+        default: 8059
+        container_port: 80
+
+  snowflake:
+    docker_compose: "dt_arena/envs/snowflake/docker-compose-hub.yml"
+    reset_endpoints:
+      admin:
+        url: "http://127.0.0.1:${SNOWFLAKE_ADMIN_PORT}/admin/reset"
+        method: POST
+    ports:
+      SNOWFLAKE_PG_PORT:
+        default: 5452
+        container_port: 5432
+      SNOWFLAKE_ADMIN_PORT:
+        default: 8454
+        container_port: 8454
+
+  databricks:
+    docker_compose: "dt_arena/envs/databricks/docker-compose-hub.yml"
+    reset_endpoints:
+      admin:
+        url: "http://127.0.0.1:${DATABRICKS_ADMIN_PORT}/admin/reset"
+        method: POST
+    ports:
+      DATABRICKS_PG_PORT:
+        default: 5453
+        container_port: 5432
+      DATABRICKS_ADMIN_PORT:
+        default: 8453
+        container_port: 8453
+
+  ers:
+    docker_compose: "dt_arena/envs/ers/docker-compose.yml"
+    ports:
+      ERS_API_PORT:
+        default: 8036
+        container_port: 8036
+  
+  ecommerce:
+    docker_compose: "dt_arena/envs/ecommerce/docker-compose.yml"
+    disable_reuse: true
+    ports:
+      ECOMMERCE_UI_PORT:
+        default: 7770
+        container_port: 80
+  
+  custom-website:
+    docker_compose: "dt_arena/envs/custom-website/docker-compose.yml"
+    disable_reuse: true
+    ports:
+      CUSTOM_WEBSITE_UI_PORT:
+        default: 7771
+        container_port: 8000
+
+  os-filesystem:
+    docker_compose: "dt_arena/envs/os-filesystem/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${OS_FILESYSTEM_API_PORT}/api/v1/reset"
+        method: POST
+    reset_scripts:
+      os-filesystem: "/scripts/reset.sh"
+    ports:
+      OS_FILESYSTEM_API_PORT:
+        default: 8034
+        container_port: 8034
+
+  atlassian:
+    docker_compose: "dt_arena/envs/atlassian/docker-compose-hub.yml"
+    ports:
+      ATLASSIAN_API_PORT:
+        default: 8040
+        container_port: 8040
+      ATLASSIAN_UI_PORT:
+        default: 8041
+        container_port: 80
+      ATLASSIAN_DB_PORT:
+        default: 5454
+        container_port: 5454
+
+  salesforce:
+    docker_compose: "dt_arena/envs/salesforce_crm/docker-compose-hub.yaml"
+    # Reset endpoints (API-based reset, preferred over scripts)
+    # URL supports ${VAR_NAME} placeholders resolved from ports config
+    reset_endpoints:
+      database:
+        url: "http://127.0.0.1:${SALESFORCE_API_PORT}/Api/V8/admin/reset-database"
+        method: POST
+    # Reset scripts per service (fallback if endpoints fail)
+    reset_scripts:
+      mariadb: "/scripts/reset.sh"
+    ports:
+      SALESFORCE_API_PORT:
+        default: 8080
+        container_port: 8080
+      SALESFORCE_DB_PORT:
+        default: 3306
+        container_port: 3306
+
+  travel:
+    docker_compose: "dt_arena/envs/travel/docker-compose.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${TRAVEL_PORT}/reset"
+        method: POST
+    # Fallback script if endpoint fails
+    reset_scripts:
+      travel-api: "/app/reset.sh"
+    ports:
+      TRAVEL_PORT:
+        default: 10300
+        container_port: 10300
+
+  telecom:
+    docker_compose: "dt_arena/envs/telecom/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${TELECOM_PORT}/reset_all"
+        method: POST
+    ports:
+      TELECOM_PORT:
+        default: 17501
+        container_port: 17501
+
+  windows:
+    docker_compose: "dt_arena/envs/windows/docker-compose.yml"
+    max_instances: 10  # Windows VM is resource-heavy, limit instances
+    health_timeout: 300  # Windows VM needs ~3-5 minutes to boot
+    reset_scripts:
+      windows-mcp: "echo 'loadvm booted' | nc -w60 localhost 7100 > /dev/null 2>&1; sleep 5"
+    reset_endpoints:
+      listeners:
+        url: "http://127.0.0.1:${EXFIL_PORT}/reset-all"
+        method: DELETE
+    reset_retries: 2
+    reset_retry_delay: 15
+    reset_script_timeout: 120
+    ports:
+      WEB_MANAGEMENT_PORT:
+        default: 8006
+        container_port: 8006
+      MCP_SERVICE_PORT:
+        default: 8005
+        container_port: 8005
+      OFFICE_SERVICE_PORT:
+        default: 8007
+        container_port: 8007
+      RDP_PORT:
+        default: 3390
+        container_port: 3389
+      RESET_PORT:
+        default: 7101
+        container_port: 7101
+      EXFIL_PORT:
+        default: 9999
+        container_port: 9999
+      FTP_API_PORT:
+        default: 9998
+        container_port: 8080
+      DNS_API_PORT:
+        default: 9997
+        container_port: 8080
+
+  hospital:
+    docker_compose: "dt_arena/envs/hospital/docker-compose-hub.yml"
+    ports:
+      HOSPITAL_PORT:
+        default: 12001
+        container_port: 12001
+      HOSPITAL_MCP_PORT:
+        default: 12002
+        container_port: 12002
+      HOSPITAL_ENV_INJECTION_MCP_PORT:
+        default: 12005
+        container_port: 12005
+
+  finance:
+    docker_compose: "dt_arena/envs/finance/docker-compose.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${FINANCE_WEB_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      FINANCE_WEB_PORT:
+        default: 5000
+        container_port: 5000
+
+  telegram:
+    docker_compose: "dt_arena/envs/telegram/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${TELEGRAM_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      TELEGRAM_PG_PORT:
+        default: 5472
+        container_port: 5432
+      TELEGRAM_API_PORT:
+        default: 8038
+        container_port: 8038
+
+  legal:
+    docker_compose: "dt_arena/envs/legal/docker-compose.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${LEGAL_WEB_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      LEGAL_WEB_PORT:
+        default: 8003
+        container_port: 8003
+
+  macos:
+    docker_compose: "dt_arena/envs/macos/docker-compose.yml"
+    vm_profile: "dt_arena/envs/macos/vm.yaml"  # ADL backend uses QemuVM
+    max_instances: 1
+    health_timeout: 300  # macOS VM needs ~3-5 minutes for first boot; loadvm ~30s
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${MCP_SERVICE_PORT}/reset"
+        method: POST
+      listeners:
+        url: "http://127.0.0.1:${EXFIL_PORT}/reset-all"
+        method: DELETE
+    reset_retries: 2
+    reset_retry_delay: 15
+    ports:
+      WEB_MANAGEMENT_PORT:
+        default: 8046
+        container_port: 8006
+      MCP_SERVICE_PORT:
+        default: 8044
+        container_port: 8005
+      EXFIL_PORT:
+        default: 9899
+        container_port: 9999
+      FTP_API_PORT:
+        default: 9898
+        container_port: 8080
+      DNS_API_PORT:
+        default: 9897
+        container_port: 8080
+
+  github:
+    docker_compose: "dt_arena/envs/github/docker/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${GITHUB_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      GITHUB_API_PORT:
+        default: 8045
+        container_port: 8083
+      GITHUB_UI_PORT:
+        default: 8046
+        container_port: 8045
+      GITHUB_PG_PORT:
+        default: 5455
+        container_port: 5432
+
+  gitlab:
+    docker_compose: "dt_arena/envs/gitlab/docker-compose-hub.yml"
+    disable_reuse: true
+    ports:
+      GITLAB_UI_PORT:
+        default: 8023
+        container_port: 80
+      GITLAB_SSH_PORT:
+        default: 2223
+        container_port: 22
+
+  customer_service:
+    docker_compose: "dt_arena/envs/customer_service/docker-compose.yml"
+    max_instances: 10
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${CUSTOMER_SERVICE_RESET_PORT}/reset"
+        method: POST
+    ports:
+      CUSTOMER_SERVICE_DB_PORT:
+        default: 5444
+        container_port: 5432
+      CUSTOMER_SERVICE_API_PORT:
+        default: 8080
+        container_port: 8080
+      CUSTOMER_SERVICE_RESET_PORT:
+        default: 8081
+        container_port: 8081
+
+  terminal:
+    docker_compose: "dt_arena/envs/terminal/docker-compose-hub.yml"
+    max_instances: 10
+    reset_script_timeout: 180
+    reset_scripts:
+      terminal-env: "/scripts/reset.sh"
+
+  research:
+    docker_compose: "dt_arena/envs/research/docker-compose-hub.yml"
+    reset_scripts:
+      research-env: "/reset.sh"
+
+  whatsapp:
+    docker_compose: "dt_arena/envs/whatsapp/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${WHATSAPP_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      WHATSAPP_PG_PORT:
+        default: 5473
+        container_port: 5473
+      WHATSAPP_API_PORT:
+        default: 8039
+        container_port: 8039
+      WHATSAPP_FRONTEND_PORT:
+        default: 8040
+        container_port: 80
+
+  googledocs:
+    docker_compose: "dt_arena/envs/googledocs/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${GOOGLEDOCS_API_PORT}/api/v1/reset"
+        method: POST
+    ports:
+      GOOGLEDOCS_PG_PORT:
+        default: 5474
+        container_port: 5474
+      GOOGLEDOCS_API_PORT:
+        default: 8041
+        container_port: 8041
+      GOOGLEDOCS_FRONTEND_PORT:
+        default: 8042
+        container_port: 80
+
+  arxiv:
+    docker_compose: "dt_arena/envs/arxiv/docker-compose-hub.yml"
+    reset_endpoints:
+      api:
+        url: "http://127.0.0.1:${ARXIV_API_PORT}/api/v1/reset"
+        method: POST
+    reset_scripts:
+      arxiv-api: "/app/reset.sh"
+    ports:
+      ARXIV_API_PORT:
+        default: 8034
+        container_port: 8033
+      ARXIV_UI_PORT:
+        default: 8058
+        container_port: 80
+
+  bigquery:
+    docker_compose: "dt_arena/envs/bigquery/docker-compose.yml"
+    disable_reuse: true
+    health_timeout: 180
+    ports:
+      BIGQUERY_API_PORT:
+        default: 9050
+        container_port: 9050
+      BIGQUERY_GRPC_PORT:
+        default: 9060
+        container_port: 9060