🐛 Fix vllm overwrite script (#98)

* 🐛 vllm dependency not always first

Signed-off-by: Evaline Ju <[email protected]>

* 🐛 Darwin dep too

Signed-off-by: Evaline Ju <[email protected]>

* 🎨 Format script

Signed-off-by: Evaline Ju <[email protected]>

---------

Signed-off-by: Evaline Ju <[email protected]>

Author: evaline-ju

scripts/update_vllm_version.py

@@ -35,9 +35,24 @@ def main():
 
     with open("pyproject.toml", "rb") as f:
         data = tomllib.load(f)
-        current_version = parse_current_version(
-            vllm_string=data["project"]["optional-dependencies"]["vllm"][0]
+        # Find appropriate index
+        vllm_deps = data["project"]["optional-dependencies"]["vllm"]
+        vllm_git_index = next(
+            (i for i, item in enumerate(vllm_deps) if item.startswith("vllm @ git")),
+            None,
         )
+        vllm_darwin_index = next(
+            (
+                i
+                for i, item in enumerate(vllm_deps)
+                if item.startswith("vllm>") or item.startswith("vllm=")
+            ),
+            None,
+        )
+        if not vllm_git_index or not vllm_darwin_index:
+            print(f"vllm dependencies to be overwritten are missing - skipping update")
+            sys.exit(0)
+        current_version = parse_current_version(vllm_string=vllm_deps[vllm_git_index])
         if current_version == args.vllm_version:
             print(
                 f"VLLM already updated to latest version, skipping update and setting 'SKIP_VERSION' output!"
@@ -47,10 +62,10 @@ def main():
             sys.exit(0)
         else:
             data["project"]["optional-dependencies"]["vllm"][
-                0
+                vllm_git_index
             ] = f"vllm @ git+https://github.com/vllm-project/vllm.git@v{args.vllm_version} ; sys_platform == 'darwin'"
             data["project"]["optional-dependencies"]["vllm"][
-                1
+                vllm_darwin_index
             ] = f"vllm=={args.vllm_version} ; sys_platform != 'darwin'"
 
             with open("pyproject.toml", "wb") as f: