diff --git a/l2mac/__init__.py b/l2mac/__init__.py index e69de29b..4f26e813 100644 --- a/l2mac/__init__.py +++ b/l2mac/__init__.py @@ -0,0 +1 @@ +from l2mac.main import generate_book, generate_codebase \ No newline at end of file diff --git a/l2mac/main.py b/l2mac/main.py index 52b2e7a0..fa4f32a6 100644 --- a/l2mac/main.py +++ b/l2mac/main.py @@ -2,7 +2,7 @@ from typing_extensions import Annotated import typer from enum import Enum -from config_loader import copy_config_to_home, load_config +from l2mac.config_loader import copy_config_to_home, load_config class Domain(str, Enum): codebase = "codebase" @@ -36,9 +36,15 @@ def run_l2mac(prompt_task: Annotated[str, typer.Argument(help="Your input prompt return None config = load_config() - - +def generate_codebase(*args, **kwargs): + kwargs['domain'] = Domain.codebase + return run_l2mac(*args, **kwargs) + +def generate_book(*args, **kwargs): + kwargs['domain'] = Domain.book + return run_l2mac(*args, **kwargs) + if __name__ == '__main__': app() \ No newline at end of file