diff --git a/tools/python/BUILD b/tools/python/BUILD
index 2181b31..32093d3 100644
--- a/tools/python/BUILD
+++ b/tools/python/BUILD
@@ -1,9 +1,31 @@
+load("@rules_python//python:defs.bzl", "py_runtime_pair")
+
 py_runtime(
-    name = "runtime",
+    name = "python3_runtime",
     files = [
         "runtime_binary.sh",
         "@python_repo//:all_files",
     ],
     interpreter = "runtime_binary.sh",
-    visibility = ["//visibility:public"],
+    python_version = "PY3",
+)
+
+py_runtime_pair(
+    name = "py_runtime",
+    py2_runtime = None,
+    py3_runtime = ":python3_runtime",
+)
+
+toolchain(
+    name = "python_toolchain",
+    target_compatible_with = [
+        "@platforms//cpu:x86_64",
+        "@platforms//os:linux",
+    ],
+    exec_compatible_with = [
+        "@platforms//cpu:x86_64",
+        "@platforms//os:linux",
+    ],
+    toolchain = ":py_runtime",
+    toolchain_type = "@rules_python//python:toolchain_type",
 )