Spaces:
Running
Running
File size: 1,420 Bytes
d202ada |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
#!/usr/bin/env python
import re
import sys
from pathlib import Path
import packaging.version
BASE_DIR = Path(__file__).parent.parent.parent
ARGUMENT_NUMBER = 2
def update_base_dep(pyproject_path: str, new_version: str) -> None:
"""Update the langflow-base dependency in pyproject.toml."""
filepath = BASE_DIR / pyproject_path
content = filepath.read_text(encoding="utf-8")
replacement = f'langflow-base-nightly = "{new_version}"'
# Updates the pattern for poetry
pattern = re.compile(r'langflow-base = \{ path = "\./src/backend/base", develop = true \}')
if not pattern.search(content):
msg = f'langflow-base poetry dependency not found in "{filepath}"'
raise ValueError(msg)
content = pattern.sub(replacement, content)
filepath.write_text(content, encoding="utf-8")
def verify_pep440(version):
"""Verify if version is PEP440 compliant.
https://github.com/pypa/packaging/blob/16.7/packaging/version.py#L191
"""
return packaging.version.Version(version)
def main() -> None:
if len(sys.argv) != ARGUMENT_NUMBER:
msg = "New version not specified"
raise ValueError(msg)
base_version = sys.argv[1]
# Strip "v" prefix from version if present
base_version = base_version.removeprefix("v")
verify_pep440(base_version)
update_base_dep("pyproject.toml", base_version)
if __name__ == "__main__":
main()
|