From d649b0ddfb89ae07937fd1ba6869f42bd1cdaa3f Mon Sep 17 00:00:00 2001
From: ulleo <ulleo.wang@gmail.com>
Date: Wed, 27 Aug 2025 17:10:47 +0800
Subject: [PATCH] feat: update pyproject.toml

---
 backend/pyproject.toml | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/backend/pyproject.toml b/backend/pyproject.toml
index 3cfc4aa7..7afa8e96 100644
--- a/backend/pyproject.toml
+++ b/backend/pyproject.toml
@@ -50,6 +50,25 @@ dependencies = [
     "dicttoxml>=1.7.16",
     "dmpython>=2.5.22; platform_system != 'Darwin'",
 ]
+
+[project.optional-dependencies]
+cpu = [
+    "torch>=2.7.0",
+]
+cu128 = [
+    "torch>=2.7.0",
+]
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
+[[tool.uv.index]]
+name = "pytorch-cu128"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true
+
 [[tool.uv.index]]
 name = "default"
 url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
@@ -62,8 +81,18 @@ explicit = true
 
 [tool.uv.sources]
 sqlbot-xpack = { index = "testpypi" }
+torch = [
+    { index = "pytorch-cpu", extra = "cpu" },
+    { index = "pytorch-cu128", extra = "cu128" },
+]
 
 [tool.uv]
+conflicts = [
+    [
+        { extra = "cpu" },
+        { extra = "cu128" },
+    ],
+]
 dev-dependencies = [
     "pytest<8.0.0,>=7.4.3",
     "mypy<2.0.0,>=1.8.0",