github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 15:19:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Autopopulate module name in cli init (#12814)

Browse Source
This commit is contained in:
Erick Friis 2023-11-02 23:45:38 -07:00 committed by GitHub
parent 98aff29fbd
commit 0da75b9ebd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/cli/langchain_cli/namespaces/template.py
View File

@ -67,6 +67,11 @@ def new(
package_dir = destination_dir / module_name package_dir = destination_dir / module_name
shutil.move(destination_dir / "package_template", package_dir) shutil.move(destination_dir / "package_template", package_dir)
# update init
init = package_dir / "__init__.py"
init_contents = init.read_text()
init.write_text(init_contents.replace("__module_name__", module_name))
# replace readme # replace readme
readme = destination_dir / "README.md" readme = destination_dir / "README.md"
readme_contents = readme.read_text() readme_contents = readme.read_text()

2
libs/cli/langchain_cli/package_template/package_template/__init__.py
View File

@ -1,3 +1,3 @@
from chain import chain from __module_name__.chain import chain
__all__ = ["chain"] __all__ = ["chain"]