cli[minor]: update code to generate migrations from langchain to community (#20946)

Updates code that generates migrations from langchain to community
This commit is contained in:
Eugene Yurtsev
2024-04-26 15:11:32 -04:00
committed by GitHub
parent 078c5d9bc6
commit 2fa0ff1a2d
4 changed files with 4423 additions and 1457 deletions

View File

@@ -5,7 +5,7 @@ import pkgutil
import click
from langchain_cli.namespaces.migrate.generate.langchain import (
generate_migrations_from_langchain_to_community,
generate_simplified_migrations,
)
from langchain_cli.namespaces.migrate.generate.partner import (
get_migrations_for_partner_package,
@@ -27,9 +27,9 @@ def cli():
def langchain(output: str) -> None:
"""Generate a migration script."""
click.echo("Migration script generated.")
migrations = generate_migrations_from_langchain_to_community()
migrations = generate_simplified_migrations()
with open(output, "w") as f:
f.write(json.dumps(migrations))
f.write(json.dumps(migrations, indent=2, sort_keys=True))
@cli.command()