CG-10473: Generate function_imports on build

.circleci/ats.sh

@@ -33,4 +33,3 @@ if [ ! -s codecov_ats/tests_to_run.txt ]; then
     run_count=1
     echo "Added ${TESTS_TO_RUN} as fallback. New run count: $run_count"
 fi
-

.circleci/config.yml

@@ -528,6 +528,3 @@ workflows:
           requires:
             - linux-wheels
             - osx-wheels
-
-
-

.github/codecov.yml

@@ -71,4 +71,3 @@ cli:
         - "--verbose"
         - "-n=auto"
       python_path: ".venv/bin/python"
-

.gitignore

@@ -65,4 +65,4 @@ graph-sitter-types/typings/**
 coverage.json
 tests/integration/verified_codemods/codemod_data/repo_commits.json
 .codegen/*
-.benchmarks/*
+.benchmarks/*

.lfsconfig

@@ -1,2 +1,2 @@
 [lfs]
-	url = [email protected]:codegen-sh/graph-sitter-private.git
+	url = [email protected]:codegen-sh/graph-sitter-private.git

.pre-commit-config.yaml

@@ -43,7 +43,7 @@ repos:
         exclude_types: [mdx, pyi]
       - id: check-yaml
       - id: end-of-file-fixer
-        files: .*/expected_diff.patch
+        exclude: (src/codemods/eval)|(tests/unit/skills/snapshots)|(tests/unit/codegen/sdk/output)|(tests/integration/verified_codemods)|(docs/)
       - id: check-merge-conflict
 
   - repo: https://github.com/python-jsonschema/check-jsonschema
@@ -80,6 +80,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:

CONTRIBUTING.md

@@ -69,4 +69,3 @@ git push origin v0.YOUR_VERSION
 ```
 
 This will trigger a release job to build this new version.
-

mypy.ini

@@ -49,4 +49,3 @@ exclude = (?x)(
           | tests/unit/skills/snapshots/*
     )
 plugins = pydantic.mypy
-

pyproject.toml

@@ -153,14 +153,7 @@ max-line-length = 200
 
 [tool.ruff]
 line-length = 200
-exclude = [
-  "app/codemod/compilation/function_imports.py", # generated import list to compile uesr code with
-  'codegen_tests/**/expected*/**',
-  'codegen_tests/**/original/**',
-  'codegen_tests/projects/**/codebase/**',
-  'docs/**',
-  'scripts/Debugging/**',
-]
+exclude = ["src/codegen/shared/compilation/function_imports.py", 'docs/**']
 
 lint.select = ["F", "E", "W", "I", "UP", "D", "RUF", "DTZ"]
 lint.ignore = [
@@ -192,9 +185,8 @@ lint.exclude = [
 ] # disable just linting for notebooks (allow for formatting)
 lint.pydocstyle.convention = "google"
 [tool.ruff.lint.isort]
-known-first-party = ["app"]
-known-third-party = ["git"]
-#known-local-folder=[]
+known-first-party = ["codegen"]
+
 [tool.coverage.run]
 branch = true
 plugins = ["Cython.Coverage"]

scripts/setup-lfs.sh

@@ -2,4 +2,4 @@
 ./scripts/install-deps.sh
 
 git lfs install
-git lfs pull
+git lfs pull

scripts/setup.sh

@@ -6,4 +6,4 @@ uv tool update-shell
 uv venv && source .venv/bin/activate
 uv sync
 pre-commit install
-pre-commit install-hooks
+pre-commit install-hooks

src/codegen/gsbuild/build.py

@@ -21,5 +21,4 @@ class SpecialBuildHook(BuildHookInterface):
     def initialize(self, version: str, build_data: dict[str, Any]) -> None:
         file = Path(self.root) / "src" / "codegen" / "sdk" / "__init__.py"
         update_init_file(file)
-
         build_data["artifacts"].append(f"/{file}")

src/codegen/gscli/__init__.py

@@ -1 +0,0 @@
-

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/runner/sandbox/executor.py

@@ -53,7 +53,7 @@ async def find_flag_groups(self, code_flags: list[CodeFlag], grouping_config: Gr
 
         # =====[ Group the code flags ]=====
         logger.info(f"> Grouping CodeFlags by config: {grouping_config}")
-        grouper = get_grouper_by_group_by(grouping_config.group_by, repo_id=self.codebase.op.repo_config.id)
+        grouper = get_grouper_by_group_by(grouping_config.group_by)
         groups = grouper.create_all_groups(flags=code_flags, repo_operator=self.codebase.op)
         logger.info(f"> Created {len(groups)} groups")
         return groups

src/codegen/shared/compilation/README.md

@@ -5,4 +5,3 @@ This includes:
 - Catching and logging any compilation errors
 - Monkey patching built-ins like print
 - etc
-

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,10 +12,11 @@ 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
 
-from codegen.sdk.codebase.control_flow import StopCodemodException
+from codegen.shared.exceptions.control_flow import StopCodemodException
 
 # GraphSitter imports (public)
 from codegen.sdk.codebase.flagging.enums import FlagKwargs