fix: allow setting custom codebase path from codegen run (#672)

Author: christinewangcw

src/codegen/cli/commands/run/main.py

@@ -13,12 +13,14 @@
 @click.command(name="run")
 @requires_init
 @click.argument("label", required=True)
+@click.option("--path", type=str, help="Path to build the codebase from. Defaults to the repo root.")
 @click.option("--web", is_flag=True, help="Run the function on the web service instead of locally")
 @click.option("--diff-preview", type=int, help="Show a preview of the first N lines of the diff")
 @click.option("--arguments", type=str, help="Arguments as a json string to pass as the function's 'arguments' parameter")
 def run_command(
     session: CodegenSession,
     label: str,
+    path: str | None = None,
     web: bool = False,
     diff_preview: int | None = None,
     arguments: str | None = None,
@@ -57,4 +59,4 @@ def run_command(
     else:
         from codegen.cli.commands.run.run_local import run_local
 
-        run_local(session, codemod, diff_preview=diff_preview)
+        run_local(session, codemod, diff_preview=diff_preview, path=path)

src/codegen/cli/commands/run/run_local.py

@@ -26,6 +26,7 @@ def run_local(
     session: CodegenSession,
     function: DecoratedFunction,
     diff_preview: int | None = None,
+    path: Path | None = None,
 ) -> None:
     """Run a function locally against the codebase.
 
@@ -35,10 +36,10 @@ def run_local(
         diff_preview: Number of lines of diff to preview (None for all)
     """
     # Parse codebase and run
-    repo_root = session.repo_path
+    codebase_path = f"{session.repo_path}/{path}" if path else session.repo_path
 
-    with Status(f"[bold]Parsing codebase at {repo_root} ...", spinner="dots") as status:
-        codebase = parse_codebase(repo_root)
+    with Status(f"[bold]Parsing codebase at {codebase_path} ...", spinner="dots") as status:
+        codebase = parse_codebase(codebase_path)
         status.update("[bold green]✓ Parsed codebase")
 
         status.update("[bold]Running codemod...")