fix(docker): handle missing ArcadeDB wheel in Docker environment

Author: tae898

bindings/python/examples/07_stackoverflow_tables_oltp.py

@@ -2029,9 +2029,26 @@ def run_in_docker(args):
             continue
         filtered_args.append(arg)
 
-    packages = ["lxml"]
+    arcadedb_wheel_mount_path = None
     if args.db == "arcadedb":
-        packages.append(f"arcadedb-embedded=={args.arcadedb_version}")
+        wheel_candidates = sorted(
+            (repo_root / "bindings/python/dist").glob("*embed*.whl")
+        )
+        if not wheel_candidates:
+            if os.environ.get("GITHUB_ACTIONS", "").lower() == "true":
+                raise RuntimeError(
+                    "ArcadeDB wheel not found in bindings/python/dist during GitHub Actions run"
+                )
+            print(
+                "[info] No local ArcadeDB wheel found in bindings/python/dist; "
+                "skipping Docker wrapper and running natively."
+            )
+            return False
+        arcadedb_wheel_mount_path = (
+            f"/workspace/bindings/python/dist/{wheel_candidates[0].name}"
+        )
+
+    packages = ["lxml"]
     if args.db == "duckdb":
         packages.append(f"duckdb=={args.duckdb_version}")
     if args.db == "postgresql":
@@ -2050,6 +2067,8 @@ def run_in_docker(args):
     inner_cmd_parts.append(". /tmp/bench-venv/bin/activate")
     inner_cmd_parts.append(f"{python_cmd} -m pip install --no-cache-dir uv")
     inner_cmd_parts.append(f"uv pip install {packages_str}")
+    if arcadedb_wheel_mount_path is not None:
+        inner_cmd_parts.append(f'uv pip install "{arcadedb_wheel_mount_path}"')
     inner_cmd_parts.append("echo 'Starting benchmark...'")
     if args.db == "postgresql":
         db_name = build_benchmark_db_name(args.dataset, args.db, args.run_label)

bindings/python/examples/08_stackoverflow_tables_olap.py

@@ -1339,9 +1339,26 @@ def run_in_docker(args):
             continue
         filtered_args.append(arg)
 
-    packages = ["lxml"]
+    arcadedb_wheel_mount_path = None
     if args.db == "arcadedb":
-        packages.append(f"arcadedb-embedded=={args.arcadedb_version}")
+        wheel_candidates = sorted(
+            (repo_root / "bindings/python/dist").glob("*embed*.whl")
+        )
+        if not wheel_candidates:
+            if os.environ.get("GITHUB_ACTIONS", "").lower() == "true":
+                raise RuntimeError(
+                    "ArcadeDB wheel not found in bindings/python/dist during GitHub Actions run"
+                )
+            print(
+                "[info] No local ArcadeDB wheel found in bindings/python/dist; "
+                "skipping Docker wrapper and running natively."
+            )
+            return False
+        arcadedb_wheel_mount_path = (
+            f"/workspace/bindings/python/dist/{wheel_candidates[0].name}"
+        )
+
+    packages = ["lxml"]
     if args.db == "duckdb":
         packages.append(f"duckdb=={args.duckdb_version}")
     if args.db == "postgresql":
@@ -1359,6 +1376,8 @@ def run_in_docker(args):
     inner_cmd_parts.append(". /tmp/bench-venv/bin/activate")
     inner_cmd_parts.append(f"{python_cmd} -m pip install --no-cache-dir uv")
     inner_cmd_parts.append(f"uv pip install {packages_str}")
+    if arcadedb_wheel_mount_path is not None:
+        inner_cmd_parts.append(f'uv pip install "{arcadedb_wheel_mount_path}"')
     inner_cmd_parts.append("echo 'Starting benchmark...'")
     inner_cmd_parts.append(
         f"{python_cmd} -u 08_stackoverflow_tables_olap.py {' '.join(filtered_args)}"

bindings/python/examples/09_stackoverflow_graph_oltp.py

@@ -7738,9 +7738,26 @@ def run_in_docker(args) -> bool:
             continue
         filtered_args.append(arg)
 
-    packages = ["lxml"]
+    arcadedb_wheel_mount_path = None
     if args.db == "arcadedb":
-        packages.append(f"arcadedb-embedded=={args.arcadedb_version}")
+        wheel_candidates = sorted(
+            (repo_root / "bindings/python/dist").glob("*embed*.whl")
+        )
+        if not wheel_candidates:
+            if os.environ.get("GITHUB_ACTIONS", "").lower() == "true":
+                raise RuntimeError(
+                    "ArcadeDB wheel not found in bindings/python/dist during GitHub Actions run"
+                )
+            print(
+                "[info] No local ArcadeDB wheel found in bindings/python/dist; "
+                "skipping Docker wrapper and running natively."
+            )
+            return False
+        arcadedb_wheel_mount_path = (
+            f"/workspace/bindings/python/dist/{wheel_candidates[0].name}"
+        )
+
+    packages = ["lxml"]
     if args.db in ("ladybug", "ladybugdb"):
         if args.ladybug_version:
             packages.append(f"real_ladybug=={args.ladybug_version}")
@@ -7757,6 +7774,8 @@ def run_in_docker(args) -> bool:
     inner_cmd_parts.append(". /tmp/bench-venv/bin/activate")
     inner_cmd_parts.append(f"{python_cmd} -m pip install --no-cache-dir uv")
     inner_cmd_parts.append(f"uv pip install {packages_str}")
+    if arcadedb_wheel_mount_path is not None:
+        inner_cmd_parts.append(f'uv pip install "{arcadedb_wheel_mount_path}"')
     inner_cmd_parts.append("echo 'Starting benchmark...'")
     inner_cmd_parts.append(
         f"{python_cmd} -u 09_stackoverflow_graph_oltp.py {' '.join(filtered_args)}"

bindings/python/examples/10_stackoverflow_graph_olap.py

@@ -4926,9 +4926,26 @@ def run_in_docker(args) -> bool:
             continue
         filtered_args.append(arg)
 
-    packages = ["lxml"]
+    arcadedb_wheel_mount_path = None
     if args.db in ("arcadedb", "arcadedb_cypher"):
-        packages.append(f"arcadedb-embedded=={args.arcadedb_version}")
+        wheel_candidates = sorted(
+            (repo_root / "bindings/python/dist").glob("*embed*.whl")
+        )
+        if not wheel_candidates:
+            if os.environ.get("GITHUB_ACTIONS", "").lower() == "true":
+                raise RuntimeError(
+                    "ArcadeDB wheel not found in bindings/python/dist during GitHub Actions run"
+                )
+            print(
+                "[info] No local ArcadeDB wheel found in bindings/python/dist; "
+                "skipping Docker wrapper and running natively."
+            )
+            return False
+        arcadedb_wheel_mount_path = (
+            f"/workspace/bindings/python/dist/{wheel_candidates[0].name}"
+        )
+
+    packages = ["lxml"]
     if args.db in ("ladybug", "ladybugdb"):
         if args.ladybug_version:
             packages.append(f"real_ladybug=={args.ladybug_version}")
@@ -4948,6 +4965,8 @@ def run_in_docker(args) -> bool:
     inner_cmd_parts.append(". /tmp/bench-venv/bin/activate")
     inner_cmd_parts.append(f"{python_cmd} -m pip install --no-cache-dir uv")
     inner_cmd_parts.append(f"uv pip install {packages_str}")
+    if arcadedb_wheel_mount_path is not None:
+        inner_cmd_parts.append(f'uv pip install "{arcadedb_wheel_mount_path}"')
     inner_cmd_parts.append("echo 'Starting benchmark...'")
     inner_cmd_parts.append(
         f"{python_cmd} -u 10_stackoverflow_graph_olap.py {' '.join(filtered_args)}"

bindings/python/examples/11_vector_index_build.py

@@ -1418,6 +1418,25 @@ def run_in_docker(args) -> bool:
         if not has_qdrant_host:
             filtered_args.extend(["--qdrant-host", "host.docker.internal"])
 
+    arcadedb_wheel_mount_path = None
+    if args.backend == "arcadedb":
+        wheel_candidates = sorted(
+            (repo_root / "bindings/python/dist").glob("*embed*.whl")
+        )
+        if not wheel_candidates:
+            if os.environ.get("GITHUB_ACTIONS", "").lower() == "true":
+                raise RuntimeError(
+                    "ArcadeDB wheel not found in bindings/python/dist during GitHub Actions run"
+                )
+            print(
+                "[info] No local ArcadeDB wheel found in bindings/python/dist; "
+                "skipping Docker wrapper and running natively."
+            )
+            return False
+        arcadedb_wheel_mount_path = (
+            f"/workspace/bindings/python/dist/{wheel_candidates[0].name}"
+        )
+
     image = args.docker_image
     if args.backend == "pgvector" and image == "python:3.12-slim":
         image = default_docker_image(args.backend)
@@ -1428,8 +1447,8 @@ def run_in_docker(args) -> bool:
                 "python -m venv /tmp/bench-venv",
                 ". /tmp/bench-venv/bin/activate",
                 "python -m pip install --no-cache-dir uv",
-                "uv pip install "
-                f"arcadedb-embedded=={args.arcadedb_version} numpy psutil",
+                "uv pip install numpy psutil",
+                f'uv pip install "{arcadedb_wheel_mount_path}"',
                 "echo 'Starting vector build benchmark...'",
                 f"python -u 11_vector_index_build.py {' '.join(filtered_args)}",
             ]

bindings/python/examples/12_vector_search.py

@@ -1734,6 +1734,25 @@ def run_in_docker(args) -> bool:
         if not has_qdrant_host:
             filtered_args.extend(["--qdrant-host", "host.docker.internal"])
 
+    arcadedb_wheel_mount_path = None
+    if args.backend == "arcadedb":
+        wheel_candidates = sorted(
+            (repo_root / "bindings/python/dist").glob("*embed*.whl")
+        )
+        if not wheel_candidates:
+            if os.environ.get("GITHUB_ACTIONS", "").lower() == "true":
+                raise RuntimeError(
+                    "ArcadeDB wheel not found in bindings/python/dist during GitHub Actions run"
+                )
+            print(
+                "[info] No local ArcadeDB wheel found in bindings/python/dist; "
+                "skipping Docker wrapper and running natively."
+            )
+            return False
+        arcadedb_wheel_mount_path = (
+            f"/workspace/bindings/python/dist/{wheel_candidates[0].name}"
+        )
+
     image = args.docker_image or default_docker_image(args.backend)
 
     if args.backend == "arcadedb":
@@ -1742,7 +1761,8 @@ def run_in_docker(args) -> bool:
                 "python -m venv /tmp/bench-venv",
                 ". /tmp/bench-venv/bin/activate",
                 "python -m pip install --no-cache-dir uv",
-                f"uv pip install arcadedb-embedded=={args.arcadedb_version} numpy psutil",
+                "uv pip install numpy psutil",
+                f'uv pip install "{arcadedb_wheel_mount_path}"',
                 "echo 'Starting vector search benchmark...'",
                 f"python -u 12_vector_search.py {' '.join(filtered_args)}",
             ]