CG-10473: Generate function_imports on build

.pre-commit-config.yaml

@@ -43,7 +43,6 @@ repos:
         exclude_types: [mdx, pyi]
       - id: check-yaml
       - id: end-of-file-fixer
-        files: .*/expected_diff.patch
       - id: check-merge-conflict
 
   - repo: https://github.com/python-jsonschema/check-jsonschema

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"]

src/codegen/gsbuild/build.py

@@ -15,11 +15,19 @@ def update_init_file(file: Path) -> None:
     file.write_text(content)
 
 
+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))
+
+
 class SpecialBuildHook(BuildHookInterface):
     PLUGIN_NAME = "codegen_build"
 
     def initialize(self, version: str, build_data: dict[str, Any]) -> None:
         file = Path(self.root) / "src" / "codegen" / "sdk" / "__init__.py"
         update_init_file(file)
-
+        update_function_imports(Path(self.root))
         build_data["artifacts"].append(f"/{file}")

src/codegen/shared/compilation/function_imports.py

@@ -15,7 +15,7 @@ def get_generated_imports():
 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