fix build hook

README.md

@@ -59,6 +59,7 @@ To install `ell`, follow these steps:
    ```sh
    poetry install
    poetry run build-hook
+   poetry shell
    ```
 
 This will install `ell` and its dependencies, including building the necessary frontend components for `ell-studio`.

build_hook.py

@@ -1,6 +1,8 @@
 import os
 import subprocess
 import shutil
+import sys
+import toml
 
 
 def run_command(command, cwd=None):
@@ -12,9 +14,13 @@ def run_command(command, cwd=None):
 
 
 def sync_version():
-    version = (
-        subprocess.check_output(["poetry", "version", "-s"]).strip().decode("utf-8")
-    )
+    with open("pyproject.toml", "r") as f:
+        pyproject = toml.load(f)
+    version = pyproject.get("tool", {}).get("poetry", {}).get("version")
+    if not version:
+        raise ValueError("Version not found in pyproject.toml")
+    version = "v" + version
+    print(f"setting {version=}")
     run_command(f"git tag -f {version}")
 
 
@@ -25,8 +31,16 @@ def python_install():
     run_command("pip uninstall -y ell")
     run_command("poetry install")
     run_command("poetry build")
-    run_command("pip install dist/*.whl")
-    shutil.rmtree("dist", ignore_errors=True)
+    # Use a wildcard pattern that works on both Windows and Unix-like systems
+    dist_dir = "dist"
+    dist_files = [
+        os.path.join(dist_dir, f) for f in os.listdir(dist_dir) if f.endswith(".whl")
+    ]
+    if not dist_files:
+        raise RuntimeError("No .whl files found in dist directory")
+    for file in dist_files:
+        run_command(f"pip install {file}")
+    shutil.rmtree(dist_dir, ignore_errors=True)
 
 
 def npm_install():
@@ -49,3 +63,7 @@ def main():
     npm_install()
     npm_build()
     python_install()
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -24,7 +24,7 @@ include = [
 ]
 
 [tool.poetry.dependencies]
-python = ">=3.9"
+python = ">=3.9,<=3.13"
 fastapi = "^0.111.1"
 numpy = "^2.0.1"
 dill = "^0.3.8"
@@ -39,6 +39,7 @@ typing-extensions = "^4.12.2"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.2"
+toml = "^0.10.2"
 
 [build-system]
 requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning>=1.0.0,<2.0.0"]