fixes

Author: codegen-bot

.pre-commit-config.yaml

@@ -79,6 +79,13 @@ repos:
         entry: scripts/disallowed-words-check.sh
         language: script
         files: ''  # Check all files
+      - id: generate-runner-imports
+        name: Generate Runner Imports
+        entry: bash -c "uv run --frozen python -m codegen.gscli.main generate runner-imports src/codegen/shared/compilation/function_imports.py"
+        language: system
+        pass_filenames: false
+        always_run: true
+
   - repo: https://github.com/renovatebot/pre-commit-hooks
     rev: 39.120.4
     hooks:

src/codegen/gsbuild/build.py

@@ -19,7 +19,6 @@ def update_function_imports(root_path: Path) -> None:
     from codegen.gscli.generate.runner_imports import _generate_runner_imports
 
     file_path = root_path / "src" / "codegen" / "shared" / "compilation" / "function_imports.py"
-    print(str(file_path))
     _generate_runner_imports(str(file_path))
 
 

src/codegen/gscli/generate/commands.py

@@ -186,3 +186,7 @@ def generate_codegen_sdk_docs(docs_dir: str) -> None:
         json.dump(mint_data, mint_file, indent=2)
 
     print(colored("Updated mint.json with new page sets", "green"))
+
+
+if __name__ == "__main__":
+    generate()

src/codegen/gscli/generate/runner_imports.py

@@ -14,6 +14,7 @@
 CODEGEN_IMPORTS = """
 from codegen.git.models.codemod_context import CodemodContext
 from codegen.git.models.github_named_user_context import GithubNamedUserContext
+from codegen.git.models.pr_options import PROptions
 from codegen.git.models.pr_part_context import PRPartContext
 from codegen.git.models.pull_request_context import PullRequestContext
 """
@@ -34,13 +35,16 @@
 {gs_public_imports}
 """.strip()
 
-IMPORT_FILE_TEMPLATE = '''
-# This file is auto-generated, do not modify manually. Edit this in codegen-backend/cli/generate/runner_imports.py.
+IMPORT_FILE_TEMPLATE = (
+    '''
+# This file is auto-generated, do not modify manually. Edit this in src/codegen/gscli/generate/runner_imports.py.
 def get_generated_imports():
     return """
 {import_str}
 """
 '''.strip()
+    + "\n"
+)
 
 
 def get_runner_imports(include_codegen=True, include_private_imports: bool = True) -> str:

src/codegen/shared/compilation/function_imports.py

@@ -1,4 +1,4 @@
-# This file is auto-generated, do not modify manually. Edit this in codegen-backend/cli/generate/runner_imports.py.
+# This file is auto-generated, do not modify manually. Edit this in src/codegen/gscli/generate/runner_imports.py.
 def get_generated_imports():
     return """
 # External imports
@@ -12,6 +12,7 @@ def get_generated_imports():
 
 from codegen.git.models.codemod_context import CodemodContext
 from codegen.git.models.github_named_user_context import GithubNamedUserContext
+from codegen.git.models.pr_options import PROptions
 from codegen.git.models.pr_part_context import PRPartContext
 from codegen.git.models.pull_request_context import PullRequestContext