feat: switch to uv

Author: gusye1234

.github/workflows/publish.yaml

@@ -30,23 +30,22 @@ jobs:
     name: Tests on ${{ matrix.os }} for ${{ matrix.python-version }}
     strategy:
       matrix:
-        python-version: [3.11]
+        python-version: [3.12]
         os: [ubuntu-latest]
     runs-on: ${{ matrix.os }}
     permissions: 
       contents: read
     steps:
       - uses: actions/checkout@v4
-      - name: Setup Python
-        uses: actions/setup-python@v5
-        with:
-          python-version: ${{ matrix.python-version }}
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
+
+      - name: Set up Python
+        run: uv python install
       - name: Install dependencies
         working-directory: ./src/server/api
         run: |
-          python -m pip install --upgrade pip
-          pip install -r requirements.txt
-          pip install -r requirements-dev.txt
+          uv sync --frozen --no-dev --no-cache-dir
       - name: Start containers
         working-directory: ./src/server
         run: |
@@ -64,7 +63,7 @@ jobs:
         working-directory: ./src/server/api
         run: |
           cp .env.example .env
-          pytest --junit-xml=junit/test-results-${{ matrix.python-version }}.xml --cov=. --cov-report=xml:coverage-${{ matrix.python-version }}.xml tests/ -s -v
+          uv run -m pytest --junit-xml=junit/test-results-${{ matrix.python-version }}.xml --cov=. --cov-report=xml:coverage-${{ matrix.python-version }}.xml tests/ -s -v
       - name: Upload pytest test results and coverage
         uses: actions/upload-artifact@v4
         with:

src/server/api/.python-version

@@ -1 +1 @@
-3.12
+3.12

src/server/api/Dockerfile

@@ -1,13 +1,28 @@
-FROM python:3.11-slim-buster AS build
+FROM python:3.12-slim-bookworm
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 
 
+# Set working directory
 WORKDIR /app
 
-COPY ./requirements.txt /app/requirements.txt
-RUN python3.11 -m pip install -r requirements.txt --no-cache-dir
+# Copy dependency files
+COPY ./pyproject.toml /app
+COPY ./uv.lock /app
 
+# Install Python dependencies using uv
+RUN uv sync --frozen --no-dev --no-cache-dir
+
+# Copy the application code
 COPY ./memobase_server /app/memobase_server
 COPY ./api.py /app
 
 
-CMD ["python3.11", "-m", "fastapi", "run", "api.py"]
+
+# Expose port 8000 (FastAPI default)
+EXPOSE 8000
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+
+# Run the FastAPI server using uvicorn
+CMD ["/app/.venv/bin/uvicorn", "api:app", "--host", "0.0.0.0", "--port", "8000"]

src/server/api/pyproject.toml

@@ -1,7 +1,32 @@
 [project]
-name = "api"
+name = "memobase_server"
 version = "0.1.0"
-description = "Add your description here"
+description = "Memobase Server"
 readme = "README.md"
-requires-python = ">=3.13"
-dependencies = []
+requires-python = ">=3.12"
+dependencies = [
+    "fastapi[standard]>=0.116.1",
+    "numpy>=2.3.1",
+    "openai>=1.97.0",
+    "opentelemetry-api>=1.35.0",
+    "opentelemetry-exporter-prometheus>=0.56b0",
+    "opentelemetry-instrumentation-fastapi>=0.56b0",
+    "opentelemetry-sdk>=1.35.0",
+    "pgvector>=0.4.1",
+    "psycopg2-binary>=2.9.10",
+    "python-dotenv>=1.1.1",
+    "pyyaml>=6.0.2",
+    "redis>=6.2.0",
+    "sqlalchemy>=2.0.41",
+    "tiktoken>=0.9.0",
+    "typeguard>=4.4.4",
+    "volcengine-python-sdk[ark]>=4.0.6",
+]
+
+[dependency-groups]
+dev = [
+    "alembic>=1.16.4",
+    "pytest>=8.4.1",
+    "pytest-asyncio>=1.1.0",
+    "pytest-cov>=6.2.1",
+]

src/server/api/requirements-dev.txt

@@ -88,9 +88,9 @@ Check out the [docs](https://docs.memobase.io/quickstart) of how to use Memobase
 
 1. Start a local DB first by `sh script/up-dev.sh`
 2. Open a new terminal window and `cd ./api`
-3. Install python deps: `pip install -r requirements.txt`
-4. To test if you got everything right, run `pytest` to see if all the tests are passed.
-5. Launch Memobase Server in dev mode: `fastapi dev --port 8019`
+3. Install python deps: `uv sync`
+4. To test if you got everything right, run `uv run pytest` to see if all the tests are passed.
+5. Launch Memobase Server in dev mode: `uv run -m fastapi dev --port 8019`
 
 > `fastapi dev` has hot-reload, so you can just modify the code and test it without relaunch the service.