fix: bundle vikingdb wheel for byteplus studio

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

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,

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",
+    ),
 )