version: '3.8'

services:
  brunix-engine:
    build:
      context: ..
      dockerfile: Docker/Dockerfile
    container_name: brunix-assistance-engine
    ports:
      - "50052:50051"
      - "8000:8000"
    volumes:
      - ./data:/data

    environment:
      ELASTICSEARCH_URL: ${ELASTICSEARCH_URL}
      ELASTICSEARCH_INDEX: ${ELASTICSEARCH_INDEX}
      POSTGRES_URL: ${POSTGRES_URL}
      LANGFUSE_HOST: ${LANGFUSE_HOST}
      LANGFUSE_PUBLIC_KEY: ${LANGFUSE_PUBLIC_KEY}
      LANGFUSE_SECRET_KEY: ${LANGFUSE_SECRET_KEY}
      OLLAMA_URL: ${OLLAMA_URL}
      OLLAMA_MODEL_NAME: ${OLLAMA_MODEL_NAME}
      OLLAMA_MODEL_NAME_CONVERSATIONAL: ${OLLAMA_MODEL_NAME_CONVERSATIONAL}
      OLLAMA_EMB_MODEL_NAME: ${OLLAMA_EMB_MODEL_NAME}
      CLASSIFIER_EXPORT_THRESHOLD: ${CLASSIFIER_EXPORT_THRESHOLD}
      CLASSIFIER_EXPORT_DIR: ${CLASSIFIER_EXPORT_DIR}
      CLASSIFIER_MODEL_PATH: ${CLASSIFIER_MODEL_PATH}
      CLASSIFIER_CONFIDENCE_THRESHOLD: ${CLASSIFIER_CONFIDENCE_THRESHOLD}
      RETRAIN_ON_EXPORT: ${RETRAIN_ON_EXPORT}
      RETRAIN_SCRIPT_PATH: ${RETRAIN_SCRIPT_PATH}
      CLASSIFIER_ARCHIVE_DIR: ${CLASSIFIER_ARCHIVE_DIR}
      CLASSIFIER_SEED_DATASET: ${CLASSIFIER_SEED_DATASET}
      CLASSIFIER_MIN_CV_ACCURACY: ${CLASSIFIER_MIN_CV_ACCURACY}
      CLASSIFIER_HELD_OUT_RATIO: ${CLASSIFIER_HELD_OUT_RATIO}
      PROXY_THREAD_WORKERS: 10

    extra_hosts:
      - "host.docker.internal:host-gateway"