mirror of
https://github.com/vegu-ai/talemate.git
synced 2025-12-16 03:37:51 +01:00
bb1cf6941b
* move memory agent to directory structure * chromadb settings rework * memory agent improvements embedding presets support switching embeddings without restart support custom sentence transformer embeddings * toggle to hide / show disabled clients * add memory debug tools * chromadb no longer needs its dedicated config entry * add missing emits * fix initial value * hidden disabled clients no longer cause enumeration issues with client actions * improve memory agent error handling and hot reloading * more memory agent error handling * DEBUG_MEMORY_REQUESTS off * relock * sim suite: fix issue with removing or changing characters * relock * fix issue where actor dialogue editor would break with multiple characters in the scene * remove cruft * implement interrupt function * margin adjustments * fix rubber banding issue in world editor when editing certain text fields * status notification when re-importing vectorb due to embeddings change * properly open new client context on agent actions * move jiggle apply to the end of prompt tune stack * narrator agent length limit and jiggle settings added - also improve post generation cleanup * progress story prompt improvements * narrator prompt and cleanup tweaks * prompt tweak * revert * autocomplete dialogue improvements * Unified process (#141) * progress to unified process * --dev arg * use gunicorn to serve built frontend * gunicorn config adjustments * remove dist from gitignore * revert * uvicorn instead * save decode * graceful shutdown * refactor unified process * clean up frontend log messages * more logging fixes * 0.27.0 * startup message * clean up scripts a bit * fixes to update.bat * fixes to install.bat * sim suite supports generation cancellation * debug * simplify narrator prompts * prompt tweaks * unified docker file * update docker compose config for unified docker file * cruft * fix startup in linux docker * download punkt so its available * prompt tweaks * fix bug when editing scene outline would wipe message history * add o1 models * add sampler, scheduler and cfg config to a1111 visualizer * update installation docs * visualizer configurable timeout * memory agent docs * docs * relock * relock * fix issue where changing embeddings on immutable scene would hang * remove debug message * take torch install out of poetry since conditionals don't work. * torch gets installed through some dependency so put it back into poetry, but reinstall with cuda if cuda support exists * fix install syntax * no need for torchvision * torch cuda install added to linux install script * add torch cuda install to update.bat * docs * docs * relock * fix install.sh * handle torch+cuda install in docker * docs * typo
86 lines
2.2 KiB
Docker
86 lines
2.2 KiB
Docker
# Stage 1: Frontend build
|
|
FROM node:21 AS frontend-build
|
|
|
|
ENV NODE_ENV=development
|
|
|
|
WORKDIR /app
|
|
|
|
# Copy the frontend directory contents into the container at /app
|
|
COPY ./talemate_frontend /app
|
|
|
|
# Install all dependencies and build
|
|
RUN npm install && npm run build
|
|
|
|
# Stage 2: Backend build
|
|
FROM python:3.11-slim AS backend-build
|
|
|
|
WORKDIR /app
|
|
|
|
# Install system dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
bash \
|
|
gcc \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install poetry
|
|
RUN pip install poetry
|
|
|
|
# Copy poetry files
|
|
COPY pyproject.toml poetry.lock* /app/
|
|
|
|
# Create a virtual environment
|
|
RUN python -m venv /app/talemate_env
|
|
|
|
# Activate virtual environment and install dependencies
|
|
RUN . /app/talemate_env/bin/activate && \
|
|
poetry config virtualenvs.create false && \
|
|
poetry install --no-dev --no-root
|
|
|
|
# Copy the Python source code
|
|
COPY ./src /app/src
|
|
|
|
# Conditional PyTorch+CUDA install
|
|
ARG CUDA_AVAILABLE=false
|
|
RUN . /app/talemate_env/bin/activate && \
|
|
if [ "$CUDA_AVAILABLE" = "true" ]; then \
|
|
echo "Installing PyTorch with CUDA support..." && \
|
|
pip uninstall torch torchaudio -y && \
|
|
pip install torch~=2.4.1 torchaudio~=2.4.1 --index-url https://download.pytorch.org/whl/cu121; \
|
|
fi
|
|
|
|
# Stage 3: Final image
|
|
FROM python:3.11-slim
|
|
|
|
WORKDIR /app
|
|
|
|
RUN apt-get update && apt-get install -y \
|
|
bash \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Copy virtual environment from backend-build stage
|
|
COPY --from=backend-build /app/talemate_env /app/talemate_env
|
|
|
|
# Copy Python source code
|
|
COPY --from=backend-build /app/src /app/src
|
|
|
|
# Copy Node.js build artifacts from frontend-build stage
|
|
COPY --from=frontend-build /app/dist /app/talemate_frontend/dist
|
|
|
|
# Copy the frontend WSGI file if it exists
|
|
COPY frontend_wsgi.py /app/frontend_wsgi.py
|
|
|
|
# Copy base config
|
|
COPY config.example.yaml /app/config.yaml
|
|
|
|
# Copy essentials
|
|
COPY scenes templates chroma* /app/
|
|
|
|
# Set PYTHONPATH to include the src directory
|
|
ENV PYTHONPATH=/app/src:$PYTHONPATH
|
|
|
|
# Make ports available to the world outside this container
|
|
EXPOSE 5050
|
|
EXPOSE 8080
|
|
|
|
# Use bash as the shell, activate the virtual environment, and run backend server
|
|
CMD ["poetry run src/talemate/server/run.py runserver --host 0.0.0.0 --port 5050 --frontend-host 0.0.0.0 --frontend-port 8080"] |