mirror of
https://github.com/hwchase17/langchain.git
synced 2026-07-16 09:06:19 +00:00
12c906f6ce
This auto generates partner migrations. At the moment the migration is from community -> partner. So one would need to run the migration script twice to go from langchain to partner.
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
"""Script to generate migrations for the migration script."""
|
|
import json
|
|
import pkgutil
|
|
|
|
import click
|
|
|
|
from langchain_cli.namespaces.migrate.generate.langchain import (
|
|
generate_migrations_from_langchain_to_community,
|
|
)
|
|
from langchain_cli.namespaces.migrate.generate.partner import (
|
|
get_migrations_for_partner_package,
|
|
)
|
|
|
|
|
|
@click.group()
|
|
def cli():
|
|
"""Migration scripts management."""
|
|
pass
|
|
|
|
|
|
@cli.command()
|
|
@click.option(
|
|
"--output",
|
|
default="langchain_migrations.json",
|
|
help="Output file for the migration script.",
|
|
)
|
|
def langchain(output: str) -> None:
|
|
"""Generate a migration script."""
|
|
click.echo("Migration script generated.")
|
|
migrations = generate_migrations_from_langchain_to_community()
|
|
with open(output, "w") as f:
|
|
f.write(json.dumps(migrations))
|
|
|
|
|
|
@cli.command()
|
|
@click.argument("pkg")
|
|
@click.option("--output", default=None, help="Output file for the migration script.")
|
|
def partner(pkg: str, output: str) -> None:
|
|
"""Generate migration scripts specifically for LangChain modules."""
|
|
click.echo("Migration script for LangChain generated.")
|
|
migrations = get_migrations_for_partner_package(pkg)
|
|
# Run with python 3.9+
|
|
output_name = f"{pkg.removeprefix('langchain_')}.json" if output is None else output
|
|
if migrations:
|
|
with open(output_name, "w") as f:
|
|
f.write(json.dumps(migrations, indent=2, sort_keys=True))
|
|
click.secho(f"LangChain migration script saved to {output_name}")
|
|
else:
|
|
click.secho(f"No migrations found for {pkg}", fg="yellow")
|
|
|
|
|
|
@cli.command()
|
|
def all_installed_partner_pkgs() -> None:
|
|
"""Generate migration scripts for all LangChain modules."""
|
|
# Will generate migrations for all pather packages.
|
|
# Define as "langchain_<partner_name>".
|
|
# First let's determine which packages are installed in the environment
|
|
# and then generate migrations for them.
|
|
langchain_pkgs = [
|
|
name
|
|
for _, name, _ in pkgutil.iter_modules()
|
|
if name.startswith("langchain_")
|
|
and name not in {"langchain_core", "langchain_cli", "langchain_community"}
|
|
]
|
|
for pkg in langchain_pkgs:
|
|
migrations = get_migrations_for_partner_package(pkg)
|
|
# Run with python 3.9+
|
|
output_name = f"{pkg.removeprefix('langchain_')}.json"
|
|
if migrations:
|
|
with open(output_name, "w") as f:
|
|
f.write(json.dumps(migrations, indent=2, sort_keys=True))
|
|
click.secho(f"LangChain migration script saved to {output_name}")
|
|
else:
|
|
click.secho(f"No migrations found for {pkg}", fg="yellow")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|