1 files changed, 11 insertions, 0 deletions
diff --git a/setup.py b/setup.py
index fa68277..875d132 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,14 @@
 from setuptools import find_packages, setup
 
+
+def sqlglotrs_version():
+    with open("sqlglotrs/Cargo.toml") as fd:
+        for line in fd.readlines():
+            if line.strip().startswith("version"):
+                return line.split("=")[1].strip().strip('"')
+    raise ValueError("Could not find version in Cargo.toml")
+
+
 setup(
     name="sqlglot",
     description="An easily customizable SQL parser and transpiler",
@@ -31,7 +40,9 @@ setup(
             "pdoc",
             "pre-commit",
             "types-python-dateutil",
+            "maturin>=1.4,<2.0",
         ],
+        "rs": [f"sqlglotrs=={sqlglotrs_version()}"],
     },
     classifiers=[
         "Development Status :: 5 - Production/Stable",