Enhance Docker setup and configuration

- Updated GitHub Actions workflow to start Colima with specific resource allocations (CPU, memory, disk).
- Modified Dockerfile to set memory configuration for PyTorch and added environment variables for memory management.
- Improved Docker image build process in DockerRepository by specifying CPU and memory limits, and naming the container for better identification.

Author: efraespada

.github/workflows/gbf_vector.yml

@@ -31,7 +31,8 @@ jobs:
         run: |
           brew install --cask docker
           brew install colima
-          brew services start colima
+          colima start --cpu 4 --memory 16 --disk 100
+          brew services restart colima
 
       - name: Set up DOCKER_HOST as output
         id: docker-env

docker/Dockerfile

@@ -3,10 +3,14 @@ FROM python:3.11-slim
 # Opcional: define la ruta de cache huggingface
 ENV HF_HOME=/cache/huggingface
 
+# Configuración de memoria para PyTorch
+ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512
+ENV MAX_MEMORY=8g
+
 # Instala dependencias necesarias
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
- && rm -rf /var/lib/apt/lists/*
+    && rm -rf /var/lib/apt/lists/*
 
 # Crea directorios
 WORKDIR /app

src/data/repository/docker_repository.ts

@@ -50,9 +50,12 @@ export class DockerRepository {
             // Build the image with explicit tagging
             const stream = await this.docker.buildImage({
                 context: this.dockerDir,
-                src: ['Dockerfile', 'requirements.txt', 'main.py']
+                src: ['Dockerfile', 'requirements.txt', 'main.py'],
             }, { 
                 t: 'fastapi-app:latest',
+                cpusetcpus: 4,
+                memory: 16384,
+                memswap: 16384,
                 dockerfile: 'Dockerfile',
                 buildargs: {},
                 nocache: true
@@ -106,7 +109,8 @@ export class DockerRepository {
                     PortBindings: {
                         '8000/tcp': [{ HostPort: '8000' }]
                     }
-                }
+                },
+                name: 'fastapi-app'
             });
 
             logDebugInfo('🐳 🟡 Starting container...');