feat: 导入 MemRelay 初始源码

2026-09-23 21:40:46 +08:00
commit 0ade3bb318
167 changed files with 53511 additions and 0 deletions
@@ -0,0 +1,84 @@
+FROM python:3.12-slim-bookworm
+
+ARG UID=1000
+ARG GID=1000
+ARG BUILD_HTTP_PROXY=""
+ARG BUILD_HTTPS_PROXY=""
+ARG BUILD_ALL_PROXY=""
+ARG BUILD_APT_MIRROR=""
+
+COPY --from=ghcr.io/astral-sh/uv:0.8.4 /uv /uvx /bin/
+
+ENV PYTHONUNBUFFERED=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    UV_PROJECT_ENVIRONMENT=/opt/venv \
+    UV_DYNAMIC_VERSIONING_BYPASS=0.22.1 \
+    BASIC_MEMORY_HOME=/data/basic-memory/memories \
+    BASIC_MEMORY_CONFIG_DIR=/data/basic-memory/config \
+    BASIC_MEMORY_PROJECT_ROOT=/data/basic-memory/memories \
+    BASIC_MEMORY_SEMANTIC_SEARCH_ENABLED=true \
+    BASIC_MEMORY_SEMANTIC_EMBEDDING_PROVIDER=fastembed \
+    BASIC_MEMORY_SEMANTIC_EMBEDDING_MODEL=sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 \
+    BASIC_MEMORY_SEMANTIC_EMBEDDING_CACHE_DIR=/data/basic-memory/cache \
+    BASIC_MEMORY_AUTO_UPDATE=false \
+    HTTP_PROXY="" \
+    HTTPS_PROXY="" \
+    ALL_PROXY="" \
+    http_proxy="" \
+    https_proxy="" \
+    all_proxy="" \
+    PATH=/opt/venv/bin:$PATH
+
+WORKDIR /app
+
+COPY third_party/basic-memory/pyproject.toml third_party/basic-memory/uv.lock ./
+COPY third_party/basic-memory/src ./src
+COPY third_party/basic-memory/README.md ./
+
+RUN HTTP_PROXY="$BUILD_HTTP_PROXY" \
+    HTTPS_PROXY="$BUILD_HTTPS_PROXY" \
+    ALL_PROXY="$BUILD_ALL_PROXY" \
+    http_proxy="$BUILD_HTTP_PROXY" \
+    https_proxy="$BUILD_HTTPS_PROXY" \
+    all_proxy="$BUILD_ALL_PROXY" \
+    uv sync --locked --no-dev --python /usr/local/bin/python
+
+COPY scripts/preload_basic_memory_model.py /usr/local/lib/memrelay/preload_basic_memory_model.py
+RUN HTTP_PROXY="$BUILD_HTTP_PROXY" \
+    HTTPS_PROXY="$BUILD_HTTPS_PROXY" \
+    ALL_PROXY="$BUILD_ALL_PROXY" \
+    http_proxy="$BUILD_HTTP_PROXY" \
+    https_proxy="$BUILD_HTTPS_PROXY" \
+    all_proxy="$BUILD_ALL_PROXY" \
+    python /usr/local/lib/memrelay/preload_basic_memory_model.py \
+      --cache-dir /opt/basic-memory-model \
+      --threads 2
+
+COPY deploy/basic-memory/entrypoint.sh /usr/local/bin/basic-memory-entrypoint
+
+RUN HTTP_PROXY="$BUILD_HTTP_PROXY" \
+    HTTPS_PROXY="$BUILD_HTTPS_PROXY" \
+    ALL_PROXY="$BUILD_ALL_PROXY" \
+    http_proxy="$BUILD_HTTP_PROXY" \
+    https_proxy="$BUILD_HTTPS_PROXY" \
+    all_proxy="$BUILD_ALL_PROXY" \
+    sh -c 'if [ -n "$BUILD_APT_MIRROR" ]; then sed -i "s|http://deb.debian.org|${BUILD_APT_MIRROR%/}|g" /etc/apt/sources.list.d/debian.sources; fi' \
+    && \
+    groupadd --gid "$GID" appgroup \
+    && useradd --uid "$UID" --gid "$GID" --create-home --shell /usr/sbin/nologin appuser \
+    && apt-get update \
+    && apt-get install --yes --no-install-recommends gosu \
+    && rm -rf /var/lib/apt/lists/* \
+    && mkdir -p /data/basic-memory /opt/basic-memory-model \
+    && chmod +x /usr/local/bin/basic-memory-entrypoint \
+    && chown -R appuser:appgroup /data/basic-memory /opt/basic-memory-model
+
+USER root
+
+EXPOSE 8000
+
+HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
+  CMD python -c "import socket; socket.create_connection(('127.0.0.1', 8000), 5).close()" || exit 1
+
+ENTRYPOINT ["basic-memory-entrypoint"]
+CMD ["basic-memory", "mcp", "--transport", "streamable-http", "--host", "0.0.0.0", "--port", "8000", "--path", "/mcp"]
@@ -0,0 +1,13 @@
+#!/bin/sh
+
+set -eu
+
+cache_dir="${BASIC_MEMORY_SEMANTIC_EMBEDDING_CACHE_DIR:-/data/basic-memory/cache}"
+mkdir -p "$cache_dir"
+
+if [ -z "$(find "$cache_dir" -mindepth 1 -maxdepth 1 -print -quit)" ]; then
+  cp -R /opt/basic-memory-model/. "$cache_dir"/
+fi
+
+chown -R appuser:appgroup /data/basic-memory
+exec gosu appuser "$@"