mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-12 14:23:58 +00:00
e7eac27241
* standardizes ruff dep version across all `pyproject.toml` files * cli: ruff rules and corrections * langchain: rules and corrections
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
"""Generate migrations for partner packages."""
|
|
|
|
import importlib
|
|
|
|
from langchain_core.documents import BaseDocumentCompressor, BaseDocumentTransformer
|
|
from langchain_core.embeddings import Embeddings
|
|
from langchain_core.language_models import BaseLanguageModel
|
|
from langchain_core.retrievers import BaseRetriever
|
|
from langchain_core.vectorstores import VectorStore
|
|
|
|
from langchain_cli.namespaces.migrate.generate.utils import (
|
|
COMMUNITY_PKG,
|
|
find_subclasses_in_module,
|
|
list_classes_by_package,
|
|
list_init_imports_by_package,
|
|
)
|
|
|
|
# PUBLIC API
|
|
|
|
|
|
def get_migrations_for_partner_package(pkg_name: str) -> list[tuple[str, str]]:
|
|
"""Generate migrations from community package to partner package.
|
|
|
|
This code works
|
|
|
|
Args:
|
|
pkg_name (str): The name of the partner package.
|
|
|
|
Returns:
|
|
List of 2-tuples containing old and new import paths.
|
|
|
|
"""
|
|
package = importlib.import_module(pkg_name)
|
|
classes_ = find_subclasses_in_module(
|
|
package,
|
|
[
|
|
BaseLanguageModel,
|
|
Embeddings,
|
|
BaseRetriever,
|
|
VectorStore,
|
|
BaseDocumentTransformer,
|
|
BaseDocumentCompressor,
|
|
],
|
|
)
|
|
community_classes = list_classes_by_package(str(COMMUNITY_PKG))
|
|
imports_for_pkg = list_init_imports_by_package(str(COMMUNITY_PKG))
|
|
|
|
old_paths = community_classes + imports_for_pkg
|
|
|
|
return [
|
|
(f"{module}.{item}", f"{pkg_name}.{item}")
|
|
for module, item in old_paths
|
|
if item in classes_
|
|
]
|