Merge pull request #3 from SharpAI/feature/api-parity-roadmap

Feature/api parity roadmap

Author: solderzzc

.github/workflows/build.yml

@@ -2,7 +2,7 @@ name: Build
 
 on:
   push:
-    branches: [main]
+    branches: [main, develop]
   pull_request:
     branches: [main]
 
@@ -15,6 +15,14 @@ jobs:
       - name: Install Metal Toolchain
         run: xcodebuild -downloadComponent MetalToolchain || true
 
+      - name: Cache Swift packages
+        uses: actions/cache@v4
+        with:
+          path: .build
+          key: ${{ runner.os }}-spm-${{ hashFiles('Package.resolved') }}
+          restore-keys: |
+            ${{ runner.os }}-spm-
+
       - name: Resolve dependencies
         run: swift package resolve
 
@@ -25,10 +33,3 @@ jobs:
         run: |
           ls -lh .build/release/mlx-server
           file .build/release/mlx-server
-
-      - name: Upload binary
-        uses: actions/upload-artifact@v4
-        with:
-          name: mlx-server-arm64
-          path: .build/release/mlx-server
-          retention-days: 30

.github/workflows/e2e-test.yml

@@ -0,0 +1,55 @@
+name: E2E Tests
+
+on:
+  push:
+    branches: [main, feature/*]
+  pull_request:
+    branches: [main]
+
+concurrency:
+  group: e2e-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  e2e:
+    runs-on: macos-15
+    timeout-minutes: 30
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Cache Swift packages
+        uses: actions/cache@v4
+        with:
+          path: .build
+          key: ${{ runner.os }}-spm-${{ hashFiles('Package.resolved') }}
+          restore-keys: |
+            ${{ runner.os }}-spm-
+
+      - name: Build (Release)
+        run: swift build -c release
+
+      - name: Install MLX Metal library
+        run: |
+          python3 -m venv /tmp/mlx_venv
+          /tmp/mlx_venv/bin/pip install --quiet mlx
+          cp /tmp/mlx_venv/lib/python*/site-packages/mlx/lib/mlx.metallib .build/release/
+
+      - name: Cache MLX model
+        uses: actions/cache@v4
+        with:
+          path: ~/.cache/huggingface
+          key: mlx-model-qwen2.5-0.5b-4bit
+
+      - name: Run E2E tests
+        run: |
+          chmod +x tests/test-server.sh
+          tests/test-server.sh .build/release/mlx-server 15413
+
+      - name: Upload test logs on failure
+        if: failure()
+        uses: actions/upload-artifact@v4
+        with:
+          name: e2e-test-logs
+          path: /tmp/mlx-server-test-*.log
+          retention-days: 7

Package.swift

@@ -23,6 +23,7 @@ let package = Package(
             dependencies: [
                 .product(name: "MLX", package: "mlx-swift"),
                 .product(name: "MLXLLM", package: "mlx-swift-lm"),
+                .product(name: "MLXVLM", package: "mlx-swift-lm"),
                 .product(name: "MLXLMCommon", package: "mlx-swift-lm"),
                 .product(name: "Transformers", package: "swift-transformers"),
                 .product(name: "Hummingbird", package: "hummingbird"),