feat: remove blackwell dockerfiles

backends/advanced/Dockerfile

@@ -19,16 +19,12 @@ WORKDIR /app
 
 # Copy package structure and dependency files first
 COPY pyproject.toml README.md ./
-COPY uv.lock .
 RUN mkdir -p src/advanced_omi_backend
 COPY src/advanced_omi_backend/__init__.py src/advanced_omi_backend/
 
-# Install dependencies using uv with deepgram extra
-# Use cache mount for BuildKit, fallback for legacy builds
-# RUN --mount=type=cache,target=/root/.cache/uv \
-#     uv sync --extra deepgram
+# Install dependencies using uv pip with deepgram extra
 # Fallback for legacy Docker builds (CI compatibility)
-RUN uv sync --extra deepgram
+RUN uv pip install --system ".[deepgram]" -f https://download.pytorch.org/whl/cpu
 
 # Copy all application code
 COPY . .
@@ -38,6 +34,7 @@ COPY . .
 COPY memory_config.yaml* ./
 COPY diarization_config.json* ./
 
+ENV PATH="/app/.venv/bin:$PATH"
 
 # Copy and make startup scripts executable
 COPY start.sh ./

backends/advanced/pyproject.toml

@@ -1,3 +1,7 @@
+[build-system]
+requires = ["setuptools>=69", "wheel"]
+build-backend = "setuptools.build_meta"
+
 [project]
 name = "advanced-omi-backend"
 version = "0.1.0"
@@ -7,7 +11,7 @@ requires-python = ">=3.12"
 dependencies = [
     "easy-audio-interfaces>=0.7.1", # we need to add local-audio for scripts/local-audio.py | If we don't need that, we can remove this, and then remove portaudio19-dev from Dockerfile
     "fastapi>=0.115.12",
-    "mem0ai", # Using main branch with PR #3250 AsyncMemory fix
+    "mem0ai @ git+https://github.com/AnkushMalaker/mem0.git@main", # Using main branch with PR #3250 AsyncMemory fix
     "langchain_neo4j",
     "motor>=3.7.1",
     "ollama>=0.4.8",
@@ -35,9 +39,26 @@ local-audio = [
     "easy-audio-interfaces[local-audio]>=0.7.1",
 ]
 
-[build-system]
-requires = ["setuptools>=61.0", "wheel"]
-build-backend = "setuptools.build_meta"
+# ---- Optional compute backends (choose ONE when installing) ----
+# Note: with plain pip, point to the PyTorch wheel index for the chosen extra (examples below).
+cpu   = ["torch", "torchaudio"]
+cu118 = ["torch", "torchaudio"]
+cu121 = ["torch", "torchaudio"]
+cu124 = ["torch", "torchaudio"]
+cu126 = ["torch", "torchaudio"]
+cu128 = ["torch", "torchaudio"]  # CUDA 12.8 / Blackwell
+
+# ---- Dev/test extras (moved from `dependency-groups` to standard extras) ----
+dev = [
+    "black>=25.1.0",
+    "isort>=6.0.1",
+    "pre-commit>=4.2.0",
+    "pre-commit-uv>=4.1.4",
+]
+test = [
+    "pytest>=8.4.1",
+    "pytest-asyncio>=1.0.0",
+]
 
 [tool.setuptools.packages.find]
 where = ["src"]
@@ -48,9 +69,6 @@ profile = "black"
 [tool.black]
 line-length = 100
 
-[tool.uv.sources]
-mem0ai = { git = "https://github.com/AnkushMalaker/mem0.git", rev = "async-client-unbound-var-fix" }
-
 [tool.poetry.dependencies]
 robotframework = "^6.1.1"
 
@@ -89,22 +107,3 @@ filterwarnings = [
     "ignore::DeprecationWarning",
     "ignore::PendingDeprecationWarning",
 ]
-
-[dependency-groups]
-dev = [
-    "black>=25.1.0",
-    "isort>=6.0.1",
-    "pre-commit>=4.2.0",
-    "pre-commit-uv>=4.1.4",
-]
-test = [
-    "pytest>=8.4.1",
-    "pytest-asyncio>=1.0.0",
-    "pytest-spec>=3.2.0",
-    "pytest-cov>=6.0.0",
-    "pytest-xdist>=3.6.0",
-    "pytest-mock>=3.14.0",
-    "requests-mock>=1.12.1",
-    "pytest-json-report>=1.5.0",
-    "pytest-html>=4.0.0",
-]

extras/asr-services/Dockerfile_Parakeet

@@ -12,8 +12,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 # Dependency manifest first for cache‑friendly installs
 COPY pyproject.toml uv.lock ./
+ARG INSTALL_EXTRAS
+
 RUN --mount=type=cache,target=/root/.cache/uv \
-    uv sync --no-install-project --group parakeet
+    uv venv && \
+    uv pip install -e "${INSTALL_EXTRAS}" --group demo --group dev --group parakeet
+
 
 # Should prepare the .venv for use :)
 
@@ -22,7 +26,7 @@ FROM python:3.12-slim-bookworm AS runtime
 ENV PYTHONUNBUFFERED=1
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y --no-install-recommends libsndfile1 build-essential portaudio19-dev curl && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y --no-install-recommends libsndfile1 build-essential ffmpeg portaudio19-dev curl && rm -rf /var/lib/apt/lists/*
 
 COPY --from=builder /app /app
 COPY . .

extras/asr-services/docker-compose.yml

@@ -3,6 +3,8 @@ services:
     build:
       context: .
       dockerfile: Dockerfile_Parakeet
+      args:
+        - INSTALL_EXTRAS=${INSTALL_EXTRAS:-.[cu121]}
     image: parakeet-asr:latest
     ports:
       - "${PARAKEET_HOST_PORT:-8767}:${PARAKEET_CONTAINER_PORT:-8765}"

extras/asr-services/env.template

@@ -0,0 +1,6 @@
+# ASR Services Environment Configuration Template
+# Copy this file to .env and configure the values for your environment
+
+# Install extras for PyTorch (CUDA version)
+# Options: .[cu118], .[cu121], .[cu128], .[cpu], or just . for CPU-only
+# INSTALL_EXTRAS=.[cu121]