github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-12 12:59:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

25
libs/cli/tests/unit_tests/migrate/generate/test_langchain_migration.py Normal file
View File

@@ -0,0 +1,25 @@
from langchain_cli.namespaces.migrate.generate.langchain import (
generate_simplified_migrations,
)
def test_create_json_agent_migration() -> None:
"""Test the migration of create_json_agent from langchain to langchain_community."""
raw_migrations = generate_simplified_migrations()
json_agent_migrations = [
migration for migration in raw_migrations if "create_json_agent" in migration[0]
]
assert json_agent_migrations == [
(
"langchain.agents.create_json_agent",
"langchain_community.agent_toolkits.create_json_agent",
),
(
"langchain.agents.agent_toolkits.create_json_agent",
"langchain_community.agent_toolkits.create_json_agent",
),
(
"langchain.agents.agent_toolkits.json.base.create_json_agent",
"langchain_community.agent_toolkits.create_json_agent",
),
]