From 7ad274729bbae106f0feb67a878816885d224afc Mon Sep 17 00:00:00 2001
From: shanchunhua <shanchunhua@bytedance.com>
Date: Tue, 22 Sep 2026 15:09:54 +0800
Subject: [PATCH] fix: bundle vikingdb wheel for byteplus studio

---
 tests/cli/test_studio_deploy_target.py |  1 +
 tests/cli/test_studio_release.py       | 13 +++++++++++++
 veadk/cli/studio_dependencies.py       |  9 +++++++++
 3 files changed, 23 insertions(+)

diff --git a/tests/cli/test_studio_deploy_target.py b/tests/cli/test_studio_deploy_target.py
index beda6a6a1..3a81b015d 100644
--- a/tests/cli/test_studio_deploy_target.py
+++ b/tests/cli/test_studio_deploy_target.py
@@ -74,6 +74,7 @@ def _stage_test_studio_dependencies(
     ]
     if provider == "byteplus":
         names.append("pydantic-2.12.5-py3-none-any.whl")
+        names.append("vikingdb_python_sdk-0.1.32-py3-none-any.whl")
     staged: list[Path] = []
     for name in names:
         wheel = destination / name
diff --git a/tests/cli/test_studio_release.py b/tests/cli/test_studio_release.py
index 8f5e40295..6c2463cf8 100644
--- a/tests/cli/test_studio_release.py
+++ b/tests/cli/test_studio_release.py
@@ -42,6 +42,7 @@
     STUDIO_DEPENDENCY_SOURCES,
     StudioDependencyWheel,
     stage_studio_dependency_wheels,
+    studio_dependency_wheels,
     write_studio_dependency_manifest,
 )
 from veadk.cli.studio_package import build_frontend_assets, studio_run_script
@@ -569,6 +570,18 @@ def _urlopen(url: str, *, timeout: int) -> io.BytesIO:
     assert urls == [dependency.url]
 
 
+def test_byteplus_dependency_wheels_include_vikingdb_sdk_pin() -> None:
+    volcengine_names = {
+        dependency.filename for dependency in studio_dependency_wheels("volcengine")
+    }
+    byteplus_names = {
+        dependency.filename for dependency in studio_dependency_wheels("byteplus")
+    }
+
+    assert "vikingdb_python_sdk-0.1.32-py3-none-any.whl" not in volcengine_names
+    assert "vikingdb_python_sdk-0.1.32-py3-none-any.whl" in byteplus_names
+
+
 def test_write_dependency_manifest_uses_pinned_wheel_metadata(
     tmp_path: Path,
     monkeypatch: pytest.MonkeyPatch,
diff --git a/veadk/cli/studio_dependencies.py b/veadk/cli/studio_dependencies.py
index 813336cbf..8d1ec5053 100644
--- a/veadk/cli/studio_dependencies.py
+++ b/veadk/cli/studio_dependencies.py
@@ -113,6 +113,15 @@ class StudioDependencySource:
         ),
         sha256="e561593fccf61e8a20fc46dfc2dfe075b8be7d0188df33f221ad1f0139180f9d",
     ),
+    StudioDependencyWheel(
+        filename="vikingdb_python_sdk-0.1.32-py3-none-any.whl",
+        url=(
+            "https://files.pythonhosted.org/packages/e7/4d/"
+            "933c322781c6a1ac4543a3442d1975e164182a6a0e35eff81118cc92995d/"
+            "vikingdb_python_sdk-0.1.32-py3-none-any.whl"
+        ),
+        sha256="7e58ba299b561bd53859cfb63a5e4af95088a3723378109548c2011390f90bfb",
+    ),
 )