Преглед изворни кода

fix: update sglang dependency version and correct version tag handling

myhloli пре 5 месеци
родитељ
комит
2c8db7edec
2 измењених фајлова са 4 додато и 4 уклоњено
  1. 1 1
      pyproject.toml
  2. 3 3
      update_version.py

+ 1 - 1
pyproject.toml

@@ -42,7 +42,7 @@ vlm = [
     "pydantic",
 ]
 sglang = [
-    "sglang[all]==0.4.6.post5",
+    "sglang[all]>=0.4.7",
 ]
 pipeline = [
     "matplotlib>=3.10,<4",

+ 3 - 3
update_version.py

@@ -7,17 +7,17 @@ def get_version():
     try:
         version = subprocess.check_output(command).decode().strip()
         version_parts = version.split("-")
-        if len(version_parts) > 1 and version_parts[0].startswith("magic_pdf"):
+        if len(version_parts) > 1 and version_parts[0].startswith("mineru"):
             return version_parts[1]
         else:
-            raise ValueError(f"Invalid version tag {version}. Expected format is magic_pdf-<version>-released.")
+            raise ValueError(f"Invalid version tag {version}. Expected format is mineru-<version>-released.")
     except Exception as e:
         print(e)
         return "0.0.0"
 
 
 def write_version_to_commons(version):
-    commons_path = os.path.join(os.path.dirname(__file__), 'magic_pdf', 'libs', 'version.py')
+    commons_path = os.path.join(os.path.dirname(__file__), 'mineru', 'version.py')
     with open(commons_path, 'w') as f:
         f.write(f'__version__ = "{version}"\n')