adding venv to prompt

2025-07-02 15:27:42 +01:00
parent 8ae5a2311b
commit 8065fed126
4 changed files with 21 additions and 6 deletions
--- a/docker/Dockerfile-uv-base
+++ b/docker/Dockerfile-uv-base
@@ -34,3 +34,6 @@ RUN uv pip install packaging setuptools wheel psutil \
    && uv pip install --no-build-isolation "causal_conv1d @ git+https://github.com/Dao-AILab/causal-conv1d.git@main" \
    && uv pip install "mamba_ssm @ git+https://github.com/state-spaces/mamba.git@main" \
    && uv pip install awscli pydantic
+
+# Configure shell to display current virtual environment
+RUN echo 'export PS1="(axolotl-venv) \u@\h:\w\$ "' >> ~/.bashrc