langchain/.github/scripts/check_prerelease_dependencies.py

"""Check that no dependencies allow prereleases unless we're releasing a prerelease."""

import sys

import tomllib

if __name__ == "__main__":
    # Get the TOML file path from the command line argument
    toml_file = sys.argv[1]

    with open(toml_file, "rb") as file:
        toml_data = tomllib.load(file)

    # See if we're releasing an rc or dev version
    version = toml_data["project"]["version"]
    releasing_rc = "rc" in version or "dev" in version

    # If not, iterate through dependencies and make sure none allow prereleases
    if not releasing_rc:
        dependencies = toml_data["project"]["dependencies"]
        for dep_version in dependencies:
            dep_version_string = (
                dep_version["version"] if isinstance(dep_version, dict) else dep_version
            )

            if "rc" in dep_version_string:
                raise ValueError(
                    f"Dependency {dep_version} has a prerelease version. Please remove this."
                )

            if isinstance(dep_version, dict) and dep_version.get(
                "allow-prereleases", False
            ):
                raise ValueError(
                    f"Dependency {dep_version} has allow-prereleases set to true. Please remove this."
                )