diff --git a/.circleci/config.yml b/.circleci/config.yml index 66521e4ee..44ee67ec4 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -9,7 +9,7 @@ executors: default_image: docker: - image: ghcr.io/astral-sh/uv:python<>-bookworm - working_directory: /graph-sitter + working_directory: /codegen-sdk environment: UV_COMPILE_BYTECODE: 1 UV_LINK_MODE: copy diff --git a/.github/CODEOWNERS b/.github/CODEOWNERS index 1c2b7a2fa..17e111575 100644 --- a/.github/CODEOWNERS +++ b/.github/CODEOWNERS @@ -1,3 +1,3 @@ -* @codegen-sh/graph-sitter @codegen-team +* @codegen-sh/codegen-sdk @codegen-team **/skills @codegen-sh/capabilities @codegen-team @codegen-sh/capabilities -**/code_generation @codegen-sh/graph-sitter @codegen-team +**/code_generation @codegen-sh/codegen-sdk @codegen-team diff --git a/.github/codecov.yml b/.github/codecov.yml index 183deb54e..a9111cff1 100644 --- a/.github/codecov.yml +++ b/.github/codecov.yml @@ -7,28 +7,28 @@ component_management: - type: project # in this case every component that doens't have a status defined will have a project type one threshold: 0 # Shouldn't remove coverage individual_components: - - component_id: graph-sitter-python # this is an identifier that should not be changed - name: graph-sitter-python # this is a display name, and can be changed freely + - component_id: codegen-sdk-python # this is an identifier that should not be changed + name: codegen-sdk-python # this is a display name, and can be changed freely paths: - - src/graph_sitter/python/** + - src/codegen/sdk/python/** statuses: - type: project # in this case every component that doens't have a status defined will have a project type one threshold: 0 # Shouldn't remove coverage - type: patch target: 50 # Language specific featues must be 100% covered - - component_id: graph-sitter-typescript - name: graph-sitter-typescript + - component_id: codegen-sdk-typescript + name: codegen-sdk-typescript paths: - - src/graph_sitter/typescript/** + - src/codegen/sdk/typescript/** statuses: - type: project # in this case every component that doens't have a status defined will have a project type one threshold: 0 # Shouldn't remove coverage - type: patch target: 50 # Language specific featues must be 100% covered - - component_id: graph-sitter-core - name: graph-sitter-core + - component_id: codegen-sdk-core + name: codegen-sdk-core paths: - - src/graph_sitter/** + - src/codegen/sdk/** - component_id: codegen-git name: codegen-git paths: diff --git a/.gitignore b/.gitignore index 47686bbc6..489e43a1e 100644 --- a/.gitignore +++ b/.gitignore @@ -55,13 +55,15 @@ alembic_versions_backup **/.virtual_documents /.nvmrc **/build/test-results/test/TEST*.xml -src/graph_sitter/__init__.py +src/codegen/sdk/__init__.py src/**/*.html .ccache/ uv-*.tar.gz .venv graph-sitter-types/out/** graph-sitter-types/typings/** +codegen-sdk/out/** +codegen-sdk/typings/** coverage.json tests/verified_codemods/codemod_data/repo_commits.json @@ -69,4 +71,4 @@ tests/verified_codemods/codemod_data/repo_commits.json # These are auto-generated docs/api-reference/core/* docs/api-reference/python/* -docs/api-reference/typescript/* \ No newline at end of file +docs/api-reference/typescript/* diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 4c3c1e7a1..43e685749 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -63,15 +63,6 @@ repos: # - tomli # files: "codegen-backend/docs/.*/.*.mdx" - # - repo: "local" - # hooks: - # - id: clean-diffs - # name: Clean codemod diffs - # entry: bash -c "cd codegen-backend && poetry run cg codemod clean-diffs" - # types: [file] - # language: system - # pass_filenames: false - # files: "codegen-backend/codegen_tests/graph_sitter/codemod/.*/expected_diff.patch" - repo: https://github.com/fpgmaas/deptry.git rev: "0.22.0" hooks: diff --git a/Dockerfile b/Dockerfile index 1432e330c..15a5a1ef2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -5,7 +5,7 @@ ENV UV_COMPILE_BYTECODE=1 UV_LINK_MODE=copy ENV GITHUB_WORKSPACE=/workspace LABEL com.circleci.preserve-entrypoint=true # RUN uv tool install keyring --with keyrings.codeartifact -## Change the working directory to the `graph-sitter` directory +## Change the working directory to the `codegen-sdk` directory FROM base_uv AS install-tools RUN apt-get update && apt-get install -y build-essential curl git RUN curl -fsSL https://deb.nodesource.com/setup_23.x -o nodesource_setup.sh @@ -15,7 +15,7 @@ RUN corepack enable RUN --mount=type=cache,target=/root/.cache/uv uv pip install --system coverage RUN --mount=type=cache,target=/root/.cache/uv uv tool install codecov-cli --python 3.10 RUN --mount=type=cache,target=/root/.cache/uv uv tool install pre-commit --with pre-commit-uv -WORKDIR /graph-sitter +WORKDIR /codegen-sdk ENTRYPOINT [ "uv", "run", "--frozen", "/bin/bash"] FROM install-tools AS base-image ## Install dependencies @@ -24,7 +24,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \ --mount=type=bind,source=pyproject.toml,target=pyproject.toml \ --mount=type=bind,source=hatch.toml,target=hatch.toml \ uv sync --frozen --no-install-workspace --all-extras -ADD . /graph-sitter +ADD . /codegen-sdk ## Sync the project RUN --mount=type=cache,target=/root/.cache/uv \ uv sync --frozen --all-extras diff --git a/codegen-git/pyproject.toml b/codegen-git/pyproject.toml index 11e767c55..c09efdb08 100644 --- a/codegen-git/pyproject.toml +++ b/codegen-git/pyproject.toml @@ -26,8 +26,8 @@ source = "vcs" raw-options = { root = ".." } [tool.hatch.metadata.hooks.vcs] -Homepage = "https://github.com/codegen-sh/graph_sitter" -source_archive = "https://github.com/codegen-sh/graph_sitter/archive/{commit_hash}.zip" +Homepage = "https://github.com/codegen-sh/codegen-sdk" +source_archive = "https://github.com/codegen-sh/codegen-sdk/archive/{commit_hash}.zip" #[tool.hatch.build.targets.wheel.hooks.mypyc] #dependencies = ["hatch-mypyc", "pydantic"] diff --git a/conftest.py b/conftest.py index df3035936..0e6da30b0 100644 --- a/conftest.py +++ b/conftest.py @@ -1,8 +1,7 @@ import logging import os -from graph_sitter.core import autocommit -from graph_sitter.testing.models import Size +from codegen.sdk.testing.models import Size def find_dirs_to_ignore(start_dir, prefix): @@ -15,11 +14,6 @@ def find_dirs_to_ignore(start_dir, prefix): return dirs_to_ignore -collect_ignore_glob = ["codegen_tests/graph_sitter/codemod/**/*.py"] -if not autocommit.enabled: - collect_ignore_glob.append("**/test_autocommit.py") - - def pytest_addoption(parser): parser.addoption( "--size", diff --git a/docs/README.md b/docs/README.md index f5668a351..2af883ca8 100644 --- a/docs/README.md +++ b/docs/README.md @@ -1,7 +1,7 @@ # Codegen Docs This unites docs from: -- graph-sitter (this repo) +- codegen-sdk (this repo) - codegen-cli ## Development @@ -17,4 +17,4 @@ You should then get hot reloading. Also recommend installing the `MDX` extension ## Adding New Pages - Edit the page as a `.mdx` doc -- Make sure to edit `mint.json` so it appears in the UI \ No newline at end of file +- Make sure to edit `mint.json` so it appears in the UI diff --git a/docs/building-with-codegen/traversing-the-call-graph.mdx b/docs/building-with-codegen/traversing-the-call-graph.mdx index d17df5a8a..b02934f7b 100644 --- a/docs/building-with-codegen/traversing-the-call-graph.mdx +++ b/docs/building-with-codegen/traversing-the-call-graph.mdx @@ -34,48 +34,48 @@ Here's how to build a directed graph of function calls using NetworkX: ```python import networkx as nx -from graph_sitter.core.interfaces.callable import FunctionCallDefinition -from graph_sitter.core.function import Function +from codegen.sdk.core.interfaces.callable import FunctionCallDefinition +from codegen.sdk.core.function import Function def create_call_graph(start_func, end_func, max_depth=5): G = nx.DiGraph() - + def traverse_calls(parent_func, current_depth): if current_depth > max_depth: return - + # Determine source node if isinstance(parent_func, Function): src_call = src_func = parent_func else: src_func = parent_func.callables[0] src_call = parent_func.call - + # Traverse all function calls for func_call_def in src_func.call_graph_successors(): call = func_call_def.call func = func_call_def.callables[0] - + # Skip recursive calls if func.name == src_func.name: continue - + # Add nodes and edges G.add_node(call) G.add_edge(src_call, call) - + # Check if we reached the target if func == end_func: G.add_edge(call, end_func) return - + # Continue traversal traverse_calls(func_call_def, current_depth + 1) - + # Initialize graph G.add_node(start_func, color="blue") # Start node G.add_node(end_func, color="red") # End node - + # Start traversal traverse_calls(start_func, 1) return G @@ -139,7 +139,7 @@ Find the longest call chain in your codebase: ```python def get_max_call_chain(function): G = nx.DiGraph() - + def build_graph(func, depth=0): if depth > 10: # Prevent infinite recursion return @@ -147,7 +147,7 @@ def get_max_call_chain(function): called_func = successor.callables[0] G.add_edge(func, called_func) build_graph(called_func, depth + 1) - + build_graph(function) return nx.dag_longest_path(G) ``` diff --git a/docs/conftest.py b/docs/conftest.py index 9c14dc425..9294bfc31 100644 --- a/docs/conftest.py +++ b/docs/conftest.py @@ -6,11 +6,11 @@ from sybil.parsers.markdown import PythonCodeBlockParser from doctest import ELLIPSIS -from graph_sitter.code_generation.current_code_codebase import get_documented_objects -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.class_definition import TSClass -from graph_sitter.typescript.file import TSFile +from codegen.sdk.code_generation.current_code_codebase import get_documented_objects +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.class_definition import TSClass +from codegen.sdk.typescript.file import TSFile from gscli.generate.runner_imports import EXTERNAL_IMPORTS SAMPLE_FILENAME = { ProgrammingLanguage.PYTHON: "sample.py", diff --git a/hatch.toml b/hatch.toml index 6857a796d..03d1e3c62 100644 --- a/hatch.toml +++ b/hatch.toml @@ -1,6 +1,6 @@ [build.hooks.vcs] dependencies = ["hatch-vcs>=0.4.0"] -version-file = "src/graph_sitter/__init__.py" +version-file = "src/codegen/sdk/__init__.py" [metadata] allow-direct-references = true @@ -32,7 +32,7 @@ dependencies = [ ] [build.targets.wheel.hooks.cython.options] -src = "graph_sitter" +src = "codegen/sdk" compile_args = [ "-O3", { platforms = [ @@ -71,7 +71,7 @@ macos-max-compat = false [build] packages = [ - "src/graph_sitter", + "src/codegen/sdk", "src/graph_visualization", "src/gscli", ] diff --git a/mypy.ini b/mypy.ini index 4e4b0b8a3..4f89d4127 100644 --- a/mypy.ini +++ b/mypy.ini @@ -46,7 +46,7 @@ exclude = (?x)( | scripts/.* | .*/codemods/.*/test_.*/ | ^/expected/ - | src/graph_sitter/skills/* + | src/codegen/sdk/skills/* ) plugins = pydantic.mypy diff --git a/pyproject.toml b/pyproject.toml index 7e56d4bae..ba0f99b6a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,5 +1,5 @@ [project] -name = "graph-sitter" +name = "codegen-sdk" dynamic = ["version", "urls"] description = "Add your description here" readme = "README.md" @@ -62,7 +62,7 @@ classifiers = [ [project.scripts] gs = "gscli.main:main" -run_string = "graph_sitter.core.main:main" +run_string = "codegen.sdk.core.main:main" [project.optional-dependencies] types = [ "types-networkx>=3.2.1.20240918", diff --git a/scripts/profiling/apis.py b/scripts/profiling/apis.py index 9a01af6a3..ac61f7e97 100644 --- a/scripts/profiling/apis.py +++ b/scripts/profiling/apis.py @@ -6,8 +6,8 @@ import networkx as nx from tabulate import tabulate -from graph_sitter.codebase.factory.get_dev_customer_codebase import get_codebase_codegen -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_dev_customer_codebase import get_codebase_codegen +from codegen.sdk.enums import ProgrammingLanguage logging.basicConfig(level=logging.INFO) codegen = get_codebase_codegen("../codegen", ".") @@ -21,7 +21,7 @@ if "test" not in file.filepath: if file.ctx.repo_name == "codegen": color = "yellow" - elif file.ctx.repo_name == "graph_sitter": + elif file.ctx.repo_name == "codegen-sdk": color = "red" if file.ctx.base_path == "codegen-git": color = "green" diff --git a/src/graph_sitter/_proxy.py b/src/codegen/sdk/_proxy.py similarity index 94% rename from src/graph_sitter/_proxy.py rename to src/codegen/sdk/_proxy.py index aecb902fa..f50f49766 100644 --- a/src/graph_sitter/_proxy.py +++ b/src/codegen/sdk/_proxy.py @@ -6,7 +6,7 @@ from lazy_object_proxy.simple import make_proxy_method try: - from graph_sitter.extensions.utils import cached_property + from codegen.sdk.extensions.utils import cached_property except ModuleNotFoundError: from functools import cached_property diff --git a/src/graph_sitter/ai/converters.py b/src/codegen/sdk/ai/converters.py similarity index 100% rename from src/graph_sitter/ai/converters.py rename to src/codegen/sdk/ai/converters.py diff --git a/src/graph_sitter/ai/helpers.py b/src/codegen/sdk/ai/helpers.py similarity index 99% rename from src/graph_sitter/ai/helpers.py rename to src/codegen/sdk/ai/helpers.py index 26450e58f..e9d0634c3 100644 --- a/src/graph_sitter/ai/helpers.py +++ b/src/codegen/sdk/ai/helpers.py @@ -13,8 +13,8 @@ from openai import OpenAI from tenacity import retry, stop_after_attempt, wait_random_exponential -from graph_sitter.ai.converters import convert_openai_messages_to_claude -from graph_sitter.utils import XMLUtils +from codegen.sdk.ai.converters import convert_openai_messages_to_claude +from codegen.sdk.utils import XMLUtils CLAUDE_OPENAI_MODEL_MAP = { "gpt-4o": "claude-3-5-sonnet-20240620", diff --git a/src/graph_sitter/code_generation/__init__.py b/src/codegen/sdk/code_generation/__init__.py similarity index 100% rename from src/graph_sitter/code_generation/__init__.py rename to src/codegen/sdk/code_generation/__init__.py diff --git a/src/codegen/sdk/code_generation/codegen_sdk_codebase.py b/src/codegen/sdk/code_generation/codegen_sdk_codebase.py new file mode 100644 index 000000000..180b0d461 --- /dev/null +++ b/src/codegen/sdk/code_generation/codegen_sdk_codebase.py @@ -0,0 +1,16 @@ +import os.path + +from codegen.sdk.code_generation.current_code_codebase import get_codegen_codebase_base_path, get_current_code_codebase +from codegen.sdk.codebase.config import CodebaseConfig +from codegen.sdk.core.codebase import Codebase + + +def get_codegen_sdk_subdirectories() -> list[str]: + base = get_codegen_codebase_base_path() + return [os.path.join(base, "codegen/sdk"), os.path.join(base, "codemods")] + + +def get_codegen_sdk_codebase() -> Codebase: + """Grabs a Codebase w/ GraphSitter content. Responsible for figuring out where it is, e.g. in Modal or local""" + codebase = get_current_code_codebase(CodebaseConfig(), subdirectories=get_codegen_sdk_subdirectories()) + return codebase diff --git a/src/graph_sitter/code_generation/current_code_codebase.py b/src/codegen/sdk/code_generation/current_code_codebase.py similarity index 65% rename from src/graph_sitter/code_generation/current_code_codebase.py rename to src/codegen/sdk/code_generation/current_code_codebase.py index 33ee87b72..07f88ac7b 100644 --- a/src/graph_sitter/code_generation/current_code_codebase.py +++ b/src/codegen/sdk/code_generation/current_code_codebase.py @@ -4,31 +4,31 @@ from pathlib import Path from typing import TypedDict +from codegen.sdk.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig +from codegen.sdk.core.codebase import Codebase, CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.writer_decorators import DocumentedObject, apidoc_objects, no_apidoc_objects, py_apidoc_objects, ts_apidoc_objects from codegen_git.repo_operator.local_repo_operator import LocalRepoOperator from codegen_git.schemas.repo_config import BaseRepoConfig -from graph_sitter.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig -from graph_sitter.core.codebase import Codebase, CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.writer_decorators import DocumentedObject, apidoc_objects, no_apidoc_objects, py_apidoc_objects, ts_apidoc_objects logger = logging.getLogger(__name__) def get_graphsitter_repo_path() -> str: """Points to base directory of the Codegen repo (.git) that is currently running""" - import graph_sitter + import codegen.sdk as sdk - filepath = graph_sitter.__file__ - codegen_base_dir = filepath.replace("/graph_sitter/__init__.py", "") + filepath = sdk.__file__ + codegen_base_dir = filepath.replace("/codegen/sdk/__init__.py", "") codegen_base_dir = codegen_base_dir.replace("/src", "") return codegen_base_dir def get_codegen_codebase_base_path() -> str: - import graph_sitter + import codegen.sdk as sdk - filepath = graph_sitter.__file__ - codegen_base_dir = filepath.replace("/graph_sitter/__init__.py", "") + filepath = sdk.__file__ + codegen_base_dir = filepath.replace("/codegen/sdk/__init__.py", "") return "src" if "src" in codegen_base_dir else "" @@ -43,19 +43,19 @@ def get_current_code_codebase(config: CodebaseConfig = DefaultConfig, subdirecto return codebase -def import_all_graph_sitter_modules(): - # for file in graph_sitter: +def import_all_codegen_sdk_modules(): + # for file in codegen.sdk: - GRAPH_SITTER_DIR = Path(get_graphsitter_repo_path()) + CODEGEN_SDK_DIR = Path(get_graphsitter_repo_path()) if base := get_codegen_codebase_base_path(): - GRAPH_SITTER_DIR /= base - GRAPH_SITTER_DIR /= "graph_sitter" - for file in GRAPH_SITTER_DIR.rglob("*.py"): - relative_path = file.relative_to(GRAPH_SITTER_DIR) + CODEGEN_SDK_DIR /= base + CODEGEN_SDK_DIR /= "codegen/sdk" + for file in CODEGEN_SDK_DIR.rglob("*.py"): + relative_path = file.relative_to(CODEGEN_SDK_DIR) # ignore braintrust_evaluator because it runs stuff on import if "__init__" in file.name or "braintrust_evaluator" in file.name: continue - module_name = "graph_sitter." + str(relative_path).replace("/", ".").removesuffix(".py") + module_name = "codegen.sdk." + str(relative_path).replace("/", ".").removesuffix(".py") try: importlib.import_module(module_name) except Exception as e: @@ -71,16 +71,16 @@ class DocumentedObjects(TypedDict): def get_documented_objects() -> DocumentedObjects: """Get all the objects decorated with apidoc, py_apidoc, ts_apidoc, and no_apidoc decorators, - by importing all graph_sitter modules and keeping track of decorated objects at import time using + by importing all codegen.sdk modules and keeping track of decorated objects at import time using the respective decorators """ - import_all_graph_sitter_modules() - from graph_sitter.core.codebase import CodebaseType, PyCodebaseType, TSCodebaseType + import_all_codegen_sdk_modules() + from codegen.sdk.core.codebase import CodebaseType, PyCodebaseType, TSCodebaseType if PyCodebaseType not in apidoc_objects: - apidoc_objects.append(DocumentedObject(name="PyCodebaseType", module="graph_sitter.core.codebase", object=PyCodebaseType)) + apidoc_objects.append(DocumentedObject(name="PyCodebaseType", module="codegen.sdk.core.codebase", object=PyCodebaseType)) if TSCodebaseType not in apidoc_objects: - apidoc_objects.append(DocumentedObject(name="TSCodebaseType", module="graph_sitter.core.codebase", object=TSCodebaseType)) + apidoc_objects.append(DocumentedObject(name="TSCodebaseType", module="codegen.sdk.core.codebase", object=TSCodebaseType)) if CodebaseType not in apidoc_objects: - apidoc_objects.append(DocumentedObject(name="CodebaseType", module="graph_sitter.core.codebase", object=CodebaseType)) + apidoc_objects.append(DocumentedObject(name="CodebaseType", module="codegen.sdk.core.codebase", object=CodebaseType)) return {"apidoc": apidoc_objects, "py_apidoc": py_apidoc_objects, "ts_apidoc": ts_apidoc_objects, "no_apidoc": no_apidoc_objects} diff --git a/src/graph_sitter/code_generation/doc_utils/__init__.py b/src/codegen/sdk/code_generation/doc_utils/__init__.py similarity index 100% rename from src/graph_sitter/code_generation/doc_utils/__init__.py rename to src/codegen/sdk/code_generation/doc_utils/__init__.py diff --git a/src/graph_sitter/code_generation/doc_utils/canonicals.py b/src/codegen/sdk/code_generation/doc_utils/canonicals.py similarity index 89% rename from src/graph_sitter/code_generation/doc_utils/canonicals.py rename to src/codegen/sdk/code_generation/doc_utils/canonicals.py index 558233b6e..edcd0081f 100644 --- a/src/graph_sitter/code_generation/doc_utils/canonicals.py +++ b/src/codegen/sdk/code_generation/doc_utils/canonicals.py @@ -1,12 +1,12 @@ import textwrap -from graph_sitter.code_generation.doc_utils.utils import ( +from codegen.sdk.code_generation.doc_utils.utils import ( format_python_codeblock, ) -from graph_sitter.code_generation.enums import DocumentationDecorators -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python.class_definition import PyClass +from codegen.sdk.code_generation.enums import DocumentationDecorators +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python.class_definition import PyClass def get_canonical_codemod_classes(codebase: Codebase, language: ProgrammingLanguage) -> dict[str, PyClass]: diff --git a/src/graph_sitter/code_generation/doc_utils/skills.py b/src/codegen/sdk/code_generation/doc_utils/skills.py similarity index 92% rename from src/graph_sitter/code_generation/doc_utils/skills.py rename to src/codegen/sdk/code_generation/doc_utils/skills.py index 98729895a..6b7045b5f 100644 --- a/src/graph_sitter/code_generation/doc_utils/skills.py +++ b/src/codegen/sdk/code_generation/doc_utils/skills.py @@ -3,14 +3,14 @@ import inflection from typing_extensions import deprecated -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python import PyClass -from graph_sitter.python.function import PyFunction -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_implementation import SkillImplementation -from graph_sitter.skills.core.utils import get_all_evaluation_skills +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python import PyClass +from codegen.sdk.python.function import PyFunction +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_implementation import SkillImplementation +from codegen.sdk.skills.core.utils import get_all_evaluation_skills def remove_leading_tab_or_spaces(text: str) -> str: diff --git a/src/graph_sitter/code_generation/doc_utils/utils.py b/src/codegen/sdk/code_generation/doc_utils/utils.py similarity index 94% rename from src/graph_sitter/code_generation/doc_utils/utils.py rename to src/codegen/sdk/code_generation/doc_utils/utils.py index cc5a859e2..959f907f3 100644 --- a/src/graph_sitter/code_generation/doc_utils/utils.py +++ b/src/codegen/sdk/code_generation/doc_utils/utils.py @@ -1,13 +1,13 @@ import re from copy import deepcopy -from graph_sitter.code_generation.enums import DocumentationDecorators -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.function import Function -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import NodeType, ProgrammingLanguage -from graph_sitter.python.class_definition import PyClass -from graph_sitter.python.function import PyFunction +from codegen.sdk.code_generation.enums import DocumentationDecorators +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.function import Function +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import NodeType, ProgrammingLanguage +from codegen.sdk.python.class_definition import PyClass +from codegen.sdk.python.function import PyFunction def get_api_classes_by_decorator( @@ -106,7 +106,7 @@ def filter_undocumented_methods_list(doc_methods: list[Function]) -> list[Functi return filtered_doc_methods -def get_graph_sitter_class_docstring(cls: PyClass, codebase: Codebase) -> str: +def get_codegen_sdk_class_docstring(cls: PyClass, codebase: Codebase) -> str: """Get the documentation for a single GraphSitter class and its methods.""" # =====[ Parent classes ]===== parent_classes = cls.parent_class_names diff --git a/src/graph_sitter/code_generation/enums.py b/src/codegen/sdk/code_generation/enums.py similarity index 100% rename from src/graph_sitter/code_generation/enums.py rename to src/codegen/sdk/code_generation/enums.py diff --git a/src/graph_sitter/code_generation/mdx_docs_generation.py b/src/codegen/sdk/code_generation/mdx_docs_generation.py similarity index 95% rename from src/graph_sitter/code_generation/mdx_docs_generation.py rename to src/codegen/sdk/code_generation/mdx_docs_generation.py index c53c05cbe..5484eb330 100644 --- a/src/graph_sitter/code_generation/mdx_docs_generation.py +++ b/src/codegen/sdk/code_generation/mdx_docs_generation.py @@ -1,17 +1,17 @@ import textwrap -from graph_sitter.code_generation.doc_utils.utils import ( +from codegen.sdk.code_generation.doc_utils.utils import ( filter_undocumented_methods_list, get_all_classes_to_document, get_nearest_parent_docstring, is_property, sanitize_mdx_mintlify_desscription, ) -from graph_sitter.core.codebase import Codebase -from graph_sitter.python.class_definition import PyClass -from graph_sitter.python.detached_symbols.parameter import PyParameter -from graph_sitter.python.function import PyFunction -from graph_sitter.python.statements.attribute import PyAttribute +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.python.class_definition import PyClass +from codegen.sdk.python.detached_symbols.parameter import PyParameter +from codegen.sdk.python.function import PyFunction +from codegen.sdk.python.statements.attribute import PyAttribute def render_mdx_for_codebase_page(codebase: Codebase) -> str: diff --git a/src/graph_sitter/code_generation/prompts/__init__.py b/src/codegen/sdk/code_generation/prompts/__init__.py similarity index 100% rename from src/graph_sitter/code_generation/prompts/__init__.py rename to src/codegen/sdk/code_generation/prompts/__init__.py diff --git a/src/graph_sitter/code_generation/prompts/api_docs.py b/src/codegen/sdk/code_generation/prompts/api_docs.py similarity index 92% rename from src/graph_sitter/code_generation/prompts/api_docs.py rename to src/codegen/sdk/code_generation/prompts/api_docs.py index d23c03ab1..ce5567d87 100644 --- a/src/graph_sitter/code_generation/prompts/api_docs.py +++ b/src/codegen/sdk/code_generation/prompts/api_docs.py @@ -1,12 +1,12 @@ import logging -from graph_sitter.code_generation.doc_utils.utils import ( +from codegen.sdk.code_generation.codegen_sdk_codebase import get_codegen_sdk_codebase +from codegen.sdk.code_generation.doc_utils.utils import ( get_api_classes_by_decorator, - get_graph_sitter_class_docstring, + get_codegen_sdk_class_docstring, ) -from graph_sitter.code_generation.graph_sitter_codebase import get_graph_sitter_codebase -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage logger = logging.getLogger(__name__) @@ -20,7 +20,7 @@ def get_docstrings_for_classes(codebase: Codebase, language: ProgrammingLanguage, classnames: list[str]) -> dict[str, str]: """Returns map of ClassName -> Docstring""" classes = get_api_classes_by_decorator(codebase=codebase, language=language) - class_docstrings = {k: get_graph_sitter_class_docstring(cls=v, codebase=codebase) for k, v in classes.items()} + class_docstrings = {k: get_codegen_sdk_class_docstring(cls=v, codebase=codebase) for k, v in classes.items()} return {k: class_docstrings[k] for k in classnames} @@ -125,7 +125,7 @@ def get_language_specific_docstring(codebase: Codebase, language: ProgrammingLan # =====[ Grab docstrings ]===== classes = get_api_classes_by_decorator(codebase=codebase, language=language) - class_docstrings = {k: get_graph_sitter_class_docstring(cls=v, codebase=codebase) for k, v in classes.items()} + class_docstrings = {k: get_codegen_sdk_class_docstring(cls=v, codebase=codebase) for k, v in classes.items()} docstrings = {k: v for k, v in class_docstrings.items() if k.startswith(prefix)} # =====[ Get mapping from e.g. File => PyFile and TFile => PyFile ]===== @@ -150,14 +150,14 @@ def get_language_specific_docstring(codebase: Codebase, language: ProgrammingLan ######################################################################################################################## -def get_graph_sitter_docs(language: ProgrammingLanguage = ProgrammingLanguage.PYTHON, codebase: Codebase | None = None) -> str: +def get_codegen_sdk_docs(language: ProgrammingLanguage = ProgrammingLanguage.PYTHON, codebase: Codebase | None = None) -> str: """Computes the GraphSitter docs from scratch""" - codebase = codebase or get_graph_sitter_codebase() + codebase = codebase or get_codegen_sdk_codebase() with codebase.session(sync_graph=False, commit=False): return f""" -# GraphSitter Docs +# Codegen SDK Docs -GraphSitter is a Python SDK for writing powerful programs that operate on codebases. In essence, it is a scriptable, multi-language language server that is optimized for fast code transformations and analytics. +Codegen SDK is a Python SDK for writing powerful programs that operate on codebases. In essence, it is a scriptable, multi-language language server that is optimized for fast code transformations and analytics. Consider the following: ```python @@ -232,7 +232,7 @@ def get_graph_sitter_docs(language: ProgrammingLanguage = ProgrammingLanguage.PY - Hierarchical logging, with filenames in single quotes - You do not need to explain to the developer the code you are going to write before calling CREATE_CODEMOD - This will just make the code -- You *DO NOT* need to import `graph_sitter`, (this module does not exist) `codebase` or any types. +- You *DO NOT* need to import `codegen.sdk`, (this module does not exist) `codebase` or any types. - All types in the library are available in the global namespace and are automatically imported, as is the `codebase` object. - You *DO NOT* need to do anything to parse the codebase. - This is done automatically by the Codegen infrastructure and pre-cached for fast execution. Just interact with the `codebase` object. diff --git a/src/graph_sitter/codebase/__init__.py b/src/codegen/sdk/codebase/__init__.py similarity index 100% rename from src/graph_sitter/codebase/__init__.py rename to src/codegen/sdk/codebase/__init__.py diff --git a/src/graph_sitter/codebase/codebase_ai.py b/src/codegen/sdk/codebase/codebase_ai.py similarity index 99% rename from src/graph_sitter/codebase/codebase_ai.py rename to src/codegen/sdk/codebase/codebase_ai.py index 80004b631..658dbe23c 100644 --- a/src/graph_sitter/codebase/codebase_ai.py +++ b/src/codegen/sdk/codebase/codebase_ai.py @@ -1,5 +1,5 @@ -from graph_sitter.core.file import File -from graph_sitter.core.interfaces.editable import Editable +from codegen.sdk.core.file import File +from codegen.sdk.core.interfaces.editable import Editable def generate_system_prompt(target: Editable | None = None, context: None | str | Editable | list[Editable] | dict[str, str | Editable | list[Editable]] = None) -> str: diff --git a/src/graph_sitter/codebase/codebase_analysis.py b/src/codegen/sdk/codebase/codebase_analysis.py similarity index 89% rename from src/graph_sitter/codebase/codebase_analysis.py rename to src/codegen/sdk/codebase/codebase_analysis.py index 2d3b05ec8..f9dffe314 100644 --- a/src/graph_sitter/codebase/codebase_analysis.py +++ b/src/codegen/sdk/codebase/codebase_analysis.py @@ -1,11 +1,11 @@ -from graph_sitter.core.class_definition import Class -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.file import SourceFile -from graph_sitter.core.function import Function -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import EdgeType, SymbolType +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.file import SourceFile +from codegen.sdk.core.function import Function +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import EdgeType, SymbolType def get_codebase_summary(codebase: Codebase) -> str: diff --git a/src/graph_sitter/codebase/codebase_graph.py b/src/codegen/sdk/codebase/codebase_graph.py similarity index 94% rename from src/graph_sitter/codebase/codebase_graph.py rename to src/codegen/sdk/codebase/codebase_graph.py index 80a295e34..7bbf7aff5 100644 --- a/src/graph_sitter/codebase/codebase_graph.py +++ b/src/codegen/sdk/codebase/codebase_graph.py @@ -15,33 +15,33 @@ from git import Commit as GitCommit from rustworkx import PyDiGraph, WeightedEdgeList +from codegen.sdk.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig, SessionOptions +from codegen.sdk.codebase.config_parser import ConfigParser, get_config_parser_for_language +from codegen.sdk.codebase.control_flow import StopCodemodException +from codegen.sdk.codebase.diff_lite import ChangeType, DiffLite +from codegen.sdk.codebase.flagging.flags import Flags +from codegen.sdk.codebase.transaction_manager import TransactionManager +from codegen.sdk.codebase.validation import get_edges, post_reset_validation +from codegen.sdk.core.autocommit import AutoCommit, commiter +from codegen.sdk.core.dataclasses.usage import Usage +from codegen.sdk.core.directory import Directory +from codegen.sdk.core.external.dependency_manager import DependencyManager, get_dependency_manager +from codegen.sdk.core.external.language_engine import LanguageEngine, get_language_engine +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.enums import Edge, EdgeType, NodeType, ProgrammingLanguage +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.extensions.utils import uncache_all +from codegen.sdk.typescript.external.ts_declassify.ts_declassify import TSDeclassify from codegen_git.repo_operator.repo_operator import RepoOperator from codegen_git.utils.stopwatch_utils import stopwatch, stopwatch_with_sentry -from graph_sitter.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig, SessionOptions -from graph_sitter.codebase.config_parser import ConfigParser, get_config_parser_for_language -from graph_sitter.codebase.control_flow import StopCodemodException -from graph_sitter.codebase.diff_lite import ChangeType, DiffLite -from graph_sitter.codebase.flagging.flags import Flags -from graph_sitter.codebase.transaction_manager import TransactionManager -from graph_sitter.codebase.validation import get_edges, post_reset_validation -from graph_sitter.core.autocommit import AutoCommit, commiter -from graph_sitter.core.dataclasses.usage import Usage -from graph_sitter.core.directory import Directory -from graph_sitter.core.external.dependency_manager import DependencyManager, get_dependency_manager -from graph_sitter.core.external.language_engine import LanguageEngine, get_language_engine -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.enums import Edge, EdgeType, NodeType, ProgrammingLanguage -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.extensions.utils import uncache_all -from graph_sitter.typescript.external.ts_declassify.ts_declassify import TSDeclassify if TYPE_CHECKING: - from graph_sitter.codebase.node_classes.node_classes import NodeClasses - from graph_sitter.core.expressions import Expression - from graph_sitter.core.external_module import ExternalModule - from graph_sitter.core.file import SourceFile - from graph_sitter.core.parser import Parser + from codegen.sdk.codebase.node_classes.node_classes import NodeClasses + from codegen.sdk.core.expressions import Expression + from codegen.sdk.core.external_module import ExternalModule + from codegen.sdk.core.file import SourceFile + from codegen.sdk.core.parser import Parser import logging @@ -63,11 +63,11 @@ class SyncType(IntEnum): def get_node_classes(programming_language: ProgrammingLanguage) -> NodeClasses: if programming_language == ProgrammingLanguage.PYTHON: - from graph_sitter.codebase.node_classes.py_node_classes import PyNodeClasses + from codegen.sdk.codebase.node_classes.py_node_classes import PyNodeClasses return PyNodeClasses elif programming_language == ProgrammingLanguage.TYPESCRIPT: - from graph_sitter.codebase.node_classes.ts_node_classes import TSNodeClasses + from codegen.sdk.codebase.node_classes.ts_node_classes import TSNodeClasses return TSNodeClasses else: @@ -114,7 +114,7 @@ def __init__( config: CodebaseConfig = DefaultConfig, ) -> None: """Initializes codebase graph and TransactionManager""" - from graph_sitter.core.parser import Parser + from codegen.sdk.core.parser import Parser self._graph = PyDiGraph() self.filepath_idx = {} @@ -436,7 +436,7 @@ def _process_diff_files(self, files_to_sync: Mapping[SyncType, list[Path]], incr node.compute_export_dependencies() to_resolve.extend(node.symbol_usages) if counter[NodeType.SYMBOL] > 0: - from graph_sitter.core.interfaces.inherits import Inherits + from codegen.sdk.core.interfaces.inherits import Inherits logger.info("> Computing superclass dependencies") for symbol in to_resolve: diff --git a/src/graph_sitter/codebase/config.py b/src/codegen/sdk/codebase/config.py similarity index 96% rename from src/graph_sitter/codebase/config.py rename to src/codegen/sdk/codebase/config.py index 0c776ada1..fa105b306 100644 --- a/src/graph_sitter/codebase/config.py +++ b/src/codegen/sdk/codebase/config.py @@ -1,8 +1,8 @@ from pydantic import BaseModel, ConfigDict, Field +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.secrets import Secrets from codegen_git.repo_operator.repo_operator import RepoOperator -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.secrets import Secrets HARD_MAX_AI_LIMIT = 500 # Global limit for AI requests diff --git a/src/graph_sitter/codebase/config_parser.py b/src/codegen/sdk/codebase/config_parser.py similarity index 71% rename from src/graph_sitter/codebase/config_parser.py rename to src/codegen/sdk/codebase/config_parser.py index 5cec6e2ed..4f8ac6cc7 100644 --- a/src/graph_sitter/codebase/config_parser.py +++ b/src/codegen/sdk/codebase/config_parser.py @@ -1,10 +1,10 @@ from abc import ABC, abstractmethod from typing import TYPE_CHECKING -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.enums import ProgrammingLanguage if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph class ConfigParser(ABC): @@ -16,7 +16,7 @@ def parse_configs(self, codebase_graph: "CodebaseGraph"): ... def get_config_parser_for_language(language: ProgrammingLanguage, codebase_graph: "CodebaseGraph") -> ConfigParser | None: - from graph_sitter.typescript.config_parser import TSConfigParser + from codegen.sdk.typescript.config_parser import TSConfigParser if language == ProgrammingLanguage.TYPESCRIPT: return TSConfigParser(codebase_graph) diff --git a/src/graph_sitter/codebase/control_flow.py b/src/codegen/sdk/codebase/control_flow.py similarity index 100% rename from src/graph_sitter/codebase/control_flow.py rename to src/codegen/sdk/codebase/control_flow.py diff --git a/src/graph_sitter/codebase/diff_lite.py b/src/codegen/sdk/codebase/diff_lite.py similarity index 100% rename from src/graph_sitter/codebase/diff_lite.py rename to src/codegen/sdk/codebase/diff_lite.py diff --git a/src/graph_sitter/codebase/factory/codebase_factory.py b/src/codegen/sdk/codebase/factory/codebase_factory.py similarity index 87% rename from src/graph_sitter/codebase/factory/codebase_factory.py rename to src/codegen/sdk/codebase/factory/codebase_factory.py index e993cd1c1..3ce79bb47 100644 --- a/src/graph_sitter/codebase/factory/codebase_factory.py +++ b/src/codegen/sdk/codebase/factory/codebase_factory.py @@ -1,11 +1,11 @@ -from codegen_git.repo_operator.local_repo_operator import LocalRepoOperator -from codegen_git.schemas.repo_config import BaseRepoConfig -from graph_sitter.codebase.config import CodebaseConfig, ProjectConfig -from graph_sitter.core.codebase import ( +from codegen.sdk.codebase.config import CodebaseConfig, ProjectConfig +from codegen.sdk.core.codebase import ( Codebase, CodebaseType, ) -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.enums import ProgrammingLanguage +from codegen_git.repo_operator.local_repo_operator import LocalRepoOperator +from codegen_git.schemas.repo_config import BaseRepoConfig class CodebaseFactory: diff --git a/src/graph_sitter/codebase/factory/get_session.ipynb b/src/codegen/sdk/codebase/factory/get_session.ipynb similarity index 91% rename from src/graph_sitter/codebase/factory/get_session.ipynb rename to src/codegen/sdk/codebase/factory/get_session.ipynb index b22c8c554..e703e7b7e 100644 --- a/src/graph_sitter/codebase/factory/get_session.ipynb +++ b/src/codegen/sdk/codebase/factory/get_session.ipynb @@ -7,7 +7,7 @@ "outputs": [], "source": [ "# test get_codebase_session\n", - "from graph_sitter.codebase.factory.get_session import get_codebase_session\n", + "from codegen.sdk.codebase.factory.get_session import get_codebase_session\n", "import time\n", "\n", "with get_codebase_session(tmpdir=f\"/tmp/{int(time.time())}\", files={\"file.py\": \"a = 1 + 2\"}) as codebase:\n", diff --git a/src/graph_sitter/codebase/factory/get_session.py b/src/codegen/sdk/codebase/factory/get_session.py similarity index 88% rename from src/graph_sitter/codebase/factory/get_session.py rename to src/codegen/sdk/codebase/factory/get_session.py index d46c3d4a0..722b30b00 100644 --- a/src/graph_sitter/codebase/factory/get_session.py +++ b/src/codegen/sdk/codebase/factory/get_session.py @@ -4,15 +4,15 @@ from contextlib import AbstractContextManager, contextmanager from typing import Any, Literal, overload +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.codebase.config import CodebaseConfig, GSFeatureFlags, ProjectConfig, SessionOptions, TestFlags +from codegen.sdk.codebase.factory.codebase_factory import CodebaseFactory +from codegen.sdk.core.codebase import Codebase, PyCodebaseType, TSCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.secrets import Secrets +from codegen.sdk.tree_sitter_parser import print_errors from codegen_git.repo_operator.local_repo_operator import LocalRepoOperator from codegen_git.schemas.repo_config import BaseRepoConfig -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.codebase.config import CodebaseConfig, GSFeatureFlags, ProjectConfig, SessionOptions, TestFlags -from graph_sitter.codebase.factory.codebase_factory import CodebaseFactory -from graph_sitter.core.codebase import Codebase, PyCodebaseType, TSCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.secrets import Secrets -from graph_sitter.tree_sitter_parser import print_errors @overload diff --git a/src/graph_sitter/codebase/flagging/code_flag.py b/src/codegen/sdk/codebase/flagging/code_flag.py similarity index 89% rename from src/graph_sitter/codebase/flagging/code_flag.py rename to src/codegen/sdk/codebase/flagging/code_flag.py index bd308415e..cb10a0057 100644 --- a/src/graph_sitter/codebase/flagging/code_flag.py +++ b/src/codegen/sdk/codebase/flagging/code_flag.py @@ -1,10 +1,10 @@ from dataclasses import dataclass from typing import TYPE_CHECKING -from graph_sitter.codebase.flagging.enums import MessageType +from codegen.sdk.codebase.flagging.enums import MessageType if TYPE_CHECKING: - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.interfaces.editable import Editable @dataclass diff --git a/src/graph_sitter/codebase/flagging/enums.py b/src/codegen/sdk/codebase/flagging/enums.py similarity index 91% rename from src/graph_sitter/codebase/flagging/enums.py rename to src/codegen/sdk/codebase/flagging/enums.py index 3da82959a..6375484b0 100644 --- a/src/graph_sitter/codebase/flagging/enums.py +++ b/src/codegen/sdk/codebase/flagging/enums.py @@ -3,7 +3,7 @@ from typing_extensions import ReadOnly -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.writer_decorators import apidoc @apidoc diff --git a/src/graph_sitter/codebase/flagging/flags.py b/src/codegen/sdk/codebase/flagging/flags.py similarity index 92% rename from src/graph_sitter/codebase/flagging/flags.py rename to src/codegen/sdk/codebase/flagging/flags.py index f7e942cbb..1c6332d89 100644 --- a/src/graph_sitter/codebase/flagging/flags.py +++ b/src/codegen/sdk/codebase/flagging/flags.py @@ -1,10 +1,10 @@ from dataclasses import dataclass, field from typing import TYPE_CHECKING -from graph_sitter.codebase.flagging.code_flag import CodeFlag -from graph_sitter.codebase.flagging.enums import MessageType -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk.codebase.flagging.code_flag import CodeFlag +from codegen.sdk.codebase.flagging.enums import MessageType +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: from app.codemod.types import Group diff --git a/src/graph_sitter/codebase/multigraph.py b/src/codegen/sdk/codebase/multigraph.py similarity index 71% rename from src/graph_sitter/codebase/multigraph.py rename to src/codegen/sdk/codebase/multigraph.py index 7e053cefd..735fc01d1 100644 --- a/src/graph_sitter/codebase/multigraph.py +++ b/src/codegen/sdk/codebase/multigraph.py @@ -1,11 +1,11 @@ from collections import defaultdict from dataclasses import dataclass, field -from graph_sitter import TYPE_CHECKING -from graph_sitter.core.detached_symbols.function_call import FunctionCall +from codegen.sdk import TYPE_CHECKING +from codegen.sdk.core.detached_symbols.function_call import FunctionCall if TYPE_CHECKING: - from graph_sitter.core.function import Function + from codegen.sdk.core.function import Function @dataclass diff --git a/src/graph_sitter/codebase/node_classes/__init__.py b/src/codegen/sdk/codebase/node_classes/__init__.py similarity index 100% rename from src/graph_sitter/codebase/node_classes/__init__.py rename to src/codegen/sdk/codebase/node_classes/__init__.py diff --git a/src/graph_sitter/codebase/node_classes/node_classes.py b/src/codegen/sdk/codebase/node_classes/node_classes.py similarity index 60% rename from src/graph_sitter/codebase/node_classes/node_classes.py rename to src/codegen/sdk/codebase/node_classes/node_classes.py index 781597ba5..579063c3a 100644 --- a/src/graph_sitter/codebase/node_classes/node_classes.py +++ b/src/codegen/sdk/codebase/node_classes/node_classes.py @@ -4,20 +4,20 @@ from functools import cached_property from typing import TYPE_CHECKING -from graph_sitter.core.interfaces.resolvable import Resolvable +from codegen.sdk.core.interfaces.resolvable import Resolvable if TYPE_CHECKING: - from graph_sitter.core.class_definition import Class - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.detached_symbols.function_call import FunctionCall - from graph_sitter.core.detached_symbols.parameter import Parameter - from graph_sitter.core.expressions import Expression - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.file import SourceFile - from graph_sitter.core.function import Function - from graph_sitter.core.import_resolution import Import - from graph_sitter.core.statements.comment import Comment - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.function_call import FunctionCall + from codegen.sdk.core.detached_symbols.parameter import Parameter + from codegen.sdk.core.expressions import Expression + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.file import SourceFile + from codegen.sdk.core.function import Function + from codegen.sdk.core.import_resolution import Import + from codegen.sdk.core.statements.comment import Comment + from codegen.sdk.core.symbol import Symbol @dataclass diff --git a/src/graph_sitter/codebase/node_classes/py_node_classes.py b/src/codegen/sdk/codebase/node_classes/py_node_classes.py similarity index 55% rename from src/graph_sitter/codebase/node_classes/py_node_classes.py rename to src/codegen/sdk/codebase/node_classes/py_node_classes.py index a29a49d0e..0bdb86b3e 100644 --- a/src/graph_sitter/codebase/node_classes/py_node_classes.py +++ b/src/codegen/sdk/codebase/node_classes/py_node_classes.py @@ -1,34 +1,34 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.node_classes.node_classes import NodeClasses -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import String, Type -from graph_sitter.core.expressions.await_expression import AwaitExpression -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.expressions.boolean import Boolean -from graph_sitter.core.expressions.comparison_expression import ComparisonExpression -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.expressions.none_type import NoneType -from graph_sitter.core.expressions.number import Number -from graph_sitter.core.expressions.parenthesized_expression import ParenthesizedExpression -from graph_sitter.core.expressions.subscript_expression import SubscriptExpression -from graph_sitter.core.expressions.unary_expression import UnaryExpression -from graph_sitter.core.expressions.unpack import Unpack -from graph_sitter.core.statements.comment import Comment -from graph_sitter.core.symbol_groups.dict import Dict -from graph_sitter.core.symbol_groups.list import List -from graph_sitter.core.symbol_groups.tuple import Tuple -from graph_sitter.core.symbol_groups.type_parameters import TypeParameters -from graph_sitter.python import PyClass, PyFile, PyFunction, PyImport, PySymbol -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.detached_symbols.parameter import PyParameter -from graph_sitter.python.expressions.chained_attribute import PyChainedAttribute -from graph_sitter.python.expressions.conditional_expression import PyConditionalExpression -from graph_sitter.python.expressions.generic_type import PyGenericType -from graph_sitter.python.expressions.named_type import PyNamedType -from graph_sitter.python.expressions.string import PyString -from graph_sitter.python.expressions.union_type import PyUnionType -from graph_sitter.python.statements.import_statement import PyImportStatement +from codegen.sdk.codebase.node_classes.node_classes import NodeClasses +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import String, Type +from codegen.sdk.core.expressions.await_expression import AwaitExpression +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.expressions.boolean import Boolean +from codegen.sdk.core.expressions.comparison_expression import ComparisonExpression +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.expressions.none_type import NoneType +from codegen.sdk.core.expressions.number import Number +from codegen.sdk.core.expressions.parenthesized_expression import ParenthesizedExpression +from codegen.sdk.core.expressions.subscript_expression import SubscriptExpression +from codegen.sdk.core.expressions.unary_expression import UnaryExpression +from codegen.sdk.core.expressions.unpack import Unpack +from codegen.sdk.core.statements.comment import Comment +from codegen.sdk.core.symbol_groups.dict import Dict +from codegen.sdk.core.symbol_groups.list import List +from codegen.sdk.core.symbol_groups.tuple import Tuple +from codegen.sdk.core.symbol_groups.type_parameters import TypeParameters +from codegen.sdk.python import PyClass, PyFile, PyFunction, PyImport, PySymbol +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.detached_symbols.parameter import PyParameter +from codegen.sdk.python.expressions.chained_attribute import PyChainedAttribute +from codegen.sdk.python.expressions.conditional_expression import PyConditionalExpression +from codegen.sdk.python.expressions.generic_type import PyGenericType +from codegen.sdk.python.expressions.named_type import PyNamedType +from codegen.sdk.python.expressions.string import PyString +from codegen.sdk.python.expressions.union_type import PyUnionType +from codegen.sdk.python.statements.import_statement import PyImportStatement def parse_subscript(node: TSNode, file_node_id, G, parent): diff --git a/src/graph_sitter/codebase/node_classes/ts_node_classes.py b/src/codegen/sdk/codebase/node_classes/ts_node_classes.py similarity index 54% rename from src/graph_sitter/codebase/node_classes/ts_node_classes.py rename to src/codegen/sdk/codebase/node_classes/ts_node_classes.py index e35355c28..5ac4c3a8b 100644 --- a/src/graph_sitter/codebase/node_classes/ts_node_classes.py +++ b/src/codegen/sdk/codebase/node_classes/ts_node_classes.py @@ -1,53 +1,53 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.node_classes.node_classes import NodeClasses -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions.await_expression import AwaitExpression -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.expressions.boolean import Boolean -from graph_sitter.core.expressions.defined_name import DefinedName -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.expressions.none_type import NoneType -from graph_sitter.core.expressions.number import Number -from graph_sitter.core.expressions.parenthesized_expression import ParenthesizedExpression -from graph_sitter.core.expressions.subscript_expression import SubscriptExpression -from graph_sitter.core.expressions.tuple_type import TupleType -from graph_sitter.core.expressions.unary_expression import UnaryExpression -from graph_sitter.core.expressions.unpack import Unpack -from graph_sitter.core.expressions.value import Value -from graph_sitter.core.statements.comment import Comment -from graph_sitter.core.symbol_groups.list import List -from graph_sitter.core.symbol_groups.type_parameters import TypeParameters -from graph_sitter.typescript.class_definition import TSClass -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.detached_symbols.jsx.element import JSXElement -from graph_sitter.typescript.detached_symbols.jsx.expression import JSXExpression -from graph_sitter.typescript.detached_symbols.parameter import TSParameter -from graph_sitter.typescript.enum_definition import TSEnum -from graph_sitter.typescript.enums import TSFunctionTypeNames -from graph_sitter.typescript.expressions.array_type import TSArrayType -from graph_sitter.typescript.expressions.chained_attribute import TSChainedAttribute -from graph_sitter.typescript.expressions.conditional_type import TSConditionalType -from graph_sitter.typescript.expressions.function_type import TSFunctionType -from graph_sitter.typescript.expressions.generic_type import TSGenericType -from graph_sitter.typescript.expressions.lookup_type import TSLookupType -from graph_sitter.typescript.expressions.named_type import TSNamedType -from graph_sitter.typescript.expressions.object_type import TSObjectType -from graph_sitter.typescript.expressions.query_type import TSQueryType -from graph_sitter.typescript.expressions.readonly_type import TSReadonlyType -from graph_sitter.typescript.expressions.string import TSString -from graph_sitter.typescript.expressions.ternary_expression import TSTernaryExpression -from graph_sitter.typescript.expressions.undefined_type import TSUndefinedType -from graph_sitter.typescript.expressions.union_type import TSUnionType -from graph_sitter.typescript.file import TSFile -from graph_sitter.typescript.function import TSFunction -from graph_sitter.typescript.import_resolution import TSImport -from graph_sitter.typescript.interface import TSInterface -from graph_sitter.typescript.namespace import TSNamespace -from graph_sitter.typescript.statements.comment import TSComment -from graph_sitter.typescript.statements.import_statement import TSImportStatement -from graph_sitter.typescript.symbol_groups.dict import TSDict -from graph_sitter.typescript.type_alias import TSTypeAlias +from codegen.sdk.codebase.node_classes.node_classes import NodeClasses +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions.await_expression import AwaitExpression +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.expressions.boolean import Boolean +from codegen.sdk.core.expressions.defined_name import DefinedName +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.expressions.none_type import NoneType +from codegen.sdk.core.expressions.number import Number +from codegen.sdk.core.expressions.parenthesized_expression import ParenthesizedExpression +from codegen.sdk.core.expressions.subscript_expression import SubscriptExpression +from codegen.sdk.core.expressions.tuple_type import TupleType +from codegen.sdk.core.expressions.unary_expression import UnaryExpression +from codegen.sdk.core.expressions.unpack import Unpack +from codegen.sdk.core.expressions.value import Value +from codegen.sdk.core.statements.comment import Comment +from codegen.sdk.core.symbol_groups.list import List +from codegen.sdk.core.symbol_groups.type_parameters import TypeParameters +from codegen.sdk.typescript.class_definition import TSClass +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.detached_symbols.jsx.element import JSXElement +from codegen.sdk.typescript.detached_symbols.jsx.expression import JSXExpression +from codegen.sdk.typescript.detached_symbols.parameter import TSParameter +from codegen.sdk.typescript.enum_definition import TSEnum +from codegen.sdk.typescript.enums import TSFunctionTypeNames +from codegen.sdk.typescript.expressions.array_type import TSArrayType +from codegen.sdk.typescript.expressions.chained_attribute import TSChainedAttribute +from codegen.sdk.typescript.expressions.conditional_type import TSConditionalType +from codegen.sdk.typescript.expressions.function_type import TSFunctionType +from codegen.sdk.typescript.expressions.generic_type import TSGenericType +from codegen.sdk.typescript.expressions.lookup_type import TSLookupType +from codegen.sdk.typescript.expressions.named_type import TSNamedType +from codegen.sdk.typescript.expressions.object_type import TSObjectType +from codegen.sdk.typescript.expressions.query_type import TSQueryType +from codegen.sdk.typescript.expressions.readonly_type import TSReadonlyType +from codegen.sdk.typescript.expressions.string import TSString +from codegen.sdk.typescript.expressions.ternary_expression import TSTernaryExpression +from codegen.sdk.typescript.expressions.undefined_type import TSUndefinedType +from codegen.sdk.typescript.expressions.union_type import TSUnionType +from codegen.sdk.typescript.file import TSFile +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.typescript.import_resolution import TSImport +from codegen.sdk.typescript.interface import TSInterface +from codegen.sdk.typescript.namespace import TSNamespace +from codegen.sdk.typescript.statements.comment import TSComment +from codegen.sdk.typescript.statements.import_statement import TSImportStatement +from codegen.sdk.typescript.symbol_groups.dict import TSDict +from codegen.sdk.typescript.type_alias import TSTypeAlias def parse_dict(node: TSNode, *args): diff --git a/src/graph_sitter/codebase/range_index.py b/src/codegen/sdk/codebase/range_index.py similarity index 93% rename from src/graph_sitter/codebase/range_index.py rename to src/codegen/sdk/codebase/range_index.py index 342622c4f..ea8cdd4b9 100644 --- a/src/graph_sitter/codebase/range_index.py +++ b/src/codegen/sdk/codebase/range_index.py @@ -4,8 +4,8 @@ from tree_sitter import Range -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.extensions.sort import sort_editables +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.extensions.sort import sort_editables class RangeIndex: diff --git a/src/codegen/sdk/codebase/resolution_stack.py b/src/codegen/sdk/codebase/resolution_stack.py new file mode 100644 index 000000000..ccce6d38a --- /dev/null +++ b/src/codegen/sdk/codebase/resolution_stack.py @@ -0,0 +1,3 @@ +from codegen.sdk.extensions.resolution import ResolutionStack + +__all__ = ["ResolutionStack"] diff --git a/src/graph_sitter/codebase/span.py b/src/codegen/sdk/codebase/span.py similarity index 97% rename from src/graph_sitter/codebase/span.py rename to src/codegen/sdk/codebase/span.py index 658000e16..59aa175b2 100644 --- a/src/graph_sitter/codebase/span.py +++ b/src/codegen/sdk/codebase/span.py @@ -4,7 +4,7 @@ from pydantic_core.core_schema import ValidationInfo from tree_sitter import Point, Range -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.writer_decorators import apidoc def validate_range(value: Any, info: ValidationInfo) -> Range: diff --git a/src/graph_sitter/codebase/transaction_manager.py b/src/codegen/sdk/codebase/transaction_manager.py similarity index 98% rename from src/graph_sitter/codebase/transaction_manager.py rename to src/codegen/sdk/codebase/transaction_manager.py index a0d9e2115..1ea184a1e 100644 --- a/src/graph_sitter/codebase/transaction_manager.py +++ b/src/codegen/sdk/codebase/transaction_manager.py @@ -3,9 +3,9 @@ from pathlib import Path from typing import TYPE_CHECKING -from graph_sitter.codebase.control_flow import MaxPreviewTimeExceeded, MaxTransactionsExceeded -from graph_sitter.codebase.diff_lite import DiffLite -from graph_sitter.codebase.transactions import ( +from codegen.sdk.codebase.control_flow import MaxPreviewTimeExceeded, MaxTransactionsExceeded +from codegen.sdk.codebase.diff_lite import DiffLite +from codegen.sdk.codebase.transactions import ( EditTransaction, FileAddTransaction, FileRemoveTransaction, @@ -16,7 +16,7 @@ ) if TYPE_CHECKING: - from graph_sitter.core.file import File + from codegen.sdk.core.file import File import logging diff --git a/src/graph_sitter/codebase/transactions.py b/src/codegen/sdk/codebase/transactions.py similarity index 98% rename from src/graph_sitter/codebase/transactions.py rename to src/codegen/sdk/codebase/transactions.py index 938cab2d5..7fcdfd8c1 100644 --- a/src/graph_sitter/codebase/transactions.py +++ b/src/codegen/sdk/codebase/transactions.py @@ -6,10 +6,10 @@ from pathlib import Path from typing import TYPE_CHECKING, Protocol, runtime_checkable -from graph_sitter.codebase.diff_lite import ChangeType, DiffLite +from codegen.sdk.codebase.diff_lite import ChangeType, DiffLite if TYPE_CHECKING: - from graph_sitter.core.file import File + from codegen.sdk.core.file import File class TransactionPriority(IntEnum): diff --git a/src/graph_sitter/codebase/validation.py b/src/codegen/sdk/codebase/validation.py similarity index 96% rename from src/graph_sitter/codebase/validation.py rename to src/codegen/sdk/codebase/validation.py index fee6af826..3ce05da38 100644 --- a/src/graph_sitter/codebase/validation.py +++ b/src/codegen/sdk/codebase/validation.py @@ -10,13 +10,13 @@ from rustworkx import PyDiGraph from tabulate import tabulate -from graph_sitter.enums import NodeType -from graph_sitter.utils import truncate_line +from codegen.sdk.enums import NodeType +from codegen.sdk.utils import truncate_line logger = logging.getLogger(__name__) if TYPE_CHECKING: - from graph_sitter.core.codebase import CodebaseType + from codegen.sdk.core.codebase import CodebaseType class PostInitValidationStatus(StrEnum): @@ -131,7 +131,7 @@ def get_nodes_error(init_nodes, nodes): ``` """ for node in set_nodes - set_init_nodes: - from graph_sitter.core.external_module import ExternalModule + from codegen.sdk.core.external_module import ExternalModule if isinstance(node, ExternalModule): message += "External Module persisted with following dependencies: " + str(list((node.G.get_node(source), edge) for source, _, edge in node.G.in_edges(node.node_id))) diff --git a/src/graph_sitter/codemod.py b/src/codegen/sdk/codemod.py similarity index 100% rename from src/graph_sitter/codemod.py rename to src/codegen/sdk/codemod.py diff --git a/src/graph_sitter/core/__init__.py b/src/codegen/sdk/core/__init__.py similarity index 100% rename from src/graph_sitter/core/__init__.py rename to src/codegen/sdk/core/__init__.py diff --git a/src/graph_sitter/core/assignment.py b/src/codegen/sdk/core/assignment.py similarity index 84% rename from src/graph_sitter/core/assignment.py rename to src/codegen/sdk/core/assignment.py index a150864ba..ea792ce5c 100644 --- a/src/graph_sitter/core/assignment.py +++ b/src/codegen/sdk/core/assignment.py @@ -5,37 +5,37 @@ from tree_sitter import Node as TSNode -from graph_sitter._proxy import proxy_property -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression, Name -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.expressions.multi_expression import MultiExpression -from graph_sitter.core.expressions.subscript_expression import SubscriptExpression -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.typeable import Typeable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.core.symbol_groups.dict import Dict -from graph_sitter.enums import SymbolType -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.typescript.expressions.object_type import TSObjectType -from graph_sitter.utils import find_index -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk._proxy import proxy_property +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression, Name +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.expressions.multi_expression import MultiExpression +from codegen.sdk.core.expressions.subscript_expression import SubscriptExpression +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.typeable import Typeable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.core.symbol_groups.dict import Dict +from codegen.sdk.enums import SymbolType +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.typescript.expressions.object_type import TSObjectType +from codegen.sdk.utils import find_index +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.statements.assignment_statement import AssignmentStatement - from graph_sitter.core.statements.export_statement import ExportStatement + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.statements.assignment_statement import AssignmentStatement + from codegen.sdk.core.statements.export_statement import ExportStatement Parent = TypeVar("Parent", bound="AssignmentStatement | ExportStatement") @@ -139,7 +139,7 @@ def is_local_variable(self) -> bool: Returns: bool: True if the assignment is a local variable, False otherwise. """ - from graph_sitter.core.file import File + from codegen.sdk.core.file import File if isinstance(self._left, ChainedAttribute): return False @@ -222,7 +222,7 @@ def _resolved_types(self) -> Generator[ResolutionStack[Self], None, None]: yield from self.with_resolution_frame(self.type, direct=False) elif self.value: resolved = False - from graph_sitter.core.statements.assignment_statement import AssignmentStatement + from codegen.sdk.core.statements.assignment_statement import AssignmentStatement if self.parent_of_type(AssignmentStatement) and len(self.parent_of_type(AssignmentStatement).assignments) > 0: name_node = self._name_node.ts_node diff --git a/src/graph_sitter/core/autocommit/__init__.py b/src/codegen/sdk/core/autocommit/__init__.py similarity index 80% rename from src/graph_sitter/core/autocommit/__init__.py rename to src/codegen/sdk/core/autocommit/__init__.py index 49e77dfa1..fabd6f4bd 100644 --- a/src/graph_sitter/core/autocommit/__init__.py +++ b/src/codegen/sdk/core/autocommit/__init__.py @@ -17,9 +17,9 @@ - We skip commits when you do raw edits and inserts, but will fall back to autocommit if needed """ -from graph_sitter.core.autocommit.constants import enabled -from graph_sitter.core.autocommit.decorators import mover, remover, repr_func, writer -from graph_sitter.core.autocommit.manager import AutoCommit -from graph_sitter.extensions.autocommit import commiter, reader +from codegen.sdk.core.autocommit.constants import enabled +from codegen.sdk.core.autocommit.decorators import mover, remover, repr_func, writer +from codegen.sdk.core.autocommit.manager import AutoCommit +from codegen.sdk.extensions.autocommit import commiter, reader __all__ = ["AutoCommit", "commiter", "enabled", "mover", "reader", "remover", "repr_func", "writer"] diff --git a/src/graph_sitter/core/autocommit/constants.py b/src/codegen/sdk/core/autocommit/constants.py similarity index 93% rename from src/graph_sitter/core/autocommit/constants.py rename to src/codegen/sdk/core/autocommit/constants.py index 73ecb1bde..2d3b50969 100644 --- a/src/graph_sitter/core/autocommit/constants.py +++ b/src/codegen/sdk/core/autocommit/constants.py @@ -2,7 +2,7 @@ from typing import TYPE_CHECKING if TYPE_CHECKING: - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.interfaces.editable import Editable REMOVED = "REMOVED" @@ -38,7 +38,7 @@ class OutdatedNodeError(Exception): def __init__(self, node: "Editable") -> None: parent = node - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.symbol import Symbol while parent is not None and not isinstance(parent, Symbol): parent = parent.parent diff --git a/src/graph_sitter/core/autocommit/decorators.py b/src/codegen/sdk/core/autocommit/decorators.py similarity index 93% rename from src/graph_sitter/core/autocommit/decorators.py rename to src/codegen/sdk/core/autocommit/decorators.py index aace8b9fb..6a5c88342 100644 --- a/src/graph_sitter/core/autocommit/decorators.py +++ b/src/codegen/sdk/core/autocommit/decorators.py @@ -3,13 +3,12 @@ from typing import TYPE_CHECKING, ParamSpec, TypeVar, Union, overload import wrapt - -from graph_sitter.core.autocommit.constants import AutoCommitState, enabled -from graph_sitter.core.node_id_factory import NodeId +from codegen.sdk.core.autocommit.constants import AutoCommitState, enabled +from codegen.sdk.core.node_id_factory import NodeId if TYPE_CHECKING: - from graph_sitter.core.interfaces.editable import Editable - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.interfaces.editable import Editable + from codegen.sdk.core.symbol import Symbol import logging diff --git a/src/graph_sitter/core/autocommit/manager.py b/src/codegen/sdk/core/autocommit/manager.py similarity index 94% rename from src/graph_sitter/core/autocommit/manager.py rename to src/codegen/sdk/core/autocommit/manager.py index 985e6f31c..52a72d678 100644 --- a/src/graph_sitter/core/autocommit/manager.py +++ b/src/codegen/sdk/core/autocommit/manager.py @@ -4,16 +4,16 @@ from pathlib import Path from typing import TYPE_CHECKING, Optional, Union -from graph_sitter.core.autocommit.constants import REMOVED, AutoCommitState, AutoCommitSymbol, IllegalWriteError, NodeNotFoundError, OutdatedNodeError -from graph_sitter.core.autocommit.utils import is_file, is_on_graph, is_symbol -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.extensions.autocommit import update_dict +from codegen.sdk.core.autocommit.constants import REMOVED, AutoCommitState, AutoCommitSymbol, IllegalWriteError, NodeNotFoundError, OutdatedNodeError +from codegen.sdk.core.autocommit.utils import is_file, is_on_graph, is_symbol +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.extensions.autocommit import update_dict if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.file import File - from graph_sitter.core.import_resolution import Import - from graph_sitter.core.symbol import Symbol + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.file import File + from codegen.sdk.core.import_resolution import Import + from codegen.sdk.core.symbol import Symbol import logging diff --git a/src/codegen/sdk/core/autocommit/ruff.toml b/src/codegen/sdk/core/autocommit/ruff.toml new file mode 100644 index 000000000..0291e8e20 --- /dev/null +++ b/src/codegen/sdk/core/autocommit/ruff.toml @@ -0,0 +1,2 @@ +extend = "../../../../../pyproject.toml" +lint.extend-select = ["G"] diff --git a/src/graph_sitter/core/autocommit/utils.py b/src/codegen/sdk/core/autocommit/utils.py similarity index 53% rename from src/graph_sitter/core/autocommit/utils.py rename to src/codegen/sdk/core/autocommit/utils.py index 1213b0c32..a7b9e5c84 100644 --- a/src/graph_sitter/core/autocommit/utils.py +++ b/src/codegen/sdk/core/autocommit/utils.py @@ -3,25 +3,25 @@ from typing import TYPE_CHECKING, Any, TypeGuard, Union if TYPE_CHECKING: - from graph_sitter.core.file import File - from graph_sitter.core.import_resolution import Import - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.file import File + from codegen.sdk.core.import_resolution import Import + from codegen.sdk.core.symbol import Symbol def is_file(node: Any) -> TypeGuard["File"]: - from graph_sitter.core.file import File + from codegen.sdk.core.file import File return isinstance(node, File) def is_symbol(node: Any) -> TypeGuard["Symbol"]: - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.symbol import Symbol return isinstance(node, Symbol) def is_on_graph(node: Any) -> TypeGuard[Union["Import", "Symbol"]]: - from graph_sitter.core.import_resolution import Import - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.import_resolution import Import + from codegen.sdk.core.symbol import Symbol return isinstance(node, Import | Symbol) diff --git a/src/graph_sitter/core/class_definition.py b/src/codegen/sdk/core/class_definition.py similarity index 90% rename from src/graph_sitter/core/class_definition.py rename to src/codegen/sdk/core/class_definition.py index 866975bd0..a2095fb5a 100644 --- a/src/graph_sitter/core/class_definition.py +++ b/src/codegen/sdk/core/class_definition.py @@ -5,37 +5,37 @@ from tree_sitter import Node as TSNode -from graph_sitter._proxy import proxy_property -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.detached_symbols.decorator import Decorator -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.expressions import Name -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.function import Function -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.interface import Interface -from graph_sitter.core.interfaces.callable import Callable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.interfaces.inherits import Inherits -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.attribute import Attribute -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.core.statements.symbol_statement import SymbolStatement -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.symbol_groups.multi_line_collection import MultiLineCollection -from graph_sitter.core.symbol_groups.parents import Parents -from graph_sitter.enums import SymbolType -from graph_sitter.extensions.utils import cached_property -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk._proxy import proxy_property +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.detached_symbols.decorator import Decorator +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.expressions import Name +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.function import Function +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.interface import Interface +from codegen.sdk.core.interfaces.callable import Callable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.interfaces.inherits import Inherits +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.attribute import Attribute +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.core.statements.symbol_statement import SymbolStatement +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection +from codegen.sdk.core.symbol_groups.parents import Parents +from codegen.sdk.enums import SymbolType +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.writer_decorators import apidoc, noapidoc from graph_visualization.enums import VizNode if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.expressions.type import Type + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.expressions.type import Type import logging diff --git a/src/graph_sitter/core/codebase.py b/src/codegen/sdk/core/codebase.py similarity index 93% rename from src/graph_sitter/core/codebase.py rename to src/codegen/sdk/core/codebase.py index 6c049b4b5..9b0894406 100644 --- a/src/graph_sitter/core/codebase.py +++ b/src/codegen/sdk/core/codebase.py @@ -19,59 +19,59 @@ from rich.console import Console from typing_extensions import deprecated +from codegen.sdk._proxy import proxy_property +from codegen.sdk.ai.helpers import AbstractAIHelper, MultiProviderAIHelper +from codegen.sdk.codebase.codebase_ai import generate_system_prompt, generate_tools +from codegen.sdk.codebase.codebase_graph import GLOBAL_FILE_IGNORE_LIST, CodebaseGraph +from codegen.sdk.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig, SessionOptions +from codegen.sdk.codebase.control_flow import MaxAIRequestsError +from codegen.sdk.codebase.diff_lite import DiffLite +from codegen.sdk.codebase.flagging.code_flag import CodeFlag +from codegen.sdk.codebase.flagging.enums import FlagKwargs +from codegen.sdk.codebase.span import Span +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.directory import Directory +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.file import File, SourceFile +from codegen.sdk.core.function import Function +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.interface import Interface +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.type_alias import TypeAlias +from codegen.sdk.enums import NodeType, ProgrammingLanguage, SymbolType +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.extensions.utils import uncache_all +from codegen.sdk.output.constants import ANGULAR_STYLE +from codegen.sdk.python.assignment import PyAssignment +from codegen.sdk.python.class_definition import PyClass +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.detached_symbols.parameter import PyParameter +from codegen.sdk.python.file import PyFile +from codegen.sdk.python.function import PyFunction +from codegen.sdk.python.import_resolution import PyImport +from codegen.sdk.python.symbol import PySymbol +from codegen.sdk.typescript.assignment import TSAssignment +from codegen.sdk.typescript.class_definition import TSClass +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.detached_symbols.parameter import TSParameter +from codegen.sdk.typescript.file import TSFile +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.typescript.import_resolution import TSImport +from codegen.sdk.typescript.interface import TSInterface +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.typescript.type_alias import TSTypeAlias +from codegen.sdk.utils import determine_project_language +from codegen.sdk.writer_decorators import apidoc, noapidoc from codegen_git.repo_operator.local_repo_operator import LocalRepoOperator from codegen_git.repo_operator.repo_operator import RepoOperator from codegen_git.schemas.enums import CheckoutResult from codegen_git.schemas.repo_config import BaseRepoConfig from codegen_git.utils.stopwatch_utils import stopwatch -from graph_sitter._proxy import proxy_property -from graph_sitter.ai.helpers import AbstractAIHelper, MultiProviderAIHelper -from graph_sitter.codebase.codebase_ai import generate_system_prompt, generate_tools -from graph_sitter.codebase.codebase_graph import GLOBAL_FILE_IGNORE_LIST, CodebaseGraph -from graph_sitter.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig, SessionOptions -from graph_sitter.codebase.control_flow import MaxAIRequestsError -from graph_sitter.codebase.diff_lite import DiffLite -from graph_sitter.codebase.flagging.code_flag import CodeFlag -from graph_sitter.codebase.flagging.enums import FlagKwargs -from graph_sitter.codebase.span import Span -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.class_definition import Class -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.directory import Directory -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.file import File, SourceFile -from graph_sitter.core.function import Function -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.interface import Interface -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.type_alias import TypeAlias -from graph_sitter.enums import NodeType, ProgrammingLanguage, SymbolType -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.extensions.utils import uncache_all -from graph_sitter.output.constants import ANGULAR_STYLE -from graph_sitter.python.assignment import PyAssignment -from graph_sitter.python.class_definition import PyClass -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.detached_symbols.parameter import PyParameter -from graph_sitter.python.file import PyFile -from graph_sitter.python.function import PyFunction -from graph_sitter.python.import_resolution import PyImport -from graph_sitter.python.symbol import PySymbol -from graph_sitter.typescript.assignment import TSAssignment -from graph_sitter.typescript.class_definition import TSClass -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.detached_symbols.parameter import TSParameter -from graph_sitter.typescript.file import TSFile -from graph_sitter.typescript.function import TSFunction -from graph_sitter.typescript.import_resolution import TSImport -from graph_sitter.typescript.interface import TSInterface -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.typescript.type_alias import TSTypeAlias -from graph_sitter.utils import determine_project_language -from graph_sitter.writer_decorators import apidoc, noapidoc from graph_visualization.visualization_manager import VisualizationManager if TYPE_CHECKING: @@ -948,7 +948,7 @@ def session(self, sync_graph: bool = True, commit: bool = True, session_options: def _enable_experimental_language_engine(self, async_start: bool = False, install_deps: bool = False, use_v8: bool = False) -> None: """Debug option to enable experimental language engine for the current codebase.""" if install_deps and not self.G.language_engine: - from graph_sitter.core.external.dependency_manager import get_dependency_manager + from codegen.sdk.core.external.dependency_manager import get_dependency_manager logger.info("Cold installing dependencies...") logger.info("This may take a while for large repos...") @@ -958,7 +958,7 @@ def _enable_experimental_language_engine(self, async_start: bool = False, instal self.G.dependency_manager.wait_until_ready(ignore_error=False) logger.info("Dependencies ready") if not self.G.language_engine: - from graph_sitter.core.external.language_engine import get_language_engine + from codegen.sdk.core.external.language_engine import get_language_engine logger.info("Cold starting language engine...") logger.info("This may take a while for large repos...") diff --git a/src/graph_sitter/core/dataclasses/usage.py b/src/codegen/sdk/core/dataclasses/usage.py similarity index 83% rename from src/graph_sitter/core/dataclasses/usage.py rename to src/codegen/sdk/core/dataclasses/usage.py index c5de777f1..d217186b8 100644 --- a/src/graph_sitter/core/dataclasses/usage.py +++ b/src/codegen/sdk/core/dataclasses/usage.py @@ -6,16 +6,16 @@ from dataclasses_json import dataclass_json -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.function_call import FunctionCall - from graph_sitter.core.export import Export - from graph_sitter.core.expressions import Name - from graph_sitter.core.expressions.chained_attribute import ChainedAttribute - from graph_sitter.core.file import SourceFile - from graph_sitter.core.import_resolution import Import - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.detached_symbols.function_call import FunctionCall + from codegen.sdk.core.export import Export + from codegen.sdk.core.expressions import Name + from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute + from codegen.sdk.core.file import SourceFile + from codegen.sdk.core.import_resolution import Import + from codegen.sdk.core.symbol import Symbol @apidoc diff --git a/src/graph_sitter/core/detached_symbols/__init__.py b/src/codegen/sdk/core/detached_symbols/__init__.py similarity index 100% rename from src/graph_sitter/core/detached_symbols/__init__.py rename to src/codegen/sdk/core/detached_symbols/__init__.py diff --git a/src/graph_sitter/core/detached_symbols/argument.py b/src/codegen/sdk/core/detached_symbols/argument.py similarity index 85% rename from src/graph_sitter/core/detached_symbols/argument.py rename to src/codegen/sdk/core/detached_symbols/argument.py index 65e3e0759..bf72dc559 100644 --- a/src/graph_sitter/core/detached_symbols/argument.py +++ b/src/codegen/sdk/core/detached_symbols/argument.py @@ -4,22 +4,22 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.multi_expression import MultiExpression -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.multi_expression import MultiExpression +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.detached_symbols.function_call import FunctionCall + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.detached_symbols.function_call import FunctionCall Parent = TypeVar("Parent", bound="FunctionCall") diff --git a/src/graph_sitter/core/detached_symbols/code_block.py b/src/codegen/sdk/core/detached_symbols/code_block.py similarity index 94% rename from src/graph_sitter/core/detached_symbols/code_block.py rename to src/codegen/sdk/core/detached_symbols/code_block.py index 992c3fe11..d13720f59 100644 --- a/src/graph_sitter/core/detached_symbols/code_block.py +++ b/src/codegen/sdk/core/detached_symbols/code_block.py @@ -8,29 +8,29 @@ from tree_sitter import Node as TSNode from typing_extensions import deprecated -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind, UsageType -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression, Value -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.statements.assignment_statement import AssignmentStatement -from graph_sitter.core.statements.attribute import Attribute -from graph_sitter.core.statements.return_statement import ReturnStatement -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.core.statements.symbol_statement import SymbolStatement -from graph_sitter.core.symbol_groups.multi_line_collection import MultiLineCollection -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.extensions.utils import find_line_start_and_end_nodes -from graph_sitter.output.ast import AST -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind, UsageType +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression, Value +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.statements.assignment_statement import AssignmentStatement +from codegen.sdk.core.statements.attribute import Attribute +from codegen.sdk.core.statements.return_statement import ReturnStatement +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.core.statements.symbol_statement import SymbolStatement +from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.extensions.utils import find_line_start_and_end_nodes +from codegen.sdk.output.ast import AST +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.interfaces.has_block import HasBlock - from graph_sitter.core.statements.comment import Comment - from graph_sitter.core.statements.if_block_statement import IfBlockStatement + from codegen.sdk.core.interfaces.has_block import HasBlock + from codegen.sdk.core.statements.comment import Comment + from codegen.sdk.core.statements.if_block_statement import IfBlockStatement Parent = TypeVar("Parent", bound="HasBlock") diff --git a/src/graph_sitter/core/detached_symbols/decorator.py b/src/codegen/sdk/core/detached_symbols/decorator.py similarity index 72% rename from src/graph_sitter/core/detached_symbols/decorator.py rename to src/codegen/sdk/core/detached_symbols/decorator.py index eb1df1b7c..0c561cc76 100644 --- a/src/graph_sitter/core/detached_symbols/decorator.py +++ b/src/codegen/sdk/core/detached_symbols/decorator.py @@ -5,18 +5,18 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.class_definition import Class - from graph_sitter.core.detached_symbols.parameter import Parameter - from graph_sitter.core.function import Function + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.detached_symbols.parameter import Parameter + from codegen.sdk.core.function import Function TClass = TypeVar("TClass", bound="Class") diff --git a/src/graph_sitter/core/detached_symbols/function_call.py b/src/codegen/sdk/core/detached_symbols/function_call.py similarity index 92% rename from src/graph_sitter/core/detached_symbols/function_call.py rename to src/codegen/sdk/core/detached_symbols/function_call.py index 91999bafe..a6a476e35 100644 --- a/src/graph_sitter/core/detached_symbols/function_call.py +++ b/src/codegen/sdk/core/detached_symbols/function_call.py @@ -5,32 +5,32 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader, remover, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.argument import Argument -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.expressions import Expression, Name, Value -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.expressions.generic_type import GenericType -from graph_sitter.core.expressions.unpack import Unpack -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.resolvable import Resolvable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.enums import NodeType, ProgrammingLanguage -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.extensions.utils import cached_property, is_descendant_of -from graph_sitter.typescript.enums import TSFunctionTypeNames -from graph_sitter.utils import find_first_ancestor -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader, remover, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.argument import Argument +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.expressions import Expression, Name, Value +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.expressions.generic_type import GenericType +from codegen.sdk.core.expressions.unpack import Unpack +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.resolvable import Resolvable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.enums import NodeType, ProgrammingLanguage +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.extensions.utils import cached_property, is_descendant_of +from codegen.sdk.typescript.enums import TSFunctionTypeNames +from codegen.sdk.utils import find_first_ancestor +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.function import Function - from graph_sitter.core.interfaces.callable import Callable + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.function import Function + from codegen.sdk.core.interfaces.callable import Callable from graph_visualization.enums import VizNode Parent = TypeVar("Parent", bound="Expression | None") @@ -92,7 +92,7 @@ def parent_function(self) -> Function | None: # HACK: This is temporary until we establish a full parent path if self.file.programming_language == ProgrammingLanguage.TYPESCRIPT: if func := find_first_ancestor(self.ts_node, [function_type.value for function_type in TSFunctionTypeNames]): - from graph_sitter.typescript.function import TSFunction + from codegen.sdk.typescript.function import TSFunction return TSFunction.from_function_type(func, self.file_node_id, self.G, self.parent) elif self.file.programming_language == ProgrammingLanguage.PYTHON: @@ -219,7 +219,7 @@ def viz(self) -> VizNode: from graph_visualization.enums import VizNode func = self.function_definition - from graph_sitter.core.function import Function + from codegen.sdk.core.function import Function if isinstance(func, Function) and func.is_method: name = f"{func.parent_class.name}.{self.name}" @@ -301,7 +301,7 @@ def set_kwarg(self, name: str, value: str, *, create_on_missing: bool = True, ov @noapidoc @reader def find_parameter_by_index(self, index: int) -> Parameter | None: - from graph_sitter.python import PyFunction + from codegen.sdk.python import PyFunction for function_definition in self.function_definitions: if function_definition.node_type == NodeType.EXTERNAL or function_definition.parameters is None: @@ -392,7 +392,7 @@ def convert_args_to_kwargs(self, exclude: int = 0) -> None: - Requires the function definition to be resolvable and have parameters """ definition = self.function_definition - from graph_sitter.core.interfaces.callable import Callable + from codegen.sdk.core.interfaces.callable import Callable if definition is None or definition.parameters is None or not isinstance(definition, Callable): return @@ -418,8 +418,8 @@ def convert_args_to_kwargs(self, exclude: int = 0) -> None: @reader @noapidoc def function_definition_frames(self) -> list[ResolutionStack[Callable]]: - from graph_sitter.core.class_definition import Class - from graph_sitter.core.interfaces.callable import Callable + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.interfaces.callable import Callable result = [] for resolution in self.get_name().resolved_type_frames: @@ -486,8 +486,8 @@ def remove(self, delete_formatting: bool = True, priority: int = 0, dedupe: bool @noapidoc @override def _resolved_types(self) -> Generator[ResolutionStack[Self], None, None]: - from graph_sitter.core.class_definition import Class - from graph_sitter.core.function import Function + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.function import Function if self.get_name().ts_node.type == "import" or self.full_name == "require": # TS imports diff --git a/src/graph_sitter/core/detached_symbols/parameter.py b/src/codegen/sdk/core/detached_symbols/parameter.py similarity index 89% rename from src/graph_sitter/core/detached_symbols/parameter.py rename to src/codegen/sdk/core/detached_symbols/parameter.py index c7a2294e8..52431695a 100644 --- a/src/graph_sitter/core/detached_symbols/parameter.py +++ b/src/codegen/sdk/core/detached_symbols/parameter.py @@ -7,25 +7,25 @@ from tree_sitter import Node as TSNode from typing_extensions import deprecated -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.typeable import Typeable -from graph_sitter.core.interfaces.usable import Usable -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.extensions.resolution import UsageKind -from graph_sitter.utils import find_first_descendant -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.typeable import Typeable +from codegen.sdk.core.interfaces.usable import Usable +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.extensions.resolution import UsageKind +from codegen.sdk.utils import find_first_descendant +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.function import Function + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.function import Function import logging diff --git a/src/graph_sitter/core/directory.py b/src/codegen/sdk/core/directory.py similarity index 94% rename from src/graph_sitter/core/directory.py rename to src/codegen/sdk/core/directory.py index ff370b9ac..a81d06f48 100644 --- a/src/graph_sitter/core/directory.py +++ b/src/codegen/sdk/core/directory.py @@ -3,15 +3,15 @@ from pathlib import Path from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.core.assignment import Assignment - from graph_sitter.core.class_definition import Class - from graph_sitter.core.file import File - from graph_sitter.core.function import Function - from graph_sitter.core.import_resolution import ImportStatement - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.assignment import Assignment + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.file import File + from codegen.sdk.core.function import Function + from codegen.sdk.core.import_resolution import ImportStatement + from codegen.sdk.core.symbol import Symbol import logging @@ -170,7 +170,7 @@ def remove_file_by_path(self, file_path: os.PathLike) -> None: def get_file(self, filename: str, ignore_case: bool = False) -> TFile | None: """Get a file by its name relative to the directory.""" - from graph_sitter.core.file import File + from codegen.sdk.core.file import File if ignore_case: return next((f for name, f in self.items.items() if name.lower() == filename.lower() and isinstance(f, File)), None) diff --git a/src/graph_sitter/core/export.py b/src/codegen/sdk/core/export.py similarity index 85% rename from src/graph_sitter/core/export.py rename to src/codegen/sdk/core/export.py index ebb863921..21d7a88a1 100644 --- a/src/graph_sitter/core/export.py +++ b/src/codegen/sdk/core/export.py @@ -5,15 +5,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.interfaces.exportable import Exportable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.interfaces.exportable import Exportable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.statements.export_statement import ExportStatement + from codegen.sdk.core.statements.export_statement import ExportStatement Parent = TypeVar("Parent", bound="Collection[Export, ExportStatement]") @@ -81,7 +81,7 @@ def is_aliased(self) -> bool: Returns: bool: True if the exported symbol has a different name than the name it is exported as, False otherwise. """ - from graph_sitter.core.import_resolution import Import + from codegen.sdk.core.import_resolution import Import if self.exported_symbol is None: return False diff --git a/src/codegen/sdk/core/expressions/__init__.py b/src/codegen/sdk/core/expressions/__init__.py new file mode 100644 index 000000000..b2add957e --- /dev/null +++ b/src/codegen/sdk/core/expressions/__init__.py @@ -0,0 +1,13 @@ +from typing import TYPE_CHECKING + +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.expressions.string import String +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.expressions.value import Value +from codegen.sdk.core.symbol_groups.dict import Dict +from codegen.sdk.core.symbol_groups.list import List + +if TYPE_CHECKING: + from codegen.sdk.core.detached_symbols.function_call import FunctionCall +__all__ = ["Dict", "Expression", "FunctionCall", "List", "Name", "String", "Type", "Value"] diff --git a/src/graph_sitter/core/expressions/await_expression.py b/src/codegen/sdk/core/expressions/await_expression.py similarity index 68% rename from src/graph_sitter/core/expressions/await_expression.py rename to src/codegen/sdk/core/expressions/await_expression.py index c3dab3bb8..85ee3a1fe 100644 --- a/src/graph_sitter/core/expressions/await_expression.py +++ b/src/codegen/sdk/core/expressions/await_expression.py @@ -1,12 +1,12 @@ from typing import Generic, TypeVar -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.wrapper_expression import IWrapper -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.wrapper_expression import IWrapper +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/expressions/binary_expression.py b/src/codegen/sdk/core/expressions/binary_expression.py similarity index 85% rename from src/graph_sitter/core/expressions/binary_expression.py rename to src/codegen/sdk/core/expressions/binary_expression.py index 29d7a005f..ead1f48f7 100644 --- a/src/graph_sitter/core/expressions/binary_expression.py +++ b/src/codegen/sdk/core/expressions/binary_expression.py @@ -4,19 +4,19 @@ from functools import cached_property from typing import Generic, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.unwrappable import Unwrappable -from graph_sitter.core.symbol_groups.expression_group import ExpressionGroup -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.unwrappable import Unwrappable +from codegen.sdk.core.symbol_groups.expression_group import ExpressionGroup +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.writer_decorators import apidoc, noapidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/expressions/boolean.py b/src/codegen/sdk/core/expressions/boolean.py similarity index 56% rename from src/graph_sitter/core/expressions/boolean.py rename to src/codegen/sdk/core/expressions/boolean.py index 57e213fa7..fc4ad74fe 100644 --- a/src/graph_sitter/core/expressions/boolean.py +++ b/src/codegen/sdk/core/expressions/boolean.py @@ -1,11 +1,11 @@ from typing import Generic, TypeVar, override -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.builtin import Builtin -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.builtin import Builtin +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc Parent = TypeVar("Parent", bound="Expression") diff --git a/src/graph_sitter/core/expressions/builtin.py b/src/codegen/sdk/core/expressions/builtin.py similarity index 60% rename from src/graph_sitter/core/expressions/builtin.py rename to src/codegen/sdk/core/expressions/builtin.py index 96f25dced..fd4feddae 100644 --- a/src/graph_sitter/core/expressions/builtin.py +++ b/src/codegen/sdk/core/expressions/builtin.py @@ -1,14 +1,14 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Self, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: - from graph_sitter.core.external_module import ExternalModule + from codegen.sdk.core.external_module import ExternalModule @noapidoc diff --git a/src/graph_sitter/core/expressions/chained_attribute.py b/src/codegen/sdk/core/expressions/chained_attribute.py similarity index 87% rename from src/graph_sitter/core/expressions/chained_attribute.py rename to src/codegen/sdk/core/expressions/chained_attribute.py index fa5517108..f071db786 100644 --- a/src/graph_sitter/core/expressions/chained_attribute.py +++ b/src/codegen/sdk/core/expressions/chained_attribute.py @@ -3,20 +3,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Name -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.core.interfaces.resolvable import Resolvable -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Name +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.core.interfaces.resolvable import Resolvable +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.interfaces.has_name import HasName - from graph_sitter.core.interfaces.importable import Importable + from codegen.sdk.core.interfaces.has_name import HasName + from codegen.sdk.core.interfaces.importable import Importable Object = TypeVar("Object", bound="Chainable") Attribute = TypeVar("Attribute", bound="Resolvable") diff --git a/src/graph_sitter/core/expressions/comparison_expression.py b/src/codegen/sdk/core/expressions/comparison_expression.py similarity index 88% rename from src/graph_sitter/core/expressions/comparison_expression.py rename to src/codegen/sdk/core/expressions/comparison_expression.py index a920ed587..dfed264c4 100644 --- a/src/graph_sitter/core/expressions/comparison_expression.py +++ b/src/codegen/sdk/core/expressions/comparison_expression.py @@ -1,10 +1,10 @@ from functools import cached_property from typing import Self, TypeVar -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.symbol_groups.expression_group import ExpressionGroup -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.symbol_groups.expression_group import ExpressionGroup +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent") diff --git a/src/graph_sitter/core/expressions/defined_name.py b/src/codegen/sdk/core/expressions/defined_name.py similarity index 63% rename from src/graph_sitter/core/expressions/defined_name.py rename to src/codegen/sdk/core/expressions/defined_name.py index 4609558ea..47d21684f 100644 --- a/src/graph_sitter/core/expressions/defined_name.py +++ b/src/codegen/sdk/core/expressions/defined_name.py @@ -1,13 +1,13 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.expressions import Name -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.expressions import Name +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.symbol import Symbol Parent = TypeVar("Parent", bound="Symbol") diff --git a/src/graph_sitter/core/expressions/expression.py b/src/codegen/sdk/core/expressions/expression.py similarity index 81% rename from src/graph_sitter/core/expressions/expression.py rename to src/codegen/sdk/core/expressions/expression.py index c8c1651b3..8e5f1ad0a 100644 --- a/src/graph_sitter/core/expressions/expression.py +++ b/src/codegen/sdk/core/expressions/expression.py @@ -2,11 +2,11 @@ from typing import Generic, TypeVar -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.enums import NodeType -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.enums import NodeType +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/expressions/generic_type.py b/src/codegen/sdk/core/expressions/generic_type.py similarity index 73% rename from src/graph_sitter/core/expressions/generic_type.py rename to src/codegen/sdk/core/expressions/generic_type.py index 3cac31737..93fe5e487 100644 --- a/src/graph_sitter/core/expressions/generic_type.py +++ b/src/codegen/sdk/core/expressions/generic_type.py @@ -4,20 +4,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.named_type import NamedType -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.resolution import ResolutionStack -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.named_type import NamedType +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.resolution import ResolutionStack +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.interfaces.importable import Importable + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.interfaces.importable import Importable Parent = TypeVar("Parent") diff --git a/src/graph_sitter/core/expressions/multi_expression.py b/src/codegen/sdk/core/expressions/multi_expression.py similarity index 70% rename from src/graph_sitter/core/expressions/multi_expression.py rename to src/codegen/sdk/core/expressions/multi_expression.py index 6e30b110c..bad2fa5e2 100644 --- a/src/graph_sitter/core/expressions/multi_expression.py +++ b/src/codegen/sdk/core/expressions/multi_expression.py @@ -4,15 +4,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound="Expression") diff --git a/src/graph_sitter/core/expressions/name.py b/src/codegen/sdk/core/expressions/name.py similarity index 73% rename from src/graph_sitter/core/expressions/name.py rename to src/codegen/sdk/core/expressions/name.py index 83f7ef04a..4e825cc44 100644 --- a/src/graph_sitter/core/expressions/name.py +++ b/src/codegen/sdk/core/expressions/name.py @@ -1,16 +1,16 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Optional, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.resolvable import Resolvable -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.resolvable import Resolvable +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.interfaces.has_name import HasName + from codegen.sdk.core.interfaces.has_name import HasName Parent = TypeVar("Parent", bound="Expression") diff --git a/src/graph_sitter/core/expressions/named_type.py b/src/codegen/sdk/core/expressions/named_type.py similarity index 71% rename from src/graph_sitter/core/expressions/named_type.py rename to src/codegen/sdk/core/expressions/named_type.py index 7146451d7..17461d8a3 100644 --- a/src/graph_sitter/core/expressions/named_type.py +++ b/src/codegen/sdk/core/expressions/named_type.py @@ -4,20 +4,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Name, String -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.resolvable import Resolvable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Name, String +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.resolvable import Resolvable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound="Editable") @@ -31,7 +31,7 @@ def __init__(self, ts_node: TSNode, file_node_id: NodeId, G: "CodebaseGraph", pa self._name_node = self._parse_expression(self._get_name_node(), default=Name) def __eq__(self, other: object) -> bool: - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.symbol import Symbol if isinstance(other, Symbol): for resolved in self.resolved_types: diff --git a/src/graph_sitter/core/expressions/none_type.py b/src/codegen/sdk/core/expressions/none_type.py similarity index 56% rename from src/graph_sitter/core/expressions/none_type.py rename to src/codegen/sdk/core/expressions/none_type.py index b675701c0..30e4b6d42 100644 --- a/src/graph_sitter/core/expressions/none_type.py +++ b/src/codegen/sdk/core/expressions/none_type.py @@ -1,13 +1,13 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/core/expressions/number.py b/src/codegen/sdk/core/expressions/number.py similarity index 51% rename from src/graph_sitter/core/expressions/number.py rename to src/codegen/sdk/core/expressions/number.py index 434ff7678..d6440a4ee 100644 --- a/src/graph_sitter/core/expressions/number.py +++ b/src/codegen/sdk/core/expressions/number.py @@ -1,11 +1,11 @@ from typing import Generic, TypeVar, override -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.builtin import Builtin -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.builtin import Builtin +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc Parent = TypeVar("Parent", bound="Expression") diff --git a/src/graph_sitter/core/expressions/parenthesized_expression.py b/src/codegen/sdk/core/expressions/parenthesized_expression.py similarity index 79% rename from src/graph_sitter/core/expressions/parenthesized_expression.py rename to src/codegen/sdk/core/expressions/parenthesized_expression.py index f37d84e9d..56bbac7f5 100644 --- a/src/graph_sitter/core/expressions/parenthesized_expression.py +++ b/src/codegen/sdk/core/expressions/parenthesized_expression.py @@ -1,16 +1,16 @@ from typing import Generic, TypeVar, override -from graph_sitter.codebase.transactions import TransactionPriority -from graph_sitter.core.autocommit import writer -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.unwrappable import Unwrappable -from graph_sitter.core.interfaces.wrapper_expression import IWrapper -from graph_sitter.extensions.autocommit import reader -from graph_sitter.typescript.statements.if_block_statement import TSIfBlockStatement -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.codebase.transactions import TransactionPriority +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.unwrappable import Unwrappable +from codegen.sdk.core.interfaces.wrapper_expression import IWrapper +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.typescript.statements.if_block_statement import TSIfBlockStatement +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/expressions/placeholder_type.py b/src/codegen/sdk/core/expressions/placeholder_type.py similarity index 58% rename from src/graph_sitter/core/expressions/placeholder_type.py rename to src/codegen/sdk/core/expressions/placeholder_type.py index a74f59435..e932a247c 100644 --- a/src/graph_sitter/core/expressions/placeholder_type.py +++ b/src/codegen/sdk/core/expressions/placeholder_type.py @@ -1,14 +1,14 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import commiter -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import commiter +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/core/expressions/string.py b/src/codegen/sdk/core/expressions/string.py similarity index 77% rename from src/graph_sitter/core/expressions/string.py rename to src/codegen/sdk/core/expressions/string.py index 7cbd1f475..b1225dd07 100644 --- a/src/graph_sitter/core/expressions/string.py +++ b/src/codegen/sdk/core/expressions/string.py @@ -2,19 +2,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.builtin import Builtin -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.builtin import Builtin +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound="Expression") diff --git a/src/graph_sitter/core/expressions/subscript_expression.py b/src/codegen/sdk/core/expressions/subscript_expression.py similarity index 70% rename from src/graph_sitter/core/expressions/subscript_expression.py rename to src/codegen/sdk/core/expressions/subscript_expression.py index ac84ce38b..41ff7f5fc 100644 --- a/src/graph_sitter/core/expressions/subscript_expression.py +++ b/src/codegen/sdk/core/expressions/subscript_expression.py @@ -1,18 +1,18 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Optional, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Name -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.resolvable import Resolvable -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Name +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.resolvable import Resolvable +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.interfaces.has_name import HasName + from codegen.sdk.core.interfaces.has_name import HasName Object = TypeVar("Object", bound="Chainable") diff --git a/src/graph_sitter/core/expressions/ternary_expression.py b/src/codegen/sdk/core/expressions/ternary_expression.py similarity index 78% rename from src/graph_sitter/core/expressions/ternary_expression.py rename to src/codegen/sdk/core/expressions/ternary_expression.py index 58fdc0da3..5918f6650 100644 --- a/src/graph_sitter/core/expressions/ternary_expression.py +++ b/src/codegen/sdk/core/expressions/ternary_expression.py @@ -2,17 +2,17 @@ from collections.abc import Generator from typing import Generic, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.unwrappable import Unwrappable -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.unwrappable import Unwrappable +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.writer_decorators import apidoc, noapidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/expressions/tuple_type.py b/src/codegen/sdk/core/expressions/tuple_type.py similarity index 74% rename from src/graph_sitter/core/expressions/tuple_type.py rename to src/codegen/sdk/core/expressions/tuple_type.py index 453fbfe5d..5eddf7127 100644 --- a/src/graph_sitter/core/expressions/tuple_type.py +++ b/src/codegen/sdk/core/expressions/tuple_type.py @@ -3,17 +3,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.interfaces.importable import Importable + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.interfaces.importable import Importable TType = TypeVar("TType", bound="Type") diff --git a/src/graph_sitter/core/expressions/type.py b/src/codegen/sdk/core/expressions/type.py similarity index 68% rename from src/graph_sitter/core/expressions/type.py rename to src/codegen/sdk/core/expressions/type.py index 1f3b28c3d..d5bea5a62 100644 --- a/src/graph_sitter/core/expressions/type.py +++ b/src/codegen/sdk/core/expressions/type.py @@ -4,16 +4,16 @@ from typing_extensions import deprecated -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.interfaces.importable import Importable - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.interfaces.importable import Importable + from codegen.sdk.core.symbol import Symbol Parent = TypeVar("Parent", bound="Editable") @@ -34,7 +34,7 @@ def _compute_dependencies(self, usage_type: UsageKind, dest: "Importable"): ... @noapidoc @reader def resolved_symbol(self) -> "Symbol | str | None": - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.symbol import Symbol for resolved in self.resolved_types: if isinstance(resolved, Symbol): diff --git a/src/graph_sitter/core/expressions/unary_expression.py b/src/codegen/sdk/core/expressions/unary_expression.py similarity index 70% rename from src/graph_sitter/core/expressions/unary_expression.py rename to src/codegen/sdk/core/expressions/unary_expression.py index c22951ce9..238639976 100644 --- a/src/graph_sitter/core/expressions/unary_expression.py +++ b/src/codegen/sdk/core/expressions/unary_expression.py @@ -1,18 +1,18 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.extensions.resolution import ResolutionStack -from graph_sitter.extensions.utils import TSNode -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.extensions.resolution import ResolutionStack +from codegen.sdk.extensions.utils import TSNode +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/core/expressions/union_type.py b/src/codegen/sdk/core/expressions/union_type.py similarity index 74% rename from src/graph_sitter/core/expressions/union_type.py rename to src/codegen/sdk/core/expressions/union_type.py index 95cb50c08..0523d08c1 100644 --- a/src/graph_sitter/core/expressions/union_type.py +++ b/src/codegen/sdk/core/expressions/union_type.py @@ -3,17 +3,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.interfaces.importable import Importable + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.interfaces.importable import Importable TType = TypeVar("TType", bound="Type") diff --git a/src/graph_sitter/core/expressions/unpack.py b/src/codegen/sdk/core/expressions/unpack.py similarity index 83% rename from src/graph_sitter/core/expressions/unpack.py rename to src/codegen/sdk/core/expressions/unpack.py index a5a372c31..7791cb18d 100644 --- a/src/graph_sitter/core/expressions/unpack.py +++ b/src/codegen/sdk/core/expressions/unpack.py @@ -1,11 +1,11 @@ from typing import Generic, TypeVar -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.unwrappable import Unwrappable -from graph_sitter.core.interfaces.wrapper_expression import IWrapper -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.unwrappable import Unwrappable +from codegen.sdk.core.interfaces.wrapper_expression import IWrapper +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent", bound="Editable") @@ -36,7 +36,7 @@ def unwrap(self, node: Expression | None = None): Returns: None """ - from graph_sitter.core.symbol_groups.dict import Dict + from codegen.sdk.core.symbol_groups.dict import Dict node = node or self._value_node if isinstance(node, Dict) and isinstance(self.parent, Dict): diff --git a/src/graph_sitter/core/expressions/value.py b/src/codegen/sdk/core/expressions/value.py similarity index 68% rename from src/graph_sitter/core/expressions/value.py rename to src/codegen/sdk/core/expressions/value.py index ea627f251..79057acdd 100644 --- a/src/graph_sitter/core/expressions/value.py +++ b/src/codegen/sdk/core/expressions/value.py @@ -2,12 +2,12 @@ from typing import TYPE_CHECKING, Generic, TypeVar -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/core/external/dependency_manager.py b/src/codegen/sdk/core/external/dependency_manager.py similarity index 76% rename from src/graph_sitter/core/external/dependency_manager.py rename to src/codegen/sdk/core/external/dependency_manager.py index c5c4c9e4d..85e250c34 100644 --- a/src/graph_sitter/core/external/dependency_manager.py +++ b/src/codegen/sdk/core/external/dependency_manager.py @@ -1,11 +1,11 @@ from abc import abstractmethod from typing import TYPE_CHECKING -from graph_sitter.core.external.external_process import ExternalProcess -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.core.external.external_process import ExternalProcess +from codegen.sdk.enums import ProgrammingLanguage if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph class DependencyManager(ExternalProcess): @@ -28,7 +28,7 @@ def remove_dependencies(self): def get_dependency_manager(language: ProgrammingLanguage, codebase_graph: "CodebaseGraph", enabled: bool = False) -> DependencyManager | None: - from graph_sitter.typescript.external.dependency_manager import TypescriptDependencyManager + from codegen.sdk.typescript.external.dependency_manager import TypescriptDependencyManager ts_enabled = enabled or codebase_graph.config.feature_flags.ts_dependency_manager if language == ProgrammingLanguage.TYPESCRIPT: diff --git a/src/graph_sitter/core/external/external_process.py b/src/codegen/sdk/core/external/external_process.py similarity index 100% rename from src/graph_sitter/core/external/external_process.py rename to src/codegen/sdk/core/external/external_process.py diff --git a/src/graph_sitter/core/external/language_engine.py b/src/codegen/sdk/core/external/language_engine.py similarity index 79% rename from src/graph_sitter/core/external/language_engine.py rename to src/codegen/sdk/core/external/language_engine.py index 8a57b097d..8e2706460 100644 --- a/src/graph_sitter/core/external/language_engine.py +++ b/src/codegen/sdk/core/external/language_engine.py @@ -1,12 +1,12 @@ from abc import abstractmethod from typing import TYPE_CHECKING -from graph_sitter.core.external.external_process import ExternalProcess -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.core.external.external_process import ExternalProcess +from codegen.sdk.enums import ProgrammingLanguage if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.interfaces.editable import Editable class LanguageEngine(ExternalProcess): @@ -22,7 +22,7 @@ def get_return_type(self, node: "Editable") -> str | None: def get_language_engine(language: ProgrammingLanguage, codebase_graph: "CodebaseGraph", use_ts: bool = False, use_v8: bool = False) -> LanguageEngine | None: - from graph_sitter.typescript.external.ts_analyzer_engine import NodeTypescriptEngine, V8TypescriptEngine + from codegen.sdk.typescript.external.ts_analyzer_engine import NodeTypescriptEngine, V8TypescriptEngine use_ts = use_ts or codebase_graph.config.feature_flags.ts_language_engine use_v8 = use_v8 or codebase_graph.config.feature_flags.v8_ts_engine diff --git a/src/graph_sitter/core/external_module.py b/src/codegen/sdk/core/external_module.py similarity index 87% rename from src/graph_sitter/core/external_module.py rename to src/codegen/sdk/core/external_module.py index 5705c75f0..6780b5cf4 100644 --- a/src/graph_sitter/core/external_module.py +++ b/src/codegen/sdk/core/external_module.py @@ -4,22 +4,22 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.interfaces.callable import Callable -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.placeholder.placeholder_stub import StubPlaceholder -from graph_sitter.enums import ImportType, NodeType -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.interfaces.callable import Callable +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.placeholder.placeholder_stub import StubPlaceholder +from codegen.sdk.enums import ImportType, NodeType +from codegen.sdk.writer_decorators import apidoc, noapidoc from graph_visualization.enums import VizNode if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.parameter import Parameter - from graph_sitter.core.import_resolution import Import + from codegen.sdk.core.detached_symbols.parameter import Parameter + from codegen.sdk.core.import_resolution import Import @apidoc diff --git a/src/graph_sitter/core/file.py b/src/codegen/sdk/core/file.py similarity index 96% rename from src/graph_sitter/core/file.py rename to src/codegen/sdk/core/file.py index 61c488deb..35623d12a 100644 --- a/src/graph_sitter/core/file.py +++ b/src/codegen/sdk/core/file.py @@ -12,32 +12,32 @@ from tree_sitter import Node as TSNode -from graph_sitter._proxy import proxy_property -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.codebase.range_index import RangeIndex -from graph_sitter.codebase.span import Range -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.autocommit import commiter, mover, reader, remover, writer -from graph_sitter.core.class_definition import Class -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.directory import Directory -from graph_sitter.core.function import Function -from graph_sitter.core.import_resolution import Import, WildcardImport -from graph_sitter.core.interface import Interface -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.usable import Usable -from graph_sitter.core.statements.import_statement import ImportStatement -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import EdgeType, ImportType, NodeType, SymbolType -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.topological_sort import pseudo_topological_sort -from graph_sitter.tree_sitter_parser import get_parser_by_filepath_or_extension, parse_file -from graph_sitter.typescript.function import TSFunction -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk._proxy import proxy_property +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.codebase.range_index import RangeIndex +from codegen.sdk.codebase.span import Range +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.autocommit import commiter, mover, reader, remover, writer +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.directory import Directory +from codegen.sdk.core.function import Function +from codegen.sdk.core.import_resolution import Import, WildcardImport +from codegen.sdk.core.interface import Interface +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.usable import Usable +from codegen.sdk.core.statements.import_statement import ImportStatement +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import EdgeType, ImportType, NodeType, SymbolType +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.topological_sort import pseudo_topological_sort +from codegen.sdk.tree_sitter_parser import get_parser_by_filepath_or_extension, parse_file +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.writer_decorators import apidoc, noapidoc from graph_visualization.enums import VizNode logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/core/function.py b/src/codegen/sdk/core/function.py similarity index 90% rename from src/graph_sitter/core/function.py rename to src/codegen/sdk/core/function.py index 4d0241ea1..d0a5cbd22 100644 --- a/src/graph_sitter/core/function.py +++ b/src/codegen/sdk/core/function.py @@ -4,31 +4,31 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.detached_symbols.decorator import Decorator -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.import_resolution import Import, WildcardImport -from graph_sitter.core.interfaces.callable import Callable -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.supports_generic import SupportsGenerics -from graph_sitter.core.statements.return_statement import ReturnStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import SymbolType -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.extensions.utils import cached_property -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.detached_symbols.decorator import Decorator +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.import_resolution import Import, WildcardImport +from codegen.sdk.core.interfaces.callable import Callable +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.supports_generic import SupportsGenerics +from codegen.sdk.core.statements.return_statement import ReturnStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import SymbolType +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.writer_decorators import apidoc, noapidoc from graph_visualization.enums import VizNode if TYPE_CHECKING: - from graph_sitter.core.export import Export - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.file import File + from codegen.sdk.core.export import Export + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.file import File TFunction = TypeVar("TFunction", bound="Function") @@ -110,7 +110,7 @@ def is_method(self) -> bool: Returns: bool: True if the function is a method within a class, False otherwise. """ - from graph_sitter.core.class_definition import Class + from codegen.sdk.core.class_definition import Class return isinstance(self.parent.parent.parent, Class) @@ -140,7 +140,7 @@ def is_async(self) -> bool: @noapidoc @reader def resolve_name(self, name: str, start_byte: int | None = None) -> Symbol | Import | WildcardImport | None: - from graph_sitter.core.class_definition import Class + from codegen.sdk.core.class_definition import Class for symbol in self.valid_symbol_names: if symbol.name == name and (start_byte is None or (symbol.start_byte if isinstance(symbol, Class | Function) else symbol.end_byte) <= start_byte): @@ -158,7 +158,7 @@ def valid_symbol_names(self) -> list[Importable]: # def resolve_name(self, name: str, start_byte: int | None = None) -> Symbol | Import | WildcardImport | None: # if symbols := self.valid_symbol_names.get(name, None): # for symbol in symbols: - # from graph_sitter.core.class_definition import Class + # from codegen.sdk.core.class_definition import Class # # if (symbol.start_byte if isinstance(symbol, Class | Function) else symbol.end_byte) <= start_byte: # return symbol diff --git a/src/graph_sitter/core/global_context.py b/src/codegen/sdk/core/global_context.py similarity index 58% rename from src/graph_sitter/core/global_context.py rename to src/codegen/sdk/core/global_context.py index 47163833a..08e430adb 100644 --- a/src/graph_sitter/core/global_context.py +++ b/src/codegen/sdk/core/global_context.py @@ -1,13 +1,13 @@ from dataclasses import dataclass, field from typing import TYPE_CHECKING -from graph_sitter.codebase.config import DefaultConfig, GraphSitterConfig -from graph_sitter.codebase.multigraph import MultiGraph -from graph_sitter.core.plugins import PLUGINS +from codegen.sdk.codebase.config import DefaultConfig, GraphSitterConfig +from codegen.sdk.codebase.multigraph import MultiGraph +from codegen.sdk.core.plugins import PLUGINS if TYPE_CHECKING: - from graph_sitter.core.codebase import Codebase - from graph_sitter.core.function import Function + from codegen.sdk.core.codebase import Codebase + from codegen.sdk.core.function import Function @dataclass diff --git a/src/graph_sitter/core/import_resolution.py b/src/codegen/sdk/core/import_resolution.py similarity index 94% rename from src/graph_sitter/core/import_resolution.py rename to src/codegen/sdk/core/import_resolution.py index 1228c8e30..b55e63dcf 100644 --- a/src/graph_sitter/core/import_resolution.py +++ b/src/codegen/sdk/core/import_resolution.py @@ -8,30 +8,30 @@ import rich.repr from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.codebase.transactions import TransactionPriority -from graph_sitter.core.autocommit import commiter, reader, remover, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.exportable import Exportable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.usable import Usable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.import_statement import ImportStatement -from graph_sitter.enums import EdgeType, ImportType, NodeType -from graph_sitter.extensions.utils import cached_property -from graph_sitter.output.constants import ANGULAR_STYLE -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.codebase.transactions import TransactionPriority +from codegen.sdk.core.autocommit import commiter, reader, remover, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.exportable import Exportable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.usable import Usable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.import_statement import ImportStatement +from codegen.sdk.enums import EdgeType, ImportType, NodeType +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.output.constants import ANGULAR_STYLE +from codegen.sdk.writer_decorators import apidoc, noapidoc from graph_visualization.enums import VizNode if TYPE_CHECKING: - from graph_sitter.core.file import SourceFile - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.file import SourceFile + from codegen.sdk.core.symbol import Symbol TSourceFile = TypeVar("TSourceFile", bound="SourceFile") @@ -325,7 +325,7 @@ def _imported_symbol(self, resolve_exports: bool = False) -> Symbol | ExternalMo """Returns the symbol directly being imported, including an indirect import and an External Module. """ - from graph_sitter.typescript.file import TSFile + from codegen.sdk.typescript.file import TSFile symbol = next(iter(self.G.successors(self.node_id, edge_type=EdgeType.IMPORT_SYMBOL_RESOLUTION, sort=False)), None) if symbol is None: @@ -533,7 +533,7 @@ def _resolved_types(self) -> Generator[ResolutionStack[Self], None, None]: @reader def _wildcards(self) -> dict[str, WildcardImport[Self]]: """A list of all imports or wildcard imports.""" - from graph_sitter.core.file import SourceFile + from codegen.sdk.core.file import SourceFile res = {} if self.is_wildcard_import(): diff --git a/src/graph_sitter/core/interface.py b/src/codegen/sdk/core/interface.py similarity index 74% rename from src/graph_sitter/core/interface.py rename to src/codegen/sdk/core/interface.py index fdec35fa5..4b45e294b 100644 --- a/src/graph_sitter/core/interface.py +++ b/src/codegen/sdk/core/interface.py @@ -2,22 +2,22 @@ from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter._proxy import proxy_property -from graph_sitter.core.autocommit import reader -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.function import Function -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.interfaces.inherits import Inherits -from graph_sitter.core.statements.attribute import Attribute -from graph_sitter.core.symbol_groups.parents import Parents -from graph_sitter.enums import SymbolType -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk._proxy import proxy_property +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.function import Function +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.interfaces.inherits import Inherits +from codegen.sdk.core.statements.attribute import Attribute +from codegen.sdk.core.symbol_groups.parents import Parents +from codegen.sdk.enums import SymbolType +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: - from graph_sitter.core.class_definition import Class + from codegen.sdk.core.class_definition import Class TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock") diff --git a/src/graph_sitter/core/interfaces/__init__.py b/src/codegen/sdk/core/interfaces/__init__.py similarity index 100% rename from src/graph_sitter/core/interfaces/__init__.py rename to src/codegen/sdk/core/interfaces/__init__.py diff --git a/src/graph_sitter/core/interfaces/callable.py b/src/codegen/sdk/core/interfaces/callable.py similarity index 85% rename from src/graph_sitter/core/interfaces/callable.py rename to src/codegen/sdk/core/interfaces/callable.py index 019d421bd..8c59e7bce 100644 --- a/src/graph_sitter/core/interfaces/callable.py +++ b/src/codegen/sdk/core/interfaces/callable.py @@ -1,20 +1,20 @@ from dataclasses import dataclass from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter.core.autocommit import reader -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.interfaces.usable import Usable -from graph_sitter.core.placeholder.placeholder import Placeholder -from graph_sitter.core.symbol_group import SymbolGroup -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.interfaces.usable import Usable +from codegen.sdk.core.placeholder.placeholder import Placeholder +from codegen.sdk.core.symbol_group import SymbolGroup +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.core.class_definition import Class - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.external_module import ExternalModule - from graph_sitter.core.function import Function - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.external_module import ExternalModule + from codegen.sdk.core.function import Function + from codegen.sdk.core.symbol import Symbol @dataclass diff --git a/src/graph_sitter/core/interfaces/chainable.py b/src/codegen/sdk/core/interfaces/chainable.py similarity index 86% rename from src/graph_sitter/core/interfaces/chainable.py rename to src/codegen/sdk/core/interfaces/chainable.py index e7e56813a..29c48f18f 100644 --- a/src/graph_sitter/core/interfaces/chainable.py +++ b/src/codegen/sdk/core/interfaces/chainable.py @@ -2,13 +2,13 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.extensions.utils import cached_property -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: - from graph_sitter.core.interfaces.has_attribute import HasAttribute + from codegen.sdk.core.interfaces.has_attribute import HasAttribute Parent = TypeVar("Parent", bound="Editable") @@ -39,7 +39,7 @@ def resolved_type_frames(self) -> list[ResolutionStack["Self"]]: @noapidoc def with_resolution_frame(self, child: Editable, *args, generic_parameters: list | None = None, generics: dict | None = None, **kwargs) -> Generator[ResolutionStack["Self"], None, None]: """Resolve the definition(s) of this object.""" - from graph_sitter.core.interfaces.supports_generic import SupportsGenerics + from codegen.sdk.core.interfaces.supports_generic import SupportsGenerics if isinstance(child, Chainable): assert child is not self diff --git a/src/graph_sitter/core/interfaces/editable.py b/src/codegen/sdk/core/interfaces/editable.py similarity index 94% rename from src/graph_sitter/core/interfaces/editable.py rename to src/codegen/sdk/core/interfaces/editable.py index 00001e019..926df1229 100644 --- a/src/graph_sitter/core/interfaces/editable.py +++ b/src/codegen/sdk/core/interfaces/editable.py @@ -14,39 +14,39 @@ from tree_sitter import Node as TSNode from tree_sitter import Point, Range -from graph_sitter.codebase.flagging.code_flag import CodeFlag -from graph_sitter.codebase.flagging.enums import FlagKwargs -from graph_sitter.codebase.span import Span -from graph_sitter.codebase.transaction_manager import TransactionManager -from graph_sitter.codebase.transactions import EditTransaction, InsertTransaction, RemoveTransaction, TransactionPriority -from graph_sitter.core.autocommit import commiter, reader, remover, repr_func, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.placeholder.placeholder import Placeholder -from graph_sitter.enums import NodeType -from graph_sitter.extensions.utils import get_all_identifiers -from graph_sitter.output.ast import AST -from graph_sitter.output.constants import ANGULAR_STYLE, MAX_STRING_LENGTH -from graph_sitter.output.jsonable import JSONable -from graph_sitter.output.utils import style_editable -from graph_sitter.utils import descendant_for_byte_range, find_all_descendants, find_first_ancestor, find_index, truncate_line -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.flagging.code_flag import CodeFlag +from codegen.sdk.codebase.flagging.enums import FlagKwargs +from codegen.sdk.codebase.span import Span +from codegen.sdk.codebase.transaction_manager import TransactionManager +from codegen.sdk.codebase.transactions import EditTransaction, InsertTransaction, RemoveTransaction, TransactionPriority +from codegen.sdk.core.autocommit import commiter, reader, remover, repr_func, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.placeholder.placeholder import Placeholder +from codegen.sdk.enums import NodeType +from codegen.sdk.extensions.utils import get_all_identifiers +from codegen.sdk.output.ast import AST +from codegen.sdk.output.constants import ANGULAR_STYLE, MAX_STRING_LENGTH +from codegen.sdk.output.jsonable import JSONable +from codegen.sdk.output.utils import style_editable +from codegen.sdk.utils import descendant_for_byte_range, find_all_descendants, find_first_ancestor, find_index, truncate_line +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.class_definition import Class - from graph_sitter.core.detached_symbols.function_call import FunctionCall - from graph_sitter.core.export import Export - from graph_sitter.core.expressions import Expression - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.file import File, SourceFile - from graph_sitter.core.function import Function - from graph_sitter.core.import_resolution import Import, WildcardImport - from graph_sitter.core.interfaces.has_name import HasName - from graph_sitter.core.interfaces.importable import Importable - from graph_sitter.core.statements.statement import Statement - from graph_sitter.core.symbol import Symbol - from graph_sitter.core.symbol_group import SymbolGroup + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.detached_symbols.function_call import FunctionCall + from codegen.sdk.core.export import Export + from codegen.sdk.core.expressions import Expression + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.file import File, SourceFile + from codegen.sdk.core.function import Function + from codegen.sdk.core.import_resolution import Import, WildcardImport + from codegen.sdk.core.interfaces.has_name import HasName + from codegen.sdk.core.interfaces.importable import Importable + from codegen.sdk.core.statements.statement import Statement + from codegen.sdk.core.symbol import Symbol + from codegen.sdk.core.symbol_group import SymbolGroup from graph_visualization.enums import VizNode CONTAINER_CHARS = (b"(", b")", b"{", b"}", b"[", b"]", b"<", b">", b"import") MAX_REPR_LEN: int = 200 @@ -275,7 +275,7 @@ def extended(self) -> SymbolGroup: SymbolGroup: A group containing this node and its extended nodes that allows batch modification through a common interface. """ - from graph_sitter.core.symbol_group import SymbolGroup + from codegen.sdk.core.symbol_group import SymbolGroup return SymbolGroup(self.file_node_id, self.G, self.parent, children=self.extended_nodes) @@ -940,8 +940,8 @@ def _compute_dependencies(self, usage_type: UsageKind, dest: HasName | None = No @commiter @noapidoc def _add_symbol_usages(self: HasName, identifiers: list[TSNode], usage_type: UsageKind, dest: HasName | None = None) -> None: - from graph_sitter.core.expressions import Name - from graph_sitter.core.interfaces.resolvable import Resolvable + from codegen.sdk.core.expressions import Name + from codegen.sdk.core.interfaces.resolvable import Resolvable if dest is None: dest = self @@ -974,7 +974,7 @@ def _log_parse(self, msg: str, *args, **kwargs): @property @noapidoc def viz(self) -> VizNode: - from graph_sitter.core.interfaces.has_name import HasName + from codegen.sdk.core.interfaces.has_name import HasName from graph_visualization.enums import VizNode if isinstance(self, HasName): @@ -1047,7 +1047,7 @@ def function_calls(self) -> list[FunctionCall]: @noapidoc def self_dest(self) -> Importable: """Returns the symbol usage resolution destination node for the symbol.""" - from graph_sitter.core.interfaces.importable import Importable + from codegen.sdk.core.interfaces.importable import Importable dest = self while dest and not isinstance(dest, Importable): @@ -1099,7 +1099,7 @@ def first_ancestors(self, type: type[T]) -> T | None: @reader def parent_statement(self) -> Statement | None: """Find the statement this node is contained in""" - from graph_sitter.core.statements.statement import Statement + from codegen.sdk.core.statements.statement import Statement return self.parent_of_type(Statement) @@ -1107,7 +1107,7 @@ def parent_statement(self) -> Statement | None: @reader def parent_function(self) -> Function | None: """Find the function this node is contained in""" - from graph_sitter.core.function import Function + from codegen.sdk.core.function import Function return self.parent_of_type(Function) @@ -1115,7 +1115,7 @@ def parent_function(self) -> Function | None: @reader def parent_class(self) -> Class | None: """Find the class this node is contained in""" - from graph_sitter.core.class_definition import Class + from codegen.sdk.core.class_definition import Class return self.parent_of_type(Class) @@ -1137,4 +1137,4 @@ def _get_ast_children(self) -> list[tuple[str | None, AST]]: @final def ast(self) -> AST: children = self._get_ast_children() - return AST(graph_sitter_type=self.__class__.__name__, span=self.span, tree_sitter_type=self.ts_node_type, children=children) + return AST(codegen_sdk_type=self.__class__.__name__, span=self.span, tree_sitter_type=self.ts_node_type, children=children) diff --git a/src/graph_sitter/core/interfaces/exportable.py b/src/codegen/sdk/core/interfaces/exportable.py similarity index 90% rename from src/graph_sitter/core/interfaces/exportable.py rename to src/codegen/sdk/core/interfaces/exportable.py index bff6d4f97..8f58f73a3 100644 --- a/src/graph_sitter/core/interfaces/exportable.py +++ b/src/codegen/sdk/core/interfaces/exportable.py @@ -4,14 +4,14 @@ from rustworkx import NoSuitableNeighbors -from graph_sitter.core.autocommit import reader -from graph_sitter.core.interfaces.usable import Usable -from graph_sitter.enums import EdgeType, ImportType, NodeType -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.interfaces.usable import Usable +from codegen.sdk.enums import EdgeType, ImportType, NodeType +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.core.export import Export - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.export import Export + from codegen.sdk.core.interfaces.editable import Editable Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/interfaces/has_attribute.py b/src/codegen/sdk/core/interfaces/has_attribute.py similarity index 84% rename from src/graph_sitter/core/interfaces/has_attribute.py rename to src/codegen/sdk/core/interfaces/has_attribute.py index d47432a01..1fcada30b 100644 --- a/src/graph_sitter/core/interfaces/has_attribute.py +++ b/src/codegen/sdk/core/interfaces/has_attribute.py @@ -2,7 +2,7 @@ from typing import TYPE_CHECKING, Generic, TypeVar if TYPE_CHECKING: - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.interfaces.editable import Editable Attribute = TypeVar("Attribute", bound="Editable") diff --git a/src/graph_sitter/core/interfaces/has_block.py b/src/codegen/sdk/core/interfaces/has_block.py similarity index 88% rename from src/graph_sitter/core/interfaces/has_block.py rename to src/codegen/sdk/core/interfaces/has_block.py index 4135a7bc6..c186e13a5 100644 --- a/src/graph_sitter/core/interfaces/has_block.py +++ b/src/codegen/sdk/core/interfaces/has_block.py @@ -5,15 +5,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.detached_symbols.decorator import Decorator -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.statements.comment import Comment -from graph_sitter.core.symbol_groups.comment_group import CommentGroup -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.detached_symbols.decorator import Decorator +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.statements.comment import Comment +from codegen.sdk.core.symbol_groups.comment_group import CommentGroup +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.writer_decorators import apidoc TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock") TDecorator = TypeVar("TDecorator", bound="Decorator") diff --git a/src/graph_sitter/core/interfaces/has_name.py b/src/codegen/sdk/core/interfaces/has_name.py similarity index 87% rename from src/graph_sitter/core/interfaces/has_name.py rename to src/codegen/sdk/core/interfaces/has_name.py index a8a171e38..ea14501fc 100644 --- a/src/graph_sitter/core/interfaces/has_name.py +++ b/src/codegen/sdk/core/interfaces/has_name.py @@ -1,11 +1,11 @@ from functools import cached_property -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.expressions.defined_name import DefinedName -from graph_sitter.core.expressions.name import Name -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.expressions.defined_name import DefinedName +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.writer_decorators import apidoc, noapidoc @apidoc @@ -39,7 +39,7 @@ def full_name(self) -> str | None: if isinstance(self._name_node, ChainedAttribute): return self._name_node.full_name if isinstance(self._name_node, DefinedName): - from graph_sitter.core.function import Function + from codegen.sdk.core.function import Function if isinstance(self, Function) and self.is_method: return self.parent_class.full_name + "." + self.name diff --git a/src/graph_sitter/core/interfaces/has_value.py b/src/codegen/sdk/core/interfaces/has_value.py similarity index 82% rename from src/graph_sitter/core/interfaces/has_value.py rename to src/codegen/sdk/core/interfaces/has_value.py index 10e6d32fc..89d82be79 100644 --- a/src/graph_sitter/core/interfaces/has_value.py +++ b/src/codegen/sdk/core/interfaces/has_value.py @@ -1,6 +1,6 @@ -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.writer_decorators import apidoc @apidoc diff --git a/src/graph_sitter/core/interfaces/importable.py b/src/codegen/sdk/core/interfaces/importable.py similarity index 83% rename from src/graph_sitter/core/interfaces/importable.py rename to src/codegen/sdk/core/interfaces/importable.py index 15fe2f973..67f685002 100644 --- a/src/graph_sitter/core/interfaces/importable.py +++ b/src/codegen/sdk/core/interfaces/importable.py @@ -2,21 +2,21 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.enums import EdgeType -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.enums import EdgeType +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.import_resolution import Import - from graph_sitter.core.symbol import Symbol + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.import_resolution import Import + from codegen.sdk.core.symbol import Symbol Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/interfaces/inherits.py b/src/codegen/sdk/core/interfaces/inherits.py similarity index 82% rename from src/graph_sitter/core/interfaces/inherits.py rename to src/codegen/sdk/core/interfaces/inherits.py index 313c53608..2d29bc1dc 100644 --- a/src/graph_sitter/core/interfaces/inherits.py +++ b/src/codegen/sdk/core/interfaces/inherits.py @@ -4,15 +4,15 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, TypeVar -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.expressions import Type -from graph_sitter.core.interfaces.supports_generic import SupportsGenerics -from graph_sitter.enums import EdgeType +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.expressions import Type +from codegen.sdk.core.interfaces.supports_generic import SupportsGenerics +from codegen.sdk.enums import EdgeType if TYPE_CHECKING: - from graph_sitter.core.class_definition import Class - from graph_sitter.core.external_module import ExternalModule - from graph_sitter.core.interface import Interface + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.external_module import ExternalModule + from codegen.sdk.core.interface import Interface TType = TypeVar("TType", bound=Type) @@ -28,8 +28,8 @@ def compute_superclass_dependencies(self) -> None: @reader def _get_superclasses(self, max_depth: int | None = None) -> list[Class | ExternalModule | Interface]: """Returns a list of all classes that this class extends, up to max_depth.""" - from graph_sitter.core.class_definition import Class - from graph_sitter.core.interface import Interface + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.interface import Interface # Implements the python MRO, IE: by level seen = set() diff --git a/src/graph_sitter/core/interfaces/parseable.py b/src/codegen/sdk/core/interfaces/parseable.py similarity index 78% rename from src/graph_sitter/core/interfaces/parseable.py rename to src/codegen/sdk/core/interfaces/parseable.py index 8d316bf0d..86185bf11 100644 --- a/src/graph_sitter/core/interfaces/parseable.py +++ b/src/codegen/sdk/core/interfaces/parseable.py @@ -2,7 +2,7 @@ from typing import TYPE_CHECKING if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph class Parseable(ABC): diff --git a/src/graph_sitter/core/interfaces/resolvable.py b/src/codegen/sdk/core/interfaces/resolvable.py similarity index 67% rename from src/graph_sitter/core/interfaces/resolvable.py rename to src/codegen/sdk/core/interfaces/resolvable.py index f6d598a32..58133dfa8 100644 --- a/src/graph_sitter/core/interfaces/resolvable.py +++ b/src/codegen/sdk/core/interfaces/resolvable.py @@ -3,10 +3,10 @@ from typing_extensions import TypeVar -from graph_sitter.core.autocommit import writer -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/core/interfaces/supports_generic.py b/src/codegen/sdk/core/interfaces/supports_generic.py similarity index 59% rename from src/graph_sitter/core/interfaces/supports_generic.py rename to src/codegen/sdk/core/interfaces/supports_generic.py index 8689859a6..73f9342f9 100644 --- a/src/graph_sitter/core/interfaces/supports_generic.py +++ b/src/codegen/sdk/core/interfaces/supports_generic.py @@ -2,14 +2,14 @@ from typing_extensions import TypeVar -from graph_sitter.core.expressions.named_type import NamedType -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.symbol_groups.type_parameters import TypeParameters -from graph_sitter.extensions.utils import cached_property -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk.core.expressions.named_type import NamedType +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.symbol_groups.type_parameters import TypeParameters +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: - from graph_sitter.core.expressions import Type + from codegen.sdk.core.expressions import Type TType = TypeVar("TType", bound="Type") diff --git a/src/graph_sitter/core/interfaces/typeable.py b/src/codegen/sdk/core/interfaces/typeable.py similarity index 73% rename from src/graph_sitter/core/interfaces/typeable.py rename to src/codegen/sdk/core/interfaces/typeable.py index 3c38f7a5d..b5315d8f3 100644 --- a/src/graph_sitter/core/interfaces/typeable.py +++ b/src/codegen/sdk/core/interfaces/typeable.py @@ -3,15 +3,15 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.placeholder.placeholder_type import TypePlaceholder -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.placeholder.placeholder_type import TypePlaceholder +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.expressions.type import Type + from codegen.sdk.core.expressions.type import Type TType = TypeVar("TType", bound="Type") diff --git a/src/graph_sitter/core/interfaces/unwrappable.py b/src/codegen/sdk/core/interfaces/unwrappable.py similarity index 81% rename from src/graph_sitter/core/interfaces/unwrappable.py rename to src/codegen/sdk/core/interfaces/unwrappable.py index e78bbc848..35f6c9245 100644 --- a/src/graph_sitter/core/interfaces/unwrappable.py +++ b/src/codegen/sdk/core/interfaces/unwrappable.py @@ -3,9 +3,9 @@ from typing_extensions import TypeVar -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent", bound=Editable) diff --git a/src/graph_sitter/core/interfaces/usable.py b/src/codegen/sdk/core/interfaces/usable.py similarity index 86% rename from src/graph_sitter/core/interfaces/usable.py rename to src/codegen/sdk/core/interfaces/usable.py index 97960c676..42e2298e2 100644 --- a/src/graph_sitter/core/interfaces/usable.py +++ b/src/codegen/sdk/core/interfaces/usable.py @@ -2,18 +2,18 @@ from typing import TYPE_CHECKING, Generic, TypeVar -from graph_sitter._proxy import proxy_property -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import Usage, UsageType -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.enums import EdgeType -from graph_sitter.writer_decorators import apidoc +from codegen.sdk._proxy import proxy_property +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import Usage, UsageType +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.enums import EdgeType +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.core.export import Export - from graph_sitter.core.import_resolution import Import - from graph_sitter.core.symbol import Symbol + from codegen.sdk.core.export import Export + from codegen.sdk.core.import_resolution import Import + from codegen.sdk.core.symbol import Symbol Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/interfaces/wrapper_expression.py b/src/codegen/sdk/core/interfaces/wrapper_expression.py similarity index 69% rename from src/graph_sitter/core/interfaces/wrapper_expression.py rename to src/codegen/sdk/core/interfaces/wrapper_expression.py index cb6166195..33586da15 100644 --- a/src/graph_sitter/core/interfaces/wrapper_expression.py +++ b/src/codegen/sdk/core/interfaces/wrapper_expression.py @@ -2,18 +2,18 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Self, final, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: - from graph_sitter.core.interfaces.importable import Importable + from codegen.sdk.core.interfaces.importable import Importable class IWrapper(Chainable, Editable): diff --git a/src/graph_sitter/core/node_id_factory.py b/src/codegen/sdk/core/node_id_factory.py similarity index 100% rename from src/graph_sitter/core/node_id_factory.py rename to src/codegen/sdk/core/node_id_factory.py diff --git a/src/graph_sitter/core/parser.py b/src/codegen/sdk/core/parser.py similarity index 79% rename from src/graph_sitter/core/parser.py rename to src/codegen/sdk/core/parser.py index 7fe4d92d9..2947b6076 100644 --- a/src/graph_sitter/core/parser.py +++ b/src/codegen/sdk/core/parser.py @@ -6,21 +6,21 @@ from rich.console import Console from tree_sitter import Node as TSNode -from graph_sitter.codebase.node_classes.node_classes import NodeClasses -from graph_sitter.core.expressions.placeholder_type import PlaceholderType -from graph_sitter.core.expressions.value import Value -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement -from graph_sitter.core.statements.symbol_statement import SymbolStatement -from graph_sitter.utils import find_first_function_descendant +from codegen.sdk.codebase.node_classes.node_classes import NodeClasses +from codegen.sdk.core.expressions.placeholder_type import PlaceholderType +from codegen.sdk.core.expressions.value import Value +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.core.statements.symbol_statement import SymbolStatement +from codegen.sdk.utils import find_first_function_descendant if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.symbol import Symbol - from graph_sitter.python.detached_symbols.code_block import PyCodeBlock - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.symbol import Symbol + from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock Parent = TypeVar("Parent", bound="Editable") @@ -91,26 +91,26 @@ def parse_type(self, node: TSNode, file_node_id: NodeId, G: CodebaseGraph, paren expr_type, node = self._process_type(expr_type, node) return expr_type(node, file_node_id, G, parent) self.log_unparsed(node) - from graph_sitter.core.expressions.placeholder_type import PlaceholderType + from codegen.sdk.core.expressions.placeholder_type import PlaceholderType return PlaceholderType(node, file_node_id, G, parent) def parse_ts_statements(self, node: TSNode, file_node_id: NodeId, G: CodebaseGraph, parent: TSCodeBlock) -> list[Statement]: - from graph_sitter.core.statements.export_statement import ExportStatement - from graph_sitter.core.statements.expression_statement import ExpressionStatement - from graph_sitter.core.statements.return_statement import ReturnStatement - from graph_sitter.core.statements.statement import Statement - from graph_sitter.core.statements.symbol_statement import SymbolStatement - from graph_sitter.typescript.function import _VALID_TYPE_NAMES - from graph_sitter.typescript.statements.assignment_statement import TSAssignmentStatement - from graph_sitter.typescript.statements.attribute import TSAttribute - from graph_sitter.typescript.statements.comment import TSComment - from graph_sitter.typescript.statements.for_loop_statement import TSForLoopStatement - from graph_sitter.typescript.statements.if_block_statement import TSIfBlockStatement - from graph_sitter.typescript.statements.labeled_statement import TSLabeledStatement - from graph_sitter.typescript.statements.switch_statement import TSSwitchStatement - from graph_sitter.typescript.statements.try_catch_statement import TSTryCatchStatement - from graph_sitter.typescript.statements.while_statement import TSWhileStatement + from codegen.sdk.core.statements.export_statement import ExportStatement + from codegen.sdk.core.statements.expression_statement import ExpressionStatement + from codegen.sdk.core.statements.return_statement import ReturnStatement + from codegen.sdk.core.statements.statement import Statement + from codegen.sdk.core.statements.symbol_statement import SymbolStatement + from codegen.sdk.typescript.function import _VALID_TYPE_NAMES + from codegen.sdk.typescript.statements.assignment_statement import TSAssignmentStatement + from codegen.sdk.typescript.statements.attribute import TSAttribute + from codegen.sdk.typescript.statements.comment import TSComment + from codegen.sdk.typescript.statements.for_loop_statement import TSForLoopStatement + from codegen.sdk.typescript.statements.if_block_statement import TSIfBlockStatement + from codegen.sdk.typescript.statements.labeled_statement import TSLabeledStatement + from codegen.sdk.typescript.statements.switch_statement import TSSwitchStatement + from codegen.sdk.typescript.statements.try_catch_statement import TSTryCatchStatement + from codegen.sdk.typescript.statements.while_statement import TSWhileStatement statements = [] @@ -193,21 +193,21 @@ def parse_ts_statements(self, node: TSNode, file_node_id: NodeId, G: CodebaseGra return statements def parse_py_statements(self, node: TSNode, file_node_id: NodeId, G: CodebaseGraph, parent: PyCodeBlock) -> list[Statement]: - from graph_sitter.core.statements.expression_statement import ExpressionStatement - from graph_sitter.core.statements.raise_statement import RaiseStatement - from graph_sitter.core.statements.return_statement import ReturnStatement - from graph_sitter.core.statements.statement import Statement - from graph_sitter.python.statements.assignment_statement import PyAssignmentStatement - from graph_sitter.python.statements.attribute import PyAttribute - from graph_sitter.python.statements.break_statement import PyBreakStatement - from graph_sitter.python.statements.comment import PyComment - from graph_sitter.python.statements.for_loop_statement import PyForLoopStatement - from graph_sitter.python.statements.if_block_statement import PyIfBlockStatement - from graph_sitter.python.statements.match_statement import PyMatchStatement - from graph_sitter.python.statements.pass_statement import PyPassStatement - from graph_sitter.python.statements.try_catch_statement import PyTryCatchStatement - from graph_sitter.python.statements.while_statement import PyWhileStatement - from graph_sitter.python.statements.with_statement import WithStatement + from codegen.sdk.core.statements.expression_statement import ExpressionStatement + from codegen.sdk.core.statements.raise_statement import RaiseStatement + from codegen.sdk.core.statements.return_statement import ReturnStatement + from codegen.sdk.core.statements.statement import Statement + from codegen.sdk.python.statements.assignment_statement import PyAssignmentStatement + from codegen.sdk.python.statements.attribute import PyAttribute + from codegen.sdk.python.statements.break_statement import PyBreakStatement + from codegen.sdk.python.statements.comment import PyComment + from codegen.sdk.python.statements.for_loop_statement import PyForLoopStatement + from codegen.sdk.python.statements.if_block_statement import PyIfBlockStatement + from codegen.sdk.python.statements.match_statement import PyMatchStatement + from codegen.sdk.python.statements.pass_statement import PyPassStatement + from codegen.sdk.python.statements.try_catch_statement import PyTryCatchStatement + from codegen.sdk.python.statements.while_statement import PyWhileStatement + from codegen.sdk.python.statements.with_statement import WithStatement statements = [] @@ -267,7 +267,7 @@ def parse_py_statements(self, node: TSNode, file_node_id: NodeId, G: CodebaseGra if var.type == "string": statements.append(PyComment.from_code_block(var, parent, pos=len(statements))) elif var.type in ["assignment", "augmented_assignment"]: - from graph_sitter.core.class_definition import Class + from codegen.sdk.core.class_definition import Class if isinstance(parent.parent, Class): statements.append(PyAttribute(child, file_node_id, G, parent, len(statements), var)) diff --git a/src/graph_sitter/core/placeholder/__init__.py b/src/codegen/sdk/core/placeholder/__init__.py similarity index 100% rename from src/graph_sitter/core/placeholder/__init__.py rename to src/codegen/sdk/core/placeholder/__init__.py diff --git a/src/graph_sitter/core/placeholder/placeholder.py b/src/codegen/sdk/core/placeholder/placeholder.py similarity index 92% rename from src/graph_sitter/core/placeholder/placeholder.py rename to src/codegen/sdk/core/placeholder/placeholder.py index 2c50c8bf2..06426af82 100644 --- a/src/graph_sitter/core/placeholder/placeholder.py +++ b/src/codegen/sdk/core/placeholder/placeholder.py @@ -1,11 +1,11 @@ from abc import ABC, abstractmethod from typing import TYPE_CHECKING, Generic, Literal, TypeVar -from graph_sitter.core.autocommit import repr_func -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.autocommit import repr_func +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.interfaces.editable import Editable Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/placeholder/placeholder_stub.py b/src/codegen/sdk/core/placeholder/placeholder_stub.py similarity index 85% rename from src/graph_sitter/core/placeholder/placeholder_stub.py rename to src/codegen/sdk/core/placeholder/placeholder_stub.py index 1be46c2e2..9f436d421 100644 --- a/src/graph_sitter/core/placeholder/placeholder_stub.py +++ b/src/codegen/sdk/core/placeholder/placeholder_stub.py @@ -1,8 +1,8 @@ from typing import Generic, TypeVar -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.placeholder.placeholder import Placeholder -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.placeholder.placeholder import Placeholder +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/core/placeholder/placeholder_type.py b/src/codegen/sdk/core/placeholder/placeholder_type.py similarity index 88% rename from src/graph_sitter/core/placeholder/placeholder_type.py rename to src/codegen/sdk/core/placeholder/placeholder_type.py index 6edb85f28..74b8c3597 100644 --- a/src/graph_sitter/core/placeholder/placeholder_type.py +++ b/src/codegen/sdk/core/placeholder/placeholder_type.py @@ -1,8 +1,8 @@ from typing import Generic, TypeVar -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.placeholder.placeholder import Placeholder -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.placeholder.placeholder import Placeholder +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/plugins/__init__.py b/src/codegen/sdk/core/plugins/__init__.py new file mode 100644 index 000000000..4cf23493c --- /dev/null +++ b/src/codegen/sdk/core/plugins/__init__.py @@ -0,0 +1,9 @@ +from codegen.sdk.core.plugins.axios import AxiosApiFinder +from codegen.sdk.core.plugins.flask import FlaskApiFinder +from codegen.sdk.core.plugins.modal import ModalApiFinder + +PLUGINS = [ + FlaskApiFinder(), + AxiosApiFinder(), + ModalApiFinder(), +] diff --git a/src/graph_sitter/core/plugins/axios.py b/src/codegen/sdk/core/plugins/axios.py similarity index 76% rename from src/graph_sitter/core/plugins/axios.py rename to src/codegen/sdk/core/plugins/axios.py index 085238337..22a4de747 100644 --- a/src/graph_sitter/core/plugins/axios.py +++ b/src/codegen/sdk/core/plugins/axios.py @@ -1,16 +1,16 @@ from logging import getLogger from typing import TYPE_CHECKING -from graph_sitter.core.detached_symbols.argument import Argument -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import String -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.plugins.plugin import Plugin -from graph_sitter.core.symbol_groups.dict import Dict -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.core.detached_symbols.argument import Argument +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import String +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.plugins.plugin import Plugin +from codegen.sdk.core.symbol_groups.dict import Dict +from codegen.sdk.enums import ProgrammingLanguage if TYPE_CHECKING: - from graph_sitter.core.codebase import TSCodebaseType + from codegen.sdk.core.codebase import TSCodebaseType logger = getLogger(__name__) diff --git a/src/graph_sitter/core/plugins/flask.py b/src/codegen/sdk/core/plugins/flask.py similarity index 92% rename from src/graph_sitter/core/plugins/flask.py rename to src/codegen/sdk/core/plugins/flask.py index adfaa7421..1a399e6fd 100644 --- a/src/graph_sitter/core/plugins/flask.py +++ b/src/codegen/sdk/core/plugins/flask.py @@ -1,11 +1,11 @@ from logging import getLogger from typing import TYPE_CHECKING -from graph_sitter.core.plugins.plugin import Plugin -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.core.plugins.plugin import Plugin +from codegen.sdk.enums import ProgrammingLanguage if TYPE_CHECKING: - from graph_sitter.core.codebase import PyCodebaseType + from codegen.sdk.core.codebase import PyCodebaseType logger = getLogger(__name__) diff --git a/src/graph_sitter/core/plugins/modal.py b/src/codegen/sdk/core/plugins/modal.py similarity index 82% rename from src/graph_sitter/core/plugins/modal.py rename to src/codegen/sdk/core/plugins/modal.py index 756677a39..d6edc2475 100644 --- a/src/graph_sitter/core/plugins/modal.py +++ b/src/codegen/sdk/core/plugins/modal.py @@ -1,11 +1,11 @@ from logging import getLogger from typing import TYPE_CHECKING -from graph_sitter.core.plugins.plugin import Plugin -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.core.plugins.plugin import Plugin +from codegen.sdk.enums import ProgrammingLanguage if TYPE_CHECKING: - from graph_sitter.core.codebase import PyCodebaseType + from codegen.sdk.core.codebase import PyCodebaseType logger = getLogger(__name__) diff --git a/src/graph_sitter/core/plugins/plugin.py b/src/codegen/sdk/core/plugins/plugin.py similarity index 64% rename from src/graph_sitter/core/plugins/plugin.py rename to src/codegen/sdk/core/plugins/plugin.py index d3114f650..79d9f6ed3 100644 --- a/src/graph_sitter/core/plugins/plugin.py +++ b/src/codegen/sdk/core/plugins/plugin.py @@ -1,11 +1,11 @@ from abc import ABC, abstractmethod from typing import TYPE_CHECKING -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.enums import ProgrammingLanguage if TYPE_CHECKING: - from graph_sitter.core.codebase import Codebase + from codegen.sdk.core.codebase import Codebase class Plugin(ABC): diff --git a/src/graph_sitter/core/statements/__init__.py b/src/codegen/sdk/core/statements/__init__.py similarity index 100% rename from src/graph_sitter/core/statements/__init__.py rename to src/codegen/sdk/core/statements/__init__.py diff --git a/src/graph_sitter/core/statements/assignment_statement.py b/src/codegen/sdk/core/statements/assignment_statement.py similarity index 72% rename from src/graph_sitter/core/statements/assignment_statement.py rename to src/codegen/sdk/core/statements/assignment_statement.py index 51916078c..7c4e84b4a 100644 --- a/src/graph_sitter/core/statements/assignment_statement.py +++ b/src/codegen/sdk/core/statements/assignment_statement.py @@ -5,23 +5,23 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.multi_expression import MultiExpression -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.multi_expression import MultiExpression +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.interfaces.has_block import HasBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.interfaces.has_block import HasBlock TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock") diff --git a/src/graph_sitter/core/statements/attribute.py b/src/codegen/sdk/core/statements/attribute.py similarity index 77% rename from src/graph_sitter/core/statements/attribute.py rename to src/codegen/sdk/core/statements/attribute.py index a464ed6ee..d81d91a0b 100644 --- a/src/graph_sitter/core/statements/attribute.py +++ b/src/codegen/sdk/core/statements/attribute.py @@ -7,21 +7,21 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.autocommit import writer -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.usable import Usable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.assignment_statement import AssignmentStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.extensions.autocommit import reader -from graph_sitter.extensions.resolution import ResolutionStack -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.usable import Usable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.assignment_statement import AssignmentStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.extensions.resolution import ResolutionStack +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock | None") TAssignment = TypeVar("TAssignment", bound="Assignment") diff --git a/src/graph_sitter/core/statements/block_statement.py b/src/codegen/sdk/core/statements/block_statement.py similarity index 76% rename from src/graph_sitter/core/statements/block_statement.py rename to src/codegen/sdk/core/statements/block_statement.py index 94bc2e8f3..872c9a2de 100644 --- a/src/graph_sitter/core/statements/block_statement.py +++ b/src/codegen/sdk/core/statements/block_statement.py @@ -5,19 +5,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.detached_symbols.code_block import CodeBlock TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock") diff --git a/src/graph_sitter/core/statements/catch_statement.py b/src/codegen/sdk/core/statements/catch_statement.py similarity index 64% rename from src/graph_sitter/core/statements/catch_statement.py rename to src/codegen/sdk/core/statements/catch_statement.py index 8ea0abd2e..28ea0314e 100644 --- a/src/graph_sitter/core/statements/catch_statement.py +++ b/src/codegen/sdk/core/statements/catch_statement.py @@ -2,15 +2,15 @@ from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.statements.block_statement import BlockStatement -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.statements.block_statement import BlockStatement +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock Parent = TypeVar("Parent", bound="CodeBlock") diff --git a/src/graph_sitter/core/statements/comment.py b/src/codegen/sdk/core/statements/comment.py similarity index 88% rename from src/graph_sitter/core/statements/comment.py rename to src/codegen/sdk/core/statements/comment.py index 3b882f957..6c24f33de 100644 --- a/src/graph_sitter/core/statements/comment.py +++ b/src/codegen/sdk/core/statements/comment.py @@ -4,16 +4,16 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock def lowest_indentation(text_blocks, skip_lines: int = 0): diff --git a/src/graph_sitter/core/statements/export_statement.py b/src/codegen/sdk/core/statements/export_statement.py similarity index 86% rename from src/graph_sitter/core/statements/export_statement.py rename to src/codegen/sdk/core/statements/export_statement.py index 0018b1a6d..014286c12 100644 --- a/src/graph_sitter/core/statements/export_statement.py +++ b/src/codegen/sdk/core/statements/export_statement.py @@ -4,19 +4,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.export import Export -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.export import TSExport -from graph_sitter.typescript.statements.import_statement import TSImportStatement -from graph_sitter.utils import find_first_ancestor -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.export import Export +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.export import TSExport +from codegen.sdk.typescript.statements.import_statement import TSImportStatement +from codegen.sdk.utils import find_first_ancestor +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/core/statements/expression_statement.py b/src/codegen/sdk/core/statements/expression_statement.py similarity index 73% rename from src/graph_sitter/core/statements/expression_statement.py rename to src/codegen/sdk/core/statements/expression_statement.py index bf4af289d..eec83ce13 100644 --- a/src/graph_sitter/core/statements/expression_statement.py +++ b/src/codegen/sdk/core/statements/expression_statement.py @@ -4,20 +4,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.wrapper_expression import IWrapper -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.wrapper_expression import IWrapper +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.interfaces.has_block import HasBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.interfaces.has_block import HasBlock Parent = TypeVar("Parent", bound="HasBlock") diff --git a/src/graph_sitter/core/statements/for_loop_statement.py b/src/codegen/sdk/core/statements/for_loop_statement.py similarity index 71% rename from src/graph_sitter/core/statements/for_loop_statement.py rename to src/codegen/sdk/core/statements/for_loop_statement.py index 82c96d3ff..87736bd6c 100644 --- a/src/graph_sitter/core/statements/for_loop_statement.py +++ b/src/codegen/sdk/core/statements/for_loop_statement.py @@ -3,18 +3,18 @@ from abc import ABC from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter.core.expressions import Expression -from graph_sitter.core.import_resolution import Import, WildcardImport -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.statements.block_statement import BlockStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.import_resolution import Import, WildcardImport +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.statements.block_statement import BlockStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock Parent = TypeVar("Parent", bound="CodeBlock") diff --git a/src/graph_sitter/core/statements/if_block_statement.py b/src/codegen/sdk/core/statements/if_block_statement.py similarity index 93% rename from src/graph_sitter/core/statements/if_block_statement.py rename to src/codegen/sdk/core/statements/if_block_statement.py index 6e3a87709..e21564601 100644 --- a/src/graph_sitter/core/statements/if_block_statement.py +++ b/src/codegen/sdk/core/statements/if_block_statement.py @@ -4,21 +4,21 @@ from functools import cached_property from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter._proxy import proxy_property -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.function import Function -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk._proxy import proxy_property +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.function import Function +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock TIfBlockStatement = TypeVar("TIfBlockStatement", bound="IfBlockStatement") diff --git a/src/graph_sitter/core/statements/import_statement.py b/src/codegen/sdk/core/statements/import_statement.py similarity index 68% rename from src/graph_sitter/core/statements/import_statement.py rename to src/codegen/sdk/core/statements/import_statement.py index 28714a22a..ba7a4d072 100644 --- a/src/graph_sitter/core/statements/import_statement.py +++ b/src/codegen/sdk/core/statements/import_statement.py @@ -4,19 +4,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.file import SourceFile - from graph_sitter.core.import_resolution import Import + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.file import SourceFile + from codegen.sdk.core.import_resolution import Import TSourceFile = TypeVar("TSourceFile", bound="SourceFile") diff --git a/src/graph_sitter/core/statements/raise_statement.py b/src/codegen/sdk/core/statements/raise_statement.py similarity index 71% rename from src/graph_sitter/core/statements/raise_statement.py rename to src/codegen/sdk/core/statements/raise_statement.py index 3c474104e..2b736a5db 100644 --- a/src/graph_sitter/core/statements/raise_statement.py +++ b/src/codegen/sdk/core/statements/raise_statement.py @@ -4,18 +4,18 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.detached_symbols.code_block import CodeBlock Parent = TypeVar("Parent", bound="CodeBlock") diff --git a/src/graph_sitter/core/statements/return_statement.py b/src/codegen/sdk/core/statements/return_statement.py similarity index 72% rename from src/graph_sitter/core/statements/return_statement.py rename to src/codegen/sdk/core/statements/return_statement.py index 97ad198a8..8351752a1 100644 --- a/src/graph_sitter/core/statements/return_statement.py +++ b/src/codegen/sdk/core/statements/return_statement.py @@ -4,19 +4,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.interfaces.has_block import HasBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.interfaces.has_block import HasBlock Parent = TypeVar("Parent", bound="HasBlock") diff --git a/src/graph_sitter/core/statements/statement.py b/src/codegen/sdk/core/statements/statement.py similarity index 85% rename from src/graph_sitter/core/statements/statement.py rename to src/codegen/sdk/core/statements/statement.py index ec35fe9d8..3602f35e9 100644 --- a/src/graph_sitter/core/statements/statement.py +++ b/src/codegen/sdk/core/statements/statement.py @@ -6,20 +6,20 @@ import rich.repr -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.multi_line_collection import MultiLineCollection -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.output.constants import ANGULAR_STYLE -from graph_sitter.utils import find_all_descendants -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.output.constants import ANGULAR_STYLE +from codegen.sdk.utils import find_all_descendants +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.detached_symbols.code_block import CodeBlock from tree_sitter import Node as TSNode @@ -131,7 +131,7 @@ def nested_statements(self) -> list[MultiLineCollection[Statement[Self], Parent] return nested_statements def _get_indent(self) -> int: - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock if isinstance(self.parent, CodeBlock): return self.parent.level * 4 diff --git a/src/graph_sitter/core/statements/switch_case.py b/src/codegen/sdk/core/statements/switch_case.py similarity index 58% rename from src/graph_sitter/core/statements/switch_case.py rename to src/codegen/sdk/core/statements/switch_case.py index 5d0057d8c..12c140726 100644 --- a/src/graph_sitter/core/statements/switch_case.py +++ b/src/codegen/sdk/core/statements/switch_case.py @@ -2,17 +2,17 @@ from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.statements.block_statement import BlockStatement -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.statements.block_statement import BlockStatement +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.assignment import Assignment - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.statements.switch_statement import SwitchStatement + from codegen.sdk.core.assignment import Assignment + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.statements.switch_statement import SwitchStatement Parent = TypeVar("Parent", bound="CodeBlock[SwitchStatement, Assignment]") diff --git a/src/graph_sitter/core/statements/switch_statement.py b/src/codegen/sdk/core/statements/switch_statement.py similarity index 78% rename from src/graph_sitter/core/statements/switch_statement.py rename to src/codegen/sdk/core/statements/switch_statement.py index 11391608b..091a3b60c 100644 --- a/src/graph_sitter/core/statements/switch_statement.py +++ b/src/codegen/sdk/core/statements/switch_statement.py @@ -2,18 +2,18 @@ from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.core.statements.switch_case import SwitchCase -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.core.statements.switch_case import SwitchCase +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock Parent = TypeVar("Parent", bound="CodeBlock") diff --git a/src/graph_sitter/core/statements/symbol_statement.py b/src/codegen/sdk/core/statements/symbol_statement.py similarity index 74% rename from src/graph_sitter/core/statements/symbol_statement.py rename to src/codegen/sdk/core/statements/symbol_statement.py index a227ebc05..91cae684e 100644 --- a/src/graph_sitter/core/statements/symbol_statement.py +++ b/src/codegen/sdk/core/statements/symbol_statement.py @@ -4,19 +4,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.interfaces.importable import Importable - from graph_sitter.core.symbol import Symbol + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.interfaces.importable import Importable + from codegen.sdk.core.symbol import Symbol Parent = TypeVar("Parent", bound="CodeBlock") diff --git a/src/graph_sitter/core/statements/try_catch_statement.py b/src/codegen/sdk/core/statements/try_catch_statement.py similarity index 65% rename from src/graph_sitter/core/statements/try_catch_statement.py rename to src/codegen/sdk/core/statements/try_catch_statement.py index 64516c648..153dc9fda 100644 --- a/src/graph_sitter/core/statements/try_catch_statement.py +++ b/src/codegen/sdk/core/statements/try_catch_statement.py @@ -3,13 +3,13 @@ from abc import ABC from typing import TYPE_CHECKING, Generic, TypeVar -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.statements.block_statement import BlockStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.statements.block_statement import BlockStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock Parent = TypeVar("Parent", bound="CodeBlock") diff --git a/src/graph_sitter/core/statements/while_statement.py b/src/codegen/sdk/core/statements/while_statement.py similarity index 75% rename from src/graph_sitter/core/statements/while_statement.py rename to src/codegen/sdk/core/statements/while_statement.py index 410a13ed9..7023ac692 100644 --- a/src/graph_sitter/core/statements/while_statement.py +++ b/src/codegen/sdk/core/statements/while_statement.py @@ -5,20 +5,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.detached_symbols.code_block import CodeBlock TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock") diff --git a/src/graph_sitter/core/symbol.py b/src/codegen/sdk/core/symbol.py similarity index 90% rename from src/graph_sitter/core/symbol.py rename to src/codegen/sdk/core/symbol.py index e7c8716d3..636e1cf09 100644 --- a/src/graph_sitter/core/symbol.py +++ b/src/codegen/sdk/core/symbol.py @@ -7,31 +7,31 @@ from rich.markup import escape from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind, UsageType -from graph_sitter.core.detached_symbols.argument import Argument -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Name, Value -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.expressions.defined_name import DefinedName -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.usable import Usable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement -from graph_sitter.core.symbol_groups.comment_group import CommentGroup -from graph_sitter.enums import ImportType, NodeType, SymbolType -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.output.constants import ANGULAR_STYLE -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind, UsageType +from codegen.sdk.core.detached_symbols.argument import Argument +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Name, Value +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.expressions.defined_name import DefinedName +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.usable import Usable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.core.symbol_groups.comment_group import CommentGroup +from codegen.sdk.enums import ImportType, NodeType, SymbolType +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.output.constants import ANGULAR_STYLE +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.code_block import CodeBlock - from graph_sitter.core.export import Export - from graph_sitter.core.file import SourceFile - from graph_sitter.core.import_resolution import Import - from graph_sitter.core.interfaces.has_block import HasBlock + from codegen.sdk.core.detached_symbols.code_block import CodeBlock + from codegen.sdk.core.export import Export + from codegen.sdk.core.file import SourceFile + from codegen.sdk.core.import_resolution import Import + from codegen.sdk.core.interfaces.has_block import HasBlock Parent = TypeVar("Parent", bound="HasBlock") TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock") @@ -59,7 +59,7 @@ def __init__( super().__init__(ts_node, file_id, G, parent) name_node = self._get_name_node(ts_node) if name_node is None else name_node self._name_node = self._parse_expression(name_node, default=name_node_type) - from graph_sitter.core.interfaces.has_block import HasBlock + from codegen.sdk.core.interfaces.has_block import HasBlock if isinstance(self, HasBlock): self.code_block = self._parse_code_block() @@ -76,7 +76,7 @@ def __rich_repr__(self) -> rich.repr.Result: @noapidoc def parent_symbol(self) -> Symbol | SourceFile | Import | Export: """Returns the parent symbol of the symbol.""" - from graph_sitter.core.export import Export + from codegen.sdk.core.export import Export parent = super().parent_symbol if parent == self.file or isinstance(parent, Export): @@ -104,7 +104,7 @@ def extended_nodes(self) -> list[Editable]: list[Editable]: A list of Editable nodes containing the current symbol and its extended symbols, sorted in the correct order. """ - from graph_sitter.core.interfaces.has_block import HasBlock + from codegen.sdk.core.interfaces.has_block import HasBlock comment_nodes = self.comment.symbols if self.comment else [] inline_comment_nodes = self.inline_comment.symbols if self.inline_comment else [] @@ -287,7 +287,7 @@ def move_to_file(self, file: SourceFile, include_dependencies: bool = True, stra @noapidoc def _move_to_file(self, file: SourceFile, encountered_symbols: set[Symbol | Import], include_dependencies: bool = True, strategy: str = "update_all_imports") -> tuple[NodeId, NodeId]: """Helper recursive function for `move_to_file`""" - from graph_sitter.core.import_resolution import Import + from codegen.sdk.core.import_resolution import Import # =====[ Arg checking ]===== if file == self.file: @@ -372,7 +372,7 @@ def _move_to_file(self, file: SourceFile, encountered_symbols: set[Symbol | Impo @noapidoc def is_top_level(self) -> bool: """Is this symbol a top-level symbol: does it have a level of 0?""" - from graph_sitter.core.file import File + from codegen.sdk.core.file import File parent = self.parent while not isinstance(parent, Symbol | Argument): @@ -411,7 +411,7 @@ def add_keyword(self, keyword: str): @property @noapidoc def descendant_symbols(self) -> list[Importable]: - from graph_sitter.core.interfaces.has_block import HasBlock + from codegen.sdk.core.interfaces.has_block import HasBlock symbols = [self] if isinstance(self, HasBlock): diff --git a/src/graph_sitter/core/symbol_group.py b/src/codegen/sdk/core/symbol_group.py similarity index 95% rename from src/graph_sitter/core/symbol_group.py rename to src/codegen/sdk/core/symbol_group.py index 0dc6a9471..7572c6983 100644 --- a/src/graph_sitter/core/symbol_group.py +++ b/src/codegen/sdk/core/symbol_group.py @@ -5,17 +5,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader, repr_func, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.output.ast import AST -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import reader, repr_func, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.output.ast import AST +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Child = TypeVar("Child", bound="Editable") diff --git a/src/graph_sitter/core/symbol_groups/__init__.py b/src/codegen/sdk/core/symbol_groups/__init__.py similarity index 100% rename from src/graph_sitter/core/symbol_groups/__init__.py rename to src/codegen/sdk/core/symbol_groups/__init__.py diff --git a/src/graph_sitter/core/symbol_groups/collection.py b/src/codegen/sdk/core/symbol_groups/collection.py similarity index 96% rename from src/graph_sitter/core/symbol_groups/collection.py rename to src/codegen/sdk/core/symbol_groups/collection.py index dc2e27152..b08f2de28 100644 --- a/src/graph_sitter/core/symbol_groups/collection.py +++ b/src/codegen/sdk/core/symbol_groups/collection.py @@ -4,15 +4,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.transactions import TransactionPriority -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_group import SymbolGroup -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk.codebase.transactions import TransactionPriority +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_group import SymbolGroup +from codegen.sdk.writer_decorators import noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Child = TypeVar("Child", bound="Editable") diff --git a/src/graph_sitter/core/symbol_groups/comment_group.py b/src/codegen/sdk/core/symbol_groups/comment_group.py similarity index 87% rename from src/graph_sitter/core/symbol_groups/comment_group.py rename to src/codegen/sdk/core/symbol_groups/comment_group.py index 7d47d884a..1c11cf999 100644 --- a/src/graph_sitter/core/symbol_groups/comment_group.py +++ b/src/codegen/sdk/core/symbol_groups/comment_group.py @@ -2,12 +2,12 @@ from typing import TypeVar -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.comment import Comment -from graph_sitter.core.symbol_group import SymbolGroup -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.comment import Comment +from codegen.sdk.core.symbol_group import SymbolGroup +from codegen.sdk.writer_decorators import apidoc, noapidoc Parent = TypeVar("Parent") diff --git a/src/graph_sitter/core/symbol_groups/dict.py b/src/codegen/sdk/core/symbol_groups/dict.py similarity index 88% rename from src/graph_sitter/core/symbol_groups/dict.py rename to src/codegen/sdk/core/symbol_groups/dict.py index abace83a2..9d55f8667 100644 --- a/src/graph_sitter/core/symbol_groups/dict.py +++ b/src/codegen/sdk/core/symbol_groups/dict.py @@ -3,22 +3,22 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.builtin import Builtin -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.expressions.string import String -from graph_sitter.core.expressions.unpack import Unpack -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.builtin import Builtin +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.expressions.string import String +from codegen.sdk.core.expressions.unpack import Unpack +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.interfaces.importable import Importable + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.interfaces.importable import Importable TExpression = TypeVar("TExpression", bound="Expression") diff --git a/src/graph_sitter/core/symbol_groups/expression_group.py b/src/codegen/sdk/core/symbol_groups/expression_group.py similarity index 85% rename from src/graph_sitter/core/symbol_groups/expression_group.py rename to src/codegen/sdk/core/symbol_groups/expression_group.py index 880e8e3dd..d8b7d7d57 100644 --- a/src/graph_sitter/core/symbol_groups/expression_group.py +++ b/src/codegen/sdk/core/symbol_groups/expression_group.py @@ -2,11 +2,11 @@ from typing import Generic, TypeVar -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.symbol_group import SymbolGroup -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.symbol_group import SymbolGroup +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc Parent = TypeVar("Parent") diff --git a/src/graph_sitter/core/symbol_groups/list.py b/src/codegen/sdk/core/symbol_groups/list.py similarity index 60% rename from src/graph_sitter/core/symbol_groups/list.py rename to src/codegen/sdk/core/symbol_groups/list.py index 02074f99b..4fc6c3729 100644 --- a/src/graph_sitter/core/symbol_groups/list.py +++ b/src/codegen/sdk/core/symbol_groups/list.py @@ -2,15 +2,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.expressions.builtin import Builtin -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.expressions.builtin import Builtin +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound=Editable) diff --git a/src/graph_sitter/core/symbol_groups/multi_line_collection.py b/src/codegen/sdk/core/symbol_groups/multi_line_collection.py similarity index 92% rename from src/graph_sitter/core/symbol_groups/multi_line_collection.py rename to src/codegen/sdk/core/symbol_groups/multi_line_collection.py index 4f2d7d465..1bc58fec8 100644 --- a/src/graph_sitter/core/symbol_groups/multi_line_collection.py +++ b/src/codegen/sdk/core/symbol_groups/multi_line_collection.py @@ -4,13 +4,13 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Child = TypeVar("Child", bound=Editable) diff --git a/src/graph_sitter/core/symbol_groups/parents.py b/src/codegen/sdk/core/symbol_groups/parents.py similarity index 79% rename from src/graph_sitter/core/symbol_groups/parents.py rename to src/codegen/sdk/core/symbol_groups/parents.py index 6eea0eb89..385125b69 100644 --- a/src/graph_sitter/core/symbol_groups/parents.py +++ b/src/codegen/sdk/core/symbol_groups/parents.py @@ -5,20 +5,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.enums import EdgeType +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.enums import EdgeType if TYPE_CHECKING: - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.interfaces.inherits import Inherits + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.interfaces.inherits import Inherits TType = TypeVar("TType", bound="Type") @@ -59,8 +59,8 @@ def _compute_dependencies(self, usage_type: UsageKind = UsageKind.SUBCLASS, dest @reader def is_subclass_of(self, parent: str | HasName, max_depth: int | None = None) -> bool: """Returns True if the class is a subclass of the given parent class.""" - from graph_sitter.core.class_definition import Class - from graph_sitter.core.interface import Interface + from codegen.sdk.core.class_definition import Class + from codegen.sdk.core.interface import Interface if isinstance(parent, HasName): parent = parent.name diff --git a/src/graph_sitter/core/symbol_groups/tuple.py b/src/codegen/sdk/core/symbol_groups/tuple.py similarity index 60% rename from src/graph_sitter/core/symbol_groups/tuple.py rename to src/codegen/sdk/core/symbol_groups/tuple.py index 417df1552..ad981982d 100644 --- a/src/graph_sitter/core/symbol_groups/tuple.py +++ b/src/codegen/sdk/core/symbol_groups/tuple.py @@ -2,15 +2,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.expressions.builtin import Builtin -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.expressions.builtin import Builtin +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound=Editable) diff --git a/src/graph_sitter/core/symbol_groups/type_parameters.py b/src/codegen/sdk/core/symbol_groups/type_parameters.py similarity index 64% rename from src/graph_sitter/core/symbol_groups/type_parameters.py rename to src/codegen/sdk/core/symbol_groups/type_parameters.py index bcb12af27..6d3079576 100644 --- a/src/graph_sitter/core/symbol_groups/type_parameters.py +++ b/src/codegen/sdk/core/symbol_groups/type_parameters.py @@ -4,13 +4,13 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection if TYPE_CHECKING: - from graph_sitter.core.expressions.type import Type - from graph_sitter.core.interfaces.supports_generic import SupportsGenerics + from codegen.sdk.core.expressions.type import Type + from codegen.sdk.core.interfaces.supports_generic import SupportsGenerics TType = TypeVar("TType", bound="Type") diff --git a/src/graph_sitter/core/type_alias.py b/src/codegen/sdk/core/type_alias.py similarity index 69% rename from src/graph_sitter/core/type_alias.py rename to src/codegen/sdk/core/type_alias.py index f958424d5..702b1b165 100644 --- a/src/graph_sitter/core/type_alias.py +++ b/src/codegen/sdk/core/type_alias.py @@ -5,21 +5,21 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.interfaces.supports_generic import SupportsGenerics -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.attribute import Attribute -from graph_sitter.core.statements.statement import Statement -from graph_sitter.enums import SymbolType -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.interfaces.supports_generic import SupportsGenerics +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.attribute import Attribute +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.enums import SymbolType +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock") diff --git a/src/graph_sitter/enums.py b/src/codegen/sdk/enums.py similarity index 98% rename from src/graph_sitter/enums.py rename to src/codegen/sdk/enums.py index 05abb8cf2..c98391b72 100644 --- a/src/graph_sitter/enums.py +++ b/src/codegen/sdk/enums.py @@ -3,7 +3,7 @@ from enum import IntEnum, StrEnum, auto from typing import TYPE_CHECKING, NamedTuple -from graph_sitter.core.dataclasses.usage import Usage +from codegen.sdk.core.dataclasses.usage import Usage if TYPE_CHECKING: pass diff --git a/src/graph_sitter/exceptions.py b/src/codegen/sdk/exceptions.py similarity index 100% rename from src/graph_sitter/exceptions.py rename to src/codegen/sdk/exceptions.py diff --git a/src/graph_sitter/extensions/autocommit.pyi b/src/codegen/sdk/extensions/autocommit.pyi similarity index 88% rename from src/graph_sitter/extensions/autocommit.pyi rename to src/codegen/sdk/extensions/autocommit.pyi index 999f2d88d..fea4fad06 100644 --- a/src/graph_sitter/extensions/autocommit.pyi +++ b/src/codegen/sdk/extensions/autocommit.pyi @@ -1,8 +1,8 @@ from collections.abc import Callable from typing import Any, ParamSpec, TypeVar, overload -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.interfaces.editable import Editable +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.interfaces.editable import Editable P = ParamSpec("P") T = TypeVar("T") diff --git a/src/graph_sitter/extensions/autocommit.pyx b/src/codegen/sdk/extensions/autocommit.pyx similarity index 96% rename from src/graph_sitter/extensions/autocommit.pyx rename to src/codegen/sdk/extensions/autocommit.pyx index 87263b68a..cde308e15 100644 --- a/src/graph_sitter/extensions/autocommit.pyx +++ b/src/codegen/sdk/extensions/autocommit.pyx @@ -4,14 +4,14 @@ from typing import Any, ParamSpec, TypeVar, Union, overload import wrapt -from graph_sitter.core.autocommit.constants import AutoCommitState, OutdatedNodeError, enabled +from codegen.sdk.core.autocommit.constants import AutoCommitState, OutdatedNodeError, enabled P = ParamSpec("P") T = TypeVar("T") def is_outdated(c) -> bool: - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.interfaces.editable import Editable if isinstance(c, Editable): return c.is_outdated @@ -120,7 +120,7 @@ def _delay_update(new_value) -> bool: def update_dict(seen: set["Editable"], obj: "Editable", new_obj: "Editable"): - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.interfaces.editable import Editable if obj in seen or obj.removed: return @@ -199,7 +199,7 @@ def commiter(wrapped: Callable[P, T] | None = None, *, reset: bool = False) -> C def wrapper(wrapped: Callable[P, T], instance: Union["Editable", "CodebaseGraph", None] = None, args: P.args = None, kwargs: P.kwargs = None) -> T: if instance is None: instance = args[0] - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph if isinstance(instance, CodebaseGraph): autocommit = instance._autocommit diff --git a/src/graph_sitter/extensions/py.typed b/src/codegen/sdk/extensions/py.typed similarity index 100% rename from src/graph_sitter/extensions/py.typed rename to src/codegen/sdk/extensions/py.typed diff --git a/src/graph_sitter/extensions/resolution.pyi b/src/codegen/sdk/extensions/resolution.pyi similarity index 87% rename from src/graph_sitter/extensions/resolution.pyi rename to src/codegen/sdk/extensions/resolution.pyi index c684ee660..e0f0c8324 100644 --- a/src/graph_sitter/extensions/resolution.pyi +++ b/src/codegen/sdk/extensions/resolution.pyi @@ -4,10 +4,10 @@ from typing import Generic from typing_extensions import TypeVar -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.dataclasses.usage import UsageKind, UsageType -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.dataclasses.usage import UsageKind, UsageType +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName NodeType = TypeVar("NodeType") diff --git a/src/graph_sitter/extensions/resolution.pyx b/src/codegen/sdk/extensions/resolution.pyx similarity index 93% rename from src/graph_sitter/extensions/resolution.pyx rename to src/codegen/sdk/extensions/resolution.pyx index eb421dd1c..47643172a 100644 --- a/src/graph_sitter/extensions/resolution.pyx +++ b/src/codegen/sdk/extensions/resolution.pyx @@ -5,12 +5,12 @@ from typing import TYPE_CHECKING, Generic from typing_extensions import TypeVar -from graph_sitter.core.dataclasses.usage import Usage, UsageKind, UsageType -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.enums import Edge, EdgeType +from codegen.sdk.core.dataclasses.usage import Usage, UsageKind, UsageType +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.enums import Edge, EdgeType if TYPE_CHECKING: - from graph_sitter.core.import_resolution import Import + from codegen.sdk.core.import_resolution import Import NodeType = TypeVar("NodeType", bound=Editable) @@ -71,7 +71,7 @@ class ResolutionStack(Generic[NodeType]): usage = Usage(kind=usage_type, match=match, usage_type=edge_usage_type, usage_symbol=dest.parent_symbol, imported_by=imported_by) yield dest.node_id, self.node.node_id, Edge(type=EdgeType.SYMBOL_USAGE, usage=usage) if self.parent_frame is not None: - from graph_sitter.core.import_resolution import Import + from codegen.sdk.core.import_resolution import Import if isinstance(self, Import): imported_by = self diff --git a/src/graph_sitter/extensions/sort.pyx b/src/codegen/sdk/extensions/sort.pyx similarity index 96% rename from src/graph_sitter/extensions/sort.pyx rename to src/codegen/sdk/extensions/sort.pyx index 9960f98df..573ff0d9c 100644 --- a/src/graph_sitter/extensions/sort.pyx +++ b/src/codegen/sdk/extensions/sort.pyx @@ -6,7 +6,7 @@ from collections.abc import Iterable, Sequence from tree_sitter import Node as TSNode from typing_extensions import TypeVar -from graph_sitter.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.editable import Editable E = TypeVar("E", bound=Editable) diff --git a/src/graph_sitter/extensions/utils.pyi b/src/codegen/sdk/extensions/utils.pyi similarity index 100% rename from src/graph_sitter/extensions/utils.pyi rename to src/codegen/sdk/extensions/utils.pyi diff --git a/src/graph_sitter/extensions/utils.pyx b/src/codegen/sdk/extensions/utils.pyx similarity index 100% rename from src/graph_sitter/extensions/utils.pyx rename to src/codegen/sdk/extensions/utils.pyx diff --git a/src/graph_sitter/fetch_codebase.py b/src/codegen/sdk/fetch_codebase.py similarity index 85% rename from src/graph_sitter/fetch_codebase.py rename to src/codegen/sdk/fetch_codebase.py index 994f9147d..27812a738 100644 --- a/src/graph_sitter/fetch_codebase.py +++ b/src/codegen/sdk/fetch_codebase.py @@ -1,10 +1,10 @@ import logging import os +from codegen.sdk.codebase.config import DefaultConfig, ProjectConfig +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.utils import determine_project_language from codegen_git.repo_operator.local_repo_operator import LocalRepoOperator -from graph_sitter.codebase.config import DefaultConfig, ProjectConfig -from graph_sitter.core.codebase import Codebase -from graph_sitter.utils import determine_project_language logger = logging.getLogger(__name__) @@ -23,16 +23,16 @@ def fetch_codebase(repo_name: str, *, tmp_dir: str | None = None, shallow: bool Codebase: A Codebase instance initialized with the cloned repository Example: ```python - import graph_sitter + import codegen.sdk as sdk import logging # Enable logging to see progress logging.basicConfig(level=logging.INFO) # Clone a repository to default location (/tmp/codegen) - codebase = graph_sitter.fetch_codebase('facebook/react') + codebase = sdk.fetch_codebase('facebook/react') # Or specify a custom directory - codebase = graph_sitter.fetch_codebase('facebook/react', tmp_dir='~/my_repos') + codebase = sdk.fetch_codebase('facebook/react', tmp_dir='~/my_repos') # Or clone a specific commit - codebase = graph_sitter.fetch_codebase('facebook/react', commit_hash='abc123') + codebase = sdk.fetch_codebase('facebook/react', commit_hash='abc123') ``` """ logger.info(f"Fetching codebase for {repo_name}") diff --git a/src/graph_sitter/output/ast.py b/src/codegen/sdk/output/ast.py similarity index 77% rename from src/graph_sitter/output/ast.py rename to src/codegen/sdk/output/ast.py index a1055652c..da23005a7 100644 --- a/src/graph_sitter/output/ast.py +++ b/src/codegen/sdk/output/ast.py @@ -3,12 +3,12 @@ from openai import BaseModel from pydantic import ConfigDict -from graph_sitter.codebase.span import Span +from codegen.sdk.codebase.span import Span class AST(BaseModel): model_config = ConfigDict(frozen=True) - graph_sitter_type: str + codegen_sdk_type: str span: Span tree_sitter_type: str children: list[tuple[str | None, Self]] diff --git a/src/graph_sitter/output/constants.py b/src/codegen/sdk/output/constants.py similarity index 100% rename from src/graph_sitter/output/constants.py rename to src/codegen/sdk/output/constants.py diff --git a/src/graph_sitter/output/inspect.py b/src/codegen/sdk/output/inspect.py similarity index 89% rename from src/graph_sitter/output/inspect.py rename to src/codegen/sdk/output/inspect.py index 749e07258..d9f4a83cf 100644 --- a/src/graph_sitter/output/inspect.py +++ b/src/codegen/sdk/output/inspect.py @@ -2,7 +2,7 @@ from collections.abc import Callable from inspect import Parameter -from graph_sitter.writer_decorators import DocumentedObject, no_apidoc_signatures +from codegen.sdk.writer_decorators import DocumentedObject, no_apidoc_signatures def only_default_args(method: Callable) -> bool: diff --git a/src/graph_sitter/output/jsonable.py b/src/codegen/sdk/output/jsonable.py similarity index 89% rename from src/graph_sitter/output/jsonable.py rename to src/codegen/sdk/output/jsonable.py index 0a8bfb37a..2d82ba8cc 100644 --- a/src/graph_sitter/output/jsonable.py +++ b/src/codegen/sdk/output/jsonable.py @@ -3,13 +3,13 @@ from tree_sitter import Node as TSNode -from graph_sitter._proxy import ProxyProperty -from graph_sitter.codebase.span import Span -from graph_sitter.output.inspect import is_noapidoc, only_default_args -from graph_sitter.output.placeholder import Placeholder -from graph_sitter.output.utils import safe_getattr -from graph_sitter.types import JSON -from graph_sitter.writer_decorators import noapidoc +from codegen.sdk._proxy import ProxyProperty +from codegen.sdk.codebase.span import Span +from codegen.sdk.output.inspect import is_noapidoc, only_default_args +from codegen.sdk.output.placeholder import Placeholder +from codegen.sdk.output.utils import safe_getattr +from codegen.sdk.types import JSON +from codegen.sdk.writer_decorators import noapidoc BLACKLIST = ["json", "G", "viz", "autocommit_cache", "ts_node", "symbol_usages", "usages"] diff --git a/src/graph_sitter/output/placeholder.py b/src/codegen/sdk/output/placeholder.py similarity index 80% rename from src/graph_sitter/output/placeholder.py rename to src/codegen/sdk/output/placeholder.py index bef362bad..f12b3516b 100644 --- a/src/graph_sitter/output/placeholder.py +++ b/src/codegen/sdk/output/placeholder.py @@ -1,6 +1,6 @@ from pydantic import BaseModel, ConfigDict -from graph_sitter.codebase.span import Span +from codegen.sdk.codebase.span import Span class Placeholder(BaseModel): diff --git a/src/graph_sitter/output/utils.py b/src/codegen/sdk/output/utils.py similarity index 98% rename from src/graph_sitter/output/utils.py rename to src/codegen/sdk/output/utils.py index 77a018394..5dfe8a21d 100644 --- a/src/graph_sitter/output/utils.py +++ b/src/codegen/sdk/output/utils.py @@ -10,7 +10,7 @@ from tree_sitter import Node as TSNode from tree_sitter import Point -from graph_sitter.output.constants import MAX_EDITABLE_LINES +from codegen.sdk.output.constants import MAX_EDITABLE_LINES def style_editable(ts_node: TSNode, filepath: PathLike, file_node: TSNode) -> RenderResult: diff --git a/src/graph_sitter/py.typed b/src/codegen/sdk/py.typed similarity index 100% rename from src/graph_sitter/py.typed rename to src/codegen/sdk/py.typed diff --git a/src/codegen/sdk/python/__init__.py b/src/codegen/sdk/python/__init__.py new file mode 100644 index 000000000..1c007891c --- /dev/null +++ b/src/codegen/sdk/python/__init__.py @@ -0,0 +1,15 @@ +from codegen.sdk.python.assignment import PyAssignment +from codegen.sdk.python.class_definition import PyClass +from codegen.sdk.python.file import PyFile +from codegen.sdk.python.function import PyFunction +from codegen.sdk.python.import_resolution import PyImport +from codegen.sdk.python.symbol import PySymbol + +__all__ = [ + "PyAssignment", + "PyClass", + "PyFile", + "PyFunction", + "PyImport", + "PySymbol", +] diff --git a/src/graph_sitter/python/assignment.py b/src/codegen/sdk/python/assignment.py similarity index 86% rename from src/graph_sitter/python/assignment.py rename to src/codegen/sdk/python/assignment.py index 30e663d0b..aff2142a7 100644 --- a/src/graph_sitter/python/assignment.py +++ b/src/codegen/sdk/python/assignment.py @@ -4,17 +4,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.expressions.multi_expression import MultiExpression -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.extensions.autocommit import reader -from graph_sitter.python.symbol import PySymbol -from graph_sitter.python.symbol_groups.comment_group import PyCommentGroup -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.expressions.multi_expression import MultiExpression +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.python.symbol import PySymbol +from codegen.sdk.python.symbol_groups.comment_group import PyCommentGroup +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: - from graph_sitter.python.statements.assignment_statement import PyAssignmentStatement + from codegen.sdk.python.statements.assignment_statement import PyAssignmentStatement @py_apidoc diff --git a/src/graph_sitter/python/class_definition.py b/src/codegen/sdk/python/class_definition.py similarity index 79% rename from src/graph_sitter/python/class_definition.py rename to src/codegen/sdk/python/class_definition.py index f757aef05..e16a2cfe9 100644 --- a/src/graph_sitter/python/class_definition.py +++ b/src/codegen/sdk/python/class_definition.py @@ -2,25 +2,25 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.class_definition import Class -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.generic_type import GenericType -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_group import SymbolGroup -from graph_sitter.core.symbol_groups.multi_line_collection import MultiLineCollection -from graph_sitter.core.symbol_groups.parents import Parents -from graph_sitter.extensions.utils import cached_property -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.detached_symbols.decorator import PyDecorator -from graph_sitter.python.detached_symbols.parameter import PyParameter -from graph_sitter.python.expressions.type import PyType -from graph_sitter.python.function import PyFunction -from graph_sitter.python.interfaces.has_block import PyHasBlock -from graph_sitter.python.symbol import PySymbol -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.generic_type import GenericType +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_group import SymbolGroup +from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection +from codegen.sdk.core.symbol_groups.parents import Parents +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.detached_symbols.decorator import PyDecorator +from codegen.sdk.python.detached_symbols.parameter import PyParameter +from codegen.sdk.python.expressions.type import PyType +from codegen.sdk.python.function import PyFunction +from codegen.sdk.python.interfaces.has_block import PyHasBlock +from codegen.sdk.python.symbol import PySymbol +from codegen.sdk.writer_decorators import noapidoc, py_apidoc @py_apidoc diff --git a/src/graph_sitter/python/detached_symbols/code_block.py b/src/codegen/sdk/python/detached_symbols/code_block.py similarity index 78% rename from src/graph_sitter/python/detached_symbols/code_block.py rename to src/codegen/sdk/python/detached_symbols/code_block.py index ba65639eb..ac2c14263 100644 --- a/src/graph_sitter/python/detached_symbols/code_block.py +++ b/src/codegen/sdk/python/detached_symbols/code_block.py @@ -2,17 +2,17 @@ from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter.core.autocommit import reader -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.statements.block_statement import BlockStatement -from graph_sitter.core.statements.import_statement import ImportStatement -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.core.symbol_groups.multi_line_collection import MultiLineCollection -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.statements.block_statement import BlockStatement +from codegen.sdk.core.statements.import_statement import ImportStatement +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: - from graph_sitter.python.interfaces.has_block import PyHasBlock - from graph_sitter.python.statements.with_statement import WithStatement + from codegen.sdk.python.interfaces.has_block import PyHasBlock + from codegen.sdk.python.statements.with_statement import WithStatement Parent = TypeVar("Parent", bound="PyHasBlock") diff --git a/src/graph_sitter/python/detached_symbols/decorator.py b/src/codegen/sdk/python/detached_symbols/decorator.py similarity index 87% rename from src/graph_sitter/python/detached_symbols/decorator.py rename to src/codegen/sdk/python/detached_symbols/decorator.py index 427f8fee1..5931b7010 100644 --- a/src/graph_sitter/python/detached_symbols/decorator.py +++ b/src/codegen/sdk/python/detached_symbols/decorator.py @@ -2,10 +2,10 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.detached_symbols.decorator import Decorator -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.detached_symbols.decorator import Decorator +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.writer_decorators import py_apidoc @py_apidoc diff --git a/src/graph_sitter/python/detached_symbols/parameter.py b/src/codegen/sdk/python/detached_symbols/parameter.py similarity index 89% rename from src/graph_sitter/python/detached_symbols/parameter.py rename to src/codegen/sdk/python/detached_symbols/parameter.py index e8a048ad5..e6196e582 100644 --- a/src/graph_sitter/python/detached_symbols/parameter.py +++ b/src/codegen/sdk/python/detached_symbols/parameter.py @@ -1,10 +1,10 @@ from typing_extensions import deprecated -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.python.expressions.type import PyType -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.python.expressions.type import PyType +from codegen.sdk.writer_decorators import py_apidoc @py_apidoc diff --git a/src/graph_sitter/python/expressions/chained_attribute.py b/src/codegen/sdk/python/expressions/chained_attribute.py similarity index 68% rename from src/graph_sitter/python/expressions/chained_attribute.py rename to src/codegen/sdk/python/expressions/chained_attribute.py index d55361351..b615be6c2 100644 --- a/src/graph_sitter/python/expressions/chained_attribute.py +++ b/src/codegen/sdk/python/expressions/chained_attribute.py @@ -1,9 +1,9 @@ from typing import Generic, TypeVar -from graph_sitter.core.expressions import Expression, Name -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.expressions import Expression, Name +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.writer_decorators import py_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/python/expressions/conditional_expression.py b/src/codegen/sdk/python/expressions/conditional_expression.py similarity index 71% rename from src/graph_sitter/python/expressions/conditional_expression.py rename to src/codegen/sdk/python/expressions/conditional_expression.py index 575e2a04b..bb9e013d3 100644 --- a/src/graph_sitter/python/expressions/conditional_expression.py +++ b/src/codegen/sdk/python/expressions/conditional_expression.py @@ -1,8 +1,8 @@ from typing import TypeVar -from graph_sitter.core.expressions.ternary_expression import TernaryExpression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.expressions.ternary_expression import TernaryExpression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.writer_decorators import py_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/python/expressions/generic_type.py b/src/codegen/sdk/python/expressions/generic_type.py similarity index 82% rename from src/graph_sitter/python/expressions/generic_type.py rename to src/codegen/sdk/python/expressions/generic_type.py index 0088a867d..661fb7327 100644 --- a/src/graph_sitter/python/expressions/generic_type.py +++ b/src/codegen/sdk/python/expressions/generic_type.py @@ -2,13 +2,13 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.expressions.generic_type import GenericType -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.python.expressions.named_type import PyNamedType -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.expressions.generic_type import GenericType +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.python.expressions.named_type import PyNamedType +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: - from graph_sitter.python.expressions.type import PyType + from codegen.sdk.python.expressions.type import PyType import logging logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/python/expressions/named_type.py b/src/codegen/sdk/python/expressions/named_type.py similarity index 75% rename from src/graph_sitter/python/expressions/named_type.py rename to src/codegen/sdk/python/expressions/named_type.py index 2b6335b13..ca7bbecd3 100644 --- a/src/graph_sitter/python/expressions/named_type.py +++ b/src/codegen/sdk/python/expressions/named_type.py @@ -2,8 +2,8 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.expressions.named_type import NamedType -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.expressions.named_type import NamedType +from codegen.sdk.writer_decorators import py_apidoc Parent = TypeVar("Parent") diff --git a/src/graph_sitter/python/expressions/string.py b/src/codegen/sdk/python/expressions/string.py similarity index 73% rename from src/graph_sitter/python/expressions/string.py rename to src/codegen/sdk/python/expressions/string.py index fdfd0dac0..5f781a7b0 100644 --- a/src/graph_sitter/python/expressions/string.py +++ b/src/codegen/sdk/python/expressions/string.py @@ -2,12 +2,12 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.expressions import Expression, String -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.expressions import Expression, String +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound="Expression") diff --git a/src/graph_sitter/python/expressions/type.py b/src/codegen/sdk/python/expressions/type.py similarity index 100% rename from src/graph_sitter/python/expressions/type.py rename to src/codegen/sdk/python/expressions/type.py diff --git a/src/graph_sitter/python/expressions/union_type.py b/src/codegen/sdk/python/expressions/union_type.py similarity index 68% rename from src/graph_sitter/python/expressions/union_type.py rename to src/codegen/sdk/python/expressions/union_type.py index 98925a4a9..f5869117e 100644 --- a/src/graph_sitter/python/expressions/union_type.py +++ b/src/codegen/sdk/python/expressions/union_type.py @@ -1,7 +1,7 @@ from typing import TYPE_CHECKING, Generic, TypeVar -from graph_sitter.core.expressions.union_type import UnionType -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.expressions.union_type import UnionType +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/python/file.py b/src/codegen/sdk/python/file.py similarity index 87% rename from src/graph_sitter/python/file.py rename to src/codegen/sdk/python/file.py index 136386ca8..6f739c947 100644 --- a/src/graph_sitter/python/file.py +++ b/src/codegen/sdk/python/file.py @@ -1,22 +1,22 @@ from __future__ import annotations -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.file import SourceFile -from graph_sitter.core.interface import Interface -from graph_sitter.enums import ImportType, ProgrammingLanguage -from graph_sitter.extensions.utils import iter_all_descendants -from graph_sitter.python import PyAssignment -from graph_sitter.python.class_definition import PyClass -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.expressions.type import PyType -from graph_sitter.python.function import PyFunction -from graph_sitter.python.import_resolution import PyImport -from graph_sitter.python.interfaces.has_block import PyHasBlock -from graph_sitter.python.statements.attribute import PyAttribute -from graph_sitter.python.statements.import_statement import PyImportStatement -from graph_sitter.python.symbol import PySymbol -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.file import SourceFile +from codegen.sdk.core.interface import Interface +from codegen.sdk.enums import ImportType, ProgrammingLanguage +from codegen.sdk.extensions.utils import iter_all_descendants +from codegen.sdk.python import PyAssignment +from codegen.sdk.python.class_definition import PyClass +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.expressions.type import PyType +from codegen.sdk.python.function import PyFunction +from codegen.sdk.python.import_resolution import PyImport +from codegen.sdk.python.interfaces.has_block import PyHasBlock +from codegen.sdk.python.statements.attribute import PyAttribute +from codegen.sdk.python.statements.import_statement import PyImportStatement +from codegen.sdk.python.symbol import PySymbol +from codegen.sdk.writer_decorators import noapidoc, py_apidoc @py_apidoc diff --git a/src/graph_sitter/python/function.py b/src/codegen/sdk/python/function.py similarity index 89% rename from src/graph_sitter/python/function.py rename to src/codegen/sdk/python/function.py index b5346222e..4c81a9705 100644 --- a/src/graph_sitter/python/function.py +++ b/src/codegen/sdk/python/function.py @@ -6,24 +6,24 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.function import Function -from graph_sitter.core.import_resolution import Import, WildcardImport -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.utils import cached_property -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.detached_symbols.decorator import PyDecorator -from graph_sitter.python.detached_symbols.parameter import PyParameter -from graph_sitter.python.expressions.type import PyType -from graph_sitter.python.interfaces.has_block import PyHasBlock -from graph_sitter.python.placeholder.placeholder_return_type import PyReturnTypePlaceholder -from graph_sitter.python.symbol import PySymbol -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.function import Function +from codegen.sdk.core.import_resolution import Import, WildcardImport +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.detached_symbols.decorator import PyDecorator +from codegen.sdk.python.detached_symbols.parameter import PyParameter +from codegen.sdk.python.expressions.type import PyType +from codegen.sdk.python.interfaces.has_block import PyHasBlock +from codegen.sdk.python.placeholder.placeholder_return_type import PyReturnTypePlaceholder +from codegen.sdk.python.symbol import PySymbol +from codegen.sdk.writer_decorators import noapidoc, py_apidoc logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/python/import_resolution.py b/src/codegen/sdk/python/import_resolution.py similarity index 95% rename from src/graph_sitter/python/import_resolution.py rename to src/codegen/sdk/python/import_resolution.py index 741bdad25..f2110ba9d 100644 --- a/src/graph_sitter/python/import_resolution.py +++ b/src/codegen/sdk/python/import_resolution.py @@ -5,19 +5,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import reader -from graph_sitter.core.expressions import Name -from graph_sitter.core.import_resolution import ExternalImportResolver, Import, ImportResolution -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.exportable import Exportable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.import_statement import ImportStatement -from graph_sitter.enums import ImportType, NodeType -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.expressions import Name +from codegen.sdk.core.import_resolution import ExternalImportResolver, Import, ImportResolution +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.exportable import Exportable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.import_statement import ImportStatement +from codegen.sdk.enums import ImportType, NodeType +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: - from graph_sitter.python.file import PyFile + from codegen.sdk.python.file import PyFile import logging diff --git a/src/graph_sitter/python/interfaces/has_block.py b/src/codegen/sdk/python/interfaces/has_block.py similarity index 87% rename from src/graph_sitter/python/interfaces/has_block.py rename to src/codegen/sdk/python/interfaces/has_block.py index ec249416e..bc44a6f21 100644 --- a/src/graph_sitter/python/interfaces/has_block.py +++ b/src/codegen/sdk/python/interfaces/has_block.py @@ -1,12 +1,12 @@ from functools import cached_property -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.detached_symbols.decorator import PyDecorator -from graph_sitter.python.statements.comment import PyComment, PyCommentType -from graph_sitter.python.symbol_groups.comment_group import PyCommentGroup -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.detached_symbols.decorator import PyDecorator +from codegen.sdk.python.statements.comment import PyComment, PyCommentType +from codegen.sdk.python.symbol_groups.comment_group import PyCommentGroup +from codegen.sdk.writer_decorators import py_apidoc @py_apidoc diff --git a/src/graph_sitter/python/placeholder/placeholder_return_type.py b/src/codegen/sdk/python/placeholder/placeholder_return_type.py similarity index 89% rename from src/graph_sitter/python/placeholder/placeholder_return_type.py rename to src/codegen/sdk/python/placeholder/placeholder_return_type.py index 190af235c..c7bd1da43 100644 --- a/src/graph_sitter/python/placeholder/placeholder_return_type.py +++ b/src/codegen/sdk/python/placeholder/placeholder_return_type.py @@ -1,8 +1,8 @@ from typing import Generic, TypeVar -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.placeholder.placeholder import Placeholder -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.placeholder.placeholder import Placeholder +from codegen.sdk.writer_decorators import py_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/python/statements/__init__.py b/src/codegen/sdk/python/statements/__init__.py similarity index 100% rename from src/graph_sitter/python/statements/__init__.py rename to src/codegen/sdk/python/statements/__init__.py diff --git a/src/graph_sitter/python/statements/assignment_statement.py b/src/codegen/sdk/python/statements/assignment_statement.py similarity index 84% rename from src/graph_sitter/python/statements/assignment_statement.py rename to src/codegen/sdk/python/statements/assignment_statement.py index ae4956688..7f277a9e3 100644 --- a/src/graph_sitter/python/statements/assignment_statement.py +++ b/src/codegen/sdk/python/statements/assignment_statement.py @@ -4,17 +4,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.expressions.multi_expression import MultiExpression -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.assignment_statement import AssignmentStatement -from graph_sitter.extensions.utils import find_all_descendants -from graph_sitter.python.assignment import PyAssignment -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.expressions.multi_expression import MultiExpression +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.assignment_statement import AssignmentStatement +from codegen.sdk.extensions.utils import find_all_descendants +from codegen.sdk.python.assignment import PyAssignment +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: - from graph_sitter.python.detached_symbols.code_block import PyCodeBlock - from graph_sitter.python.interfaces.has_block import PyHasBlock + from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock + from codegen.sdk.python.interfaces.has_block import PyHasBlock import logging @@ -57,10 +57,10 @@ def from_assignment(cls, ts_node: TSNode, file_node_id: NodeId, G: CodebaseGraph if assignment_node.type not in cls.assignment_types: raise ValueError(f"Invalid assignment node type: {assignment_node.type}") - from graph_sitter.python.class_definition import PyClass + from codegen.sdk.python.class_definition import PyClass if isinstance(parent, PyClass): - from graph_sitter.python.statements.attribute import PyAttribute + from codegen.sdk.python.statements.attribute import PyAttribute return PyAttribute(ts_node, file_node_id, G, parent, pos, assignment_node=assignment_node) return cls(ts_node, file_node_id, G, parent, pos, assignment_node=assignment_node) diff --git a/src/graph_sitter/python/statements/attribute.py b/src/codegen/sdk/python/statements/attribute.py similarity index 87% rename from src/graph_sitter/python/statements/attribute.py rename to src/codegen/sdk/python/statements/attribute.py index ffc34985d..4e3fd52d4 100644 --- a/src/graph_sitter/python/statements/attribute.py +++ b/src/codegen/sdk/python/statements/attribute.py @@ -2,17 +2,17 @@ from tree_sitter import Node as TSNode -from graph_sitter._proxy import proxy_property -from graph_sitter.core.autocommit import reader -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.statements.attribute import Attribute -from graph_sitter.exceptions import APINotApplicableForLanguageError -from graph_sitter.python.assignment import PyAssignment -from graph_sitter.python.statements.assignment_statement import PyAssignmentStatement -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk._proxy import proxy_property +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.statements.attribute import Attribute +from codegen.sdk.exceptions import APINotApplicableForLanguageError +from codegen.sdk.python.assignment import PyAssignment +from codegen.sdk.python.statements.assignment_statement import PyAssignmentStatement +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: - from graph_sitter.python.class_definition import PyClass + from codegen.sdk.python.class_definition import PyClass @py_apidoc @@ -96,7 +96,7 @@ def attribute_docstring(self) -> str: @reader def docstring(self, base_class: "PyClass") -> str | None: """Parse the docstring of the attribute from it's parent class docstrings.""" - from graph_sitter.python.class_definition import PyClass + from codegen.sdk.python.class_definition import PyClass to_search = [base_class] to_search.extend(base_class.superclasses()) diff --git a/src/graph_sitter/python/statements/block_statement.py b/src/codegen/sdk/python/statements/block_statement.py similarity index 66% rename from src/graph_sitter/python/statements/block_statement.py rename to src/codegen/sdk/python/statements/block_statement.py index 926be9de7..0684be0ce 100644 --- a/src/graph_sitter/python/statements/block_statement.py +++ b/src/codegen/sdk/python/statements/block_statement.py @@ -2,11 +2,11 @@ from typing import TYPE_CHECKING, Generic, TypeVar -from graph_sitter.core.statements.block_statement import BlockStatement -from graph_sitter.extensions.autocommit import reader -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.interfaces.has_block import PyHasBlock -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.statements.block_statement import BlockStatement +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.interfaces.has_block import PyHasBlock +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/python/statements/break_statement.py b/src/codegen/sdk/python/statements/break_statement.py similarity index 58% rename from src/graph_sitter/python/statements/break_statement.py rename to src/codegen/sdk/python/statements/break_statement.py index d2a6c80f0..abb4ec853 100644 --- a/src/graph_sitter/python/statements/break_statement.py +++ b/src/codegen/sdk/python/statements/break_statement.py @@ -2,11 +2,11 @@ from typing import TYPE_CHECKING, override -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/python/statements/catch_statement.py b/src/codegen/sdk/python/statements/catch_statement.py similarity index 61% rename from src/graph_sitter/python/statements/catch_statement.py rename to src/codegen/sdk/python/statements/catch_statement.py index 28ad2a3d3..e089d7365 100644 --- a/src/graph_sitter/python/statements/catch_statement.py +++ b/src/codegen/sdk/python/statements/catch_statement.py @@ -4,14 +4,14 @@ from tree_sitter import Node as PyNode -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.catch_statement import CatchStatement -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.statements.block_statement import PyBlockStatement -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.catch_statement import CatchStatement +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.statements.block_statement import PyBlockStatement +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph @py_apidoc diff --git a/src/graph_sitter/python/statements/comment.py b/src/codegen/sdk/python/statements/comment.py similarity index 97% rename from src/graph_sitter/python/statements/comment.py rename to src/codegen/sdk/python/statements/comment.py index e4e36c3a3..a044db8ac 100644 --- a/src/graph_sitter/python/statements/comment.py +++ b/src/codegen/sdk/python/statements/comment.py @@ -3,9 +3,9 @@ from enum import StrEnum from typing import TYPE_CHECKING -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.statements.comment import Comment, lowest_indentation -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.statements.comment import Comment, lowest_indentation +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/python/statements/for_loop_statement.py b/src/codegen/sdk/python/statements/for_loop_statement.py similarity index 73% rename from src/graph_sitter/python/statements/for_loop_statement.py rename to src/codegen/sdk/python/statements/for_loop_statement.py index 84ad2fdc9..7331c2771 100644 --- a/src/graph_sitter/python/statements/for_loop_statement.py +++ b/src/codegen/sdk/python/statements/for_loop_statement.py @@ -4,20 +4,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.for_loop_statement import ForLoopStatement -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.python.statements.block_statement import PyBlockStatement -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.for_loop_statement import ForLoopStatement +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.python.statements.block_statement import PyBlockStatement +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.python.detached_symbols.code_block import PyCodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock @py_apidoc diff --git a/src/graph_sitter/python/statements/if_block_statement.py b/src/codegen/sdk/python/statements/if_block_statement.py similarity index 86% rename from src/graph_sitter/python/statements/if_block_statement.py rename to src/codegen/sdk/python/statements/if_block_statement.py index c74886aac..b7cb62614 100644 --- a/src/graph_sitter/python/statements/if_block_statement.py +++ b/src/codegen/sdk/python/statements/if_block_statement.py @@ -4,15 +4,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.if_block_statement import IfBlockStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.if_block_statement import IfBlockStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.python.detached_symbols.code_block import PyCodeBlock + from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock Parent = TypeVar("Parent", bound="PyCodeBlock") @@ -43,7 +43,7 @@ def __init__(self, ts_node: TSNode, file_node_id: NodeId, G: CodebaseGraph, pare @reader def _parse_consequence_block(self) -> PyCodeBlock: - from graph_sitter.python.detached_symbols.code_block import PyCodeBlock + from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock body_node = self.ts_node.child_by_field_name("body") if self.is_else_statement else self.ts_node.child_by_field_name("consequence") return PyCodeBlock(body_node, self.parent.level + 1, self.parent, self) diff --git a/src/graph_sitter/python/statements/import_statement.py b/src/codegen/sdk/python/statements/import_statement.py similarity index 71% rename from src/graph_sitter/python/statements/import_statement.py rename to src/codegen/sdk/python/statements/import_statement.py index a8f823492..bcd9e3f28 100644 --- a/src/graph_sitter/python/statements/import_statement.py +++ b/src/codegen/sdk/python/statements/import_statement.py @@ -4,13 +4,13 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.import_statement import ImportStatement -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.import_resolution import PyImport -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.import_statement import ImportStatement +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.import_resolution import PyImport +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/python/statements/match_case.py b/src/codegen/sdk/python/statements/match_case.py similarity index 55% rename from src/graph_sitter/python/statements/match_case.py rename to src/codegen/sdk/python/statements/match_case.py index a183d2ce1..1a4861d2a 100644 --- a/src/graph_sitter/python/statements/match_case.py +++ b/src/codegen/sdk/python/statements/match_case.py @@ -2,14 +2,14 @@ from tree_sitter import Node as PyNode -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.switch_case import SwitchCase -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.statements.block_statement import PyBlockStatement -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.switch_case import SwitchCase +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.statements.block_statement import PyBlockStatement +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph @py_apidoc diff --git a/src/graph_sitter/python/statements/match_statement.py b/src/codegen/sdk/python/statements/match_statement.py similarity index 67% rename from src/graph_sitter/python/statements/match_statement.py rename to src/codegen/sdk/python/statements/match_statement.py index 9146308a3..b777f93ac 100644 --- a/src/graph_sitter/python/statements/match_statement.py +++ b/src/codegen/sdk/python/statements/match_statement.py @@ -4,14 +4,14 @@ from tree_sitter import Node as PyNode -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.switch_statement import SwitchStatement -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.statements.match_case import PyMatchCase -from graph_sitter.writer_decorators import py_apidoc +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.switch_statement import SwitchStatement +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.statements.match_case import PyMatchCase +from codegen.sdk.writer_decorators import py_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph @py_apidoc diff --git a/src/graph_sitter/python/statements/pass_statement.py b/src/codegen/sdk/python/statements/pass_statement.py similarity index 58% rename from src/graph_sitter/python/statements/pass_statement.py rename to src/codegen/sdk/python/statements/pass_statement.py index 9d2415e49..8a31e650b 100644 --- a/src/graph_sitter/python/statements/pass_statement.py +++ b/src/codegen/sdk/python/statements/pass_statement.py @@ -2,11 +2,11 @@ from typing import TYPE_CHECKING, override -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/python/statements/try_catch_statement.py b/src/codegen/sdk/python/statements/try_catch_statement.py similarity index 80% rename from src/graph_sitter/python/statements/try_catch_statement.py rename to src/codegen/sdk/python/statements/try_catch_statement.py index 307e1c8dc..bc7459663 100644 --- a/src/graph_sitter/python/statements/try_catch_statement.py +++ b/src/codegen/sdk/python/statements/try_catch_statement.py @@ -4,20 +4,20 @@ from tree_sitter import Node as PyNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.try_catch_statement import TryCatchStatement -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.python.detached_symbols.code_block import PyCodeBlock -from graph_sitter.python.statements.block_statement import PyBlockStatement -from graph_sitter.python.statements.catch_statement import PyCatchStatement -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.try_catch_statement import TryCatchStatement +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock +from codegen.sdk.python.statements.block_statement import PyBlockStatement +from codegen.sdk.python.statements.catch_statement import PyCatchStatement +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.interfaces.importable import Importable + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.interfaces.importable import Importable @py_apidoc diff --git a/src/graph_sitter/python/statements/while_statement.py b/src/codegen/sdk/python/statements/while_statement.py similarity index 78% rename from src/graph_sitter/python/statements/while_statement.py rename to src/codegen/sdk/python/statements/while_statement.py index 111ef47ee..d9e7ca9af 100644 --- a/src/graph_sitter/python/statements/while_statement.py +++ b/src/codegen/sdk/python/statements/while_statement.py @@ -4,21 +4,21 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.while_statement import WhileStatement -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.python.interfaces.has_block import PyHasBlock -from graph_sitter.python.statements.if_block_statement import PyIfBlockStatement -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.while_statement import WhileStatement +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.python.interfaces.has_block import PyHasBlock +from codegen.sdk.python.statements.if_block_statement import PyIfBlockStatement +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.python.detached_symbols.code_block import PyCodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock @py_apidoc diff --git a/src/graph_sitter/python/statements/with_statement.py b/src/codegen/sdk/python/statements/with_statement.py similarity index 76% rename from src/graph_sitter/python/statements/with_statement.py rename to src/codegen/sdk/python/statements/with_statement.py index 3eb8777a9..6fb3d04c7 100644 --- a/src/graph_sitter/python/statements/with_statement.py +++ b/src/codegen/sdk/python/statements/with_statement.py @@ -5,21 +5,21 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.core.symbol_groups.expression_group import ExpressionGroup -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.python.interfaces.has_block import PyHasBlock -from graph_sitter.writer_decorators import apidoc, noapidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.core.symbol_groups.expression_group import ExpressionGroup +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.python.interfaces.has_block import PyHasBlock +from codegen.sdk.writer_decorators import apidoc, noapidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.python.detached_symbols.code_block import PyCodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock @apidoc diff --git a/src/graph_sitter/python/symbol.py b/src/codegen/sdk/python/symbol.py similarity index 93% rename from src/graph_sitter/python/symbol.py rename to src/codegen/sdk/python/symbol.py index e15bb64f7..a07317576 100644 --- a/src/graph_sitter/python/symbol.py +++ b/src/codegen/sdk/python/symbol.py @@ -4,15 +4,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import ImportType -from graph_sitter.python.statements.comment import PyComment, PyCommentType -from graph_sitter.python.symbol_groups.comment_group import PyCommentGroup -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import ImportType +from codegen.sdk.python.statements.comment import PyComment, PyCommentType +from codegen.sdk.python.symbol_groups.comment_group import PyCommentGroup +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/python/symbol_groups/comment_group.py b/src/codegen/sdk/python/symbol_groups/comment_group.py similarity index 95% rename from src/graph_sitter/python/symbol_groups/comment_group.py rename to src/codegen/sdk/python/symbol_groups/comment_group.py index 128b892e6..7be70b6a6 100644 --- a/src/graph_sitter/python/symbol_groups/comment_group.py +++ b/src/codegen/sdk/python/symbol_groups/comment_group.py @@ -6,16 +6,16 @@ from docstring_parser import Docstring, DocstringStyle, parse from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.core.symbol_groups.comment_group import CommentGroup -from graph_sitter.enums import SymbolType -from graph_sitter.python.statements.comment import PyComment -from graph_sitter.writer_decorators import noapidoc, py_apidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.core.symbol_groups.comment_group import CommentGroup +from codegen.sdk.enums import SymbolType +from codegen.sdk.python.statements.comment import PyComment +from codegen.sdk.writer_decorators import noapidoc, py_apidoc if TYPE_CHECKING: - from graph_sitter.python.function import PyFunction - from graph_sitter.python.symbol import PySymbol + from codegen.sdk.python.function import PyFunction + from codegen.sdk.python.symbol import PySymbol @py_apidoc @@ -49,7 +49,7 @@ def from_symbol_comments(cls, symbol: PySymbol): else: break # Stop if a non-comment node is encountered - from graph_sitter.python.class_definition import PyClass + from codegen.sdk.python.class_definition import PyClass # Check if the function node is a method if symbol.symbol_type == SymbolType.Function: diff --git a/src/graph_sitter/secrets.py b/src/codegen/sdk/secrets.py similarity index 100% rename from src/graph_sitter/secrets.py rename to src/codegen/sdk/secrets.py diff --git a/src/graph_sitter/skills/__init__.py b/src/codegen/sdk/skills/__init__.py similarity index 100% rename from src/graph_sitter/skills/__init__.py rename to src/codegen/sdk/skills/__init__.py diff --git a/src/graph_sitter/skills/ai_skills.py b/src/codegen/sdk/skills/ai_skills.py similarity index 96% rename from src/graph_sitter/skills/ai_skills.py rename to src/codegen/sdk/skills/ai_skills.py index 5f0f25f42..9f16e7ead 100644 --- a/src/graph_sitter/skills/ai_skills.py +++ b/src/codegen/sdk/skills/ai_skills.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl ######################################################################################################################## # Codebase AI Skills diff --git a/src/graph_sitter/skills/asyncify_skills.py b/src/codegen/sdk/skills/asyncify_skills.py similarity index 92% rename from src/graph_sitter/skills/asyncify_skills.py rename to src/codegen/sdk/skills/asyncify_skills.py index 206af55f1..37cb294d3 100644 --- a/src/graph_sitter/skills/asyncify_skills.py +++ b/src/codegen/sdk/skills/asyncify_skills.py @@ -1,11 +1,11 @@ from collections import deque -from graph_sitter.core.codebase import CodebaseType, PyCodebaseType, TSCodebaseType -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType, PyCodebaseType, TSCodebaseType +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl py_input1 = """ def func_to_convert(a: int) -> int: diff --git a/src/graph_sitter/skills/code_block_skills.py b/src/codegen/sdk/skills/code_block_skills.py similarity index 96% rename from src/graph_sitter/skills/code_block_skills.py rename to src/codegen/sdk/skills/code_block_skills.py index 250a49093..b2872e469 100644 --- a/src/graph_sitter/skills/code_block_skills.py +++ b/src/codegen/sdk/skills/code_block_skills.py @@ -1,13 +1,13 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.statements.expression_statement import ExpressionStatement -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.statements.expression_statement import ExpressionStatement +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl py_input = """ def main(): diff --git a/src/graph_sitter/skills/comment_skills.py b/src/codegen/sdk/skills/comment_skills.py similarity index 92% rename from src/graph_sitter/skills/comment_skills.py rename to src/codegen/sdk/skills/comment_skills.py index ec9479ffe..a1ce98c10 100644 --- a/src/graph_sitter/skills/comment_skills.py +++ b/src/codegen/sdk/skills/comment_skills.py @@ -1,10 +1,10 @@ import re -from graph_sitter.core.codebase import TSCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import TSCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl # Case 1: all rules disabled in a SINGLE line comment are no longer required - should remove entire comment ts_case_1_input = """ diff --git a/src/graph_sitter/skills/core/__init__.py b/src/codegen/sdk/skills/core/__init__.py similarity index 100% rename from src/graph_sitter/skills/core/__init__.py rename to src/codegen/sdk/skills/core/__init__.py diff --git a/src/graph_sitter/skills/core/skill.py b/src/codegen/sdk/skills/core/skill.py similarity index 93% rename from src/graph_sitter/skills/core/skill.py rename to src/codegen/sdk/skills/core/skill.py index e19d131fe..fc490ff85 100644 --- a/src/graph_sitter/skills/core/skill.py +++ b/src/codegen/sdk/skills/core/skill.py @@ -2,9 +2,9 @@ from abc import ABC from pathlib import Path -from graph_sitter.core.codebase import Codebase, CodebaseType, PyCodebaseType, TSCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill_implementation import SkillImplementation +from codegen.sdk.core.codebase import Codebase, CodebaseType, PyCodebaseType, TSCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill_implementation import SkillImplementation class Skill(ABC): diff --git a/src/graph_sitter/skills/core/skill_implementation.py b/src/codegen/sdk/skills/core/skill_implementation.py similarity index 93% rename from src/graph_sitter/skills/core/skill_implementation.py rename to src/codegen/sdk/skills/core/skill_implementation.py index 362e6a642..912cb2b36 100644 --- a/src/graph_sitter/skills/core/skill_implementation.py +++ b/src/codegen/sdk/skills/core/skill_implementation.py @@ -3,12 +3,12 @@ import textwrap from collections.abc import Callable +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill_test import SkillTestCase +from codegen.sdk.skills.utils.utils import verify_skill_output from codegen_git.schemas.repo_config import BaseRepoConfig -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill_test import SkillTestCase -from graph_sitter.skills.utils.utils import verify_skill_output class SkillImplementation: diff --git a/src/graph_sitter/skills/core/skill_test.py b/src/codegen/sdk/skills/core/skill_test.py similarity index 96% rename from src/graph_sitter/skills/core/skill_test.py rename to src/codegen/sdk/skills/core/skill_test.py index c152c284b..7558425b1 100644 --- a/src/graph_sitter/skills/core/skill_test.py +++ b/src/codegen/sdk/skills/core/skill_test.py @@ -1,8 +1,8 @@ from dataclasses import dataclass from pathlib import Path -from graph_sitter.python import PyFile -from graph_sitter.typescript.file import TSFile +from codegen.sdk.python import PyFile +from codegen.sdk.typescript.file import TSFile @dataclass diff --git a/src/graph_sitter/skills/core/utils.py b/src/codegen/sdk/skills/core/utils.py similarity index 94% rename from src/graph_sitter/skills/core/utils.py rename to src/codegen/sdk/skills/core/utils.py index 66a200cca..d95f7b011 100644 --- a/src/graph_sitter/skills/core/utils.py +++ b/src/codegen/sdk/skills/core/utils.py @@ -5,11 +5,11 @@ import inflection -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_implementation import SkillImplementation -from graph_sitter.skills.core.skill_test import SkillTestCase +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_implementation import SkillImplementation +from codegen.sdk.skills.core.skill_test import SkillTestCase ExecuteFuncType = Callable[[CodebaseType], None] @@ -157,7 +157,7 @@ def import_skills_from(path: Path, package: str): def import_all_skills(): - import_skills_from(SKILLS_PATH, "graph_sitter.skills") + import_skills_from(SKILLS_PATH, "codegen.sdk.skills") import_skills_from(CODEMODS_PATH, "codemods.canonical") @@ -167,7 +167,7 @@ def import_all_guide_skill_files(): relative_path = file.relative_to(SKILLS_PATH) if "evaluation" in relative_path.parts or "__init__" in file.name or "guides" not in relative_path.parts: continue - module_name = "graph_sitter.skills." + str(relative_path).replace("/", ".").removesuffix(".py") + module_name = "codegen.sdk.skills." + str(relative_path).replace("/", ".").removesuffix(".py") importlib.import_module(module_name) diff --git a/src/graph_sitter/skills/decorator_skills.py b/src/codegen/sdk/skills/decorator_skills.py similarity index 95% rename from src/graph_sitter/skills/decorator_skills.py rename to src/codegen/sdk/skills/decorator_skills.py index e6909d306..39ce37481 100644 --- a/src/graph_sitter/skills/decorator_skills.py +++ b/src/codegen/sdk/skills/decorator_skills.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl ######################################################################################################################## # ADD FUNCTION DECORATOR diff --git a/src/graph_sitter/skills/eval_skills.py b/src/codegen/sdk/skills/eval_skills.py similarity index 97% rename from src/graph_sitter/skills/eval_skills.py rename to src/codegen/sdk/skills/eval_skills.py index ba0513ed0..12217317c 100644 --- a/src/graph_sitter/skills/eval_skills.py +++ b/src/codegen/sdk/skills/eval_skills.py @@ -2,14 +2,14 @@ from abc import ABC from pathlib import Path -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python import PyFunction -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase -from graph_sitter.skills.core.utils import skill, skill_impl - -EVAL_SKILLS_TEST_DIR = Path(__file__).parents[3] / "src" / "codemods" / "eval" / "test_files" +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python import PyFunction +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase +from codegen.sdk.skills.core.utils import skill, skill_impl + +EVAL_SKILLS_TEST_DIR = Path(__file__).parent.parents[3] / "src" / "codemods" / "eval" / "test_files" ######################################################################################################################## diff --git a/src/graph_sitter/skills/example_skills.py b/src/codegen/sdk/skills/example_skills.py similarity index 98% rename from src/graph_sitter/skills/example_skills.py rename to src/codegen/sdk/skills/example_skills.py index 87d9c58de..09ebacdc4 100644 --- a/src/graph_sitter/skills/example_skills.py +++ b/src/codegen/sdk/skills/example_skills.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl ######################################################################################################################## # GLOBAL VARS diff --git a/src/graph_sitter/skills/export_skills.py b/src/codegen/sdk/skills/export_skills.py similarity index 85% rename from src/graph_sitter/skills/export_skills.py rename to src/codegen/sdk/skills/export_skills.py index 61330110b..9a6f69b2a 100644 --- a/src/graph_sitter/skills/export_skills.py +++ b/src/codegen/sdk/skills/export_skills.py @@ -1,8 +1,8 @@ -from graph_sitter.core.codebase import TSCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import TSCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl ts_input1 = """ export default A = function() {b()}; diff --git a/src/graph_sitter/skills/expressions/dict.py b/src/codegen/sdk/skills/expressions/dict.py similarity index 76% rename from src/graph_sitter/skills/expressions/dict.py rename to src/codegen/sdk/skills/expressions/dict.py index 1b96a19c0..8ce27b5d4 100644 --- a/src/graph_sitter/skills/expressions/dict.py +++ b/src/codegen/sdk/skills/expressions/dict.py @@ -1,9 +1,9 @@ -from graph_sitter.core.codebase import PyCodebaseType -from graph_sitter.core.symbol_groups.dict import Dict -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import PyCodebaseType +from codegen.sdk.core.symbol_groups.dict import Dict +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl test_cases = [ SkillTestCase(files=[SkillTestCasePyFile(input='a={"foo": "bar"}', output="a=Schema(foo=bar)")]), diff --git a/src/graph_sitter/skills/expressions/list.py b/src/codegen/sdk/skills/expressions/list.py similarity index 94% rename from src/graph_sitter/skills/expressions/list.py rename to src/codegen/sdk/skills/expressions/list.py index f1ad69ae3..42fe1ae24 100644 --- a/src/graph_sitter/skills/expressions/list.py +++ b/src/codegen/sdk/skills/expressions/list.py @@ -1,14 +1,14 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import ( +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import ( SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile, ) -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.skills.core.utils import skill, skill_impl # Test cases for appending to list test_cases_append_py = [ diff --git a/src/graph_sitter/skills/expressions/type.py b/src/codegen/sdk/skills/expressions/type.py similarity index 88% rename from src/graph_sitter/skills/expressions/type.py rename to src/codegen/sdk/skills/expressions/type.py index 48a3398a2..9f0130c5b 100644 --- a/src/graph_sitter/skills/expressions/type.py +++ b/src/codegen/sdk/skills/expressions/type.py @@ -1,11 +1,11 @@ -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.core.expressions.union_type import UnionType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python.assignment import PyAssignment -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.typescript.assignment import TSAssignment +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.core.expressions.union_type import UnionType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python.assignment import PyAssignment +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.typescript.assignment import TSAssignment test_cases_append_py = [ SkillTestCase(files=[SkillTestCasePyFile(input="a: int | None", output="a: int | None | str")]), diff --git a/src/graph_sitter/skills/feature_flag_deletion_skill.py b/src/codegen/sdk/skills/feature_flag_deletion_skill.py similarity index 90% rename from src/graph_sitter/skills/feature_flag_deletion_skill.py rename to src/codegen/sdk/skills/feature_flag_deletion_skill.py index f8005cebc..ae6eebc70 100644 --- a/src/graph_sitter/skills/feature_flag_deletion_skill.py +++ b/src/codegen/sdk/skills/feature_flag_deletion_skill.py @@ -1,14 +1,14 @@ from abc import ABC -from graph_sitter.core.codebase import PyCodebaseType, TSCodebaseType -from graph_sitter.core.function import Function -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.statements.if_block_statement import IfBlockStatement -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python.statements.with_statement import WithStatement -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import PyCodebaseType, TSCodebaseType +from codegen.sdk.core.function import Function +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.statements.if_block_statement import IfBlockStatement +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python.statements.with_statement import WithStatement +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl py_feature_flag_def = """ ROLLED_OUT_FLAG_TO_DELETE = False diff --git a/src/graph_sitter/skills/function_skills.py b/src/codegen/sdk/skills/function_skills.py similarity index 84% rename from src/graph_sitter/skills/function_skills.py rename to src/codegen/sdk/skills/function_skills.py index 290002542..2faf3053c 100644 --- a/src/graph_sitter/skills/function_skills.py +++ b/src/codegen/sdk/skills/function_skills.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import PyCodebaseType, TSCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import PyCodebaseType, TSCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl py_function_input = """ def foo(a: int): diff --git a/src/graph_sitter/skills/global_var_skills.py b/src/codegen/sdk/skills/global_var_skills.py similarity index 90% rename from src/graph_sitter/skills/global_var_skills.py rename to src/codegen/sdk/skills/global_var_skills.py index 697b79a51..b9c2871c1 100644 --- a/src/graph_sitter/skills/global_var_skills.py +++ b/src/codegen/sdk/skills/global_var_skills.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import PyCodebaseType, TSCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import PyCodebaseType, TSCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl ######################################################################################################################## # Py test cases diff --git a/src/graph_sitter/skills/graph_viz/graph_viz_app_imports.py b/src/codegen/sdk/skills/graph_viz/graph_viz_app_imports.py similarity index 87% rename from src/graph_sitter/skills/graph_viz/graph_viz_app_imports.py rename to src/codegen/sdk/skills/graph_viz/graph_viz_app_imports.py index 46a6308d5..0c5511fb2 100644 --- a/src/graph_sitter/skills/graph_viz/graph_viz_app_imports.py +++ b/src/codegen/sdk/skills/graph_viz/graph_viz_app_imports.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl FileAppImportGraphTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/graph_viz/graph_viz_call_graph.py b/src/codegen/sdk/skills/graph_viz/graph_viz_call_graph.py similarity index 95% rename from src/graph_sitter/skills/graph_viz/graph_viz_call_graph.py rename to src/codegen/sdk/skills/graph_viz/graph_viz_call_graph.py index 88f6e1594..029a9b2de 100644 --- a/src/graph_sitter/skills/graph_viz/graph_viz_call_graph.py +++ b/src/codegen/sdk/skills/graph_viz/graph_viz_call_graph.py @@ -2,16 +2,16 @@ import networkx as nx -from graph_sitter.core.class_definition import Class -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.function import Function -from graph_sitter.core.interfaces.callable import Callable -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.function import Function +from codegen.sdk.core.interfaces.callable import Callable +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl CallGraphFromNodeTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/graph_viz/graph_viz_dead_code.py b/src/codegen/sdk/skills/graph_viz/graph_viz_dead_code.py similarity index 91% rename from src/graph_sitter/skills/graph_viz/graph_viz_dead_code.py rename to src/codegen/sdk/skills/graph_viz/graph_viz_dead_code.py index d97355911..1fd4b5154 100644 --- a/src/graph_sitter/skills/graph_viz/graph_viz_dead_code.py +++ b/src/codegen/sdk/skills/graph_viz/graph_viz_dead_code.py @@ -2,14 +2,14 @@ import networkx as nx -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.core.function import Function -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.core.function import Function +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl PyDeadCodeTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/graph_viz/graph_viz_dir_tree.py b/src/codegen/sdk/skills/graph_viz/graph_viz_dir_tree.py similarity index 91% rename from src/graph_sitter/skills/graph_viz/graph_viz_dir_tree.py rename to src/codegen/sdk/skills/graph_viz/graph_viz_dir_tree.py index 1bc3b743b..79a52f208 100644 --- a/src/graph_sitter/skills/graph_viz/graph_viz_dir_tree.py +++ b/src/codegen/sdk/skills/graph_viz/graph_viz_dir_tree.py @@ -2,11 +2,11 @@ import networkx as nx -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl PyRepoDirTreeTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/graph_viz/graph_viz_foreign_key.py b/src/codegen/sdk/skills/graph_viz/graph_viz_foreign_key.py similarity index 96% rename from src/graph_sitter/skills/graph_viz/graph_viz_foreign_key.py rename to src/codegen/sdk/skills/graph_viz/graph_viz_foreign_key.py index a21dcdd63..0a2b6c563 100644 --- a/src/graph_sitter/skills/graph_viz/graph_viz_foreign_key.py +++ b/src/codegen/sdk/skills/graph_viz/graph_viz_foreign_key.py @@ -2,11 +2,11 @@ import networkx as nx -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl PyForeignKeyGraphTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/code-quality-metrics/import-loops.py b/src/codegen/sdk/skills/guides/code-quality-metrics/import-loops.py similarity index 93% rename from src/graph_sitter/skills/guides/code-quality-metrics/import-loops.py rename to src/codegen/sdk/skills/guides/code-quality-metrics/import-loops.py index dbe9cbdbd..a03bcdf4c 100644 --- a/src/graph_sitter/skills/guides/code-quality-metrics/import-loops.py +++ b/src/codegen/sdk/skills/guides/code-quality-metrics/import-loops.py @@ -3,11 +3,11 @@ import networkx from networkx import DiGraph -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl ImportCycleDetectionAndVisualizationTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/code-quality-metrics/large-files.py b/src/codegen/sdk/skills/guides/code-quality-metrics/large-files.py similarity index 89% rename from src/graph_sitter/skills/guides/code-quality-metrics/large-files.py rename to src/codegen/sdk/skills/guides/code-quality-metrics/large-files.py index 6946869d6..cc6a6d3f4 100644 --- a/src/graph_sitter/skills/guides/code-quality-metrics/large-files.py +++ b/src/codegen/sdk/skills/guides/code-quality-metrics/large-files.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl CountLargeFilesPyTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/code-quality-metrics/missing-documentation.py b/src/codegen/sdk/skills/guides/code-quality-metrics/missing-documentation.py similarity index 90% rename from src/graph_sitter/skills/guides/code-quality-metrics/missing-documentation.py rename to src/codegen/sdk/skills/guides/code-quality-metrics/missing-documentation.py index 530162edb..72d9f542c 100644 --- a/src/graph_sitter/skills/guides/code-quality-metrics/missing-documentation.py +++ b/src/codegen/sdk/skills/guides/code-quality-metrics/missing-documentation.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl AutoDocstringGeneratorTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/code-quality-metrics/unnamed-kwargs.py b/src/codegen/sdk/skills/guides/code-quality-metrics/unnamed-kwargs.py similarity index 90% rename from src/graph_sitter/skills/guides/code-quality-metrics/unnamed-kwargs.py rename to src/codegen/sdk/skills/guides/code-quality-metrics/unnamed-kwargs.py index 2209af0d3..cf5d5260e 100644 --- a/src/graph_sitter/skills/guides/code-quality-metrics/unnamed-kwargs.py +++ b/src/codegen/sdk/skills/guides/code-quality-metrics/unnamed-kwargs.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl CountUnnamedKeywordArgumentsTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/code-quality-metrics/untyped-attributes.py b/src/codegen/sdk/skills/guides/code-quality-metrics/untyped-attributes.py similarity index 89% rename from src/graph_sitter/skills/guides/code-quality-metrics/untyped-attributes.py rename to src/codegen/sdk/skills/guides/code-quality-metrics/untyped-attributes.py index c56865dfa..aff3eaed5 100644 --- a/src/graph_sitter/skills/guides/code-quality-metrics/untyped-attributes.py +++ b/src/codegen/sdk/skills/guides/code-quality-metrics/untyped-attributes.py @@ -1,11 +1,11 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.core.placeholder.placeholder_type import TypePlaceholder -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.core.placeholder.placeholder_type import TypePlaceholder +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl CountUntypedAttributesInCodebasePyTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/code-quality-metrics/untyped-parameters.py b/src/codegen/sdk/skills/guides/code-quality-metrics/untyped-parameters.py similarity index 89% rename from src/graph_sitter/skills/guides/code-quality-metrics/untyped-parameters.py rename to src/codegen/sdk/skills/guides/code-quality-metrics/untyped-parameters.py index 70f4783f5..32b02d1d1 100644 --- a/src/graph_sitter/skills/guides/code-quality-metrics/untyped-parameters.py +++ b/src/codegen/sdk/skills/guides/code-quality-metrics/untyped-parameters.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl CountUntypedParametersTest = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/code-quality-metrics/untyped-return-types.py b/src/codegen/sdk/skills/guides/code-quality-metrics/untyped-return-types.py similarity index 85% rename from src/graph_sitter/skills/guides/code-quality-metrics/untyped-return-types.py rename to src/codegen/sdk/skills/guides/code-quality-metrics/untyped-return-types.py index 41e06a72b..b1fd41c6a 100644 --- a/src/graph_sitter/skills/guides/code-quality-metrics/untyped-return-types.py +++ b/src/codegen/sdk/skills/guides/code-quality-metrics/untyped-return-types.py @@ -1,11 +1,11 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.core.placeholder.placeholder import Placeholder -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.core.placeholder.placeholder import Placeholder +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl CountUntypedReturnStatementsTest = SkillTestCase( files=[ diff --git a/src/graph_sitter/skills/guides/codebase-ai.py b/src/codegen/sdk/skills/guides/codebase-ai.py similarity index 97% rename from src/graph_sitter/skills/guides/codebase-ai.py rename to src/codegen/sdk/skills/guides/codebase-ai.py index 8b6f2133b..226e7401b 100644 --- a/src/graph_sitter/skills/guides/codebase-ai.py +++ b/src/codegen/sdk/skills/guides/codebase-ai.py @@ -1,11 +1,11 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl MethodSummaryGeneratorPyTestCase = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/complex-renaming.py b/src/codegen/sdk/skills/guides/complex-renaming.py similarity index 97% rename from src/graph_sitter/skills/guides/complex-renaming.py rename to src/codegen/sdk/skills/guides/complex-renaming.py index c9dbdde33..5a36f00ba 100644 --- a/src/graph_sitter/skills/guides/complex-renaming.py +++ b/src/codegen/sdk/skills/guides/complex-renaming.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl RenameFunctionAndUpdateReferencesPyTestCase = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/creating-documentation.py b/src/codegen/sdk/skills/guides/creating-documentation.py similarity index 97% rename from src/graph_sitter/skills/guides/creating-documentation.py rename to src/codegen/sdk/skills/guides/creating-documentation.py index c16ee036c..dcaa671f4 100644 --- a/src/graph_sitter/skills/guides/creating-documentation.py +++ b/src/codegen/sdk/skills/guides/creating-documentation.py @@ -1,11 +1,11 @@ import textwrap from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl UpdateFunctionDocstringPyTestCase = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/deleting-dead-code.py b/src/codegen/sdk/skills/guides/deleting-dead-code.py similarity index 98% rename from src/graph_sitter/skills/guides/deleting-dead-code.py rename to src/codegen/sdk/skills/guides/deleting-dead-code.py index e69d208b4..548472f57 100644 --- a/src/graph_sitter/skills/guides/deleting-dead-code.py +++ b/src/codegen/sdk/skills/guides/deleting-dead-code.py @@ -1,11 +1,11 @@ import re from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl RemoveUnusedSymbolsPyTestCase = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/increase-type-coverage.py b/src/codegen/sdk/skills/guides/increase-type-coverage.py similarity index 98% rename from src/graph_sitter/skills/guides/increase-type-coverage.py rename to src/codegen/sdk/skills/guides/increase-type-coverage.py index d2c533670..db5ebb25f 100644 --- a/src/graph_sitter/skills/guides/increase-type-coverage.py +++ b/src/codegen/sdk/skills/guides/increase-type-coverage.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl CountTypedParametersPyTestCase = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/manipulating-collections.py b/src/codegen/sdk/skills/guides/manipulating-collections.py similarity index 96% rename from src/graph_sitter/skills/guides/manipulating-collections.py rename to src/codegen/sdk/skills/guides/manipulating-collections.py index e42d92cb9..44ba69676 100644 --- a/src/graph_sitter/skills/guides/manipulating-collections.py +++ b/src/codegen/sdk/skills/guides/manipulating-collections.py @@ -1,11 +1,11 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.core.symbol_groups.list import List -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.core.symbol_groups.list import List +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl AddParameterToFunctionPyTestCase = SkillTestCase( [ diff --git a/src/graph_sitter/skills/guides/organize-your-codebase.py b/src/codegen/sdk/skills/guides/organize-your-codebase.py similarity index 97% rename from src/graph_sitter/skills/guides/organize-your-codebase.py rename to src/codegen/sdk/skills/guides/organize-your-codebase.py index a782481d1..d7dca9431 100644 --- a/src/graph_sitter/skills/guides/organize-your-codebase.py +++ b/src/codegen/sdk/skills/guides/organize-your-codebase.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import CodebaseType, TSCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType, TSCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl SplitFunctionsIntoSeparateFilesPyTestCase = SkillTestCase( [ diff --git a/src/graph_sitter/skills/if_statement_skills.py b/src/codegen/sdk/skills/if_statement_skills.py similarity index 92% rename from src/graph_sitter/skills/if_statement_skills.py rename to src/codegen/sdk/skills/if_statement_skills.py index 87b3225b7..9cc179de6 100644 --- a/src/graph_sitter/skills/if_statement_skills.py +++ b/src/codegen/sdk/skills/if_statement_skills.py @@ -1,10 +1,10 @@ from abc import ABC -from graph_sitter.core.codebase import PyCodebaseType, TSCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import PyCodebaseType, TSCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl py_if_input = """ def foo(): diff --git a/src/graph_sitter/skills/move_symbols_skills.py b/src/codegen/sdk/skills/move_symbols_skills.py similarity index 90% rename from src/graph_sitter/skills/move_symbols_skills.py rename to src/codegen/sdk/skills/move_symbols_skills.py index c40f6312f..91f60dd8e 100644 --- a/src/graph_sitter/skills/move_symbols_skills.py +++ b/src/codegen/sdk/skills/move_symbols_skills.py @@ -1,8 +1,8 @@ -from graph_sitter.core.codebase import CodebaseType, PyCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.codebase import CodebaseType, PyCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCasePyFile +from codegen.sdk.skills.core.utils import skill, skill_impl ######################################################################################################################## # Py test cases diff --git a/src/graph_sitter/skills/type_inheritance_skills.py b/src/codegen/sdk/skills/type_inheritance_skills.py similarity index 94% rename from src/graph_sitter/skills/type_inheritance_skills.py rename to src/codegen/sdk/skills/type_inheritance_skills.py index 961662c43..c22b20341 100644 --- a/src/graph_sitter/skills/type_inheritance_skills.py +++ b/src/codegen/sdk/skills/type_inheritance_skills.py @@ -1,14 +1,14 @@ -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.class_definition import Class -from graph_sitter.core.codebase import CodebaseType, PyCodebaseType, TSCodebaseType -from graph_sitter.core.function import Function -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.symbol_groups.dict import Dict -from graph_sitter.core.type_alias import TypeAlias -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.skill_test import SkillTestCase, SkillTestCaseTSFile -from graph_sitter.skills.core.utils import skill, skill_impl +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.codebase import CodebaseType, PyCodebaseType, TSCodebaseType +from codegen.sdk.core.function import Function +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.symbol_groups.dict import Dict +from codegen.sdk.core.type_alias import TypeAlias +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.skill_test import SkillTestCase, SkillTestCaseTSFile +from codegen.sdk.skills.core.utils import skill, skill_impl ts_input1 = """ export type MyMapper = { diff --git a/src/graph_sitter/skills/utils/__init__.py b/src/codegen/sdk/skills/utils/__init__.py similarity index 100% rename from src/graph_sitter/skills/utils/__init__.py rename to src/codegen/sdk/skills/utils/__init__.py diff --git a/src/graph_sitter/skills/utils/utils.py b/src/codegen/sdk/skills/utils/utils.py similarity index 98% rename from src/graph_sitter/skills/utils/utils.py rename to src/codegen/sdk/skills/utils/utils.py index c8dd938bb..e73a82e33 100644 --- a/src/graph_sitter/skills/utils/utils.py +++ b/src/codegen/sdk/skills/utils/utils.py @@ -6,8 +6,8 @@ import isort import jsbeautifier -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage def get_jsbeautifier_options() -> jsbeautifier.default_options: diff --git a/src/graph_sitter/testing/__init__.py b/src/codegen/sdk/testing/__init__.py similarity index 100% rename from src/graph_sitter/testing/__init__.py rename to src/codegen/sdk/testing/__init__.py diff --git a/src/graph_sitter/testing/constants.py b/src/codegen/sdk/testing/constants.py similarity index 100% rename from src/graph_sitter/testing/constants.py rename to src/codegen/sdk/testing/constants.py diff --git a/src/graph_sitter/testing/models.py b/src/codegen/sdk/testing/models.py similarity index 90% rename from src/graph_sitter/testing/models.py rename to src/codegen/sdk/testing/models.py index d7ecf226b..65a45b8f7 100644 --- a/src/graph_sitter/testing/models.py +++ b/src/codegen/sdk/testing/models.py @@ -9,14 +9,14 @@ from pydantic import BaseModel, ConfigDict +from codegen.sdk.codebase.config import GSFeatureFlags +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.testing.constants import DIFF_FILEPATH +from codegen.sdk.testing.verified_codemod_utils import CodemodAPI from codegen_git.repo_operator.local_repo_operator import LocalRepoOperator -from graph_sitter.codebase.config import GSFeatureFlags -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.testing.constants import DIFF_FILEPATH -from graph_sitter.testing.verified_codemod_utils import CodemodAPI BASE_TMP_DIR = Path(os.getenv("GITHUB_WORKSPACE", tempfile.gettempdir())) -BASE_PATH: Path = Path(__file__).parent.parent.parent.parent +BASE_PATH: Path = Path(__file__).parent.parent.parent.parent.parent TEST_DIR: Path = BASE_PATH / "tests" / "codemod" CODEMOD_PATH: Path = BASE_PATH / "src" / "codemods" VERIFIED_CODEMOD_DIR: Path = BASE_PATH / "tests" / "verified_codemods" @@ -29,7 +29,7 @@ except (FileNotFoundError, json.decoder.JSONDecodeError): REPO_ID_TO_URL = {} if TYPE_CHECKING: - from graph_sitter.codemod import Codemod3 + from codegen.sdk.codemod import Codemod3 @unique diff --git a/src/graph_sitter/testing/test_discovery.py b/src/codegen/sdk/testing/test_discovery.py similarity index 94% rename from src/graph_sitter/testing/test_discovery.py rename to src/codegen/sdk/testing/test_discovery.py index 73a272c66..2c56159f2 100644 --- a/src/graph_sitter/testing/test_discovery.py +++ b/src/codegen/sdk/testing/test_discovery.py @@ -10,11 +10,11 @@ import emoji from loguru import logger -from graph_sitter.codemod import Codemod3 -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.testing.constants import DIFF_FILEPATH -from graph_sitter.testing.models import BASE_PATH, CODEMOD_PATH, REPO_ID_TO_URL, TEST_DIR, VERIFIED_CODEMOD_DATA_DIR, VERIFIED_CODEMOD_DIFFS, ClonedRepoTestCase, CodemodMetadata, Repo, Size -from graph_sitter.testing.verified_codemod_utils import CodemodAPI, RepoCodemodMetadata, SkillTestConfig, anonymize_id +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.testing.constants import DIFF_FILEPATH +from codegen.sdk.testing.models import BASE_PATH, CODEMOD_PATH, REPO_ID_TO_URL, TEST_DIR, VERIFIED_CODEMOD_DATA_DIR, VERIFIED_CODEMOD_DIFFS, ClonedRepoTestCase, CodemodMetadata, Repo, Size +from codegen.sdk.testing.verified_codemod_utils import CodemodAPI, RepoCodemodMetadata, SkillTestConfig, anonymize_id from gscli.generate.runner_imports import get_runner_imports if TYPE_CHECKING: diff --git a/src/graph_sitter/testing/verified_codemod_utils.py b/src/codegen/sdk/testing/verified_codemod_utils.py similarity index 97% rename from src/graph_sitter/testing/verified_codemod_utils.py rename to src/codegen/sdk/testing/verified_codemod_utils.py index be55c153f..2ec1c5bf0 100644 --- a/src/graph_sitter/testing/verified_codemod_utils.py +++ b/src/codegen/sdk/testing/verified_codemod_utils.py @@ -8,8 +8,8 @@ import requests from pydantic import BaseModel -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.testing.constants import GET_CODEMODS_URL_SUFFIX, UPDATE_CODEMOD_DIFF_URL_SUFFIX +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.testing.constants import GET_CODEMODS_URL_SUFFIX, UPDATE_CODEMOD_DIFF_URL_SUFFIX logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/topological_sort.py b/src/codegen/sdk/topological_sort.py similarity index 100% rename from src/graph_sitter/topological_sort.py rename to src/codegen/sdk/topological_sort.py diff --git a/src/graph_sitter/tree_sitter_parser.py b/src/codegen/sdk/tree_sitter_parser.py similarity index 98% rename from src/graph_sitter/tree_sitter_parser.py rename to src/codegen/sdk/tree_sitter_parser.py index c758a25b0..c9a04c6cc 100644 --- a/src/graph_sitter/tree_sitter_parser.py +++ b/src/codegen/sdk/tree_sitter_parser.py @@ -9,7 +9,7 @@ from tree_sitter import Language, Parser from tree_sitter import Node as TSNode -from graph_sitter.output.utils import stylize_error +from codegen.sdk.output.utils import stylize_error PY_LANGUAGE = Language(ts_python.language()) JS_LANGUAGE = Language(ts_javascript.language()) diff --git a/src/graph_sitter/types.py b/src/codegen/sdk/types.py similarity index 100% rename from src/graph_sitter/types.py rename to src/codegen/sdk/types.py diff --git a/src/graph_sitter/typescript/__init__.py b/src/codegen/sdk/typescript/__init__.py similarity index 100% rename from src/graph_sitter/typescript/__init__.py rename to src/codegen/sdk/typescript/__init__.py diff --git a/src/graph_sitter/typescript/assignment.py b/src/codegen/sdk/typescript/assignment.py similarity index 88% rename from src/graph_sitter/typescript/assignment.py rename to src/codegen/sdk/typescript/assignment.py index 831aab4fd..ca7b4bd69 100644 --- a/src/graph_sitter/typescript/assignment.py +++ b/src/codegen/sdk/typescript/assignment.py @@ -4,16 +4,16 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.autocommit import writer -from graph_sitter.core.expressions.multi_expression import MultiExpression -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.expressions.multi_expression import MultiExpression +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.statements.assignment_statement import TSAssignmentStatement + from codegen.sdk.typescript.statements.assignment_statement import TSAssignmentStatement @ts_apidoc diff --git a/src/graph_sitter/typescript/class_definition.py b/src/codegen/sdk/typescript/class_definition.py similarity index 87% rename from src/graph_sitter/typescript/class_definition.py rename to src/codegen/sdk/typescript/class_definition.py index 95e1abfa3..88832178e 100644 --- a/src/graph_sitter/typescript/class_definition.py +++ b/src/codegen/sdk/typescript/class_definition.py @@ -5,26 +5,26 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.class_definition import Class -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.generic_type import GenericType -from graph_sitter.core.expressions.placeholder_type import PlaceholderType -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.symbol_statement import SymbolStatement -from graph_sitter.core.symbol_group import SymbolGroup -from graph_sitter.core.symbol_groups.multi_line_collection import MultiLineCollection -from graph_sitter.core.symbol_groups.parents import Parents -from graph_sitter.typescript.detached_symbols.decorator import TSDecorator -from graph_sitter.typescript.detached_symbols.parameter import TSParameter -from graph_sitter.typescript.expressions.type import TSType -from graph_sitter.typescript.function import TSFunction -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.generic_type import GenericType +from codegen.sdk.core.expressions.placeholder_type import PlaceholderType +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.symbol_statement import SymbolStatement +from codegen.sdk.core.symbol_group import SymbolGroup +from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection +from codegen.sdk.core.symbol_groups.parents import Parents +from codegen.sdk.typescript.detached_symbols.decorator import TSDecorator +from codegen.sdk.typescript.detached_symbols.parameter import TSParameter +from codegen.sdk.typescript.expressions.type import TSType +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc @ts_apidoc diff --git a/src/graph_sitter/typescript/config_parser.py b/src/codegen/sdk/typescript/config_parser.py similarity index 87% rename from src/graph_sitter/typescript/config_parser.py rename to src/codegen/sdk/typescript/config_parser.py index 93a4d42a6..2f5e79861 100644 --- a/src/graph_sitter/typescript/config_parser.py +++ b/src/codegen/sdk/typescript/config_parser.py @@ -1,14 +1,14 @@ from pathlib import Path from typing import TYPE_CHECKING -from graph_sitter.codebase.config_parser import ConfigParser -from graph_sitter.core.file import File -from graph_sitter.enums import NodeType -from graph_sitter.typescript.ts_config import TSConfig +from codegen.sdk.codebase.config_parser import ConfigParser +from codegen.sdk.core.file import File +from codegen.sdk.enums import NodeType +from codegen.sdk.typescript.ts_config import TSConfig if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.typescript.file import TSFile + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.typescript.file import TSFile import os from functools import cache diff --git a/src/graph_sitter/typescript/detached_symbols/code_block.py b/src/codegen/sdk/typescript/detached_symbols/code_block.py similarity index 85% rename from src/graph_sitter/typescript/detached_symbols/code_block.py rename to src/codegen/sdk/typescript/detached_symbols/code_block.py index ebb7f6eae..5d1592b33 100644 --- a/src/graph_sitter/typescript/detached_symbols/code_block.py +++ b/src/codegen/sdk/typescript/detached_symbols/code_block.py @@ -1,15 +1,15 @@ from typing import TYPE_CHECKING, Generic, Self, TypeVar -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.statements.statement import Statement -from graph_sitter.core.symbol_groups.multi_line_collection import MultiLineCollection -from graph_sitter.extensions.utils import find_line_start_and_end_nodes -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection +from codegen.sdk.extensions.utils import find_line_start_and_end_nodes +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.interfaces.has_block import TSHasBlock + from codegen.sdk.typescript.interfaces.has_block import TSHasBlock Parent = TypeVar("Parent", bound="TSHasBlock") diff --git a/src/graph_sitter/typescript/detached_symbols/decorator.py b/src/codegen/sdk/typescript/detached_symbols/decorator.py similarity index 87% rename from src/graph_sitter/typescript/detached_symbols/decorator.py rename to src/codegen/sdk/typescript/detached_symbols/decorator.py index 2cd45f8f6..194eb1a45 100644 --- a/src/graph_sitter/typescript/detached_symbols/decorator.py +++ b/src/codegen/sdk/typescript/detached_symbols/decorator.py @@ -2,10 +2,10 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.detached_symbols.decorator import Decorator -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.detached_symbols.decorator import Decorator +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.writer_decorators import ts_apidoc @ts_apidoc diff --git a/src/graph_sitter/typescript/detached_symbols/jsx/element.py b/src/codegen/sdk/typescript/detached_symbols/jsx/element.py similarity index 90% rename from src/graph_sitter/typescript/detached_symbols/jsx/element.py rename to src/codegen/sdk/typescript/detached_symbols/jsx/element.py index 5a18152c4..9619d67d6 100644 --- a/src/graph_sitter/typescript/detached_symbols/jsx/element.py +++ b/src/codegen/sdk/typescript/detached_symbols/jsx/element.py @@ -5,19 +5,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression, Value -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.typescript.detached_symbols.jsx.expression import JSXExpression -from graph_sitter.typescript.detached_symbols.jsx.prop import JSXProp -from graph_sitter.utils import find_all_descendants -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression, Value +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.typescript.detached_symbols.jsx.expression import JSXExpression +from codegen.sdk.typescript.detached_symbols.jsx.prop import JSXProp +from codegen.sdk.utils import find_all_descendants +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/typescript/detached_symbols/jsx/expression.py b/src/codegen/sdk/typescript/detached_symbols/jsx/expression.py similarity index 76% rename from src/graph_sitter/typescript/detached_symbols/jsx/expression.py rename to src/codegen/sdk/typescript/detached_symbols/jsx/expression.py index 13a5a79c1..fe10a3c2d 100644 --- a/src/graph_sitter/typescript/detached_symbols/jsx/expression.py +++ b/src/codegen/sdk/typescript/detached_symbols/jsx/expression.py @@ -1,14 +1,14 @@ from functools import cached_property from typing import TYPE_CHECKING, Self, override -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.unwrappable import Unwrappable -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.unwrappable import Unwrappable +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: pass @@ -65,8 +65,8 @@ def unwrap(self, node: Expression | None = None): Returns: None """ - from graph_sitter.typescript.detached_symbols.jsx.element import JSXElement - from graph_sitter.typescript.detached_symbols.jsx.prop import JSXProp + from codegen.sdk.typescript.detached_symbols.jsx.element import JSXElement + from codegen.sdk.typescript.detached_symbols.jsx.prop import JSXProp if node is None: node = self diff --git a/src/graph_sitter/typescript/detached_symbols/jsx/prop.py b/src/codegen/sdk/typescript/detached_symbols/jsx/prop.py similarity index 86% rename from src/graph_sitter/typescript/detached_symbols/jsx/prop.py rename to src/codegen/sdk/typescript/detached_symbols/jsx/prop.py index 525d207f6..48b91ca37 100644 --- a/src/graph_sitter/typescript/detached_symbols/jsx/prop.py +++ b/src/codegen/sdk/typescript/detached_symbols/jsx/prop.py @@ -2,19 +2,19 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.typescript.detached_symbols.jsx.expression import JSXExpression -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.typescript.detached_symbols.jsx.expression import JSXExpression +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.core.function import Function - from graph_sitter.typescript.detached_symbols.jsx.element import JSXElement + from codegen.sdk.core.function import Function + from codegen.sdk.typescript.detached_symbols.jsx.element import JSXElement @ts_apidoc diff --git a/src/graph_sitter/typescript/detached_symbols/parameter.py b/src/codegen/sdk/typescript/detached_symbols/parameter.py similarity index 88% rename from src/graph_sitter/typescript/detached_symbols/parameter.py rename to src/codegen/sdk/typescript/detached_symbols/parameter.py index 0cfdabe7a..b142d220b 100644 --- a/src/graph_sitter/typescript/detached_symbols/parameter.py +++ b/src/codegen/sdk/typescript/detached_symbols/parameter.py @@ -4,22 +4,22 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import reader -from graph_sitter.core.autocommit.decorators import writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.expressions.union_type import UnionType -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.placeholder.placeholder import Placeholder -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.extensions.autocommit import commiter -from graph_sitter.typescript.expressions.object_type import TSObjectType -from graph_sitter.typescript.expressions.type import TSType -from graph_sitter.typescript.symbol_groups.dict import TSPair -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.autocommit.decorators import writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.expressions.union_type import UnionType +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.placeholder.placeholder import Placeholder +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.extensions.autocommit import commiter +from codegen.sdk.typescript.expressions.object_type import TSObjectType +from codegen.sdk.typescript.expressions.type import TSType +from codegen.sdk.typescript.symbol_groups.dict import TSPair +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.function import TSFunction + from codegen.sdk.typescript.function import TSFunction @ts_apidoc diff --git a/src/graph_sitter/typescript/enum_definition.py b/src/codegen/sdk/typescript/enum_definition.py similarity index 71% rename from src/graph_sitter/typescript/enum_definition.py rename to src/codegen/sdk/typescript/enum_definition.py index 411ddec23..26c332c92 100644 --- a/src/graph_sitter/typescript/enum_definition.py +++ b/src/codegen/sdk/typescript/enum_definition.py @@ -4,24 +4,24 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement -from graph_sitter.enums import SymbolType -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.typescript.statements.attribute import TSAttribute -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.enums import SymbolType +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.typescript.statements.attribute import TSAttribute +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock Parent = TypeVar("Parent", bound="TSHasBlock") diff --git a/src/graph_sitter/typescript/enums.py b/src/codegen/sdk/typescript/enums.py similarity index 100% rename from src/graph_sitter/typescript/enums.py rename to src/codegen/sdk/typescript/enums.py diff --git a/src/graph_sitter/typescript/export.py b/src/codegen/sdk/typescript/export.py similarity index 94% rename from src/graph_sitter/typescript/export.py rename to src/codegen/sdk/typescript/export.py index a42d0e24f..25e0b29d7 100644 --- a/src/graph_sitter/typescript/export.py +++ b/src/codegen/sdk/typescript/export.py @@ -5,40 +5,40 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.autocommit.decorators import writer -from graph_sitter.core.dataclasses.usage import UsageKind, UsageType -from graph_sitter.core.export import Export -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.interfaces.chainable import Chainable -from graph_sitter.core.interfaces.exportable import Exportable -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.has_value import HasValue -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.enums import EdgeType, ImportType, NodeType -from graph_sitter.extensions.utils import cached_property -from graph_sitter.typescript.assignment import TSAssignment -from graph_sitter.typescript.class_definition import TSClass -from graph_sitter.typescript.enum_definition import TSEnum -from graph_sitter.typescript.enums import TSFunctionTypeNames -from graph_sitter.typescript.function import TSFunction -from graph_sitter.typescript.import_resolution import TSImport -from graph_sitter.typescript.interface import TSInterface -from graph_sitter.typescript.namespace import TSNamespace -from graph_sitter.typescript.statements.assignment_statement import TSAssignmentStatement -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.typescript.type_alias import TSTypeAlias -from graph_sitter.utils import find_all_descendants -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.autocommit.decorators import writer +from codegen.sdk.core.dataclasses.usage import UsageKind, UsageType +from codegen.sdk.core.export import Export +from codegen.sdk.core.expressions.name import Name +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.interfaces.chainable import Chainable +from codegen.sdk.core.interfaces.exportable import Exportable +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.has_value import HasValue +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.enums import EdgeType, ImportType, NodeType +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.typescript.assignment import TSAssignment +from codegen.sdk.typescript.class_definition import TSClass +from codegen.sdk.typescript.enum_definition import TSEnum +from codegen.sdk.typescript.enums import TSFunctionTypeNames +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.typescript.import_resolution import TSImport +from codegen.sdk.typescript.interface import TSInterface +from codegen.sdk.typescript.namespace import TSNamespace +from codegen.sdk.typescript.statements.assignment_statement import TSAssignmentStatement +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.typescript.type_alias import TSTypeAlias +from codegen.sdk.utils import find_all_descendants +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.core.statements.export_statement import ExportStatement + from codegen.sdk.core.statements.export_statement import ExportStatement @ts_apidoc diff --git a/src/graph_sitter/typescript/expressions/array_type.py b/src/codegen/sdk/typescript/expressions/array_type.py similarity index 73% rename from src/graph_sitter/typescript/expressions/array_type.py rename to src/codegen/sdk/typescript/expressions/array_type.py index a1dbcccff..953e4a476 100644 --- a/src/graph_sitter/typescript/expressions/array_type.py +++ b/src/codegen/sdk/typescript/expressions/array_type.py @@ -2,8 +2,8 @@ from tree_sitter import Node as TSNode -from graph_sitter.typescript.expressions.named_type import TSNamedType -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.typescript.expressions.named_type import TSNamedType +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/typescript/expressions/chained_attribute.py b/src/codegen/sdk/typescript/expressions/chained_attribute.py similarity index 79% rename from src/graph_sitter/typescript/expressions/chained_attribute.py rename to src/codegen/sdk/typescript/expressions/chained_attribute.py index 79d1b9934..3deb659c6 100644 --- a/src/graph_sitter/typescript/expressions/chained_attribute.py +++ b/src/codegen/sdk/typescript/expressions/chained_attribute.py @@ -1,11 +1,11 @@ from typing import Generic, TypeVar -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression, Name -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression, Name +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/typescript/expressions/conditional_type.py b/src/codegen/sdk/typescript/expressions/conditional_type.py similarity index 72% rename from src/graph_sitter/typescript/expressions/conditional_type.py rename to src/codegen/sdk/typescript/expressions/conditional_type.py index 4a26f51a4..0f98dce32 100644 --- a/src/graph_sitter/typescript/expressions/conditional_type.py +++ b/src/codegen/sdk/typescript/expressions/conditional_type.py @@ -3,17 +3,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.typescript.expressions.type import TSType + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.typescript.expressions.type import TSType Parent = TypeVar("Parent") diff --git a/src/graph_sitter/typescript/expressions/expression_type.py b/src/codegen/sdk/typescript/expressions/expression_type.py similarity index 63% rename from src/graph_sitter/typescript/expressions/expression_type.py rename to src/codegen/sdk/typescript/expressions/expression_type.py index 7be9ef444..11a68711e 100644 --- a/src/graph_sitter/typescript/expressions/expression_type.py +++ b/src/codegen/sdk/typescript/expressions/expression_type.py @@ -2,12 +2,12 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.typescript.expressions.named_type import TSNamedType -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.typescript.expressions.named_type import TSNamedType +from codegen.sdk.writer_decorators import ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/typescript/expressions/function_type.py b/src/codegen/sdk/typescript/expressions/function_type.py similarity index 78% rename from src/graph_sitter/typescript/expressions/function_type.py rename to src/codegen/sdk/typescript/expressions/function_type.py index 24524bcdd..262636c26 100644 --- a/src/graph_sitter/typescript/expressions/function_type.py +++ b/src/codegen/sdk/typescript/expressions/function_type.py @@ -3,20 +3,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.typescript.detached_symbols.parameter import TSParameter -from graph_sitter.typescript.placeholder.placeholder_return_type import TSReturnTypePlaceholder -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.typescript.detached_symbols.parameter import TSParameter +from codegen.sdk.typescript.placeholder.placeholder_return_type import TSReturnTypePlaceholder +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.typescript.expressions.type import TSType + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.typescript.expressions.type import TSType Parent = TypeVar("Parent") diff --git a/src/graph_sitter/typescript/expressions/generic_type.py b/src/codegen/sdk/typescript/expressions/generic_type.py similarity index 75% rename from src/graph_sitter/typescript/expressions/generic_type.py rename to src/codegen/sdk/typescript/expressions/generic_type.py index ce02b4fde..ae423227b 100644 --- a/src/graph_sitter/typescript/expressions/generic_type.py +++ b/src/codegen/sdk/typescript/expressions/generic_type.py @@ -2,10 +2,10 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.expressions.generic_type import GenericType -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.core.symbol_groups.dict import Dict -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.expressions.generic_type import GenericType +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.core.symbol_groups.dict import Dict +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/typescript/expressions/lookup_type.py b/src/codegen/sdk/typescript/expressions/lookup_type.py similarity index 73% rename from src/graph_sitter/typescript/expressions/lookup_type.py rename to src/codegen/sdk/typescript/expressions/lookup_type.py index 9c356c286..825f4ee95 100644 --- a/src/graph_sitter/typescript/expressions/lookup_type.py +++ b/src/codegen/sdk/typescript/expressions/lookup_type.py @@ -3,18 +3,18 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.typescript.expressions.type import TSType + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.typescript.expressions.type import TSType Parent = TypeVar("Parent") diff --git a/src/graph_sitter/typescript/expressions/named_type.py b/src/codegen/sdk/typescript/expressions/named_type.py similarity index 73% rename from src/graph_sitter/typescript/expressions/named_type.py rename to src/codegen/sdk/typescript/expressions/named_type.py index 0c9527727..ce955d268 100644 --- a/src/graph_sitter/typescript/expressions/named_type.py +++ b/src/codegen/sdk/typescript/expressions/named_type.py @@ -2,8 +2,8 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.expressions.named_type import NamedType -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.expressions.named_type import NamedType +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/typescript/expressions/object_type.py b/src/codegen/sdk/typescript/expressions/object_type.py similarity index 82% rename from src/graph_sitter/typescript/expressions/object_type.py rename to src/codegen/sdk/typescript/expressions/object_type.py index 09919b520..188bc581f 100644 --- a/src/graph_sitter/typescript/expressions/object_type.py +++ b/src/codegen/sdk/typescript/expressions/object_type.py @@ -2,17 +2,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.expressions.value import Value -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.typescript.symbol_groups.dict import TSDict, TSPair -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.expression import Expression +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.expressions.value import Value +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.typescript.symbol_groups.dict import TSDict, TSPair +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph import logging @@ -30,7 +30,7 @@ class TSObjectPair(TSPair, Generic[Parent]): """ def _get_key_value(self) -> tuple[Expression[Self] | None, Expression[Self] | None]: - from graph_sitter.typescript.expressions.function_type import TSFunctionType + from codegen.sdk.typescript.expressions.function_type import TSFunctionType key, value = None, None if self.ts_node_type == "property_signature": diff --git a/src/graph_sitter/typescript/expressions/query_type.py b/src/codegen/sdk/typescript/expressions/query_type.py similarity index 70% rename from src/graph_sitter/typescript/expressions/query_type.py rename to src/codegen/sdk/typescript/expressions/query_type.py index 45d3b68ac..e9352d4ec 100644 --- a/src/graph_sitter/typescript/expressions/query_type.py +++ b/src/codegen/sdk/typescript/expressions/query_type.py @@ -3,17 +3,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.typescript.expressions.type import TSType + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.typescript.expressions.type import TSType Parent = TypeVar("Parent") diff --git a/src/graph_sitter/typescript/expressions/readonly_type.py b/src/codegen/sdk/typescript/expressions/readonly_type.py similarity index 70% rename from src/graph_sitter/typescript/expressions/readonly_type.py rename to src/codegen/sdk/typescript/expressions/readonly_type.py index 55b270140..0e3d8f134 100644 --- a/src/graph_sitter/typescript/expressions/readonly_type.py +++ b/src/codegen/sdk/typescript/expressions/readonly_type.py @@ -3,17 +3,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.autocommit import reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.typescript.expressions.type import TSType + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.typescript.expressions.type import TSType Parent = TypeVar("Parent") diff --git a/src/graph_sitter/typescript/expressions/string.py b/src/codegen/sdk/typescript/expressions/string.py similarity index 83% rename from src/graph_sitter/typescript/expressions/string.py rename to src/codegen/sdk/typescript/expressions/string.py index c5511a9a8..7e05e9cf3 100644 --- a/src/graph_sitter/typescript/expressions/string.py +++ b/src/codegen/sdk/typescript/expressions/string.py @@ -2,12 +2,12 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.expressions import Expression, String -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.expressions import Expression, String +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound="Expression") diff --git a/src/graph_sitter/typescript/expressions/ternary_expression.py b/src/codegen/sdk/typescript/expressions/ternary_expression.py similarity index 75% rename from src/graph_sitter/typescript/expressions/ternary_expression.py rename to src/codegen/sdk/typescript/expressions/ternary_expression.py index 6a1b16f1f..ab00561a5 100644 --- a/src/graph_sitter/typescript/expressions/ternary_expression.py +++ b/src/codegen/sdk/typescript/expressions/ternary_expression.py @@ -1,8 +1,8 @@ from typing import TypeVar -from graph_sitter.core.expressions.ternary_expression import TernaryExpression -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.expressions.ternary_expression import TernaryExpression +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.writer_decorators import ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/typescript/expressions/type.py b/src/codegen/sdk/typescript/expressions/type.py similarity index 100% rename from src/graph_sitter/typescript/expressions/type.py rename to src/codegen/sdk/typescript/expressions/type.py diff --git a/src/graph_sitter/typescript/expressions/undefined_type.py b/src/codegen/sdk/typescript/expressions/undefined_type.py similarity index 61% rename from src/graph_sitter/typescript/expressions/undefined_type.py rename to src/codegen/sdk/typescript/expressions/undefined_type.py index 1b1211c00..281c41702 100644 --- a/src/graph_sitter/typescript/expressions/undefined_type.py +++ b/src/codegen/sdk/typescript/expressions/undefined_type.py @@ -1,12 +1,12 @@ from collections.abc import Generator from typing import TYPE_CHECKING, Generic, Self, TypeVar, override -from graph_sitter.codebase.resolution_stack import ResolutionStack -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.resolution_stack import ResolutionStack +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/typescript/expressions/union_type.py b/src/codegen/sdk/typescript/expressions/union_type.py similarity index 69% rename from src/graph_sitter/typescript/expressions/union_type.py rename to src/codegen/sdk/typescript/expressions/union_type.py index f05a50f4d..6084f6c02 100644 --- a/src/graph_sitter/typescript/expressions/union_type.py +++ b/src/codegen/sdk/typescript/expressions/union_type.py @@ -1,7 +1,7 @@ from typing import TYPE_CHECKING, Generic, TypeVar -from graph_sitter.core.expressions.union_type import UnionType -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.expressions.union_type import UnionType +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/typescript/external/dependency_manager.py b/src/codegen/sdk/typescript/external/dependency_manager.py similarity index 99% rename from src/graph_sitter/typescript/external/dependency_manager.py rename to src/codegen/sdk/typescript/external/dependency_manager.py index fcd885a80..053fd00e5 100644 --- a/src/graph_sitter/typescript/external/dependency_manager.py +++ b/src/codegen/sdk/typescript/external/dependency_manager.py @@ -11,8 +11,8 @@ import pyjson5 import requests -from graph_sitter.core.external.dependency_manager import DependencyManager -from graph_sitter.utils import shadow_files +from codegen.sdk.core.external.dependency_manager import DependencyManager +from codegen.sdk.utils import shadow_files logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/typescript/external/mega_racer.py b/src/codegen/sdk/typescript/external/mega_racer.py similarity index 100% rename from src/graph_sitter/typescript/external/mega_racer.py rename to src/codegen/sdk/typescript/external/mega_racer.py diff --git a/src/graph_sitter/typescript/external/ts_analyzer_engine.py b/src/codegen/sdk/typescript/external/ts_analyzer_engine.py similarity index 97% rename from src/graph_sitter/typescript/external/ts_analyzer_engine.py rename to src/codegen/sdk/typescript/external/ts_analyzer_engine.py index dfed5d8c5..89b321554 100644 --- a/src/graph_sitter/typescript/external/ts_analyzer_engine.py +++ b/src/codegen/sdk/typescript/external/ts_analyzer_engine.py @@ -12,12 +12,12 @@ from py_mini_racer._objects import JSMappedObject from py_mini_racer._types import JSEvalException -from graph_sitter.core.external.language_engine import LanguageEngine -from graph_sitter.typescript.external.mega_racer import MegaRacer +from codegen.sdk.core.external.language_engine import LanguageEngine +from codegen.sdk.typescript.external.mega_racer import MegaRacer if TYPE_CHECKING: - from graph_sitter.core.external.dependency_manager import DependencyManager - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.external.dependency_manager import DependencyManager + from codegen.sdk.core.interfaces.editable import Editable logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/typescript/external/ts_declassify/ts_declassify.py b/src/codegen/sdk/typescript/external/ts_declassify/ts_declassify.py similarity index 98% rename from src/graph_sitter/typescript/external/ts_declassify/ts_declassify.py rename to src/codegen/sdk/typescript/external/ts_declassify/ts_declassify.py index 4e2ffcb50..312908f78 100644 --- a/src/graph_sitter/typescript/external/ts_declassify/ts_declassify.py +++ b/src/codegen/sdk/typescript/external/ts_declassify/ts_declassify.py @@ -3,7 +3,7 @@ import shutil import subprocess -from graph_sitter.core.external.external_process import ExternalProcess +from codegen.sdk.core.external.external_process import ExternalProcess logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/.gitignore b/src/codegen/sdk/typescript/external/typescript_analyzer/.gitignore similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/.gitignore rename to src/codegen/sdk/typescript/external/typescript_analyzer/.gitignore diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/package-lock.json b/src/codegen/sdk/typescript/external/typescript_analyzer/package-lock.json similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/package-lock.json rename to src/codegen/sdk/typescript/external/typescript_analyzer/package-lock.json diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/package.json b/src/codegen/sdk/typescript/external/typescript_analyzer/package.json similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/package.json rename to src/codegen/sdk/typescript/external/typescript_analyzer/package.json diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/rollup.config.js b/src/codegen/sdk/typescript/external/typescript_analyzer/rollup.config.js similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/rollup.config.js rename to src/codegen/sdk/typescript/external/typescript_analyzer/rollup.config.js diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/src/analyzer.ts b/src/codegen/sdk/typescript/external/typescript_analyzer/src/analyzer.ts similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/src/analyzer.ts rename to src/codegen/sdk/typescript/external/typescript_analyzer/src/analyzer.ts diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/src/fs_proxy.ts b/src/codegen/sdk/typescript/external/typescript_analyzer/src/fs_proxy.ts similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/src/fs_proxy.ts rename to src/codegen/sdk/typescript/external/typescript_analyzer/src/fs_proxy.ts diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/src/fsi.ts b/src/codegen/sdk/typescript/external/typescript_analyzer/src/fsi.ts similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/src/fsi.ts rename to src/codegen/sdk/typescript/external/typescript_analyzer/src/fsi.ts diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/src/get_type_at_position.ts b/src/codegen/sdk/typescript/external/typescript_analyzer/src/get_type_at_position.ts similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/src/get_type_at_position.ts rename to src/codegen/sdk/typescript/external/typescript_analyzer/src/get_type_at_position.ts diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/src/index.ts b/src/codegen/sdk/typescript/external/typescript_analyzer/src/index.ts similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/src/index.ts rename to src/codegen/sdk/typescript/external/typescript_analyzer/src/index.ts diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/src/run_full.ts b/src/codegen/sdk/typescript/external/typescript_analyzer/src/run_full.ts similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/src/run_full.ts rename to src/codegen/sdk/typescript/external/typescript_analyzer/src/run_full.ts diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/src/test_fsi.ts b/src/codegen/sdk/typescript/external/typescript_analyzer/src/test_fsi.ts similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/src/test_fsi.ts rename to src/codegen/sdk/typescript/external/typescript_analyzer/src/test_fsi.ts diff --git a/src/graph_sitter/typescript/external/typescript_analyzer/tsconfig.json b/src/codegen/sdk/typescript/external/typescript_analyzer/tsconfig.json similarity index 100% rename from src/graph_sitter/typescript/external/typescript_analyzer/tsconfig.json rename to src/codegen/sdk/typescript/external/typescript_analyzer/tsconfig.json diff --git a/src/graph_sitter/typescript/file.py b/src/codegen/sdk/typescript/file.py similarity index 92% rename from src/graph_sitter/typescript/file.py rename to src/codegen/sdk/typescript/file.py index 673eb2814..93edcd24a 100644 --- a/src/graph_sitter/typescript/file.py +++ b/src/codegen/sdk/typescript/file.py @@ -3,30 +3,30 @@ import os from typing import TYPE_CHECKING -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, mover, reader, writer -from graph_sitter.core.file import SourceFile -from graph_sitter.core.interfaces.exportable import Exportable -from graph_sitter.core.statements.export_statement import ExportStatement -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import ImportType, NodeType, ProgrammingLanguage, SymbolType -from graph_sitter.extensions.sort import sort_editables -from graph_sitter.extensions.utils import cached_property -from graph_sitter.typescript.assignment import TSAssignment -from graph_sitter.typescript.class_definition import TSClass -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.export import TSExport -from graph_sitter.typescript.function import TSFunction -from graph_sitter.typescript.import_resolution import TSImport -from graph_sitter.typescript.interface import TSInterface -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.typescript.namespace import TSNamespace -from graph_sitter.typescript.statements.import_statement import TSImportStatement -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.typescript.ts_config import TSConfig -from graph_sitter.typescript.type_alias import TSTypeAlias -from graph_sitter.utils import calculate_base_path, find_all_descendants -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, mover, reader, writer +from codegen.sdk.core.file import SourceFile +from codegen.sdk.core.interfaces.exportable import Exportable +from codegen.sdk.core.statements.export_statement import ExportStatement +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import ImportType, NodeType, ProgrammingLanguage, SymbolType +from codegen.sdk.extensions.sort import sort_editables +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.typescript.assignment import TSAssignment +from codegen.sdk.typescript.class_definition import TSClass +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.export import TSExport +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.typescript.import_resolution import TSImport +from codegen.sdk.typescript.interface import TSInterface +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.typescript.namespace import TSNamespace +from codegen.sdk.typescript.statements.import_statement import TSImportStatement +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.typescript.ts_config import TSConfig +from codegen.sdk.typescript.type_alias import TSTypeAlias +from codegen.sdk.utils import calculate_base_path, find_all_descendants +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/typescript/function.py b/src/codegen/sdk/typescript/function.py similarity index 92% rename from src/graph_sitter/typescript/function.py rename to src/codegen/sdk/typescript/function.py index 5249c28f3..736231fd8 100644 --- a/src/graph_sitter/typescript/function.py +++ b/src/codegen/sdk/typescript/function.py @@ -6,28 +6,28 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader, writer -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.function import Function -from graph_sitter.core.import_resolution import Import, WildcardImport -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.typescript.detached_symbols.decorator import TSDecorator -from graph_sitter.typescript.detached_symbols.parameter import TSParameter -from graph_sitter.typescript.enums import TSFunctionTypeNames -from graph_sitter.typescript.expressions.type import TSType -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.typescript.placeholder.placeholder_return_type import TSReturnTypePlaceholder -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.utils import find_all_descendants -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader, writer +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.function import Function +from codegen.sdk.core.import_resolution import Import, WildcardImport +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.typescript.detached_symbols.decorator import TSDecorator +from codegen.sdk.typescript.detached_symbols.parameter import TSParameter +from codegen.sdk.typescript.enums import TSFunctionTypeNames +from codegen.sdk.typescript.expressions.type import TSType +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.typescript.placeholder.placeholder_return_type import TSReturnTypePlaceholder +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.utils import find_all_descendants +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.core.statements.export_statement import ExportStatement - from graph_sitter.core.statements.symbol_statement import SymbolStatement + from codegen.sdk.core.statements.export_statement import ExportStatement + from codegen.sdk.core.statements.symbol_statement import SymbolStatement _VALID_TYPE_NAMES = {function_type.value for function_type in TSFunctionTypeNames} logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/typescript/import_resolution.py b/src/codegen/sdk/typescript/import_resolution.py similarity index 96% rename from src/graph_sitter/typescript/import_resolution.py rename to src/codegen/sdk/typescript/import_resolution.py index 12c68dca6..8d83f6c64 100644 --- a/src/graph_sitter/typescript/import_resolution.py +++ b/src/codegen/sdk/typescript/import_resolution.py @@ -7,23 +7,23 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import reader -from graph_sitter.core.expressions import Name -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.import_resolution import Import, ImportResolution, WildcardImport -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.exportable import Exportable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.import_statement import ImportStatement -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import ImportType, NodeType -from graph_sitter.utils import find_all_descendants, find_first_ancestor, find_first_descendant -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.expressions import Name +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.import_resolution import Import, ImportResolution, WildcardImport +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.exportable import Exportable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.import_statement import ImportStatement +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import ImportType, NodeType +from codegen.sdk.utils import find_all_descendants, find_first_ancestor, find_first_descendant +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.file import TSFile - from graph_sitter.typescript.statements.import_statement import TSImportStatement + from codegen.sdk.typescript.file import TSFile + from codegen.sdk.typescript.statements.import_statement import TSImportStatement @ts_apidoc diff --git a/src/graph_sitter/typescript/interface.py b/src/codegen/sdk/typescript/interface.py similarity index 70% rename from src/graph_sitter/typescript/interface.py rename to src/codegen/sdk/typescript/interface.py index 801433c8f..120480df2 100644 --- a/src/graph_sitter/typescript/interface.py +++ b/src/codegen/sdk/typescript/interface.py @@ -4,22 +4,22 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.interface import Interface -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement -from graph_sitter.core.symbol_groups.parents import Parents -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.expressions.type import TSType -from graph_sitter.typescript.function import TSFunction -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.typescript.statements.attribute import TSAttribute -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.interface import Interface +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.core.symbol_groups.parents import Parents +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.expressions.type import TSType +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.typescript.statements.attribute import TSAttribute +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc Parent = TypeVar("Parent", bound="TSHasBlock") @@ -40,7 +40,7 @@ def __init__( G: CodebaseGraph, parent: Statement[CodeBlock[Parent, ...]], ) -> None: - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock super().__init__(ts_node, file_id, G, parent) body_node = ts_node.child_by_field_name("body") diff --git a/src/graph_sitter/typescript/interfaces/has_block.py b/src/codegen/sdk/typescript/interfaces/has_block.py similarity index 90% rename from src/graph_sitter/typescript/interfaces/has_block.py rename to src/codegen/sdk/typescript/interfaces/has_block.py index fd1688b16..82cd20291 100644 --- a/src/graph_sitter/typescript/interfaces/has_block.py +++ b/src/codegen/sdk/typescript/interfaces/has_block.py @@ -3,19 +3,19 @@ from functools import cached_property from typing import TYPE_CHECKING, Self -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.detached_symbols.code_block import CodeBlock -from graph_sitter.core.interfaces.has_block import HasBlock -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.extensions.utils import find_all_descendants -from graph_sitter.typescript.detached_symbols.decorator import TSDecorator -from graph_sitter.typescript.statements.comment import TSComment, TSCommentType -from graph_sitter.typescript.symbol_groups.comment_group import TSCommentGroup -from graph_sitter.utils import find_index -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.detached_symbols.code_block import CodeBlock +from codegen.sdk.core.interfaces.has_block import HasBlock +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.extensions.utils import find_all_descendants +from codegen.sdk.typescript.detached_symbols.decorator import TSDecorator +from codegen.sdk.typescript.statements.comment import TSComment, TSCommentType +from codegen.sdk.typescript.symbol_groups.comment_group import TSCommentGroup +from codegen.sdk.utils import find_index +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.detached_symbols.jsx.element import JSXElement + from codegen.sdk.typescript.detached_symbols.jsx.element import JSXElement @ts_apidoc diff --git a/src/graph_sitter/typescript/namespace.py b/src/codegen/sdk/typescript/namespace.py similarity index 86% rename from src/graph_sitter/typescript/namespace.py rename to src/codegen/sdk/typescript/namespace.py index ae3116872..c9df745eb 100644 --- a/src/graph_sitter/typescript/namespace.py +++ b/src/codegen/sdk/typescript/namespace.py @@ -4,27 +4,27 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import commiter -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement -from graph_sitter.core.statements.symbol_statement import SymbolStatement -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import SymbolType -from graph_sitter.extensions.utils import cached_property -from graph_sitter.typescript.class_definition import TSClass -from graph_sitter.typescript.enum_definition import TSEnum -from graph_sitter.typescript.function import TSFunction -from graph_sitter.typescript.interface import TSInterface -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.typescript.type_alias import TSTypeAlias -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import commiter +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement +from codegen.sdk.core.statements.symbol_statement import SymbolStatement +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import SymbolType +from codegen.sdk.extensions.utils import cached_property +from codegen.sdk.typescript.class_definition import TSClass +from codegen.sdk.typescript.enum_definition import TSEnum +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.typescript.interface import TSInterface +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.typescript.type_alias import TSTypeAlias +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock @ts_apidoc diff --git a/src/graph_sitter/typescript/placeholder/placeholder_return_type.py b/src/codegen/sdk/typescript/placeholder/placeholder_return_type.py similarity index 90% rename from src/graph_sitter/typescript/placeholder/placeholder_return_type.py rename to src/codegen/sdk/typescript/placeholder/placeholder_return_type.py index a7c1993e7..346c6f31d 100644 --- a/src/graph_sitter/typescript/placeholder/placeholder_return_type.py +++ b/src/codegen/sdk/typescript/placeholder/placeholder_return_type.py @@ -1,8 +1,8 @@ from typing import Generic, TypeVar -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.placeholder.placeholder import Placeholder -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.placeholder.placeholder import Placeholder +from codegen.sdk.writer_decorators import ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/graph_sitter/typescript/statements/__init__.py b/src/codegen/sdk/typescript/statements/__init__.py similarity index 100% rename from src/graph_sitter/typescript/statements/__init__.py rename to src/codegen/sdk/typescript/statements/__init__.py diff --git a/src/graph_sitter/typescript/statements/assignment_statement.py b/src/codegen/sdk/typescript/statements/assignment_statement.py similarity index 92% rename from src/graph_sitter/typescript/statements/assignment_statement.py rename to src/codegen/sdk/typescript/statements/assignment_statement.py index a379df1e3..5a48e451e 100644 --- a/src/graph_sitter/typescript/statements/assignment_statement.py +++ b/src/codegen/sdk/typescript/statements/assignment_statement.py @@ -5,17 +5,17 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.expressions.multi_expression import MultiExpression -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.assignment_statement import AssignmentStatement -from graph_sitter.extensions.autocommit import reader -from graph_sitter.typescript.assignment import TSAssignment -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.expressions.multi_expression import MultiExpression +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.assignment_statement import AssignmentStatement +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.typescript.assignment import TSAssignment +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock - from graph_sitter.typescript.interfaces.has_block import TSHasBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.typescript.interfaces.has_block import TSHasBlock import logging diff --git a/src/graph_sitter/typescript/statements/attribute.py b/src/codegen/sdk/typescript/statements/attribute.py similarity index 81% rename from src/graph_sitter/typescript/statements/attribute.py rename to src/codegen/sdk/typescript/statements/attribute.py index 06eb8de2f..93ae3c4e9 100644 --- a/src/graph_sitter/typescript/statements/attribute.py +++ b/src/codegen/sdk/typescript/statements/attribute.py @@ -4,19 +4,19 @@ from tree_sitter import Node as TSNode -from graph_sitter._proxy import proxy_property -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import reader -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.attribute import Attribute -from graph_sitter.typescript.assignment import TSAssignment -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.statements.assignment_statement import TSAssignmentStatement -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk._proxy import proxy_property +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import reader +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.attribute import Attribute +from codegen.sdk.typescript.assignment import TSAssignment +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.statements.assignment_statement import TSAssignmentStatement +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.interfaces.has_block import TSHasBlock + from codegen.sdk.typescript.interfaces.has_block import TSHasBlock @ts_apidoc diff --git a/src/graph_sitter/typescript/statements/block_statement.py b/src/codegen/sdk/typescript/statements/block_statement.py similarity index 52% rename from src/graph_sitter/typescript/statements/block_statement.py rename to src/codegen/sdk/typescript/statements/block_statement.py index 41d41cb2e..96cd02f94 100644 --- a/src/graph_sitter/typescript/statements/block_statement.py +++ b/src/codegen/sdk/typescript/statements/block_statement.py @@ -2,10 +2,10 @@ from typing import TYPE_CHECKING, Generic, TypeVar -from graph_sitter.core.statements.block_statement import BlockStatement -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.statements.block_statement import BlockStatement +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/typescript/statements/catch_statement.py b/src/codegen/sdk/typescript/statements/catch_statement.py similarity index 64% rename from src/graph_sitter/typescript/statements/catch_statement.py rename to src/codegen/sdk/typescript/statements/catch_statement.py index 6d2916eea..10e021fdc 100644 --- a/src/graph_sitter/typescript/statements/catch_statement.py +++ b/src/codegen/sdk/typescript/statements/catch_statement.py @@ -4,14 +4,14 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.catch_statement import CatchStatement -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.statements.block_statement import TSBlockStatement -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.catch_statement import CatchStatement +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.statements.block_statement import TSBlockStatement +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound="TSCodeBlock") diff --git a/src/graph_sitter/typescript/statements/comment.py b/src/codegen/sdk/typescript/statements/comment.py similarity index 97% rename from src/graph_sitter/typescript/statements/comment.py rename to src/codegen/sdk/typescript/statements/comment.py index d0b99557b..7559e3d0c 100644 --- a/src/graph_sitter/typescript/statements/comment.py +++ b/src/codegen/sdk/typescript/statements/comment.py @@ -3,9 +3,9 @@ from enum import StrEnum from typing import TYPE_CHECKING -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.statements.comment import Comment, lowest_indentation -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.statements.comment import Comment, lowest_indentation +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: pass diff --git a/src/graph_sitter/typescript/statements/for_loop_statement.py b/src/codegen/sdk/typescript/statements/for_loop_statement.py similarity index 84% rename from src/graph_sitter/typescript/statements/for_loop_statement.py rename to src/codegen/sdk/typescript/statements/for_loop_statement.py index b0de7a916..819ba15f5 100644 --- a/src/graph_sitter/typescript/statements/for_loop_statement.py +++ b/src/codegen/sdk/typescript/statements/for_loop_statement.py @@ -4,20 +4,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Expression -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.for_loop_statement import ForLoopStatement -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.typescript.statements.block_statement import TSBlockStatement -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.for_loop_statement import ForLoopStatement +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.typescript.statements.block_statement import TSBlockStatement +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock @ts_apidoc diff --git a/src/graph_sitter/typescript/statements/if_block_statement.py b/src/codegen/sdk/typescript/statements/if_block_statement.py similarity index 89% rename from src/graph_sitter/typescript/statements/if_block_statement.py rename to src/codegen/sdk/typescript/statements/if_block_statement.py index 747ce01af..8c0db8a0f 100644 --- a/src/graph_sitter/typescript/statements/if_block_statement.py +++ b/src/codegen/sdk/typescript/statements/if_block_statement.py @@ -4,15 +4,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.if_block_statement import IfBlockStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.writer_decorators import apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.if_block_statement import IfBlockStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.writer_decorators import apidoc if TYPE_CHECKING: - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock import logging @@ -61,7 +61,7 @@ def __init__( @reader def _parse_consequence_block(self) -> TSCodeBlock: - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock if self.is_if_statement or self.is_elif_statement: consequence_node = self.ts_node.child_by_field_name("consequence") diff --git a/src/graph_sitter/typescript/statements/import_statement.py b/src/codegen/sdk/typescript/statements/import_statement.py similarity index 77% rename from src/graph_sitter/typescript/statements/import_statement.py rename to src/codegen/sdk/typescript/statements/import_statement.py index d5964b094..ea643ea48 100644 --- a/src/graph_sitter/typescript/statements/import_statement.py +++ b/src/codegen/sdk/typescript/statements/import_statement.py @@ -4,16 +4,16 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.expressions.builtin import Builtin -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.import_statement import ImportStatement -from graph_sitter.core.symbol_groups.collection import Collection -from graph_sitter.typescript.import_resolution import TSImport -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.expressions.builtin import Builtin +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.import_statement import ImportStatement +from codegen.sdk.core.symbol_groups.collection import Collection +from codegen.sdk.typescript.import_resolution import TSImport +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock @ts_apidoc diff --git a/src/graph_sitter/typescript/statements/labeled_statement.py b/src/codegen/sdk/typescript/statements/labeled_statement.py similarity index 78% rename from src/graph_sitter/typescript/statements/labeled_statement.py rename to src/codegen/sdk/typescript/statements/labeled_statement.py index 96748bc1b..bd9ae110a 100644 --- a/src/graph_sitter/typescript/statements/labeled_statement.py +++ b/src/codegen/sdk/typescript/statements/labeled_statement.py @@ -4,15 +4,15 @@ from tree_sitter import Node as TSNode -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.core.expressions import Expression, Name -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.statement import Statement, StatementType -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.core.expressions import Expression, Name +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.statement import Statement, StatementType +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock Parent = TypeVar("Parent", bound="TSCodeBlock") diff --git a/src/graph_sitter/typescript/statements/switch_case.py b/src/codegen/sdk/typescript/statements/switch_case.py similarity index 61% rename from src/graph_sitter/typescript/statements/switch_case.py rename to src/codegen/sdk/typescript/statements/switch_case.py index f0c86b89b..e0f08fa8a 100644 --- a/src/graph_sitter/typescript/statements/switch_case.py +++ b/src/codegen/sdk/typescript/statements/switch_case.py @@ -2,14 +2,14 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.switch_case import SwitchCase -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.statements.block_statement import TSBlockStatement -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.switch_case import SwitchCase +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.statements.block_statement import TSBlockStatement +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph @ts_apidoc diff --git a/src/graph_sitter/typescript/statements/switch_statement.py b/src/codegen/sdk/typescript/statements/switch_statement.py similarity index 65% rename from src/graph_sitter/typescript/statements/switch_statement.py rename to src/codegen/sdk/typescript/statements/switch_statement.py index bfbae0424..ae2bf9293 100644 --- a/src/graph_sitter/typescript/statements/switch_statement.py +++ b/src/codegen/sdk/typescript/statements/switch_statement.py @@ -4,14 +4,14 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.switch_statement import SwitchStatement -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.statements.switch_case import TSSwitchCase -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.switch_statement import SwitchStatement +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.statements.switch_case import TSSwitchCase +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph @ts_apidoc diff --git a/src/graph_sitter/typescript/statements/try_catch_statement.py b/src/codegen/sdk/typescript/statements/try_catch_statement.py similarity index 79% rename from src/graph_sitter/typescript/statements/try_catch_statement.py rename to src/codegen/sdk/typescript/statements/try_catch_statement.py index 86b2c9094..3d2e3e3dc 100644 --- a/src/graph_sitter/typescript/statements/try_catch_statement.py +++ b/src/codegen/sdk/typescript/statements/try_catch_statement.py @@ -4,20 +4,20 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.try_catch_statement import TryCatchStatement -from graph_sitter.extensions.autocommit import commiter, reader -from graph_sitter.typescript.statements.block_statement import TSBlockStatement -from graph_sitter.typescript.statements.catch_statement import TSCatchStatement -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.try_catch_statement import TryCatchStatement +from codegen.sdk.extensions.autocommit import commiter, reader +from codegen.sdk.typescript.statements.block_statement import TSBlockStatement +from codegen.sdk.typescript.statements.catch_statement import TSCatchStatement +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.core.interfaces.importable import Importable - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.core.interfaces.importable import Importable + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock @ts_apidoc diff --git a/src/graph_sitter/typescript/statements/while_statement.py b/src/codegen/sdk/typescript/statements/while_statement.py similarity index 71% rename from src/graph_sitter/typescript/statements/while_statement.py rename to src/codegen/sdk/typescript/statements/while_statement.py index cfd86e9de..8374fab8c 100644 --- a/src/graph_sitter/typescript/statements/while_statement.py +++ b/src/codegen/sdk/typescript/statements/while_statement.py @@ -4,14 +4,14 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.statements.while_statement import WhileStatement -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.statements.while_statement import WhileStatement +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph - from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock + from codegen.sdk.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock @ts_apidoc diff --git a/src/graph_sitter/typescript/symbol.py b/src/codegen/sdk/typescript/symbol.py similarity index 94% rename from src/graph_sitter/typescript/symbol.py rename to src/codegen/sdk/typescript/symbol.py index b163ea7ec..2a8dace5d 100644 --- a/src/graph_sitter/typescript/symbol.py +++ b/src/codegen/sdk/typescript/symbol.py @@ -4,28 +4,28 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.autocommit import reader, writer -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.detached_symbols.parameter import Parameter -from graph_sitter.core.expressions import Value -from graph_sitter.core.expressions.chained_attribute import ChainedAttribute -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.exportable import Exportable -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol import Symbol -from graph_sitter.core.type_alias import TypeAlias -from graph_sitter.enums import ImportType, NodeType -from graph_sitter.typescript.import_resolution import TSImport -from graph_sitter.typescript.statements.comment import TSComment, TSCommentType -from graph_sitter.typescript.symbol_groups.comment_group import TSCommentGroup -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.autocommit import reader, writer +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.detached_symbols.parameter import Parameter +from codegen.sdk.core.expressions import Value +from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.exportable import Exportable +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.core.type_alias import TypeAlias +from codegen.sdk.enums import ImportType, NodeType +from codegen.sdk.typescript.import_resolution import TSImport +from codegen.sdk.typescript.statements.comment import TSComment, TSCommentType +from codegen.sdk.typescript.symbol_groups.comment_group import TSCommentGroup +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.core.file import SourceFile + from codegen.sdk.core.file import SourceFile @ts_apidoc diff --git a/src/graph_sitter/typescript/symbol_groups/comment_group.py b/src/codegen/sdk/typescript/symbol_groups/comment_group.py similarity index 94% rename from src/graph_sitter/typescript/symbol_groups/comment_group.py rename to src/codegen/sdk/typescript/symbol_groups/comment_group.py index 730ebe03b..09a8d0d0d 100644 --- a/src/graph_sitter/typescript/symbol_groups/comment_group.py +++ b/src/codegen/sdk/typescript/symbol_groups/comment_group.py @@ -2,12 +2,12 @@ from typing import TYPE_CHECKING -from graph_sitter.core.symbol_groups.comment_group import CommentGroup -from graph_sitter.typescript.statements.comment import TSComment, TSCommentType -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.symbol_groups.comment_group import CommentGroup +from codegen.sdk.typescript.statements.comment import TSComment, TSCommentType +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.symbol import TSSymbol + from codegen.sdk.typescript.symbol import TSSymbol @ts_apidoc diff --git a/src/graph_sitter/typescript/symbol_groups/dict.py b/src/codegen/sdk/typescript/symbol_groups/dict.py similarity index 88% rename from src/graph_sitter/typescript/symbol_groups/dict.py rename to src/codegen/sdk/typescript/symbol_groups/dict.py index f160c95a3..9cbc57368 100644 --- a/src/graph_sitter/typescript/symbol_groups/dict.py +++ b/src/codegen/sdk/typescript/symbol_groups/dict.py @@ -3,18 +3,18 @@ from tree_sitter import Node as TSNode -from graph_sitter.core.autocommit import writer -from graph_sitter.core.expressions import Expression -from graph_sitter.core.expressions.string import String -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.has_attribute import HasAttribute -from graph_sitter.core.node_id_factory import NodeId -from graph_sitter.core.symbol_groups.dict import Dict, Pair -from graph_sitter.extensions.autocommit import reader -from graph_sitter.writer_decorators import apidoc, noapidoc, ts_apidoc +from codegen.sdk.core.autocommit import writer +from codegen.sdk.core.expressions import Expression +from codegen.sdk.core.expressions.string import String +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.has_attribute import HasAttribute +from codegen.sdk.core.node_id_factory import NodeId +from codegen.sdk.core.symbol_groups.dict import Dict, Pair +from codegen.sdk.extensions.autocommit import reader +from codegen.sdk.writer_decorators import apidoc, noapidoc, ts_apidoc if TYPE_CHECKING: - from graph_sitter.codebase.codebase_graph import CodebaseGraph + from codegen.sdk.codebase.codebase_graph import CodebaseGraph Parent = TypeVar("Parent", bound="Editable") TExpression = TypeVar("TExpression", bound=Expression) @@ -42,7 +42,7 @@ def __init__(self, ts_node: TSNode, file_node_id: NodeId, G: "CodebaseGraph", pa self.shorthand = ts_node.type == "shorthand_property_identifier" def _get_key_value(self) -> tuple[Expression[Self] | None, Expression[Self] | None]: - from graph_sitter.typescript.function import TSFunction + from codegen.sdk.typescript.function import TSFunction key, value = None, None diff --git a/src/graph_sitter/typescript/ts_config.py b/src/codegen/sdk/typescript/ts_config.py similarity index 98% rename from src/graph_sitter/typescript/ts_config.py rename to src/codegen/sdk/typescript/ts_config.py index bc0262f07..7cb841ef6 100644 --- a/src/graph_sitter/typescript/ts_config.py +++ b/src/codegen/sdk/typescript/ts_config.py @@ -6,13 +6,13 @@ import pyjson5 -from graph_sitter.core.directory import Directory -from graph_sitter.core.file import File -from graph_sitter.writer_decorators import ts_apidoc +from codegen.sdk.core.directory import Directory +from codegen.sdk.core.file import File +from codegen.sdk.writer_decorators import ts_apidoc if TYPE_CHECKING: - from graph_sitter.typescript.config_parser import TSConfigParser - from graph_sitter.typescript.file import TSFile + from codegen.sdk.typescript.config_parser import TSConfigParser + from codegen.sdk.typescript.file import TSFile logger = logging.getLogger(__name__) diff --git a/src/graph_sitter/typescript/type_alias.py b/src/codegen/sdk/typescript/type_alias.py similarity index 80% rename from src/graph_sitter/typescript/type_alias.py rename to src/codegen/sdk/typescript/type_alias.py index 63984fe10..2c85b8c51 100644 --- a/src/graph_sitter/typescript/type_alias.py +++ b/src/codegen/sdk/typescript/type_alias.py @@ -1,13 +1,13 @@ -from graph_sitter.core.autocommit import commiter, reader -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.core.interfaces.has_name import HasName -from graph_sitter.core.type_alias import TypeAlias -from graph_sitter.enums import SymbolType -from graph_sitter.typescript.detached_symbols.code_block import TSCodeBlock -from graph_sitter.typescript.interfaces.has_block import TSHasBlock -from graph_sitter.typescript.statements.attribute import TSAttribute -from graph_sitter.typescript.symbol import TSSymbol -from graph_sitter.writer_decorators import noapidoc, ts_apidoc +from codegen.sdk.core.autocommit import commiter, reader +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.core.interfaces.has_name import HasName +from codegen.sdk.core.type_alias import TypeAlias +from codegen.sdk.enums import SymbolType +from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock +from codegen.sdk.typescript.interfaces.has_block import TSHasBlock +from codegen.sdk.typescript.statements.attribute import TSAttribute +from codegen.sdk.typescript.symbol import TSSymbol +from codegen.sdk.writer_decorators import noapidoc, ts_apidoc @ts_apidoc diff --git a/src/graph_sitter/utils.py b/src/codegen/sdk/utils.py similarity index 95% rename from src/graph_sitter/utils.py rename to src/codegen/sdk/utils.py index b2c2401da..421246de9 100644 --- a/src/graph_sitter/utils.py +++ b/src/codegen/sdk/utils.py @@ -14,12 +14,12 @@ import xmltodict from tree_sitter import Node as TSNode -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.extensions.utils import find_all_descendants, find_first_descendant, get_all_identifiers -from graph_sitter.typescript.enums import TSFunctionTypeNames +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.extensions.utils import find_all_descendants, find_first_descendant, get_all_identifiers +from codegen.sdk.typescript.enums import TSFunctionTypeNames if TYPE_CHECKING: - from graph_sitter.core.interfaces.editable import Editable + from codegen.sdk.core.interfaces.editable import Editable """ Utility functions for traversing the tree sitter structure. Do not include language specific traversals, or string manipulations here. @@ -240,8 +240,8 @@ def calculate_base_path(full_path, relative_path): def get_language_file_extensions(language: ProgrammingLanguage): """Returns the file extensions for the given language.""" - from graph_sitter.python import PyFile - from graph_sitter.typescript.file import TSFile + from codegen.sdk.python import PyFile + from codegen.sdk.typescript.file import TSFile if language == ProgrammingLanguage.PYTHON: return set(PyFile.get_extensions()) @@ -250,8 +250,8 @@ def get_language_file_extensions(language: ProgrammingLanguage): def determine_project_language(folder_path: str): - from graph_sitter.python import PyFile - from graph_sitter.typescript.file import TSFile + from codegen.sdk.python import PyFile + from codegen.sdk.typescript.file import TSFile EXTENSIONS = { ProgrammingLanguage.PYTHON: PyFile.get_extensions(), diff --git a/src/graph_sitter/writer_decorators.py b/src/codegen/sdk/writer_decorators.py similarity index 97% rename from src/graph_sitter/writer_decorators.py rename to src/codegen/sdk/writer_decorators.py index 86aac3977..0d39945cc 100644 --- a/src/graph_sitter/writer_decorators.py +++ b/src/codegen/sdk/writer_decorators.py @@ -20,7 +20,7 @@ def signature(self) -> str: def canonical(codemod): """Decorator for canonical Codemods that will be used for AI-agent prompts.""" - from graph_sitter.enums import ProgrammingLanguage + from codegen.sdk.enums import ProgrammingLanguage codemod._canonical = True if not hasattr(codemod, "language") or codemod.language not in (ProgrammingLanguage.PYTHON, ProgrammingLanguage.TYPESCRIPT): diff --git a/src/codemods/canonical/add_function_parameter_type_annotations/add_function_parameter_type_annotations.py b/src/codemods/canonical/add_function_parameter_type_annotations/add_function_parameter_type_annotations.py index d80d61acf..6b11fb185 100644 --- a/src/codemods/canonical/add_function_parameter_type_annotations/add_function_parameter_type_annotations.py +++ b/src/codemods/canonical/add_function_parameter_type_annotations/add_function_parameter_type_annotations.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/add_internal_to_non_exported_components/add_internal_to_non_exported_components.py b/src/codemods/canonical/add_internal_to_non_exported_components/add_internal_to_non_exported_components.py index bcc3d0c7e..c220b6cbe 100644 --- a/src/codemods/canonical/add_internal_to_non_exported_components/add_internal_to_non_exported_components.py +++ b/src/codemods/canonical/add_internal_to_non_exported_components/add_internal_to_non_exported_components.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/bang_bang_to_boolean/bang_bang_to_boolean.py b/src/codemods/canonical/bang_bang_to_boolean/bang_bang_to_boolean.py index 5ab4922ea..4562e3815 100644 --- a/src/codemods/canonical/bang_bang_to_boolean/bang_bang_to_boolean.py +++ b/src/codemods/canonical/bang_bang_to_boolean/bang_bang_to_boolean.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/built_in_type_annotation/built_in_type_annotation.py b/src/codemods/canonical/built_in_type_annotation/built_in_type_annotation.py index dff396281..498ae4a47 100644 --- a/src/codemods/canonical/built_in_type_annotation/built_in_type_annotation.py +++ b/src/codemods/canonical/built_in_type_annotation/built_in_type_annotation.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/change_component_tag_names/change_component_tag_names.py b/src/codemods/canonical/change_component_tag_names/change_component_tag_names.py index eb52d6d06..98f400435 100644 --- a/src/codemods/canonical/change_component_tag_names/change_component_tag_names.py +++ b/src/codemods/canonical/change_component_tag_names/change_component_tag_names.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/classnames_to_backtick.py b/src/codemods/canonical/classnames_to_backtick.py index 80a5de915..13fb1fa3c 100644 --- a/src/codemods/canonical/classnames_to_backtick.py +++ b/src/codemods/canonical/classnames_to_backtick.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/convert_array_type_to_square_bracket/convert_array_type_to_square_bracket.py b/src/codemods/canonical/convert_array_type_to_square_bracket/convert_array_type_to_square_bracket.py index d9a99f840..95e417b9f 100644 --- a/src/codemods/canonical/convert_array_type_to_square_bracket/convert_array_type_to_square_bracket.py +++ b/src/codemods/canonical/convert_array_type_to_square_bracket/convert_array_type_to_square_bracket.py @@ -1,10 +1,10 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.expressions.generic_type import GenericType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.expressions.generic_type import GenericType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/convert_attribute_to_decorator/convert_attribute_to_decorator.py b/src/codemods/canonical/convert_attribute_to_decorator/convert_attribute_to_decorator.py index d256bb1d4..824a87a12 100644 --- a/src/codemods/canonical/convert_attribute_to_decorator/convert_attribute_to_decorator.py +++ b/src/codemods/canonical/convert_attribute_to_decorator/convert_attribute_to_decorator.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/convert_comments_to_JSDoc_style/convert_comments_to_JSDoc_style.py b/src/codemods/canonical/convert_comments_to_JSDoc_style/convert_comments_to_JSDoc_style.py index 990f1192c..186b5dcbb 100644 --- a/src/codemods/canonical/convert_comments_to_JSDoc_style/convert_comments_to_JSDoc_style.py +++ b/src/codemods/canonical/convert_comments_to_JSDoc_style/convert_comments_to_JSDoc_style.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/convert_docstring_to_google_style/convert_docstring_to_google_style.py b/src/codemods/canonical/convert_docstring_to_google_style/convert_docstring_to_google_style.py index 82649a50c..89f072ff3 100644 --- a/src/codemods/canonical/convert_docstring_to_google_style/convert_docstring_to_google_style.py +++ b/src/codemods/canonical/convert_docstring_to_google_style/convert_docstring_to_google_style.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/delete_unused_functions/delete_unused_functions.py b/src/codemods/canonical/delete_unused_functions/delete_unused_functions.py index a26eb44a8..b04d809b4 100644 --- a/src/codemods/canonical/delete_unused_functions/delete_unused_functions.py +++ b/src/codemods/canonical/delete_unused_functions/delete_unused_functions.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/emojify_py_files_codemod/emojify_py_files_codemod.py b/src/codemods/canonical/emojify_py_files_codemod/emojify_py_files_codemod.py index 6b04e3f87..b8b361590 100644 --- a/src/codemods/canonical/emojify_py_files_codemod/emojify_py_files_codemod.py +++ b/src/codemods/canonical/emojify_py_files_codemod/emojify_py_files_codemod.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/enum_mover/enum_mover.py b/src/codemods/canonical/enum_mover/enum_mover.py index a7594e69b..cef525a65 100644 --- a/src/codemods/canonical/enum_mover/enum_mover.py +++ b/src/codemods/canonical/enum_mover/enum_mover.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/insert_arguments_to_decorator/insert_arguments_to_decorator.py b/src/codemods/canonical/insert_arguments_to_decorator/insert_arguments_to_decorator.py index 9c25cc6fb..dcfd10264 100644 --- a/src/codemods/canonical/insert_arguments_to_decorator/insert_arguments_to_decorator.py +++ b/src/codemods/canonical/insert_arguments_to_decorator/insert_arguments_to_decorator.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/invite_factory_create_params/invite_factory_create_params.py b/src/codemods/canonical/invite_factory_create_params/invite_factory_create_params.py index 44c031d5d..758612b34 100644 --- a/src/codemods/canonical/invite_factory_create_params/invite_factory_create_params.py +++ b/src/codemods/canonical/invite_factory_create_params/invite_factory_create_params.py @@ -1,10 +1,10 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/js_to_esm_codemod/js_to_esm_codemod.py b/src/codemods/canonical/js_to_esm_codemod/js_to_esm_codemod.py index 0bc15c3c7..e950371a5 100644 --- a/src/codemods/canonical/js_to_esm_codemod/js_to_esm_codemod.py +++ b/src/codemods/canonical/js_to_esm_codemod/js_to_esm_codemod.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/mark_as_internal_codemod/mark_as_internal_codemod.py b/src/codemods/canonical/mark_as_internal_codemod/mark_as_internal_codemod.py index de28eac59..0e062890e 100644 --- a/src/codemods/canonical/mark_as_internal_codemod/mark_as_internal_codemod.py +++ b/src/codemods/canonical/mark_as_internal_codemod/mark_as_internal_codemod.py @@ -1,11 +1,11 @@ from pathlib import Path -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/mark_internal_to_module/mark_internal_to_module.py b/src/codemods/canonical/mark_internal_to_module/mark_internal_to_module.py index 9f9b22c5c..9f4d9d67c 100644 --- a/src/codemods/canonical/mark_internal_to_module/mark_internal_to_module.py +++ b/src/codemods/canonical/mark_internal_to_module/mark_internal_to_module.py @@ -1,10 +1,10 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.dataclasses.usage import UsageKind -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.dataclasses.usage import UsageKind +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/mark_is_boolean/mark_is_boolean.py b/src/codemods/canonical/mark_is_boolean/mark_is_boolean.py index a767abdb8..fb5c8f92c 100644 --- a/src/codemods/canonical/mark_is_boolean/mark_is_boolean.py +++ b/src/codemods/canonical/mark_is_boolean/mark_is_boolean.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/migrate_class_attributes/migrate_class_attributes.py b/src/codemods/canonical/migrate_class_attributes/migrate_class_attributes.py index 2b18e753b..58aeca942 100644 --- a/src/codemods/canonical/migrate_class_attributes/migrate_class_attributes.py +++ b/src/codemods/canonical/migrate_class_attributes/migrate_class_attributes.py @@ -1,12 +1,12 @@ import logging import textwrap -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import PyCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import PyCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical logger = logging.getLogger(__name__) diff --git a/src/codemods/canonical/move_enums_codemod/move_enums_codemod.py b/src/codemods/canonical/move_enums_codemod/move_enums_codemod.py index da8e81d98..17425dd90 100644 --- a/src/codemods/canonical/move_enums_codemod/move_enums_codemod.py +++ b/src/codemods/canonical/move_enums_codemod/move_enums_codemod.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/move_functions_to_new_file/move_functions_to_new_file.py b/src/codemods/canonical/move_functions_to_new_file/move_functions_to_new_file.py index 52e70060d..039376546 100644 --- a/src/codemods/canonical/move_functions_to_new_file/move_functions_to_new_file.py +++ b/src/codemods/canonical/move_functions_to_new_file/move_functions_to_new_file.py @@ -1,10 +1,10 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.file import SourceFile -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.file import SourceFile +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/openapi_add_response_none/openapi_add_response_none.py b/src/codemods/canonical/openapi_add_response_none/openapi_add_response_none.py index 23e820877..e8adca0c6 100644 --- a/src/codemods/canonical/openapi_add_response_none/openapi_add_response_none.py +++ b/src/codemods/canonical/openapi_add_response_none/openapi_add_response_none.py @@ -1,11 +1,11 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.detached_symbols.decorator import Decorator -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.detached_symbols.decorator import Decorator +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/openapi_no_reference_request/openapi_no_reference_request.py b/src/codemods/canonical/openapi_no_reference_request/openapi_no_reference_request.py index 65493d5db..c4dd0e2be 100644 --- a/src/codemods/canonical/openapi_no_reference_request/openapi_no_reference_request.py +++ b/src/codemods/canonical/openapi_no_reference_request/openapi_no_reference_request.py @@ -1,11 +1,11 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.detached_symbols.decorator import Decorator -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.detached_symbols.decorator import Decorator +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/pascal_case_symbols/pascal_case_symbols.py b/src/codemods/canonical/pascal_case_symbols/pascal_case_symbols.py index fd12642f5..71948f050 100644 --- a/src/codemods/canonical/pascal_case_symbols/pascal_case_symbols.py +++ b/src/codemods/canonical/pascal_case_symbols/pascal_case_symbols.py @@ -1,12 +1,12 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.class_definition import Class -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.interface import Interface -from graph_sitter.core.type_alias import TypeAlias -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.interface import Interface +from codegen.sdk.core.type_alias import TypeAlias +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/pivot_return_types/pivot_return_types.py b/src/codemods/canonical/pivot_return_types/pivot_return_types.py index 95cec62fa..500dc3459 100644 --- a/src/codemods/canonical/pivot_return_types/pivot_return_types.py +++ b/src/codemods/canonical/pivot_return_types/pivot_return_types.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/refactor_react_components_into_separate_files/refactor_react_components_into_separate_files.py b/src/codemods/canonical/refactor_react_components_into_separate_files/refactor_react_components_into_separate_files.py index a7552871c..e95a1e6d9 100644 --- a/src/codemods/canonical/refactor_react_components_into_separate_files/refactor_react_components_into_separate_files.py +++ b/src/codemods/canonical/refactor_react_components_into_separate_files/refactor_react_components_into_separate_files.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/remove_indirect_imports/remove_indirect_imports.py b/src/codemods/canonical/remove_indirect_imports/remove_indirect_imports.py index 67dfa5083..1bb54b60b 100644 --- a/src/codemods/canonical/remove_indirect_imports/remove_indirect_imports.py +++ b/src/codemods/canonical/remove_indirect_imports/remove_indirect_imports.py @@ -1,12 +1,12 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/rename_function_parameters/rename_function_parameters.py b/src/codemods/canonical/rename_function_parameters/rename_function_parameters.py index d83b57cd8..c543cc439 100644 --- a/src/codemods/canonical/rename_function_parameters/rename_function_parameters.py +++ b/src/codemods/canonical/rename_function_parameters/rename_function_parameters.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/rename_local_variables/rename_local_variables.py b/src/codemods/canonical/rename_local_variables/rename_local_variables.py index 4eee8e192..6c24aeba7 100644 --- a/src/codemods/canonical/rename_local_variables/rename_local_variables.py +++ b/src/codemods/canonical/rename_local_variables/rename_local_variables.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/replace_prop_values/replace_prop_values.py b/src/codemods/canonical/replace_prop_values/replace_prop_values.py index 1ab42b3c9..0e7b892a1 100644 --- a/src/codemods/canonical/replace_prop_values/replace_prop_values.py +++ b/src/codemods/canonical/replace_prop_values/replace_prop_values.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/return_none_type_annotation/return_none_type_annotation.py b/src/codemods/canonical/return_none_type_annotation/return_none_type_annotation.py index e63c1c8ec..1b339d3c9 100644 --- a/src/codemods/canonical/return_none_type_annotation/return_none_type_annotation.py +++ b/src/codemods/canonical/return_none_type_annotation/return_none_type_annotation.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/split_decorators/split_decorators.py b/src/codemods/canonical/split_decorators/split_decorators.py index b1d7d62cd..b564dd3a0 100644 --- a/src/codemods/canonical/split_decorators/split_decorators.py +++ b/src/codemods/canonical/split_decorators/split_decorators.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/split_file/split_file.py b/src/codemods/canonical/split_file/split_file.py index 501dbec8d..8a059903e 100644 --- a/src/codemods/canonical/split_file/split_file.py +++ b/src/codemods/canonical/split_file/split_file.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/split_file_and_rename_symbols/split_file_and_rename_symbols.py b/src/codemods/canonical/split_file_and_rename_symbols/split_file_and_rename_symbols.py index e7092e3d6..86beb7ffb 100644 --- a/src/codemods/canonical/split_file_and_rename_symbols/split_file_and_rename_symbols.py +++ b/src/codemods/canonical/split_file_and_rename_symbols/split_file_and_rename_symbols.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import CodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import CodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/split_large_files/split_large_files.py b/src/codemods/canonical/split_large_files/split_large_files.py index c1156d9cc..90be05026 100644 --- a/src/codemods/canonical/split_large_files/split_large_files.py +++ b/src/codemods/canonical/split_large_files/split_large_files.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/swap_call_site_imports/swap_call_site_imports.py b/src/codemods/canonical/swap_call_site_imports/swap_call_site_imports.py index c2b3df9d9..2f2e67794 100644 --- a/src/codemods/canonical/swap_call_site_imports/swap_call_site_imports.py +++ b/src/codemods/canonical/swap_call_site_imports/swap_call_site_imports.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/swap_class_attribute_usages/swap_class_attribute_usages.py b/src/codemods/canonical/swap_class_attribute_usages/swap_class_attribute_usages.py index 379348820..555b423a9 100644 --- a/src/codemods/canonical/swap_class_attribute_usages/swap_class_attribute_usages.py +++ b/src/codemods/canonical/swap_class_attribute_usages/swap_class_attribute_usages.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/update_optional_type_annotations/update_optional_type_annotations.py b/src/codemods/canonical/update_optional_type_annotations/update_optional_type_annotations.py index 59e5ad56d..d54acc1e5 100644 --- a/src/codemods/canonical/update_optional_type_annotations/update_optional_type_annotations.py +++ b/src/codemods/canonical/update_optional_type_annotations/update_optional_type_annotations.py @@ -1,12 +1,12 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.expressions import Type -from graph_sitter.core.expressions.generic_type import GenericType -from graph_sitter.core.expressions.union_type import UnionType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.expressions import Type +from codegen.sdk.core.expressions.generic_type import GenericType +from codegen.sdk.core.expressions.union_type import UnionType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/update_union_types/update_union_types.py b/src/codemods/canonical/update_union_types/update_union_types.py index dc3cb352a..413ce3c7a 100644 --- a/src/codemods/canonical/update_union_types/update_union_types.py +++ b/src/codemods/canonical/update_union_types/update_union_types.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/use_named_kwargs/use_named_kwargs.py b/src/codemods/canonical/use_named_kwargs/use_named_kwargs.py index 8a1cc26ce..63a95cd3f 100644 --- a/src/codemods/canonical/use_named_kwargs/use_named_kwargs.py +++ b/src/codemods/canonical/use_named_kwargs/use_named_kwargs.py @@ -1,11 +1,11 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.external_module import ExternalModule -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python.class_definition import PyClass -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.external_module import ExternalModule +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python.class_definition import PyClass +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/wrap_with_component/wrap_with_component.py b/src/codemods/canonical/wrap_with_component/wrap_with_component.py index 19a83d7d3..c5a8f0d45 100644 --- a/src/codemods/canonical/wrap_with_component/wrap_with_component.py +++ b/src/codemods/canonical/wrap_with_component/wrap_with_component.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/codemods/canonical/wrap_with_statement/wrap_with_statement.py b/src/codemods/canonical/wrap_with_statement/wrap_with_statement.py index 65a36a24c..b89f54d77 100644 --- a/src/codemods/canonical/wrap_with_statement/wrap_with_statement.py +++ b/src/codemods/canonical/wrap_with_statement/wrap_with_statement.py @@ -1,9 +1,9 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.skills.core.skill import Skill -from graph_sitter.skills.core.utils import skill, skill_impl -from graph_sitter.writer_decorators import canonical +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.skills.core.skill import Skill +from codegen.sdk.skills.core.utils import skill, skill_impl +from codegen.sdk.writer_decorators import canonical @skill( diff --git a/src/graph_sitter/code_generation/graph_sitter_codebase.py b/src/graph_sitter/code_generation/graph_sitter_codebase.py deleted file mode 100644 index 0d455c68e..000000000 --- a/src/graph_sitter/code_generation/graph_sitter_codebase.py +++ /dev/null @@ -1,16 +0,0 @@ -import os.path - -from graph_sitter.code_generation.current_code_codebase import get_codegen_codebase_base_path, get_current_code_codebase -from graph_sitter.codebase.config import CodebaseConfig -from graph_sitter.core.codebase import Codebase - - -def get_graph_sitter_subdirectories() -> list[str]: - base = get_codegen_codebase_base_path() - return [os.path.join(base, "graph_sitter"), os.path.join(base, "codemods")] - - -def get_graph_sitter_codebase() -> Codebase: - """Grabs a Codebase w/ GraphSitter content. Responsible for figuring out where it is, e.g. in Modal or local""" - codebase = get_current_code_codebase(CodebaseConfig(), subdirectories=get_graph_sitter_subdirectories()) - return codebase diff --git a/src/graph_sitter/codebase/resolution_stack.py b/src/graph_sitter/codebase/resolution_stack.py deleted file mode 100644 index f908bee32..000000000 --- a/src/graph_sitter/codebase/resolution_stack.py +++ /dev/null @@ -1,3 +0,0 @@ -from graph_sitter.extensions.resolution import ResolutionStack - -__all__ = ["ResolutionStack"] diff --git a/src/graph_sitter/core/autocommit/ruff.toml b/src/graph_sitter/core/autocommit/ruff.toml deleted file mode 100644 index 9e62776f8..000000000 --- a/src/graph_sitter/core/autocommit/ruff.toml +++ /dev/null @@ -1,2 +0,0 @@ -extend = "../../../../pyproject.toml" -lint.extend-select = ["G"] \ No newline at end of file diff --git a/src/graph_sitter/core/expressions/__init__.py b/src/graph_sitter/core/expressions/__init__.py deleted file mode 100644 index 268e23e41..000000000 --- a/src/graph_sitter/core/expressions/__init__.py +++ /dev/null @@ -1,13 +0,0 @@ -from typing import TYPE_CHECKING - -from graph_sitter.core.expressions.expression import Expression -from graph_sitter.core.expressions.name import Name -from graph_sitter.core.expressions.string import String -from graph_sitter.core.expressions.type import Type -from graph_sitter.core.expressions.value import Value -from graph_sitter.core.symbol_groups.dict import Dict -from graph_sitter.core.symbol_groups.list import List - -if TYPE_CHECKING: - from graph_sitter.core.detached_symbols.function_call import FunctionCall -__all__ = ["Dict", "Expression", "FunctionCall", "List", "Name", "String", "Type", "Value"] diff --git a/src/graph_sitter/core/plugins/__init__.py b/src/graph_sitter/core/plugins/__init__.py deleted file mode 100644 index 358bc9a22..000000000 --- a/src/graph_sitter/core/plugins/__init__.py +++ /dev/null @@ -1,9 +0,0 @@ -from graph_sitter.core.plugins.axios import AxiosApiFinder -from graph_sitter.core.plugins.flask import FlaskApiFinder -from graph_sitter.core.plugins.modal import ModalApiFinder - -PLUGINS = [ - FlaskApiFinder(), - AxiosApiFinder(), - ModalApiFinder(), -] diff --git a/src/graph_sitter/python/__init__.py b/src/graph_sitter/python/__init__.py deleted file mode 100644 index db904aeb4..000000000 --- a/src/graph_sitter/python/__init__.py +++ /dev/null @@ -1,15 +0,0 @@ -from graph_sitter.python.assignment import PyAssignment -from graph_sitter.python.class_definition import PyClass -from graph_sitter.python.file import PyFile -from graph_sitter.python.function import PyFunction -from graph_sitter.python.import_resolution import PyImport -from graph_sitter.python.symbol import PySymbol - -__all__ = [ - "PyAssignment", - "PyClass", - "PyFile", - "PyFunction", - "PyImport", - "PySymbol", -] diff --git a/src/graph_sitter_build/build.py b/src/graph_sitter_build/build.py index 1270e41ea..f88580cb7 100644 --- a/src/graph_sitter_build/build.py +++ b/src/graph_sitter_build/build.py @@ -19,7 +19,7 @@ class SpecialBuildHook(BuildHookInterface): PLUGIN_NAME = "graph_sitter_build" def initialize(self, version: str, build_data: dict[str, Any]) -> None: - file = Path(self.root) / "src" / "graph_sitter" / "__init__.py" + file = Path(self.root) / "src" / "codegen" / "sdk" / "__init__.py" update_init_file(file) build_data["artifacts"].append(f"/{file}") diff --git a/src/graph_visualization/visualization_manager.py b/src/graph_visualization/visualization_manager.py index 4d672306f..fc8996f6f 100644 --- a/src/graph_visualization/visualization_manager.py +++ b/src/graph_visualization/visualization_manager.py @@ -4,8 +4,8 @@ import plotly.graph_objects as go from networkx import Graph +from codegen.sdk.core.interfaces.editable import Editable from codegen_git.repo_operator.repo_operator import RepoOperator -from graph_sitter.core.interfaces.editable import Editable from graph_visualization.viz_utils import graph_to_json logger = logging.getLogger(__name__) diff --git a/src/graph_visualization/viz_utils.py b/src/graph_visualization/viz_utils.py index d25efef13..5c03d9fed 100644 --- a/src/graph_visualization/viz_utils.py +++ b/src/graph_visualization/viz_utils.py @@ -6,10 +6,10 @@ import networkx as nx from networkx import DiGraph, Graph +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.core.interfaces.importable import Importable +from codegen.sdk.output.utils import DeterministicJSONEncoder from codegen_git.repo_operator.repo_operator import RepoOperator -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.core.interfaces.importable import Importable -from graph_sitter.output.utils import DeterministicJSONEncoder from graph_visualization.enums import GraphJson, GraphType if TYPE_CHECKING: diff --git a/src/gscli/codemod/commands.py b/src/gscli/codemod/commands.py index f4b7f7944..7ee8748b5 100644 --- a/src/gscli/codemod/commands.py +++ b/src/gscli/codemod/commands.py @@ -9,10 +9,10 @@ from rich.console import Console from rich.table import Table -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.testing.models import BASE_TMP_DIR, REPO_ID_TO_URL, VERIFIED_CODEMOD_DATA_DIR, ClonedRepoTestCase, Size -from graph_sitter.testing.test_discovery import filter_repos, find_codemod_test_cases, find_codemods, find_repos, find_verified_codemod_repos -from graph_sitter.testing.verified_codemod_utils import CodemodAPI, RepoCodemodMetadata +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.testing.models import BASE_TMP_DIR, REPO_ID_TO_URL, VERIFIED_CODEMOD_DATA_DIR, ClonedRepoTestCase, Size +from codegen.sdk.testing.test_discovery import filter_repos, find_codemod_test_cases, find_codemods, find_repos, find_verified_codemod_repos +from codegen.sdk.testing.verified_codemod_utils import CodemodAPI, RepoCodemodMetadata @click.group() diff --git a/src/gscli/generate/commands.py b/src/gscli/generate/commands.py index c9d42153a..c9f9739ee 100644 --- a/src/gscli/generate/commands.py +++ b/src/gscli/generate/commands.py @@ -8,15 +8,15 @@ import inflection from termcolor import colored -from graph_sitter.code_generation.doc_utils.canonicals import get_canonical_codemod_class_mdx, get_canonical_codemod_classes -from graph_sitter.code_generation.doc_utils.skills import format_all_skills -from graph_sitter.code_generation.doc_utils.utils import get_all_classes_to_document -from graph_sitter.code_generation.mdx_docs_generation import render_mdx_for_codebase_page, render_mdx_page_for_class -from graph_sitter.code_generation.prompts.api_docs import get_graph_sitter_codebase -from graph_sitter.core.codebase import PyCodebaseType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python import PyClass -from graph_sitter.skills.core.utils import get_all_skills, get_guide_skills_dict +from codegen.sdk.code_generation.doc_utils.canonicals import get_canonical_codemod_class_mdx, get_canonical_codemod_classes +from codegen.sdk.code_generation.doc_utils.skills import format_all_skills +from codegen.sdk.code_generation.doc_utils.utils import get_all_classes_to_document +from codegen.sdk.code_generation.mdx_docs_generation import render_mdx_for_codebase_page, render_mdx_page_for_class +from codegen.sdk.code_generation.prompts.api_docs import get_codegen_sdk_codebase +from codegen.sdk.core.codebase import PyCodebaseType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python import PyClass +from codegen.sdk.skills.core.utils import get_all_skills, get_guide_skills_dict from gscli.generate.runner_imports import _generate_runner_imports from gscli.generate.utils import LanguageType, generate_builtins_file @@ -80,7 +80,7 @@ def _generate_codebase_typestubs() -> None: # remove typings dir if it exists shutil.rmtree(frontend_typestubs_dir) # generate typestubs in codegen-frontend/assets/typestubs/graphsitter using pyright - os.system("uv run pyright -p . --createstub graph_sitter.core.codebase") + os.system("uv run pyright -p . --createstub codegen.sdk.core.codebase") os.system("uv run pyright -p . --createstub codegen.git") os.system("uv run pyright -p . --createstub networkx") # also generate for codemod context model and all its nested models @@ -110,8 +110,8 @@ def generate_docs(docs_dir: str) -> None: This will generate docs using the codebase locally, including any unstaged changes """ - codebase = get_graph_sitter_codebase() - generate_graph_sitter_docs(docs_dir, codebase) + codebase = get_codegen_sdk_codebase() + generate_codegen_sdk_docs(docs_dir, codebase) # generate_canonical_codemod_docs(docs_dir, codebase) generate_skills_docs(docs_dir) generate_guides(docs_dir) @@ -143,9 +143,9 @@ def get_snippet_pattern(target_name: str) -> str: return pattern -def generate_graph_sitter_docs(docs_dir: str, codebase: PyCodebaseType) -> None: - """Generate the docs for the graph_sitter API and update the mint.json""" - print(colored("Generating GraphSitter docs", "green")) +def generate_codegen_sdk_docs(docs_dir: str, codebase: PyCodebaseType) -> None: + """Generate the docs for the codegen_sdk API and update the mint.json""" + print(colored("Generating codegen_sdk docs", "green")) # Generate docs page for codebase api and write to the file system mdx_page = render_mdx_for_codebase_page(codebase) diff --git a/src/gscli/generate/runner_imports.py b/src/gscli/generate/runner_imports.py index 1da9f6814..df4863bcc 100644 --- a/src/gscli/generate/runner_imports.py +++ b/src/gscli/generate/runner_imports.py @@ -2,7 +2,7 @@ from termcolor import colored -from graph_sitter.code_generation.current_code_codebase import get_documented_objects +from codegen.sdk.code_generation.current_code_codebase import get_documented_objects EXTERNAL_IMPORTS = """ import os @@ -19,7 +19,7 @@ """ # TODO: these should also be made public (i.e. included in the docs site) GS_PRIVATE_IMPORTS = """ -from graph_sitter.codebase.control_flow import StopCodemodException +from codegen.sdk.codebase.control_flow import StopCodemodException """.strip() IMPORT_STRING_TEMPLATE = """ diff --git a/src/gscli/generate/utils.py b/src/gscli/generate/utils.py index ea2fcd5f3..8cca3f849 100644 --- a/src/gscli/generate/utils.py +++ b/src/gscli/generate/utils.py @@ -2,8 +2,8 @@ from enum import StrEnum from itertools import chain -from graph_sitter.code_generation.current_code_codebase import get_documented_objects -from graph_sitter.core import codebase +from codegen.sdk.code_generation.current_code_codebase import get_documented_objects +from codegen.sdk.core import codebase class LanguageType(StrEnum): @@ -37,7 +37,7 @@ def generate_builtins_file(path_to_builtins: str, language_type: LanguageType): from app.codemod.compilation.models.github_named_user_context import GithubNamedUserContext from app.codemod.compilation.models.pr_part_context import PRPartContext from app.codemod.compilation.models.pull_request_context import PullRequestContext -from graph_sitter.codebase.flagging.code_flag import MessageType as MessageType +from codegen.sdk.codebase.flagging.code_flag import MessageType as MessageType {"\n".join(inspect.getsource(codebase).splitlines()[-2:])} CodebaseType = {codebase_type} diff --git a/tests/README.md b/tests/README.md index efd6a96aa..464090415 100644 --- a/tests/README.md +++ b/tests/README.md @@ -1,43 +1 @@ -# Overview - -`codegen-backend/codegen_tests` are where we store unit tests. We store integration tests in `codegen-backend/codegen_integration`. - -TODO: update the test structure to: - -``` -/tests --> /unit --> /integration -``` - -## Test Categories - -There are three categories of tests (each in their own folder): - -- [codegen_tests/graph_sitter](graph_sitter/README.md) -> tests for GraphSitter and Codebase -- [codegen_tests/rojects](projects/README.md) -> tests for hardcoded Codemods (i.e. the ones stored in `codegen-backend/app/projects`) -- [codegen_tests/app](app/README.md) -> everything else 💀 - -## Helpers - -Lots of overlap between these two and at some point we should consolidate into one `/utils` folder. - -- [codegen_tests/utils](utils) -> utils for setting up/writing tests (ex: creating mock files) -- [codegen_tests/mocks](mocks) -> common mocks - -## Testing Guidelines - -### Category Guidelines - -Category specific testing guidelines can be found in the respective READMEs. - -- [graph_sitter README](graph_sitter/README.md) -- [projects README](projects/README.md) -- [app README](app/README.md) - -### General Guidelines - -The following are guidelines that apply to all tests: - -- Do not instantiate a DB in unit tests (ex: with `db = get_session_local()` or with `with db_context() as db`, etc) -- Use [tmpdir](https://docs.pytest.org/en/6.2.x/tmpdir.html#the-tmpdir-fixture) where possible (handles cleaning up test files) +# TODO diff --git a/tests/codemod/conftest.py b/tests/codemod/conftest.py index ea80871d8..f6bb791dc 100644 --- a/tests/codemod/conftest.py +++ b/tests/codemod/conftest.py @@ -9,14 +9,14 @@ from _pytest.python import Metafunc from pyinstrument import Profiler +from codegen.sdk.codebase.config import CodebaseConfig, GSFeatureFlags, ProjectConfig +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.testing.constants import DIFF_FILEPATH +from codegen.sdk.testing.models import BASE_PATH, BASE_TMP_DIR, VERIFIED_CODEMOD_DIFFS, CodemodMetadata, Repo, Size +from codegen.sdk.testing.test_discovery import find_codemod_test_cases, find_repos, find_verified_codemod_cases from codegen_git.repo_operator.local_repo_operator import LocalRepoOperator from codegen_git.repo_operator.repo_operator import RepoOperator -from graph_sitter.codebase.config import CodebaseConfig, GSFeatureFlags, ProjectConfig -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.testing.constants import DIFF_FILEPATH -from graph_sitter.testing.models import BASE_PATH, BASE_TMP_DIR, VERIFIED_CODEMOD_DIFFS, CodemodMetadata, Repo, Size -from graph_sitter.testing.test_discovery import find_codemod_test_cases, find_repos, find_verified_codemod_cases from tests.utils.recursion import set_recursion_limit logger = logging.getLogger(__name__) diff --git a/tests/codemod/test_codemods.py b/tests/codemod/test_codemods.py index 5dd86e105..a7dece3b6 100644 --- a/tests/codemod/test_codemods.py +++ b/tests/codemod/test_codemods.py @@ -4,9 +4,9 @@ import pytest from pytest_snapshot.plugin import Snapshot -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.testing.models import BASE_PATH +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.testing.models import BASE_PATH from tests.utils.codebase_comparison_utils import compare_codebase_diff DIFF_ROOT = BASE_PATH / ".diffs" diff --git a/tests/codemod/test_diffs.py b/tests/codemod/test_diffs.py index a9a5c128b..75a35a668 100644 --- a/tests/codemod/test_diffs.py +++ b/tests/codemod/test_diffs.py @@ -3,11 +3,11 @@ from pathlib import Path from typing import TypeVar -from graph_sitter.codebase.diff_lite import DiffLite -from graph_sitter.core.codebase import Codebase -from graph_sitter.output.utils import stylize_error -from graph_sitter.testing.models import BASE_PATH -from graph_sitter.tree_sitter_parser import print_errors +from codegen.sdk.codebase.diff_lite import DiffLite +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.output.utils import stylize_error +from codegen.sdk.testing.models import BASE_PATH +from codegen.sdk.tree_sitter_parser import print_errors from tests.utils.codebase_comparison_utils import gather_modified_files DIFF_ROOT = BASE_PATH / ".diffs" diff --git a/tests/codemod/test_parse.py b/tests/codemod/test_parse.py index 0338c3f13..9de43f6be 100644 --- a/tests/codemod/test_parse.py +++ b/tests/codemod/test_parse.py @@ -5,11 +5,11 @@ import psutil import pytest +from codegen.sdk.codebase.config import CodebaseConfig, DefaultFlags, ProjectConfig +from codegen.sdk.codebase.validation import PostInitValidationStatus, post_init_validation +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.testing.models import Repo from codegen_git.repo_operator.repo_operator import RepoOperator -from graph_sitter.codebase.config import CodebaseConfig, DefaultFlags, ProjectConfig -from graph_sitter.codebase.validation import PostInitValidationStatus, post_init_validation -from graph_sitter.core.codebase import Codebase -from graph_sitter.testing.models import Repo from tests.utils.recursion import set_recursion_limit BYTES_IN_GIGABYTE = 1024**3 diff --git a/tests/codemod/test_verified_codemods.py b/tests/codemod/test_verified_codemods.py index 3ec6f4dce..ff1530e86 100644 --- a/tests/codemod/test_verified_codemods.py +++ b/tests/codemod/test_verified_codemods.py @@ -5,9 +5,9 @@ import pytest from pytest_snapshot.plugin import Snapshot -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase -from graph_sitter.testing.models import BASE_PATH +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.testing.models import BASE_PATH from tests.utils.codebase_comparison_utils import compare_codebase_diff from tests.utils.recursion import set_recursion_limit diff --git a/tests/unit/api_doc_generation/test_api_doc_generation.py b/tests/unit/api_doc_generation/test_api_doc_generation.py index 61fa57fa2..6816eef87 100644 --- a/tests/unit/api_doc_generation/test_api_doc_generation.py +++ b/tests/unit/api_doc_generation/test_api_doc_generation.py @@ -1,22 +1,22 @@ import pytest -from graph_sitter.ai.helpers import count_tokens -from graph_sitter.code_generation.doc_utils.utils import get_decorator_for_language -from graph_sitter.code_generation.prompts.api_docs import get_graph_sitter_codebase, get_graph_sitter_docs -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.ai.helpers import count_tokens +from codegen.sdk.code_generation.doc_utils.utils import get_decorator_for_language +from codegen.sdk.code_generation.prompts.api_docs import get_codegen_sdk_codebase, get_codegen_sdk_docs +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import ProgrammingLanguage @pytest.fixture(scope="module") def codebase(): - return get_graph_sitter_codebase() + return get_codegen_sdk_codebase() @pytest.mark.xdist_group("codegen") def test_basic_docs(codebase) -> None: # =====[ Grab codebase ]===== language = ProgrammingLanguage.PYTHON - api_docs = get_graph_sitter_docs(language=language, codebase=codebase) + api_docs = get_codegen_sdk_docs(language=language, codebase=codebase) assert "class File" in api_docs assert "class PyFile" in api_docs assert "class Function" in api_docs @@ -39,7 +39,7 @@ def test_api_doc_generation_sanity(codebase, language: ProgrammingLanguage) -> N lang = "Py" if language == ProgrammingLanguage.PYTHON else "TS" other_lang = "TS" if language == ProgrammingLanguage.PYTHON else "Py" # =====[ Python ]===== - docs = get_graph_sitter_docs(language=language, codebase=codebase) + docs = get_codegen_sdk_docs(language=language, codebase=codebase) assert count_tokens(docs) < 50500 assert f"{lang}Function" in docs assert f"{lang}Class" in docs @@ -49,7 +49,7 @@ def test_api_doc_generation_sanity(codebase, language: ProgrammingLanguage) -> N @pytest.mark.xdist_group("codegen") @pytest.mark.parametrize("language", [ProgrammingLanguage.PYTHON, ProgrammingLanguage.TYPESCRIPT]) -def test_get_graph_sitter_codebase(codebase, language) -> None: +def test_get_codegen_sdk_codebase(codebase, language) -> None: """Make sure we can get the current codebase for GraphSitter, and that imports get resolved correctly""" cls = codebase.get_symbol("PyClass") assert cls is not None @@ -62,7 +62,7 @@ def test_get_graph_sitter_codebase(codebase, language) -> None: @pytest.mark.xdist_group("codegen") @pytest.mark.parametrize("language", [ProgrammingLanguage.PYTHON, ProgrammingLanguage.TYPESCRIPT]) def test_api_doc_generation(codebase, language) -> None: - api_docs = get_graph_sitter_docs(language=language, codebase=codebase) + api_docs = get_codegen_sdk_docs(language=language, codebase=codebase) decorator = get_decorator_for_language(language).value for cls in codebase.classes: diff --git a/tests/unit/api_doc_generation/test_codemod_writer_decorators.py b/tests/unit/api_doc_generation/test_codemod_writer_decorators.py index 60375ea75..39e2fc20b 100644 --- a/tests/unit/api_doc_generation/test_codemod_writer_decorators.py +++ b/tests/unit/api_doc_generation/test_codemod_writer_decorators.py @@ -4,12 +4,12 @@ import pytest -from graph_sitter.code_generation.current_code_codebase import import_all_graph_sitter_modules +from codegen.sdk.code_generation.current_code_codebase import import_all_codegen_sdk_modules @pytest.mark.skip("broken after repo split, fix or move over") def test_get_documented_objects(): - """Test that importing everything in graph-sitter + """Test that importing everything in codegen-sdk doesn't invoke any functionality (incase someone leaves actual functionality at the top level of a file). @@ -22,7 +22,7 @@ def test_get_documented_objects(): logger = logging.getLogger("test_logger") with redirect_stdout(stdout), redirect_stderr(stderr): - import_all_graph_sitter_modules() + import_all_codegen_sdk_modules() assert stdout.getvalue() == "", "Logs were written to stdout" assert stderr.getvalue() == "", "Logs were written to stderr" diff --git a/tests/unit/codebase/code_flag/test_code_flag.py b/tests/unit/codebase/code_flag/test_code_flag.py index b0dca801f..d7beba6ec 100644 --- a/tests/unit/codebase/code_flag/test_code_flag.py +++ b/tests/unit/codebase/code_flag/test_code_flag.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.codebase.flagging.enums import MessageType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.flagging.enums import MessageType def test_code_flag_properties(tmpdir): diff --git a/tests/unit/codebase/codebase_graph/test_codebase_graph.py b/tests/unit/codebase/codebase_graph/test_codebase_graph.py index 003793225..d61355239 100644 --- a/tests/unit/codebase/codebase_graph/test_codebase_graph.py +++ b/tests/unit/codebase/codebase_graph/test_codebase_graph.py @@ -2,9 +2,9 @@ import itertools -from graph_sitter.codebase.codebase_graph import CodebaseGraph -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import EdgeType +from codegen.sdk.codebase.codebase_graph import CodebaseGraph +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import EdgeType def test_codebase_with_wrapper(tmpdir) -> None: diff --git a/tests/unit/codebase/codebase_graph/test_validation.py b/tests/unit/codebase/codebase_graph/test_validation.py index 3da52f64e..da5316dcf 100644 --- a/tests/unit/codebase/codebase_graph/test_validation.py +++ b/tests/unit/codebase/codebase_graph/test_validation.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.codebase.validation import PostInitValidationStatus, post_init_validation -from graph_sitter.enums import NodeType -from graph_sitter.extensions.utils import uncache_all +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.validation import PostInitValidationStatus, post_init_validation +from codegen.sdk.enums import NodeType +from codegen.sdk.extensions.utils import uncache_all def test_post_init_validation_valid_graph(tmpdir) -> None: diff --git a/tests/unit/codebase/file/test_file.py b/tests/unit/codebase/file/test_file.py index c819ebc25..c3a6fdce2 100644 --- a/tests/unit/codebase/file/test_file.py +++ b/tests/unit/codebase/file/test_file.py @@ -2,8 +2,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.file import File, SourceFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.file import File, SourceFile def test_file(tmpdir) -> None: diff --git a/tests/unit/codebase/file/test_get_relative_path.py b/tests/unit/codebase/file/test_get_relative_path.py index e4fda718e..aa0cba529 100644 --- a/tests/unit/codebase/file/test_get_relative_path.py +++ b/tests/unit/codebase/file/test_get_relative_path.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_get_relative_path_same_directory(tmpdir) -> None: diff --git a/tests/unit/codebase/file_graph/test_symbol_parent_statement.py b/tests/unit/codebase/file_graph/test_symbol_parent_statement.py index cb81797f5..7f5fdb06d 100644 --- a/tests/unit/codebase/file_graph/test_symbol_parent_statement.py +++ b/tests/unit/codebase/file_graph/test_symbol_parent_statement.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.python import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.python import PyFile def test_parent_statement_function(tmpdir): diff --git a/tests/unit/codebase/transaction_manager/test_transaction_manager.py b/tests/unit/codebase/transaction_manager/test_transaction_manager.py index 2a9133fed..d30ae37d4 100644 --- a/tests/unit/codebase/transaction_manager/test_transaction_manager.py +++ b/tests/unit/codebase/transaction_manager/test_transaction_manager.py @@ -1,12 +1,12 @@ from os import PathLike from pathlib import Path -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.codebase.transaction_manager import ( +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.transaction_manager import ( TransactionError, TransactionManager, ) -from graph_sitter.codebase.transactions import EditTransaction, InsertTransaction, RemoveTransaction +from codegen.sdk.codebase.transactions import EditTransaction, InsertTransaction, RemoveTransaction class MockFile: diff --git a/tests/unit/codebase/transaction_manager/test_transaction_mangager_checks.py b/tests/unit/codebase/transaction_manager/test_transaction_mangager_checks.py index f6b2f9066..e9d5b8aae 100644 --- a/tests/unit/codebase/transaction_manager/test_transaction_mangager_checks.py +++ b/tests/unit/codebase/transaction_manager/test_transaction_mangager_checks.py @@ -1,9 +1,9 @@ import pytest -from graph_sitter.codebase.config import SessionOptions -from graph_sitter.codebase.control_flow import MaxPreviewTimeExceeded, MaxTransactionsExceeded -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.config import SessionOptions +from codegen.sdk.codebase.control_flow import MaxPreviewTimeExceeded, MaxTransactionsExceeded +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_check_max_preview_time_exceeded(tmpdir): diff --git a/tests/unit/output/snapshots/test_ast/test_ast_basic/ast.json b/tests/unit/output/snapshots/test_ast/test_ast_basic/ast.json index efe79df64..1d7784baf 100644 --- a/tests/unit/output/snapshots/test_ast/test_ast_basic/ast.json +++ b/tests/unit/output/snapshots/test_ast/test_ast_basic/ast.json @@ -1,5 +1,5 @@ { - "graph_sitter_type": "PyClass", + "codegen_sdk_type": "PyClass", "span": { "range": { "start_byte": 1, @@ -20,7 +20,7 @@ [ "get_name()", { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 7, @@ -43,7 +43,7 @@ [ "methods", { - "graph_sitter_type": "PyCodeBlock", + "codegen_sdk_type": "PyCodeBlock", "span": { "range": { "start_byte": 16, @@ -64,7 +64,7 @@ [ "statements", { - "graph_sitter_type": "MultiLineCollection", + "codegen_sdk_type": "MultiLineCollection", "span": { "range": { "start_byte": 16, @@ -85,7 +85,7 @@ [ null, { - "graph_sitter_type": "SymbolStatement", + "codegen_sdk_type": "SymbolStatement", "span": { "range": { "start_byte": 16, @@ -106,7 +106,7 @@ [ null, { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 20, @@ -129,7 +129,7 @@ [ null, { - "graph_sitter_type": "Collection", + "codegen_sdk_type": "Collection", "span": { "range": { "start_byte": 23, @@ -150,7 +150,7 @@ [ null, { - "graph_sitter_type": "PyParameter", + "codegen_sdk_type": "PyParameter", "span": { "range": { "start_byte": 24, @@ -176,7 +176,7 @@ [ null, { - "graph_sitter_type": "PyCodeBlock", + "codegen_sdk_type": "PyCodeBlock", "span": { "range": { "start_byte": 38, @@ -197,7 +197,7 @@ [ "statements", { - "graph_sitter_type": "MultiLineCollection", + "codegen_sdk_type": "MultiLineCollection", "span": { "range": { "start_byte": 38, @@ -218,7 +218,7 @@ [ null, { - "graph_sitter_type": "ReturnStatement", + "codegen_sdk_type": "ReturnStatement", "span": { "range": { "start_byte": 38, @@ -239,7 +239,7 @@ [ "value", { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 45, @@ -280,7 +280,7 @@ [ null, { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 20, @@ -303,7 +303,7 @@ [ null, { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 24, @@ -324,4 +324,4 @@ } ] ] -} \ No newline at end of file +} diff --git a/tests/unit/output/snapshots/test_ast/test_ast_nested/ast-nested.json b/tests/unit/output/snapshots/test_ast/test_ast_nested/ast-nested.json index 836137b1e..3079c70e1 100644 --- a/tests/unit/output/snapshots/test_ast/test_ast_nested/ast-nested.json +++ b/tests/unit/output/snapshots/test_ast/test_ast_nested/ast-nested.json @@ -1,5 +1,5 @@ { - "graph_sitter_type": "PyClass", + "codegen_sdk_type": "PyClass", "span": { "range": { "start_byte": 1, @@ -20,7 +20,7 @@ [ "get_name()", { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 7, @@ -43,7 +43,7 @@ [ "methods", { - "graph_sitter_type": "PyCodeBlock", + "codegen_sdk_type": "PyCodeBlock", "span": { "range": { "start_byte": 16, @@ -64,7 +64,7 @@ [ "statements", { - "graph_sitter_type": "MultiLineCollection", + "codegen_sdk_type": "MultiLineCollection", "span": { "range": { "start_byte": 16, @@ -85,7 +85,7 @@ [ null, { - "graph_sitter_type": "SymbolStatement", + "codegen_sdk_type": "SymbolStatement", "span": { "range": { "start_byte": 16, @@ -106,7 +106,7 @@ [ null, { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 22, @@ -129,7 +129,7 @@ [ null, { - "graph_sitter_type": "PyCodeBlock", + "codegen_sdk_type": "PyCodeBlock", "span": { "range": { "start_byte": 35, @@ -150,7 +150,7 @@ [ "statements", { - "graph_sitter_type": "MultiLineCollection", + "codegen_sdk_type": "MultiLineCollection", "span": { "range": { "start_byte": 35, @@ -171,7 +171,7 @@ [ null, { - "graph_sitter_type": "SymbolStatement", + "codegen_sdk_type": "SymbolStatement", "span": { "range": { "start_byte": 35, @@ -192,7 +192,7 @@ [ null, { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 39, @@ -215,7 +215,7 @@ [ null, { - "graph_sitter_type": "Collection", + "codegen_sdk_type": "Collection", "span": { "range": { "start_byte": 42, @@ -236,7 +236,7 @@ [ null, { - "graph_sitter_type": "PyParameter", + "codegen_sdk_type": "PyParameter", "span": { "range": { "start_byte": 43, @@ -262,7 +262,7 @@ [ null, { - "graph_sitter_type": "PyCodeBlock", + "codegen_sdk_type": "PyCodeBlock", "span": { "range": { "start_byte": 62, @@ -283,7 +283,7 @@ [ "statements", { - "graph_sitter_type": "MultiLineCollection", + "codegen_sdk_type": "MultiLineCollection", "span": { "range": { "start_byte": 62, @@ -304,7 +304,7 @@ [ null, { - "graph_sitter_type": "PyPassStatement", + "codegen_sdk_type": "PyPassStatement", "span": { "range": { "start_byte": 62, @@ -351,7 +351,7 @@ [ null, { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 22, @@ -374,7 +374,7 @@ [ null, { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 39, @@ -397,7 +397,7 @@ [ null, { - "graph_sitter_type": "Name", + "codegen_sdk_type": "Name", "span": { "range": { "start_byte": 43, @@ -418,4 +418,4 @@ } ] ] -} \ No newline at end of file +} diff --git a/tests/unit/output/test_ast.py b/tests/unit/output/test_ast.py index f914171f6..a8d3485de 100644 --- a/tests/unit/output/test_ast.py +++ b/tests/unit/output/test_ast.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_ast_basic(tmpdir: str, snapshot) -> None: @@ -15,7 +15,7 @@ class Bar: file = codebase.get_file("test.py") foo_class = file.get_class("Foo") ast = foo_class.ast() - snapshot.assert_match(ast.model_dump_json(indent=4), "ast.json") + snapshot.assert_match(ast.model_dump_json(indent=4) + "\n", "ast.json") def test_ast_nested(tmpdir: str, snapshot) -> None: @@ -33,4 +33,4 @@ def baz(self): file = codebase.get_file("test.py") foo_class = file.get_class("Foo") ast = foo_class.ast() - snapshot.assert_match(ast.model_dump_json(indent=4), "ast-nested.json") + snapshot.assert_match(ast.model_dump_json(indent=4) + "\n", "ast-nested.json") diff --git a/tests/unit/output/test_json.py b/tests/unit/output/test_json.py index 23fe56871..8ce63459e 100644 --- a/tests/unit/output/test_json.py +++ b/tests/unit/output/test_json.py @@ -1,6 +1,6 @@ import pprint -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_json_basic(tmpdir, snapshot) -> None: diff --git a/tests/unit/parser/test_get_parser_by_filepath_or_extension.py b/tests/unit/parser/test_get_parser_by_filepath_or_extension.py index c0bd9ec41..ba49c5c12 100644 --- a/tests/unit/parser/test_get_parser_by_filepath_or_extension.py +++ b/tests/unit/parser/test_get_parser_by_filepath_or_extension.py @@ -1,4 +1,4 @@ -from graph_sitter.tree_sitter_parser import get_parser_by_filepath_or_extension +from codegen.sdk.tree_sitter_parser import get_parser_by_filepath_or_extension def test_languages() -> None: diff --git a/tests/unit/python/autocommit/test_autocommit.py b/tests/unit/python/autocommit/test_autocommit.py index e9073fce9..13f56737a 100644 --- a/tests/unit/python/autocommit/test_autocommit.py +++ b/tests/unit/python/autocommit/test_autocommit.py @@ -1,10 +1,11 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.autocommit.constants import NodeNotFoundError, OutdatedNodeError -from graph_sitter.core.statements.statement import StatementType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.autocommit.constants import NodeNotFoundError, OutdatedNodeError +from codegen.sdk.core.statements.statement import StatementType +@pytest.mark.skip("No Autocommit") def test_autocommit_rename_move(tmpdir) -> None: file1_name = "file1.py" # language=python @@ -59,6 +60,7 @@ def bar2(): ) +@pytest.mark.skip("No Autocommit") def test_autocommit_move_rename(tmpdir) -> None: file1_name = "file1.py" # language=python @@ -113,6 +115,7 @@ def bar2(): ) +@pytest.mark.skip("No Autocommit") def test_autocommit_insert_remove(tmpdir) -> None: file1_name = "file1.py" content1 = "" @@ -124,6 +127,7 @@ def test_autocommit_insert_remove(tmpdir) -> None: assert file1.content == "a" +@pytest.mark.skip("No Autocommit") @pytest.mark.skip(reason="wip") @pytest.mark.parametrize("edit_block", [True, False]) def test_autocommit_file_edit(tmpdir, edit_block: bool) -> None: @@ -152,6 +156,7 @@ def a(): assert file1.content.strip() == "import os" +@pytest.mark.skip("No Autocommit") @pytest.mark.skip(reason="wip") @pytest.mark.parametrize("edit_block", [True, False]) def test_autocommit_param_edit(tmpdir, edit_block: bool) -> None: @@ -178,6 +183,7 @@ def a(a: int): assert file1.content.strip() == "" +@pytest.mark.skip("No Autocommit") @pytest.mark.skip(reason="wip") @pytest.mark.parametrize("edit_block", [True, False]) def test_autocommit_param_edit_file(tmpdir, edit_block: bool) -> None: @@ -205,6 +211,7 @@ def a(a: int): assert file1.content.strip() == "import os" +@pytest.mark.skip("No Autocommit") @pytest.mark.skip(reason="wip") def test_autocommit_param_edit_other(tmpdir) -> None: file1_name = "file1.py" @@ -234,6 +241,7 @@ def b(a: int): assert file1.content.strip() == "import os" +@pytest.mark.skip("No Autocommit") @pytest.mark.skip("Log propagate is off") def test_autocommit_remove_edit(tmpdir, caplog) -> None: file1_name = "file1.py" @@ -250,6 +258,7 @@ def a(): assert "Editing a removed node" in caplog.text +@pytest.mark.skip("No Autocommit") @pytest.mark.skip(reason="wip") def test_autocommit_not_found(tmpdir) -> None: file1_name = "file1.py" @@ -267,6 +276,7 @@ def a(): fun.prepend_statements("a") +@pytest.mark.skip("No Autocommit") def test_autocommit_outdated(tmpdir) -> None: file1_name = "file1.py" # language=python @@ -299,6 +309,7 @@ def bar(): statement.remove() +@pytest.mark.skip("No Autocommit") @pytest.mark.skip("Log propagate is off") def test_autocommit_nocommit_edit(tmpdir, caplog) -> None: """Test ability to handle transaction errors and switch to autocommit.""" @@ -322,6 +333,7 @@ def a(): assert "Committing" in caplog.text +@pytest.mark.skip("No Autocommit") def test_autocommit_repr(tmpdir) -> None: file1_name = "file1.py" # language=python @@ -335,6 +347,7 @@ def a(): repr(fun) +@pytest.mark.skip("No Autocommit") def test_autocommit_double_edit(tmpdir) -> None: file1_name = "file1.py" # language=python @@ -351,6 +364,7 @@ def b(): function.commit() +@pytest.mark.skip("No Autocommit") @pytest.mark.skip("Log propagate is off") def test_autocommit_ignore_removed(tmpdir, caplog) -> None: """Test ability for removes to not mess up ordering.""" diff --git a/tests/unit/python/class_definition/test_class_add_attribute.py b/tests/unit/python/class_definition/test_class_add_attribute.py index 95e3cdb3a..01e907998 100644 --- a/tests/unit/python/class_definition/test_class_add_attribute.py +++ b/tests/unit/python/class_definition/test_class_add_attribute.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_class_definition_add_attribute_adds_source(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_add_comment.py b/tests/unit/python/class_definition/test_class_add_comment.py index c86380c85..a55d7af42 100644 --- a/tests/unit/python/class_definition/test_class_add_comment.py +++ b/tests/unit/python/class_definition/test_class_add_comment.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_set_comment_to_method(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_add_decorator.py b/tests/unit/python/class_definition/test_class_add_decorator.py index db751953d..736568e4e 100644 --- a/tests/unit/python/class_definition/test_class_add_decorator.py +++ b/tests/unit/python/class_definition/test_class_add_decorator.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_add_decorator_to_class(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_add_method.py b/tests/unit/python/class_definition/test_class_add_method.py index b255898f7..eb0b0d84a 100644 --- a/tests/unit/python/class_definition/test_class_add_method.py +++ b/tests/unit/python/class_definition/test_class_add_method.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.python import PyClass, PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.python import PyClass, PyFile def test_add_method_basic(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_constructor_usages.py b/tests/unit/python/class_definition/test_class_constructor_usages.py index 639be433e..67b15d6d7 100644 --- a/tests/unit/python/class_definition/test_class_constructor_usages.py +++ b/tests/unit/python/class_definition/test_class_constructor_usages.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.class_definition import Class -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_parent_class_names_single(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_dependencies.py b/tests/unit/python/class_definition/test_class_dependencies.py index 021c1f96d..b027f000f 100644 --- a/tests/unit/python/class_definition/test_class_dependencies.py +++ b/tests/unit/python/class_definition/test_class_dependencies.py @@ -1,8 +1,8 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageKind, UsageType -from graph_sitter.core.import_resolution import Import -from graph_sitter.core.symbol import Symbol -from graph_sitter.enums import SymbolType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageKind, UsageType +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.core.symbol import Symbol +from codegen.sdk.enums import SymbolType def test_class_dependencies(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_editable.py b/tests/unit/python/class_definition/test_class_editable.py index 7bdcc3f0a..af625e0d4 100644 --- a/tests/unit/python/class_definition/test_class_editable.py +++ b/tests/unit/python/class_definition/test_class_editable.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_decorated_class_source(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_extended.py b/tests/unit/python/class_definition/test_class_extended.py index e7eacf95c..9b96e3003 100644 --- a/tests/unit/python/class_definition/test_class_extended.py +++ b/tests/unit/python/class_definition/test_class_extended.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_class_extended_nodes(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_filter.py b/tests/unit/python/class_definition/test_class_filter.py index 1f0e654c9..bfd8d3eb1 100644 --- a/tests/unit/python/class_definition/test_class_filter.py +++ b/tests/unit/python/class_definition/test_class_filter.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_class_definition_filter(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_inherited_methods.py b/tests/unit/python/class_definition/test_class_inherited_methods.py index 7bf948b40..700e1337b 100644 --- a/tests/unit/python/class_definition/test_class_inherited_methods.py +++ b/tests/unit/python/class_definition/test_class_inherited_methods.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_class_inherited_method_parent_detection(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_method_usages.py b/tests/unit/python/class_definition/test_class_method_usages.py index 0d05df347..5ccd9b152 100644 --- a/tests/unit/python/class_definition/test_class_method_usages.py +++ b/tests/unit/python/class_definition/test_class_method_usages.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_parent_class_names_single(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_parent_class_names.py b/tests/unit/python/class_definition/test_class_parent_class_names.py index 926c8fcca..fce6839e6 100644 --- a/tests/unit/python/class_definition/test_class_parent_class_names.py +++ b/tests/unit/python/class_definition/test_class_parent_class_names.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.interfaces.editable import Editable +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.interfaces.editable import Editable def test_class_edit_parent_class_names(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_parent_classes.py b/tests/unit/python/class_definition/test_class_parent_classes.py index ed04937e2..e606617e7 100644 --- a/tests/unit/python/class_definition/test_class_parent_classes.py +++ b/tests/unit/python/class_definition/test_class_parent_classes.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_parent_classes_single(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_properties.py b/tests/unit/python/class_definition/test_class_properties.py index 6c5d72918..7b4fcd2a1 100644 --- a/tests/unit/python/class_definition/test_class_properties.py +++ b/tests/unit/python/class_definition/test_class_properties.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.python import PyClass +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.python import PyClass def test_classes(tmpdir) -> None: diff --git a/tests/unit/python/class_definition/test_class_rename.py b/tests/unit/python/class_definition/test_class_rename.py index bc08f9807..957bd9280 100644 --- a/tests/unit/python/class_definition/test_class_rename.py +++ b/tests/unit/python/class_definition/test_class_rename.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_class_rename(tmpdir) -> None: diff --git a/tests/unit/python/codebase/test_codebase.py b/tests/unit/python/codebase/test_codebase.py index 3b1c2dd93..10d88fa5f 100644 --- a/tests/unit/python/codebase/test_codebase.py +++ b/tests/unit/python/codebase/test_codebase.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_codebase_symbols(tmpdir) -> None: diff --git a/tests/unit/python/codebase/test_codebase_auto_commit.py b/tests/unit/python/codebase/test_codebase_auto_commit.py index 5ea2b3a91..533e82fde 100644 --- a/tests/unit/python/codebase/test_codebase_auto_commit.py +++ b/tests/unit/python/codebase/test_codebase_auto_commit.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage @pytest.mark.skip(reason="We are disabling auto commit for performance reasons") diff --git a/tests/unit/python/codebase/test_codebase_create_file.py b/tests/unit/python/codebase/test_codebase_create_file.py index 41d31e4db..2f3d03f07 100644 --- a/tests/unit/python/codebase/test_codebase_create_file.py +++ b/tests/unit/python/codebase/test_codebase_create_file.py @@ -2,9 +2,9 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.file import File +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.file import File from tests.unit.python.utils.test_file_contents import file1_content, file2_content diff --git a/tests/unit/python/codebase/test_codebase_git.py b/tests/unit/python/codebase/test_codebase_git.py index dff6590a7..681e39fe2 100644 --- a/tests/unit/python/codebase/test_codebase_git.py +++ b/tests/unit/python/codebase/test_codebase_git.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage @pytest.mark.parametrize("commit, sync", [(True, True), (True, False), (False, False)]) diff --git a/tests/unit/python/codebase/test_codebase_log.py b/tests/unit/python/codebase/test_codebase_log.py index dbde4a137..eb09b87a3 100644 --- a/tests/unit/python/codebase/test_codebase_log.py +++ b/tests/unit/python/codebase/test_codebase_log.py @@ -2,10 +2,10 @@ import pytest -from graph_sitter.codebase.config import SessionOptions -from graph_sitter.codebase.control_flow import MaxPreviewTimeExceeded -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.config import SessionOptions +from codegen.sdk.codebase.control_flow import MaxPreviewTimeExceeded +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_log_writes_to_console(tmpdir) -> None: @@ -19,7 +19,7 @@ def test_log_writes_to_console(tmpdir) -> None: assert console == "This is a test log\nThis is another test log\n" -@patch("graph_sitter.core.codebase.MAX_LINES", 1) +@patch("codegen.sdk.core.codebase.MAX_LINES", 1) def test_log_truncate_at_max_lines(tmpdir) -> None: with get_codebase_session( tmpdir=tmpdir, @@ -33,7 +33,7 @@ def test_log_truncate_at_max_lines(tmpdir) -> None: assert console == "1\n" -@patch("graph_sitter.core.codebase.MAX_LINES", 1) +@patch("codegen.sdk.core.codebase.MAX_LINES", 1) def test_log_max_lines_reached_continues_execution(tmpdir) -> None: with get_codebase_session( tmpdir=tmpdir, @@ -64,7 +64,7 @@ def test_get_finalized_logs_returns_all_logs(tmpdir) -> None: assert console == "1\n2\n3\n4\n" -@patch("graph_sitter.core.codebase.MAX_LINES", 1) +@patch("codegen.sdk.core.codebase.MAX_LINES", 1) def test_get_finalized_logs_truncate_at_max_lines(tmpdir) -> None: with get_codebase_session( tmpdir=tmpdir, diff --git a/tests/unit/python/codebase/test_codebase_meta.py b/tests/unit/python/codebase/test_codebase_meta.py index 2ed036579..76b0d2d83 100644 --- a/tests/unit/python/codebase/test_codebase_meta.py +++ b/tests/unit/python/codebase/test_codebase_meta.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_codebase_meta(tmpdir) -> None: diff --git a/tests/unit/python/codebase/test_codebase_reset.py b/tests/unit/python/codebase/test_codebase_reset.py index e79f22653..a848626c6 100644 --- a/tests/unit/python/codebase/test_codebase_reset.py +++ b/tests/unit/python/codebase/test_codebase_reset.py @@ -1,8 +1,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.codebase.validation import get_edges -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.validation import get_edges +from codegen.sdk.enums import ProgrammingLanguage def test_codebase_reset_exception(tmpdir, caplog) -> None: diff --git a/tests/unit/python/codebase/test_codebase_resolution.py b/tests/unit/python/codebase/test_codebase_resolution.py index 6be079f37..bab4a2f8f 100644 --- a/tests/unit/python/codebase/test_codebase_resolution.py +++ b/tests/unit/python/codebase/test_codebase_resolution.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.enums import ProgrammingLanguage def test_codebase_resolution(tmpdir) -> None: diff --git a/tests/unit/python/codebase/test_control_flow.py b/tests/unit/python/codebase/test_control_flow.py index 19cf9464e..a5759697d 100644 --- a/tests/unit/python/codebase/test_control_flow.py +++ b/tests/unit/python/codebase/test_control_flow.py @@ -1,10 +1,10 @@ import pytest -from graph_sitter.codebase.config import SessionOptions -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.codebase.transaction_manager import MaxPreviewTimeExceeded, MaxTransactionsExceeded -from graph_sitter.core.codebase import MaxAIRequestsError -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.config import SessionOptions +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.transaction_manager import MaxPreviewTimeExceeded, MaxTransactionsExceeded +from codegen.sdk.core.codebase import MaxAIRequestsError +from codegen.sdk.enums import ProgrammingLanguage def test_max_transactions_exceeded_reached_set_threshold(tmpdir): diff --git a/tests/unit/python/comment_docstring/test_comment.py b/tests/unit/python/comment_docstring/test_comment.py index a13f3c384..5364de68a 100644 --- a/tests/unit/python/comment_docstring/test_comment.py +++ b/tests/unit/python/comment_docstring/test_comment.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_comment_basic(tmpdir) -> None: diff --git a/tests/unit/python/comment_docstring/test_comment_edit.py b/tests/unit/python/comment_docstring/test_comment_edit.py index 7772f8339..7c8419e9b 100644 --- a/tests/unit/python/comment_docstring/test_comment_edit.py +++ b/tests/unit/python/comment_docstring/test_comment_edit.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_comment_edit_basic(tmpdir) -> None: diff --git a/tests/unit/python/comment_docstring/test_comment_insert_add.py b/tests/unit/python/comment_docstring/test_comment_insert_add.py index 9adf40b0d..9e6da470d 100644 --- a/tests/unit/python/comment_docstring/test_comment_insert_add.py +++ b/tests/unit/python/comment_docstring/test_comment_insert_add.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_set_comment(tmpdir) -> None: diff --git a/tests/unit/python/comment_docstring/test_docstring.py b/tests/unit/python/comment_docstring/test_docstring.py index a9043e781..823d6ebbe 100644 --- a/tests/unit/python/comment_docstring/test_docstring.py +++ b/tests/unit/python/comment_docstring/test_docstring.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_no_docstring(tmpdir) -> None: diff --git a/tests/unit/python/comment_docstring/test_docstring_edit.py b/tests/unit/python/comment_docstring/test_docstring_edit.py index de36d8f11..0ffbc83db 100644 --- a/tests/unit/python/comment_docstring/test_docstring_edit.py +++ b/tests/unit/python/comment_docstring/test_docstring_edit.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_oneline_docstring(tmpdir) -> None: diff --git a/tests/unit/python/comment_docstring/test_docstring_insert_add.py b/tests/unit/python/comment_docstring/test_docstring_insert_add.py index 4515ef327..297d4a843 100644 --- a/tests/unit/python/comment_docstring/test_docstring_insert_add.py +++ b/tests/unit/python/comment_docstring/test_docstring_insert_add.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_set_docstring(tmpdir) -> None: diff --git a/tests/unit/python/comment_docstring/test_inline_comment.py b/tests/unit/python/comment_docstring/test_inline_comment.py index 7cdf213b6..589d87407 100644 --- a/tests/unit/python/comment_docstring/test_inline_comment.py +++ b/tests/unit/python/comment_docstring/test_inline_comment.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_comment_inline(tmpdir) -> None: diff --git a/tests/unit/python/comment_docstring/test_inline_comment_edit.py b/tests/unit/python/comment_docstring/test_inline_comment_edit.py index 415ce9c56..ce5c43e63 100644 --- a/tests/unit/python/comment_docstring/test_inline_comment_edit.py +++ b/tests/unit/python/comment_docstring/test_inline_comment_edit.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_comment_inline(tmpdir) -> None: diff --git a/tests/unit/python/comment_docstring/test_inline_comment_insert_add.py b/tests/unit/python/comment_docstring/test_inline_comment_insert_add.py index de714d9d6..68d32d64d 100644 --- a/tests/unit/python/comment_docstring/test_inline_comment_insert_add.py +++ b/tests/unit/python/comment_docstring/test_inline_comment_insert_add.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_set_comment_inline(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/argument/test_argument_edit.py b/tests/unit/python/detached_symbols/argument/test_argument_edit.py index 5af13aa91..da457e859 100644 --- a/tests/unit/python/detached_symbols/argument/test_argument_edit.py +++ b/tests/unit/python/detached_symbols/argument/test_argument_edit.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_argument_edit_updates_source(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/argument/test_argument_remove.py b/tests/unit/python/detached_symbols/argument/test_argument_remove.py index acff34742..cbfe4150f 100644 --- a/tests/unit/python/detached_symbols/argument/test_argument_remove.py +++ b/tests/unit/python/detached_symbols/argument/test_argument_remove.py @@ -1,6 +1,6 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session @pytest.mark.skip("BROKEN! todo @edward") diff --git a/tests/unit/python/detached_symbols/code_block/test_code_block.py b/tests/unit/python/detached_symbols/code_block/test_code_block.py index 1db83b386..610054cff 100644 --- a/tests/unit/python/detached_symbols/code_block/test_code_block.py +++ b/tests/unit/python/detached_symbols/code_block/test_code_block.py @@ -1,9 +1,9 @@ import re -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python import PyFile def test_code_block(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/code_block/test_code_block_indent.py b/tests/unit/python/detached_symbols/code_block/test_code_block_indent.py index 05dcdd9ee..c626d77aa 100644 --- a/tests/unit/python/detached_symbols/code_block/test_code_block_indent.py +++ b/tests/unit/python/detached_symbols/code_block/test_code_block_indent.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.python import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.python import PyFile def test_indent_left_once(tmpdir): diff --git a/tests/unit/python/detached_symbols/code_block/test_code_block_insert.py b/tests/unit/python/detached_symbols/code_block/test_code_block_insert.py index e3a40448d..1b29bb754 100644 --- a/tests/unit/python/detached_symbols/code_block/test_code_block_insert.py +++ b/tests/unit/python/detached_symbols/code_block/test_code_block_insert.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_code_block_append_single_element(tmpdir): diff --git a/tests/unit/python/detached_symbols/code_block/test_code_block_remove.py b/tests/unit/python/detached_symbols/code_block/test_code_block_remove.py index aa2aac18a..5c58933eb 100644 --- a/tests/unit/python/detached_symbols/code_block/test_code_block_remove.py +++ b/tests/unit/python/detached_symbols/code_block/test_code_block_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_code_block_remove_multiple_element(tmpdir): diff --git a/tests/unit/python/detached_symbols/code_block/test_code_block_unwrap.py b/tests/unit/python/detached_symbols/code_block/test_code_block_unwrap.py index 92d0d770b..c64a05a0c 100644 --- a/tests/unit/python/detached_symbols/code_block/test_code_block_unwrap.py +++ b/tests/unit/python/detached_symbols/code_block/test_code_block_unwrap.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.python import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.python import PyFile def test_unwrap_flat_statements(tmpdir): diff --git a/tests/unit/python/detached_symbols/code_block/test_code_block_wrap.py b/tests/unit/python/detached_symbols/code_block/test_code_block_wrap.py index 270e006fc..1691c69e5 100644 --- a/tests/unit/python/detached_symbols/code_block/test_code_block_wrap.py +++ b/tests/unit/python/detached_symbols/code_block/test_code_block_wrap.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.python import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.python import PyFile def test_wrap_with_with_statement(tmpdir): diff --git a/tests/unit/python/detached_symbols/decorator/test_decorator_call.py b/tests/unit/python/detached_symbols/decorator/test_decorator_call.py index aa3d8dbf3..e3c526f2f 100644 --- a/tests/unit/python/detached_symbols/decorator/test_decorator_call.py +++ b/tests/unit/python/detached_symbols/decorator/test_decorator_call.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session def test_decorator_call_1(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/decorator/test_decorator_name.py b/tests/unit/python/detached_symbols/decorator/test_decorator_name.py index 705cbde2a..d2ea2fb0b 100644 --- a/tests/unit/python/detached_symbols/decorator/test_decorator_name.py +++ b/tests/unit/python/detached_symbols/decorator/test_decorator_name.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_decorator_simple_name(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/decorator/test_decorator_remove.py b/tests/unit/python/detached_symbols/decorator/test_decorator_remove.py index 4755c69f4..3059ef5c0 100644 --- a/tests/unit/python/detached_symbols/decorator/test_decorator_remove.py +++ b/tests/unit/python/detached_symbols/decorator/test_decorator_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_decorator_remove(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/decorator/test_insert_argument.py b/tests/unit/python/detached_symbols/decorator/test_insert_argument.py index d067ec6a2..c07f256a5 100644 --- a/tests/unit/python/detached_symbols/decorator/test_insert_argument.py +++ b/tests/unit/python/detached_symbols/decorator/test_insert_argument.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_insert_single_arg(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/function_call/test_function_call.py b/tests/unit/python/detached_symbols/function_call/test_function_call.py index b6ffecb57..7e25c83ff 100644 --- a/tests/unit/python/detached_symbols/function_call/test_function_call.py +++ b/tests/unit/python/detached_symbols/function_call/test_function_call.py @@ -1,6 +1,6 @@ # TODO: this file needs to be broken up into API specific tests -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.python import PyFunction +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.python import PyFunction def test_basic(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/function_call/test_function_call_add_argument.py b/tests/unit/python/detached_symbols/function_call/test_function_call_add_argument.py index eb54c38f4..17bb5cc7d 100644 --- a/tests/unit/python/detached_symbols/function_call/test_function_call_add_argument.py +++ b/tests/unit/python/detached_symbols/function_call/test_function_call_add_argument.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function def test_function_call_insert_argument(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/function_call/test_function_call_from_usage.py b/tests/unit/python/detached_symbols/function_call/test_function_call_from_usage.py index e40fe4788..6f68b3280 100644 --- a/tests/unit/python/detached_symbols/function_call/test_function_call_from_usage.py +++ b/tests/unit/python/detached_symbols/function_call/test_function_call_from_usage.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.detached_symbols.function_call import FunctionCall def test_function_call_from_usage(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/function_call/test_function_call_is_awaited.py b/tests/unit/python/detached_symbols/function_call/test_function_call_is_awaited.py index 571e16d13..b6d81f3ae 100644 --- a/tests/unit/python/detached_symbols/function_call/test_function_call_is_awaited.py +++ b/tests/unit/python/detached_symbols/function_call/test_function_call_is_awaited.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_call_is_awaited_basic(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/function_call/test_function_calls_from_symbol.py b/tests/unit/python/detached_symbols/function_call/test_function_calls_from_symbol.py index bc98c6e88..9dd786afa 100644 --- a/tests/unit/python/detached_symbols/function_call/test_function_calls_from_symbol.py +++ b/tests/unit/python/detached_symbols/function_call/test_function_calls_from_symbol.py @@ -1,13 +1,13 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.detached_symbols.argument import Argument -from graph_sitter.core.statements.expression_statement import ExpressionStatement -from graph_sitter.core.statements.return_statement import ReturnStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.python import PyAssignment -from graph_sitter.python.detached_symbols.decorator import PyDecorator -from graph_sitter.python.statements.if_block_statement import PyIfBlockStatement +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.detached_symbols.argument import Argument +from codegen.sdk.core.statements.expression_statement import ExpressionStatement +from codegen.sdk.core.statements.return_statement import ReturnStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.python import PyAssignment +from codegen.sdk.python.detached_symbols.decorator import PyDecorator +from codegen.sdk.python.statements.if_block_statement import PyIfBlockStatement def test_function_calls_from_file(tmpdir): diff --git a/tests/unit/python/detached_symbols/function_call/test_manipulations_function_call.py b/tests/unit/python/detached_symbols/function_call/test_manipulations_function_call.py index 6e2672a92..ef0d591b5 100644 --- a/tests/unit/python/detached_symbols/function_call/test_manipulations_function_call.py +++ b/tests/unit/python/detached_symbols/function_call/test_manipulations_function_call.py @@ -1,9 +1,9 @@ # TODO: convert all of this to use mock files/tmpdir -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.file import SourceFile -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.file import SourceFile +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_replace_function_call_arg(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/parameter/test_parameter_edit.py b/tests/unit/python/detached_symbols/parameter/test_parameter_edit.py index a50486764..dc5b077c4 100644 --- a/tests/unit/python/detached_symbols/parameter/test_parameter_edit.py +++ b/tests/unit/python/detached_symbols/parameter/test_parameter_edit.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_edit_parameter_in_function_definition(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/parameter/test_parameter_editable.py b/tests/unit/python/detached_symbols/parameter/test_parameter_editable.py index 4d1e2f92b..95de8b7cc 100644 --- a/tests/unit/python/detached_symbols/parameter/test_parameter_editable.py +++ b/tests/unit/python/detached_symbols/parameter/test_parameter_editable.py @@ -1,6 +1,6 @@ import re -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session from tests.unit.python.utils.test_file_contents import file1_content, file2_content diff --git a/tests/unit/python/detached_symbols/parameter/test_parameter_is_optional.py b/tests/unit/python/detached_symbols/parameter/test_parameter_is_optional.py index 4fd4df36a..e31075178 100644 --- a/tests/unit/python/detached_symbols/parameter/test_parameter_is_optional.py +++ b/tests/unit/python/detached_symbols/parameter/test_parameter_is_optional.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_parameter_is_optional_should_return_true(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/parameter/test_parameter_is_variadic.py b/tests/unit/python/detached_symbols/parameter/test_parameter_is_variadic.py index 52afe6c9e..02c058281 100644 --- a/tests/unit/python/detached_symbols/parameter/test_parameter_is_variadic.py +++ b/tests/unit/python/detached_symbols/parameter/test_parameter_is_variadic.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_parameter_args_is_variadic_should_return_true(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/parameter/test_parameter_properties.py b/tests/unit/python/detached_symbols/parameter/test_parameter_properties.py index 7fe86782c..ab33a5514 100644 --- a/tests/unit/python/detached_symbols/parameter/test_parameter_properties.py +++ b/tests/unit/python/detached_symbols/parameter/test_parameter_properties.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_parameter_basic_properties(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/parameter/test_parameter_remove.py b/tests/unit/python/detached_symbols/parameter/test_parameter_remove.py index 7cc244f23..9c0d40031 100644 --- a/tests/unit/python/detached_symbols/parameter/test_parameter_remove.py +++ b/tests/unit/python/detached_symbols/parameter/test_parameter_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_parameter_remove(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/parameter/test_parameter_rename.py b/tests/unit/python/detached_symbols/parameter/test_parameter_rename.py index 98de8588e..32ed6e421 100644 --- a/tests/unit/python/detached_symbols/parameter/test_parameter_rename.py +++ b/tests/unit/python/detached_symbols/parameter/test_parameter_rename.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_rename_and_type_annotation(tmpdir) -> None: diff --git a/tests/unit/python/detached_symbols/parameter/test_parameter_set_type_annotation.py b/tests/unit/python/detached_symbols/parameter/test_parameter_set_type_annotation.py index a26c2f8fa..d72523017 100644 --- a/tests/unit/python/detached_symbols/parameter/test_parameter_set_type_annotation.py +++ b/tests/unit/python/detached_symbols/parameter/test_parameter_set_type_annotation.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session DEFAULT_FILE_NAME = "test_for_one_test.py" diff --git a/tests/unit/python/directory/test_directory.py b/tests/unit/python/directory/test_directory.py index 1edd61286..f49a8b02f 100644 --- a/tests/unit/python/directory/test_directory.py +++ b/tests/unit/python/directory/test_directory.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_directory_creation(tmpdir) -> None: diff --git a/tests/unit/python/directory/test_directory_move.py b/tests/unit/python/directory/test_directory_move.py index 5dc7e257f..86fce4a17 100644 --- a/tests/unit/python/directory/test_directory_move.py +++ b/tests/unit/python/directory/test_directory_move.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_directory_file_move_simple(tmpdir) -> None: diff --git a/tests/unit/python/directory/test_directory_remove.py b/tests/unit/python/directory/test_directory_remove.py index e69693bb2..7514fa920 100644 --- a/tests/unit/python/directory/test_directory_remove.py +++ b/tests/unit/python/directory/test_directory_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_directory_file_move_simple(tmpdir) -> None: diff --git a/tests/unit/python/editable/test_replace.py b/tests/unit/python/editable/test_replace.py index 420b21a01..6a6c8a8e2 100644 --- a/tests/unit/python/editable/test_replace.py +++ b/tests/unit/python/editable/test_replace.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_replace_all(tmpdir) -> None: diff --git a/tests/unit/python/expressions/test_binary_expression.py b/tests/unit/python/expressions/test_binary_expression.py index a03d78ca6..8f1869c1e 100644 --- a/tests/unit/python/expressions/test_binary_expression.py +++ b/tests/unit/python/expressions/test_binary_expression.py @@ -1,8 +1,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.python import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.python import PyFile def test_all_binary_expression_types(tmpdir): diff --git a/tests/unit/python/expressions/test_binary_reduce_condition.py b/tests/unit/python/expressions/test_binary_reduce_condition.py index 9870bd975..96341a81c 100644 --- a/tests/unit/python/expressions/test_binary_reduce_condition.py +++ b/tests/unit/python/expressions/test_binary_reduce_condition.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python.file import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python.file import PyFile def test_reduce_binary_simple(tmpdir): diff --git a/tests/unit/python/expressions/test_boolean.py b/tests/unit/python/expressions/test_boolean.py index f8a2e3e9f..51ac48ecb 100644 --- a/tests/unit/python/expressions/test_boolean.py +++ b/tests/unit/python/expressions/test_boolean.py @@ -1,8 +1,8 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.expressions.boolean import Boolean -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python.assignment import PyAssignment +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.expressions.boolean import Boolean +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python.assignment import PyAssignment def test_boolean_parse(tmpdir): diff --git a/tests/unit/python/expressions/test_conditional_properties.py b/tests/unit/python/expressions/test_conditional_properties.py index d37c9eb44..6de9751d0 100644 --- a/tests/unit/python/expressions/test_conditional_properties.py +++ b/tests/unit/python/expressions/test_conditional_properties.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python.file import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python.file import PyFile def test_parse_simple_conditional_expression(tmpdir: str) -> None: diff --git a/tests/unit/python/expressions/test_conditional_reduce_condition.py b/tests/unit/python/expressions/test_conditional_reduce_condition.py index 00e62b854..fc89932ad 100644 --- a/tests/unit/python/expressions/test_conditional_reduce_condition.py +++ b/tests/unit/python/expressions/test_conditional_reduce_condition.py @@ -1,8 +1,8 @@ from typing import Any -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.python.file import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.python.file import PyFile def test_reduce_ternary_condition_to_true(tmpdir: Any) -> None: diff --git a/tests/unit/python/expressions/test_dict.py b/tests/unit/python/expressions/test_dict.py index 456f5ff0a..8d08bdd04 100644 --- a/tests/unit/python/expressions/test_dict.py +++ b/tests/unit/python/expressions/test_dict.py @@ -2,8 +2,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.symbol_groups.dict import Dict +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.symbol_groups.dict import Dict def test_dict_basic(tmpdir) -> None: diff --git a/tests/unit/python/expressions/test_list.py b/tests/unit/python/expressions/test_list.py index ef288dc57..57ba0db51 100644 --- a/tests/unit/python/expressions/test_list.py +++ b/tests/unit/python/expressions/test_list.py @@ -2,8 +2,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.symbol_groups.list import List +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.symbol_groups.list import List def test_list_basic(tmpdir) -> None: diff --git a/tests/unit/python/expressions/test_number.py b/tests/unit/python/expressions/test_number.py index 31be49d26..1452f1477 100644 --- a/tests/unit/python/expressions/test_number.py +++ b/tests/unit/python/expressions/test_number.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.expressions.number import Number +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.expressions.number import Number def test_number_parse(tmpdir): diff --git a/tests/unit/python/expressions/test_string.py b/tests/unit/python/expressions/test_string.py index 0920f5fff..4871bad76 100644 --- a/tests/unit/python/expressions/test_string.py +++ b/tests/unit/python/expressions/test_string.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions import String +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions import String def test_string_edit(tmpdir) -> None: diff --git a/tests/unit/python/expressions/test_tuple.py b/tests/unit/python/expressions/test_tuple.py index 7de626b0b..c54d1f2b2 100644 --- a/tests/unit/python/expressions/test_tuple.py +++ b/tests/unit/python/expressions/test_tuple.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.symbol_groups.tuple import Tuple +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.symbol_groups.tuple import Tuple def test_tuple_basic(tmpdir) -> None: diff --git a/tests/unit/python/expressions/test_type.py b/tests/unit/python/expressions/test_type.py index 2139dc730..b67cac15f 100644 --- a/tests/unit/python/expressions/test_type.py +++ b/tests/unit/python/expressions/test_type.py @@ -1,8 +1,8 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.none_type import NoneType -from graph_sitter.core.expressions.type import Type -from graph_sitter.python import PyAssignment, PyClass, PyFunction -from graph_sitter.python.expressions.union_type import PyUnionType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.none_type import NoneType +from codegen.sdk.core.expressions.type import Type +from codegen.sdk.python import PyAssignment, PyClass, PyFunction +from codegen.sdk.python.expressions.union_type import PyUnionType def test_type_basic(tmpdir): diff --git a/tests/unit/python/expressions/test_value.py b/tests/unit/python/expressions/test_value.py index 065845879..b15848047 100644 --- a/tests/unit/python/expressions/test_value.py +++ b/tests/unit/python/expressions/test_value.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions import String +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions import String @pytest.mark.skip(reason="CG-9463: Fix resolved types to be start byte aware") diff --git a/tests/unit/python/file/test_ast_grep.py b/tests/unit/python/file/test_ast_grep.py index 6550a7b2a..bf3dee92a 100644 --- a/tests/unit/python/file/test_ast_grep.py +++ b/tests/unit/python/file/test_ast_grep.py @@ -1,6 +1,6 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session @pytest.mark.skip(reason="Not implementing AST Grep for now. Much better ways of doing this.") diff --git a/tests/unit/python/file/test_file_add_import_from_import_string.py b/tests/unit/python/file/test_file_add_import_from_import_string.py index 9cd9e9578..1d905332a 100644 --- a/tests/unit/python/file/test_file_add_import_from_import_string.py +++ b/tests/unit/python/file/test_file_add_import_from_import_string.py @@ -1,6 +1,6 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_file_add_symbol_import_from_string_adds_after_future(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_add_symbol_from_source.py b/tests/unit/python/file/test_file_add_symbol_from_source.py index 350e4762b..1f17b4fc3 100644 --- a/tests/unit/python/file/test_file_add_symbol_from_source.py +++ b/tests/unit/python/file/test_file_add_symbol_from_source.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.python import PyAssignment +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.python import PyAssignment def test_file_add_symbol_from_source_updates_graph(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_add_symbol_import.py b/tests/unit/python/file/test_file_add_symbol_import.py index 7752731d9..7088f2c1e 100644 --- a/tests/unit/python/file/test_file_add_symbol_import.py +++ b/tests/unit/python/file/test_file_add_symbol_import.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_file_add_symbol_import_updates_source(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_editable.py b/tests/unit/python/file/test_file_editable.py index c4e141b25..3d56ec062 100644 --- a/tests/unit/python/file/test_file_editable.py +++ b/tests/unit/python/file/test_file_editable.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_file_find_string_literals_fuzzy_match(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_exportable.py b/tests/unit/python/file/test_file_exportable.py index ad23b7323..1429e5e42 100644 --- a/tests/unit/python/file/test_file_exportable.py +++ b/tests/unit/python/file/test_file_exportable.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.core.symbol import Symbol +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.core.symbol import Symbol def test_file_usages_basic(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_get_function_calls.py b/tests/unit/python/file/test_file_get_function_calls.py index a4e1403d7..e21330b89 100644 --- a/tests/unit/python/file/test_file_get_function_calls.py +++ b/tests/unit/python/file/test_file_get_function_calls.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_file_get_function_calls(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_get_import_string.py b/tests/unit/python/file/test_file_get_import_string.py index f3d690049..a4f41a466 100644 --- a/tests/unit/python/file/test_file_get_import_string.py +++ b/tests/unit/python/file/test_file_get_import_string.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_file_get_import_string_no_params(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_global_calls.py b/tests/unit/python/file/test_file_global_calls.py index af4b5ec5b..b039f5b5a 100644 --- a/tests/unit/python/file/test_file_global_calls.py +++ b/tests/unit/python/file/test_file_global_calls.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_global_function_calls(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_owners.py b/tests/unit/python/file/test_file_owners.py index c740e43a9..4016dc0a2 100644 --- a/tests/unit/python/file/test_file_owners.py +++ b/tests/unit/python/file/test_file_owners.py @@ -1,6 +1,6 @@ from unittest.mock import MagicMock, patch -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_file_owners_non_codeowners_parser_returns_empty_set(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_properties.py b/tests/unit/python/file/test_file_properties.py index b997a298f..6f7a2279a 100644 --- a/tests/unit/python/file/test_file_properties.py +++ b/tests/unit/python/file/test_file_properties.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import SymbolType -from graph_sitter.python import PyAssignment, PyClass, PyFile, PyFunction, PyImport +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import SymbolType +from codegen.sdk.python import PyAssignment, PyClass, PyFile, PyFunction, PyImport default_content = """ from some_file import x, y, z diff --git a/tests/unit/python/file/test_file_remove.py b/tests/unit/python/file/test_file_remove.py index afe618ef1..13fea7f23 100644 --- a/tests/unit/python/file/test_file_remove.py +++ b/tests/unit/python/file/test_file_remove.py @@ -1,6 +1,6 @@ import os -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_file_remove_after_create(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_reparse.py b/tests/unit/python/file/test_file_reparse.py index a551909b0..f7bb442e7 100644 --- a/tests/unit/python/file/test_file_reparse.py +++ b/tests/unit/python/file/test_file_reparse.py @@ -1,9 +1,9 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.codebase import Codebase -from graph_sitter.core.file import File -from graph_sitter.enums import EdgeType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.core.file import File +from codegen.sdk.enums import EdgeType from tests.unit.python.utils.test_file_contents import file1_content, file2_content diff --git a/tests/unit/python/file/test_file_statements.py b/tests/unit/python/file/test_file_statements.py index e412c61e6..d3372d29b 100644 --- a/tests/unit/python/file/test_file_statements.py +++ b/tests/unit/python/file/test_file_statements.py @@ -1,6 +1,6 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session @pytest.mark.skip("TODO: @caroljung fix this once code block for file and symbols are unified") diff --git a/tests/unit/python/file/test_file_unicode.py b/tests/unit/python/file/test_file_unicode.py index 721e7fc73..af1c0e73a 100644 --- a/tests/unit/python/file/test_file_unicode.py +++ b/tests/unit/python/file/test_file_unicode.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_unicode_move_symbol(tmpdir) -> None: diff --git a/tests/unit/python/file/test_file_update_filepath.py b/tests/unit/python/file/test_file_update_filepath.py index 8984e70a7..e7a9bb1ca 100644 --- a/tests/unit/python/file/test_file_update_filepath.py +++ b/tests/unit/python/file/test_file_update_filepath.py @@ -1,6 +1,6 @@ import os -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_rename_file(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_async.py b/tests/unit/python/function/test_function_async.py index 442868860..796b323b7 100644 --- a/tests/unit/python/function/test_function_async.py +++ b/tests/unit/python/function/test_function_async.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_is_async_basic(tmpdir): diff --git a/tests/unit/python/function/test_function_call_sites.py b/tests/unit/python/function/test_function_call_sites.py index bafbb1a90..614eaf1dd 100644 --- a/tests/unit/python/function/test_function_call_sites.py +++ b/tests/unit/python/function/test_function_call_sites.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.python import PyFile, PyFunction +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.python import PyFile, PyFunction def test_function_callsites(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_dependencies.py b/tests/unit/python/function/test_function_dependencies.py index d52c3500a..d75588ac7 100644 --- a/tests/unit/python/function/test_function_dependencies.py +++ b/tests/unit/python/function/test_function_dependencies.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_function_dependencies_symbol_usage(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_editable.py b/tests/unit/python/function/test_function_editable.py index 197ffbdf5..558969f13 100644 --- a/tests/unit/python/function/test_function_editable.py +++ b/tests/unit/python/function/test_function_editable.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_extended_function_source(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_eq.py b/tests/unit/python/function/test_function_eq.py index ed2184a11..f95a17510 100644 --- a/tests/unit/python/function/test_function_eq.py +++ b/tests/unit/python/function/test_function_eq.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_function_statements_contains(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_exportable.py b/tests/unit/python/function/test_function_exportable.py index e71e23a2b..5d5c3e87a 100644 --- a/tests/unit/python/function/test_function_exportable.py +++ b/tests/unit/python/function/test_function_exportable.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType def test_function_usages_in_main(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_function_calls.py b/tests/unit/python/function/test_function_function_calls.py index 73f733b1c..e6cca748a 100644 --- a/tests/unit/python/function/test_function_function_calls.py +++ b/tests/unit/python/function/test_function_function_calls.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.python import PyFunction +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.python import PyFunction def test_function_calls(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_get_parameter_by_index.py b/tests/unit/python/function/test_function_get_parameter_by_index.py index 3e6d52827..70f7ace06 100644 --- a/tests/unit/python/function/test_function_get_parameter_by_index.py +++ b/tests/unit/python/function/test_function_get_parameter_by_index.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_function_get_parameter_by_index(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_importable.py b/tests/unit/python/function/test_function_importable.py index 2dbcfb69b..0ff168195 100644 --- a/tests/unit/python/function/test_function_importable.py +++ b/tests/unit/python/function/test_function_importable.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.import_resolution import Import +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.import_resolution import Import def test_function_symbol_dependencies(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_insert_statements.py b/tests/unit/python/function/test_function_insert_statements.py index 0f2e458f1..d126a98f2 100644 --- a/tests/unit/python/function/test_function_insert_statements.py +++ b/tests/unit/python/function/test_function_insert_statements.py @@ -1,6 +1,6 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_function_insert_statements_add_source(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_move_to_file.py b/tests/unit/python/function/test_function_move_to_file.py index 5721c2746..779a3e5e6 100644 --- a/tests/unit/python/function/test_function_move_to_file.py +++ b/tests/unit/python/function/test_function_move_to_file.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function def test_move_to_file_add_back_edge(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_nested_functions.py b/tests/unit/python/function/test_function_nested_functions.py index 953dd18eb..0f2e4ff4e 100644 --- a/tests/unit/python/function/test_function_nested_functions.py +++ b/tests/unit/python/function/test_function_nested_functions.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_nested_base(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_parse.py b/tests/unit/python/function/test_function_parse.py index dd3da5c62..48d2bad30 100644 --- a/tests/unit/python/function/test_function_parse.py +++ b/tests/unit/python/function/test_function_parse.py @@ -1,7 +1,7 @@ # TODO: break-up this file into one for each API -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.file import SourceFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.file import SourceFile def test_local_import(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_prepend_statements.py b/tests/unit/python/function/test_function_prepend_statements.py index 1199fe9e3..2ef317c2d 100644 --- a/tests/unit/python/function/test_function_prepend_statements.py +++ b/tests/unit/python/function/test_function_prepend_statements.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_function_prepend_statement(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_properties.py b/tests/unit/python/function/test_function_properties.py index 7b27ed990..1d6fcbf53 100644 --- a/tests/unit/python/function/test_function_properties.py +++ b/tests/unit/python/function/test_function_properties.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_functions(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_remove.py b/tests/unit/python/function/test_function_remove.py index e940826c5..f6b76e1cd 100644 --- a/tests/unit/python/function/test_function_remove.py +++ b/tests/unit/python/function/test_function_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_function_remove(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_rename.py b/tests/unit/python/function/test_function_rename.py index 5b19e13f7..285c41213 100644 --- a/tests/unit/python/function/test_function_rename.py +++ b/tests/unit/python/function/test_function_rename.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_function_rename_updates_usages(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_set_docstring.py b/tests/unit/python/function/test_function_set_docstring.py index 854081c23..6bf130881 100644 --- a/tests/unit/python/function/test_function_set_docstring.py +++ b/tests/unit/python/function/test_function_set_docstring.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_set_docstring(tmpdir) -> None: diff --git a/tests/unit/python/function/test_function_usages.py b/tests/unit/python/function/test_function_usages.py index b924dcbc4..29e623392 100644 --- a/tests/unit/python/function/test_function_usages.py +++ b/tests/unit/python/function/test_function_usages.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType def test_function_usages_recursive_function(tmpdir) -> None: diff --git a/tests/unit/python/function/test_manipulations_function.py b/tests/unit/python/function/test_manipulations_function.py index aefcf5417..48094053f 100644 --- a/tests/unit/python/function/test_manipulations_function.py +++ b/tests/unit/python/function/test_manipulations_function.py @@ -1,7 +1,7 @@ # TODO: break-up these tests into API specific tests, ex: one test file for set_return_type -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_prepend_to_function_body_with_docstring(tmpdir) -> None: diff --git a/tests/unit/python/global_var/test_global_var_editable.py b/tests/unit/python/global_var/test_global_var_editable.py index 469427829..94ef6f67f 100644 --- a/tests/unit/python/global_var/test_global_var_editable.py +++ b/tests/unit/python/global_var/test_global_var_editable.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session from tests.unit.python.utils.test_file_contents import file1_content, file2_content diff --git a/tests/unit/python/global_var/test_global_var_exportable.py b/tests/unit/python/global_var/test_global_var_exportable.py index 356512b44..6d7a6d259 100644 --- a/tests/unit/python/global_var/test_global_var_exportable.py +++ b/tests/unit/python/global_var/test_global_var_exportable.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType def test_global_var_usages(tmpdir) -> None: diff --git a/tests/unit/python/global_var/test_global_var_has_value.py b/tests/unit/python/global_var/test_global_var_has_value.py index d79bd4f79..0e37a0845 100644 --- a/tests/unit/python/global_var/test_global_var_has_value.py +++ b/tests/unit/python/global_var/test_global_var_has_value.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_global_var_single_var(tmpdir) -> None: diff --git a/tests/unit/python/global_var/test_global_var_importable.py b/tests/unit/python/global_var/test_global_var_importable.py index 9f9b3db9d..4cf6391e2 100644 --- a/tests/unit/python/global_var/test_global_var_importable.py +++ b/tests/unit/python/global_var/test_global_var_importable.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_global_var_dependencies_none(tmpdir) -> None: diff --git a/tests/unit/python/global_var/test_global_var_properties.py b/tests/unit/python/global_var/test_global_var_properties.py index a2f42234a..741978f5f 100644 --- a/tests/unit/python/global_var/test_global_var_properties.py +++ b/tests/unit/python/global_var/test_global_var_properties.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_global_vars(tmpdir) -> None: diff --git a/tests/unit/python/global_var/test_global_var_remove.py b/tests/unit/python/global_var/test_global_var_remove.py index 19b8142f0..6afaa06b0 100644 --- a/tests/unit/python/global_var/test_global_var_remove.py +++ b/tests/unit/python/global_var/test_global_var_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_global_var_remove(tmpdir) -> None: diff --git a/tests/unit/python/global_var/test_global_var_rename.py b/tests/unit/python/global_var/test_global_var_rename.py index 368e91e53..7a3f66955 100644 --- a/tests/unit/python/global_var/test_global_var_rename.py +++ b/tests/unit/python/global_var/test_global_var_rename.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_global_var_rename_updates_usages_in_method(tmpdir) -> None: diff --git a/tests/unit/python/global_var/test_global_var_special.py b/tests/unit/python/global_var/test_global_var_special.py index 24256b3a0..d87211300 100644 --- a/tests/unit/python/global_var/test_global_var_special.py +++ b/tests/unit/python/global_var/test_global_var_special.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_global_var_attribute(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_has_import.py b/tests/unit/python/import_resolution/test_has_import.py index f75cf751c..5770a8a81 100644 --- a/tests/unit/python/import_resolution/test_has_import.py +++ b/tests/unit/python/import_resolution/test_has_import.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_import_rename_usage_with_alias(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_editable.py b/tests/unit/python/import_resolution/test_import_editable.py index e6d2113ba..4b42ef6c6 100644 --- a/tests/unit/python/import_resolution/test_import_editable.py +++ b/tests/unit/python/import_resolution/test_import_editable.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_import_insert_before(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_exportable.py b/tests/unit/python/import_resolution/test_import_exportable.py index bce11a599..18ebce6b1 100644 --- a/tests/unit/python/import_resolution/test_import_exportable.py +++ b/tests/unit/python/import_resolution/test_import_exportable.py @@ -1,6 +1,6 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_import_usages_symbol_redefinition(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_from_import_statements.py b/tests/unit/python/import_resolution/test_import_from_import_statements.py index 03fcf5b55..9bed3265a 100644 --- a/tests/unit/python/import_resolution/test_import_from_import_statements.py +++ b/tests/unit/python/import_resolution/test_import_from_import_statements.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ImportType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ImportType def test_parse_import_statement(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_relative.py b/tests/unit/python/import_resolution/test_import_relative.py index c2284e7a8..6ee0f8069 100644 --- a/tests/unit/python/import_resolution/test_import_relative.py +++ b/tests/unit/python/import_resolution/test_import_relative.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_import_relative(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_remove.py b/tests/unit/python/import_resolution/test_import_remove.py index 2c3f9c3dd..d0f218f60 100644 --- a/tests/unit/python/import_resolution/test_import_remove.py +++ b/tests/unit/python/import_resolution/test_import_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_import_remove_no_usage(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_rename.py b/tests/unit/python/import_resolution/test_import_rename.py index 1599b4ed3..8f99a2d10 100644 --- a/tests/unit/python/import_resolution/test_import_rename.py +++ b/tests/unit/python/import_resolution/test_import_rename.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_import_rename_usage(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_resolution.py b/tests/unit/python/import_resolution/test_import_resolution.py index 84a103c65..4df8fbbb3 100644 --- a/tests/unit/python/import_resolution/test_import_resolution.py +++ b/tests/unit/python/import_resolution/test_import_resolution.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.file import SourceFile -from graph_sitter.core.import_resolution import Import, ImportResolution +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.file import SourceFile +from codegen.sdk.core.import_resolution import Import, ImportResolution def test_import_properties_basic(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_resolved_symbol.py b/tests/unit/python/import_resolution/test_import_resolved_symbol.py index 85a7beb2f..71784a554 100644 --- a/tests/unit/python/import_resolution/test_import_resolved_symbol.py +++ b/tests/unit/python/import_resolution/test_import_resolved_symbol.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.external_module import ExternalModule +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.external_module import ExternalModule def test_import_resolved_symbol(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_set_import_module.py b/tests/unit/python/import_resolution/test_import_set_import_module.py index 79b3d4578..623a2837a 100644 --- a/tests/unit/python/import_resolution/test_import_set_import_module.py +++ b/tests/unit/python/import_resolution/test_import_set_import_module.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_update_import_module(tmpdir) -> None: diff --git a/tests/unit/python/import_resolution/test_import_set_import_symbol_alias.py b/tests/unit/python/import_resolution/test_import_set_import_symbol_alias.py index 39ba990ac..5ad882b79 100644 --- a/tests/unit/python/import_resolution/test_import_set_import_symbol_alias.py +++ b/tests/unit/python/import_resolution/test_import_set_import_symbol_alias.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_import_set_import_symbol_alias_from_no_alias(tmpdir) -> None: diff --git a/tests/unit/python/statements/assignment_statement/test_assignment_statement_remove.py b/tests/unit/python/statements/assignment_statement/test_assignment_statement_remove.py index f4a64d7b0..b84e86c64 100644 --- a/tests/unit/python/statements/assignment_statement/test_assignment_statement_remove.py +++ b/tests/unit/python/statements/assignment_statement/test_assignment_statement_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_assignment_statement_remove(tmpdir) -> None: diff --git a/tests/unit/python/statements/assignment_statement/test_assignment_statement_rename.py b/tests/unit/python/statements/assignment_statement/test_assignment_statement_rename.py index 5fad23ab8..71468f0ca 100644 --- a/tests/unit/python/statements/assignment_statement/test_assignment_statement_rename.py +++ b/tests/unit/python/statements/assignment_statement/test_assignment_statement_rename.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_rename_local_variables_exact_match_no_change(tmpdir) -> None: diff --git a/tests/unit/python/statements/assignment_statement/test_assignment_statement_set_assignment_value.py b/tests/unit/python/statements/assignment_statement/test_assignment_statement_set_assignment_value.py index f6ed9a743..dff5461e8 100644 --- a/tests/unit/python/statements/assignment_statement/test_assignment_statement_set_assignment_value.py +++ b/tests/unit/python/statements/assignment_statement/test_assignment_statement_set_assignment_value.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_assignment_statement_set_assignment_value(tmpdir) -> None: diff --git a/tests/unit/python/statements/attribute/test_attribute_assignment_value.py b/tests/unit/python/statements/attribute/test_attribute_assignment_value.py index f164b838b..d9a7648a0 100644 --- a/tests/unit/python/statements/attribute/test_attribute_assignment_value.py +++ b/tests/unit/python/statements/attribute/test_attribute_assignment_value.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_attribute_set_assignment_value(tmpdir) -> None: diff --git a/tests/unit/python/statements/attribute/test_attribute_get_usages.py b/tests/unit/python/statements/attribute/test_attribute_get_usages.py index f6b1b06e9..c7e7d3c7c 100644 --- a/tests/unit/python/statements/attribute/test_attribute_get_usages.py +++ b/tests/unit/python/statements/attribute/test_attribute_get_usages.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_attribute_get_local_usages(tmpdir) -> None: diff --git a/tests/unit/python/statements/attribute/test_attribute_properties.py b/tests/unit/python/statements/attribute/test_attribute_properties.py index a35895265..01fee6fea 100644 --- a/tests/unit/python/statements/attribute/test_attribute_properties.py +++ b/tests/unit/python/statements/attribute/test_attribute_properties.py @@ -1,8 +1,8 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions.number import Number -from graph_sitter.python.statements.assignment_statement import PyAssignmentStatement -from graph_sitter.python.statements.attribute import PyAttribute +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions.number import Number +from codegen.sdk.python.statements.assignment_statement import PyAssignmentStatement +from codegen.sdk.python.statements.attribute import PyAttribute def test_attribute(tmpdir) -> None: diff --git a/tests/unit/python/statements/attribute/test_attribute_remove.py b/tests/unit/python/statements/attribute/test_attribute_remove.py index 86551a80a..0c1317246 100644 --- a/tests/unit/python/statements/attribute/test_attribute_remove.py +++ b/tests/unit/python/statements/attribute/test_attribute_remove.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_attribute_remove(tmpdir) -> None: diff --git a/tests/unit/python/statements/attribute/test_attribute_rename.py b/tests/unit/python/statements/attribute/test_attribute_rename.py index b9ffb77a6..53b4c37ec 100644 --- a/tests/unit/python/statements/attribute/test_attribute_rename.py +++ b/tests/unit/python/statements/attribute/test_attribute_rename.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_attribute_rename(tmpdir) -> None: diff --git a/tests/unit/python/statements/attribute/test_attribute_set_type_annotation.py b/tests/unit/python/statements/attribute/test_attribute_set_type_annotation.py index 49cf082d8..393b2a374 100644 --- a/tests/unit/python/statements/attribute/test_attribute_set_type_annotation.py +++ b/tests/unit/python/statements/attribute/test_attribute_set_type_annotation.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_attribute_set_type_annotation_from_none(tmpdir) -> None: diff --git a/tests/unit/python/statements/for_loop_statement/test_for_loop_statement.py b/tests/unit/python/statements/for_loop_statement/test_for_loop_statement.py index 250260012..88aa5f28c 100644 --- a/tests/unit/python/statements/for_loop_statement/test_for_loop_statement.py +++ b/tests/unit/python/statements/for_loop_statement/test_for_loop_statement.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_for_loop_statement_parse(tmpdir) -> None: diff --git a/tests/unit/python/statements/if_block_statement/test_if_block_reduce_block.py b/tests/unit/python/statements/if_block_statement/test_if_block_reduce_block.py index 7a7bd1d45..7dd308d41 100644 --- a/tests/unit/python/statements/if_block_statement/test_if_block_reduce_block.py +++ b/tests/unit/python/statements/if_block_statement/test_if_block_reduce_block.py @@ -1,8 +1,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.python import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.python import PyFile def test_reduce_condition_to_true_elif(tmpdir): diff --git a/tests/unit/python/statements/if_block_statement/test_if_block_statement_properties.py b/tests/unit/python/statements/if_block_statement/test_if_block_statement_properties.py index 0c10379e5..1cc7170c3 100644 --- a/tests/unit/python/statements/if_block_statement/test_if_block_statement_properties.py +++ b/tests/unit/python/statements/if_block_statement/test_if_block_statement_properties.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.python import PyFile -from graph_sitter.python.statements.if_block_statement import PyIfBlockStatement +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.python import PyFile +from codegen.sdk.python.statements.if_block_statement import PyIfBlockStatement def test_parse_conditional_statement_from_codeblock(tmpdir) -> None: diff --git a/tests/unit/python/statements/import_statement/test_import_statement.py b/tests/unit/python/statements/import_statement/test_import_statement.py index 2a409a199..86e3f64e5 100644 --- a/tests/unit/python/statements/import_statement/test_import_statement.py +++ b/tests/unit/python/statements/import_statement/test_import_statement.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_import_statements(tmpdir) -> None: diff --git a/tests/unit/python/statements/match_statement/test_try_catch_statement.py b/tests/unit/python/statements/match_statement/test_try_catch_statement.py index 999fd14b6..0a1928551 100644 --- a/tests/unit/python/statements/match_statement/test_try_catch_statement.py +++ b/tests/unit/python/statements/match_statement/test_try_catch_statement.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.try_catch_statement import TryCatchStatement +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.try_catch_statement import TryCatchStatement def test_try_except_statement_parse(tmpdir) -> None: diff --git a/tests/unit/python/statements/try_catch_statement/test_match_statement.py b/tests/unit/python/statements/try_catch_statement/test_match_statement.py index 6d4fc425b..3a07f1d81 100644 --- a/tests/unit/python/statements/try_catch_statement/test_match_statement.py +++ b/tests/unit/python/statements/try_catch_statement/test_match_statement.py @@ -1,4 +1,4 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session +from codegen.sdk.codebase.factory.get_session import get_codebase_session def test_match_switch_statement_parse(tmpdir) -> None: diff --git a/tests/unit/python/statements/while_statement/test_while_statement.py b/tests/unit/python/statements/while_statement/test_while_statement.py index 655f24aa3..73b21348a 100644 --- a/tests/unit/python/statements/while_statement/test_while_statement.py +++ b/tests/unit/python/statements/while_statement/test_while_statement.py @@ -1,9 +1,9 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.boolean import Boolean -from graph_sitter.core.expressions.comparison_expression import ComparisonExpression -from graph_sitter.core.statements.if_block_statement import IfBlockStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.core.statements.while_statement import WhileStatement +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.boolean import Boolean +from codegen.sdk.core.expressions.comparison_expression import ComparisonExpression +from codegen.sdk.core.statements.if_block_statement import IfBlockStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.core.statements.while_statement import WhileStatement def test_while_statement_parse(tmpdir) -> None: diff --git a/tests/unit/python/statements/with_statement/test_with_statement_properties.py b/tests/unit/python/statements/with_statement/test_with_statement_properties.py index 9c201ca66..61c8a68ca 100644 --- a/tests/unit/python/statements/with_statement/test_with_statement_properties.py +++ b/tests/unit/python/statements/with_statement/test_with_statement_properties.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions import Name, Value -from graph_sitter.python import PyFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions import Name, Value +from codegen.sdk.python import PyFile def test_with_statement_single_var(tmpdir): diff --git a/tests/unit/python/test_print.py b/tests/unit/python/test_print.py index c1fbc4b16..715f5fc52 100644 --- a/tests/unit/python/test_print.py +++ b/tests/unit/python/test_print.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.class_definition import Class +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.class_definition import Class def test_import_rename_usage_with_alias(tmpdir) -> None: diff --git a/tests/unit/skills/test_skills.py b/tests/unit/skills/test_skills.py index 28edf537c..611d3fb75 100644 --- a/tests/unit/skills/test_skills.py +++ b/tests/unit/skills/test_skills.py @@ -1,11 +1,11 @@ import pytest +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.skills.core.skill_implementation import SkillImplementation +from codegen.sdk.skills.core.skill_test import SkillTestCase +from codegen.sdk.skills.core.utils import get_all_skill_implementations +from codegen.sdk.skills.utils.utils import verify_skill_output from codegen_git.schemas.repo_config import BaseRepoConfig -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.skills.core.skill_implementation import SkillImplementation -from graph_sitter.skills.core.skill_test import SkillTestCase -from graph_sitter.skills.core.utils import get_all_skill_implementations -from graph_sitter.skills.utils.utils import verify_skill_output skill_implementations = get_all_skill_implementations() cases = [(skill, test_case, idx) for skill in skill_implementations for idx, test_case in enumerate(skill.test_cases) if not skill.external] diff --git a/tests/unit/test_codemod.py b/tests/unit/test_codemod.py index a78261322..0d42b51da 100644 --- a/tests/unit/test_codemod.py +++ b/tests/unit/test_codemod.py @@ -1,5 +1,5 @@ -from graph_sitter.codemod import Codemod3 -from graph_sitter.core.codebase import Codebase +from codegen.sdk.codemod import Codemod3 +from codegen.sdk.core.codebase import Codebase def test_codemod_execute(): diff --git a/tests/unit/typescript/autocommit/test_autocommit.py b/tests/unit/typescript/autocommit/test_autocommit.py index b44756fcd..3dc575e8d 100644 --- a/tests/unit/typescript/autocommit/test_autocommit.py +++ b/tests/unit/typescript/autocommit/test_autocommit.py @@ -1,7 +1,10 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +import pytest +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage + +@pytest.mark.skip("No Autocommit") def test_autocommit_move_rename(tmpdir) -> None: file1_name = "file1.ts" # language=typescript diff --git a/tests/unit/typescript/class_definition/test_abstract_class.py b/tests/unit/typescript/class_definition/test_abstract_class.py index 4fab57757..fb01fb24c 100644 --- a/tests/unit/typescript/class_definition/test_abstract_class.py +++ b/tests/unit/typescript/class_definition/test_abstract_class.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_abstract_class_inheritance(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_add_attribute.py b/tests/unit/typescript/class_definition/test_class_add_attribute.py index d60df857d..5862eb858 100644 --- a/tests/unit/typescript/class_definition/test_class_add_attribute.py +++ b/tests/unit/typescript/class_definition/test_class_add_attribute.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_add_attribute_from_source(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_add_decorator.py b/tests/unit/typescript/class_definition/test_class_add_decorator.py index c02defb95..ede79ac5c 100644 --- a/tests/unit/typescript/class_definition/test_class_add_decorator.py +++ b/tests/unit/typescript/class_definition/test_class_add_decorator.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_add_decorator_to_class(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_add_method.py b/tests/unit/typescript/class_definition/test_class_add_method.py index 2991e1391..6980b817a 100644 --- a/tests/unit/typescript/class_definition/test_class_add_method.py +++ b/tests/unit/typescript/class_definition/test_class_add_method.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.class_definition import TSClass -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.class_definition import TSClass +from codegen.sdk.typescript.file import TSFile def test_add_method_basic(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_attributes.py b/tests/unit/typescript/class_definition/test_class_attributes.py index ea6f3cb5c..9db3c429c 100644 --- a/tests/unit/typescript/class_definition/test_class_attributes.py +++ b/tests/unit/typescript/class_definition/test_class_attributes.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_attributes_finds_correct_number(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_code_block.py b/tests/unit/typescript/class_definition/test_class_code_block.py index d6547f3af..4b029e67c 100644 --- a/tests/unit/typescript/class_definition/test_class_code_block.py +++ b/tests/unit/typescript/class_definition/test_class_code_block.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_code_block_returns_non_empty(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_comment.py b/tests/unit/typescript/class_definition/test_class_comment.py index 5cf2fb800..fdae5e973 100644 --- a/tests/unit/typescript/class_definition/test_class_comment.py +++ b/tests/unit/typescript/class_definition/test_class_comment.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_comment_single_line(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_constructor_usages.py b/tests/unit/typescript/class_definition/test_class_constructor_usages.py index d4562702e..53a555113 100644 --- a/tests/unit/typescript/class_definition/test_class_constructor_usages.py +++ b/tests/unit/typescript/class_definition/test_class_constructor_usages.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.class_definition import Class -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.class_definition import Class +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_parent_class_names_single(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_definition_is_jsx.py b/tests/unit/typescript/class_definition/test_class_definition_is_jsx.py index 02ee1a2b4..b878f69c8 100644 --- a/tests/unit/typescript/class_definition/test_class_definition_is_jsx.py +++ b/tests/unit/typescript/class_definition/test_class_definition_is_jsx.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_without_parents(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_dependencies.py b/tests/unit/typescript/class_definition/test_class_dependencies.py index 52cd996a5..fb6cc49e8 100644 --- a/tests/unit/typescript/class_definition/test_class_dependencies.py +++ b/tests/unit/typescript/class_definition/test_class_dependencies.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage @pytest.mark.skip(reason="CG-9539 TODO @caroljung") diff --git a/tests/unit/typescript/class_definition/test_class_filter.py b/tests/unit/typescript/class_definition/test_class_filter.py index 7e81c6e94..d90c5ac08 100644 --- a/tests/unit/typescript/class_definition/test_class_filter.py +++ b/tests/unit/typescript/class_definition/test_class_filter.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_filter(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_is_exported.py b/tests/unit/typescript/class_definition/test_class_is_exported.py index 7b8db204d..bd2e1eee7 100644 --- a/tests/unit/typescript/class_definition/test_class_is_exported.py +++ b/tests/unit/typescript/class_definition/test_class_is_exported.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_export_function_is_exported(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_method_usages.py b/tests/unit/typescript/class_definition/test_class_method_usages.py index b914c4550..979e98c28 100644 --- a/tests/unit/typescript/class_definition/test_class_method_usages.py +++ b/tests/unit/typescript/class_definition/test_class_method_usages.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_parent_class_names_single(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_parameters.py b/tests/unit/typescript/class_definition/test_class_parameters.py index 88a188440..7f841fba9 100644 --- a/tests/unit/typescript/class_definition/test_class_parameters.py +++ b/tests/unit/typescript/class_definition/test_class_parameters.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_parameters_gets_non_empty(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_parent_class_names.py b/tests/unit/typescript/class_definition/test_class_parent_class_names.py index 0d00b3331..b3b57f83d 100644 --- a/tests/unit/typescript/class_definition/test_class_parent_class_names.py +++ b/tests/unit/typescript/class_definition/test_class_parent_class_names.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.interfaces.editable import Editable -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.interfaces.editable import Editable +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_parent_class_names_single(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_parent_classes.py b/tests/unit/typescript/class_definition/test_class_parent_classes.py index 1afadcc16..bcb39521c 100644 --- a/tests/unit/typescript/class_definition/test_class_parent_classes.py +++ b/tests/unit/typescript/class_definition/test_class_parent_classes.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_parent_classes_single(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_rename.py b/tests/unit/typescript/class_definition/test_class_rename.py index 6183f54b9..54fc0478c 100644 --- a/tests/unit/typescript/class_definition/test_class_rename.py +++ b/tests/unit/typescript/class_definition/test_class_rename.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_rename(tmpdir) -> None: diff --git a/tests/unit/typescript/class_definition/test_class_rename_class_attributes.py b/tests/unit/typescript/class_definition/test_class_rename_class_attributes.py index f50b942fb..54a9a33e1 100644 --- a/tests/unit/typescript/class_definition/test_class_rename_class_attributes.py +++ b/tests/unit/typescript/class_definition/test_class_rename_class_attributes.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_class_definition_rename_class_attribute_throws(tmpdir) -> None: diff --git a/tests/unit/typescript/codebase/test_codebase_resolution.py b/tests/unit/typescript/codebase/test_codebase_resolution.py index 6b45102c5..196f43c15 100644 --- a/tests/unit/typescript/codebase/test_codebase_resolution.py +++ b/tests/unit/typescript/codebase/test_codebase_resolution.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_codebase_resolution(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_comment.py b/tests/unit/typescript/comment_docstring/test_comment.py index 185e5a050..736c93617 100644 --- a/tests/unit/typescript/comment_docstring/test_comment.py +++ b/tests/unit/typescript/comment_docstring/test_comment.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_comment_basic(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_comment_edit.py b/tests/unit/typescript/comment_docstring/test_comment_edit.py index b533bac35..2d3fe535a 100644 --- a/tests/unit/typescript/comment_docstring/test_comment_edit.py +++ b/tests/unit/typescript/comment_docstring/test_comment_edit.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_comment_basic(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_comment_insert_add.py b/tests/unit/typescript/comment_docstring/test_comment_insert_add.py index 2e34e6e89..9dd668ad6 100644 --- a/tests/unit/typescript/comment_docstring/test_comment_insert_add.py +++ b/tests/unit/typescript/comment_docstring/test_comment_insert_add.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_set_comment(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_docstring.py b/tests/unit/typescript/comment_docstring/test_docstring.py index c7dec2986..8801b664b 100644 --- a/tests/unit/typescript/comment_docstring/test_docstring.py +++ b/tests/unit/typescript/comment_docstring/test_docstring.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_docstring_no_docstring(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_docstring_edit.py b/tests/unit/typescript/comment_docstring/test_docstring_edit.py index e9bf29e44..cbc4f3d25 100644 --- a/tests/unit/typescript/comment_docstring/test_docstring_edit.py +++ b/tests/unit/typescript/comment_docstring/test_docstring_edit.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_docstring_block(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_docstring_insert_add.py b/tests/unit/typescript/comment_docstring/test_docstring_insert_add.py index 718e51ba5..06789354d 100644 --- a/tests/unit/typescript/comment_docstring/test_docstring_insert_add.py +++ b/tests/unit/typescript/comment_docstring/test_docstring_insert_add.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_set_docstring(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_inline_comment.py b/tests/unit/typescript/comment_docstring/test_inline_comment.py index aff42f405..d40a63259 100644 --- a/tests/unit/typescript/comment_docstring/test_inline_comment.py +++ b/tests/unit/typescript/comment_docstring/test_inline_comment.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_comment_inline(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_inline_comment_edit.py b/tests/unit/typescript/comment_docstring/test_inline_comment_edit.py index b6f35e445..6e3934a8f 100644 --- a/tests/unit/typescript/comment_docstring/test_inline_comment_edit.py +++ b/tests/unit/typescript/comment_docstring/test_inline_comment_edit.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_comment_inline(tmpdir) -> None: diff --git a/tests/unit/typescript/comment_docstring/test_inline_comment_insert_add.py b/tests/unit/typescript/comment_docstring/test_inline_comment_insert_add.py index 38b7bdc2c..dcf5e076c 100644 --- a/tests/unit/typescript/comment_docstring/test_inline_comment_insert_add.py +++ b/tests/unit/typescript/comment_docstring/test_inline_comment_insert_add.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_set_comment_inline(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/argument/test_argument_edit.py b/tests/unit/typescript/detached_symbols/argument/test_argument_edit.py index 6fd633535..b936fff08 100644 --- a/tests/unit/typescript/detached_symbols/argument/test_argument_edit.py +++ b/tests/unit/typescript/detached_symbols/argument/test_argument_edit.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.file import SourceFile -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.file import SourceFile +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_edit_arg_in_function_call(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/argument/test_fetch_parametes.py b/tests/unit/typescript/detached_symbols/argument/test_fetch_parametes.py index 19890f315..7dc4d3997 100644 --- a/tests/unit/typescript/detached_symbols/argument/test_fetch_parametes.py +++ b/tests/unit/typescript/detached_symbols/argument/test_fetch_parametes.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_destructured_parameters(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_local_var_assignment.py b/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_local_var_assignment.py index bb9bca420..346caa7d0 100644 --- a/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_local_var_assignment.py +++ b/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_local_var_assignment.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_get_local_var_assignment_primitive(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_local_var_assignments.py b/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_local_var_assignments.py index ece377a4e..0b7fb785b 100644 --- a/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_local_var_assignments.py +++ b/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_local_var_assignments.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_get_and_rename_get_local_var_assignments(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_variable_usages.py b/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_variable_usages.py index d705d9736..09884d461 100644 --- a/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_variable_usages.py +++ b/tests/unit/typescript/detached_symbols/code_block/test_code_block_get_variable_usages.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_get_and_edit_variable_usages(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/code_block/test_code_block_indent.py b/tests/unit/typescript/detached_symbols/code_block/test_code_block_indent.py index 4cf0f87da..35e7622be 100644 --- a/tests/unit/typescript/detached_symbols/code_block/test_code_block_indent.py +++ b/tests/unit/typescript/detached_symbols/code_block/test_code_block_indent.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage def test_indent_once_to_left(tmpdir): diff --git a/tests/unit/typescript/detached_symbols/code_block/test_code_block_local_var_assignments.py b/tests/unit/typescript/detached_symbols/code_block/test_code_block_local_var_assignments.py index 70a908ed5..2822a25d2 100644 --- a/tests/unit/typescript/detached_symbols/code_block/test_code_block_local_var_assignments.py +++ b/tests/unit/typescript/detached_symbols/code_block/test_code_block_local_var_assignments.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_get_local_var_assignments_try_catch(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/code_block/test_code_block_unwrap.py b/tests/unit/typescript/detached_symbols/code_block/test_code_block_unwrap.py index 198f6c533..7ed1f1de6 100644 --- a/tests/unit/typescript/detached_symbols/code_block/test_code_block_unwrap.py +++ b/tests/unit/typescript/detached_symbols/code_block/test_code_block_unwrap.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_unwrap_flat_statements(tmpdir): diff --git a/tests/unit/typescript/detached_symbols/code_block/test_code_block_wrap.py b/tests/unit/typescript/detached_symbols/code_block/test_code_block_wrap.py index e325dd09b..0a52dece6 100644 --- a/tests/unit/typescript/detached_symbols/code_block/test_code_block_wrap.py +++ b/tests/unit/typescript/detached_symbols/code_block/test_code_block_wrap.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_wrap_with_if_statement(tmpdir): diff --git a/tests/unit/typescript/detached_symbols/decorator/test_decorator_call.py b/tests/unit/typescript/detached_symbols/decorator/test_decorator_call.py index 1f09d7b92..be1dd82f8 100644 --- a/tests/unit/typescript/detached_symbols/decorator/test_decorator_call.py +++ b/tests/unit/typescript/detached_symbols/decorator/test_decorator_call.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_decorator_call_should_find_expression(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/decorator/test_decorator_name.py b/tests/unit/typescript/detached_symbols/decorator/test_decorator_name.py index d062e4cda..2f448e855 100644 --- a/tests/unit/typescript/detached_symbols/decorator/test_decorator_name.py +++ b/tests/unit/typescript/detached_symbols/decorator/test_decorator_name.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_decorator_name_returns_source(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/function_call/test_function_call_extended_source.py b/tests/unit/typescript/detached_symbols/function_call/test_function_call_extended_source.py index d6d6f4d30..9b27c5703 100644 --- a/tests/unit/typescript/detached_symbols/function_call/test_function_call_extended_source.py +++ b/tests/unit/typescript/detached_symbols/function_call/test_function_call_extended_source.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_call_extended_source_in_chain(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/function_call/test_function_call_get_function_definition.py b/tests/unit/typescript/detached_symbols/function_call/test_function_call_get_function_definition.py index 092ec619e..be2ef6505 100644 --- a/tests/unit/typescript/detached_symbols/function_call/test_function_call_get_function_definition.py +++ b/tests/unit/typescript/detached_symbols/function_call/test_function_call_get_function_definition.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage # ED_TODO: Requires fix to get_function_definition. Check if that is possible diff --git a/tests/unit/typescript/detached_symbols/function_call/test_function_call_is_awaited.py b/tests/unit/typescript/detached_symbols/function_call/test_function_call_is_awaited.py index ca43c7d19..8ee567738 100644 --- a/tests/unit/typescript/detached_symbols/function_call/test_function_call_is_awaited.py +++ b/tests/unit/typescript/detached_symbols/function_call/test_function_call_is_awaited.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_call_is_awaited_basic(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/function_call/test_function_call_predecessor.py b/tests/unit/typescript/detached_symbols/function_call/test_function_call_predecessor.py index 6bb6db4bb..8181f9a72 100644 --- a/tests/unit/typescript/detached_symbols/function_call/test_function_call_predecessor.py +++ b/tests/unit/typescript/detached_symbols/function_call/test_function_call_predecessor.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_call_predecessor_single_line(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/function_call/test_function_call_rename.py b/tests/unit/typescript/detached_symbols/function_call/test_function_call_rename.py index ba090a907..77c5fe4b1 100644 --- a/tests/unit/typescript/detached_symbols/function_call/test_function_call_rename.py +++ b/tests/unit/typescript/detached_symbols/function_call/test_function_call_rename.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_call_rename_updates_source(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/function_call/test_function_call_type_resolution.py b/tests/unit/typescript/detached_symbols/function_call/test_function_call_type_resolution.py index 86c99cd8e..f6b51998a 100644 --- a/tests/unit/typescript/detached_symbols/function_call/test_function_call_type_resolution.py +++ b/tests/unit/typescript/detached_symbols/function_call/test_function_call_type_resolution.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_call_name_resolution_unary_expression(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/parameter/test_parameter_deconstructed.py b/tests/unit/typescript/detached_symbols/parameter/test_parameter_deconstructed.py index e204ca001..ec6f20a9e 100644 --- a/tests/unit/typescript/detached_symbols/parameter/test_parameter_deconstructed.py +++ b/tests/unit/typescript/detached_symbols/parameter/test_parameter_deconstructed.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile -from graph_sitter.typescript.function import TSFunction +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile +from codegen.sdk.typescript.function import TSFunction def test_edit_parameter_in_function_definition(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/parameter/test_parameter_edit.py b/tests/unit/typescript/detached_symbols/parameter/test_parameter_edit.py index 41317e7d0..bf9cf0323 100644 --- a/tests/unit/typescript/detached_symbols/parameter/test_parameter_edit.py +++ b/tests/unit/typescript/detached_symbols/parameter/test_parameter_edit.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_edit_parameter_in_function_definition(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/parameter/test_parameter_is_optional.py b/tests/unit/typescript/detached_symbols/parameter/test_parameter_is_optional.py index f6d8b1684..2d7171df7 100644 --- a/tests/unit/typescript/detached_symbols/parameter/test_parameter_is_optional.py +++ b/tests/unit/typescript/detached_symbols/parameter/test_parameter_is_optional.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_parameter_is_optional_should_return_true(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/parameter/test_parameter_is_variadic.py b/tests/unit/typescript/detached_symbols/parameter/test_parameter_is_variadic.py index 0203579fb..757e0ef56 100644 --- a/tests/unit/typescript/detached_symbols/parameter/test_parameter_is_variadic.py +++ b/tests/unit/typescript/detached_symbols/parameter/test_parameter_is_variadic.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_parameter_is_variadic_should_return_true(tmpdir) -> None: diff --git a/tests/unit/typescript/detached_symbols/parameter/test_parameter_value.py b/tests/unit/typescript/detached_symbols/parameter/test_parameter_value.py index 6ef40a470..142f60cc6 100644 --- a/tests/unit/typescript/detached_symbols/parameter/test_parameter_value.py +++ b/tests/unit/typescript/detached_symbols/parameter/test_parameter_value.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_parameter_default_value(tmpdir) -> None: diff --git a/tests/unit/typescript/export/test_export_aliased.py b/tests/unit/typescript/export/test_export_aliased.py index e56960d1f..abee95928 100644 --- a/tests/unit/typescript/export/test_export_aliased.py +++ b/tests/unit/typescript/export/test_export_aliased.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_export_aliased(tmpdir): diff --git a/tests/unit/typescript/export/test_export_export_statement.py b/tests/unit/typescript/export/test_export_export_statement.py index 6aa1529f7..05329e4b4 100644 --- a/tests/unit/typescript/export/test_export_export_statement.py +++ b/tests/unit/typescript/export/test_export_export_statement.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_export_export_statement(tmpdir): diff --git a/tests/unit/typescript/export/test_export_get_import_string.py b/tests/unit/typescript/export/test_export_get_import_string.py index c009726fd..c9b1e8fce 100644 --- a/tests/unit/typescript/export/test_export_get_import_string.py +++ b/tests/unit/typescript/export/test_export_get_import_string.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ImportType, ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ImportType, ProgrammingLanguage def test_export_get_import_string(tmpdir) -> None: diff --git a/tests/unit/typescript/export/test_export_get_reexport_source.py b/tests/unit/typescript/export/test_export_get_reexport_source.py index 06925eb2e..a520f0e1f 100644 --- a/tests/unit/typescript/export/test_export_get_reexport_source.py +++ b/tests/unit/typescript/export/test_export_get_reexport_source.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_reexport_symbol(tmpdir) -> None: diff --git a/tests/unit/typescript/export/test_export_rename.py b/tests/unit/typescript/export/test_export_rename.py index 227a7f292..7ac15a18b 100644 --- a/tests/unit/typescript/export/test_export_rename.py +++ b/tests/unit/typescript/export/test_export_rename.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_rename_export_simple(tmpdir) -> None: diff --git a/tests/unit/typescript/export/test_export_resolve_export.py b/tests/unit/typescript/export/test_export_resolve_export.py index ee2279d39..ca00a0dda 100644 --- a/tests/unit/typescript/export/test_export_resolve_export.py +++ b/tests/unit/typescript/export/test_export_resolve_export.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_resolve_export_with_declared_symbols(tmpdir) -> None: diff --git a/tests/unit/typescript/export/test_exportable_usages.py b/tests/unit/typescript/export/test_exportable_usages.py index a5b269ae3..8543710c1 100644 --- a/tests/unit/typescript/export/test_exportable_usages.py +++ b/tests/unit/typescript/export/test_exportable_usages.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.enums import ImportType, ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.enums import ImportType, ProgrammingLanguage def test_exportable_usage_symbol_export(tmpdir) -> None: diff --git a/tests/unit/typescript/export/test_external_export.py b/tests/unit/typescript/export/test_external_export.py index ab0645332..ed81ffb1d 100644 --- a/tests/unit/typescript/export/test_external_export.py +++ b/tests/unit/typescript/export/test_external_export.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_is_external_export_true(tmpdir): diff --git a/tests/unit/typescript/export/test_get_reexported_exports.py b/tests/unit/typescript/export/test_get_reexported_exports.py index 55adaa187..1363b8aac 100644 --- a/tests/unit/typescript/export/test_get_reexported_exports.py +++ b/tests/unit/typescript/export/test_get_reexported_exports.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_get_reexported_exports(tmpdir): diff --git a/tests/unit/typescript/expressions/ternary_expression/test_ternary_properties.py b/tests/unit/typescript/expressions/ternary_expression/test_ternary_properties.py index de7f888a8..78d8b8207 100644 --- a/tests/unit/typescript/expressions/ternary_expression/test_ternary_properties.py +++ b/tests/unit/typescript/expressions/ternary_expression/test_ternary_properties.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_parse_simple_ternary_expression(tmpdir) -> None: diff --git a/tests/unit/typescript/expressions/ternary_expression/test_ternary_reduce_condition.py b/tests/unit/typescript/expressions/ternary_expression/test_ternary_reduce_condition.py index 27b3c88f7..1df81e572 100644 --- a/tests/unit/typescript/expressions/ternary_expression/test_ternary_reduce_condition.py +++ b/tests/unit/typescript/expressions/ternary_expression/test_ternary_reduce_condition.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_reduce_ternary_condition_to_true(tmpdir): diff --git a/tests/unit/typescript/expressions/test_binary_expression.py b/tests/unit/typescript/expressions/test_binary_expression.py index 3a275d002..86371ec1e 100644 --- a/tests/unit/typescript/expressions/test_binary_expression.py +++ b/tests/unit/typescript/expressions/test_binary_expression.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_binary_expressions(tmpdir): diff --git a/tests/unit/typescript/expressions/test_binary_reduce_condition.py b/tests/unit/typescript/expressions/test_binary_reduce_condition.py index 8700b31c5..15a55d22a 100644 --- a/tests/unit/typescript/expressions/test_binary_reduce_condition.py +++ b/tests/unit/typescript/expressions/test_binary_reduce_condition.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_reduce_binary_simple(tmpdir): diff --git a/tests/unit/typescript/expressions/test_boolean.py b/tests/unit/typescript/expressions/test_boolean.py index 22798dc50..7ce20bc67 100644 --- a/tests/unit/typescript/expressions/test_boolean.py +++ b/tests/unit/typescript/expressions/test_boolean.py @@ -1,8 +1,8 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.expressions.boolean import Boolean -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.assignment import TSAssignment +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.expressions.boolean import Boolean +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.assignment import TSAssignment def test_boolean_parse(tmpdir): diff --git a/tests/unit/typescript/expressions/test_dict.py b/tests/unit/typescript/expressions/test_dict.py index ad849736f..4478340dc 100644 --- a/tests/unit/typescript/expressions/test_dict.py +++ b/tests/unit/typescript/expressions/test_dict.py @@ -2,11 +2,11 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.unpack import Unpack -from graph_sitter.core.symbol_groups.dict import Dict -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.function import TSFunction +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.unpack import Unpack +from codegen.sdk.core.symbol_groups.dict import Dict +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.function import TSFunction def test_dict_basic(tmpdir): diff --git a/tests/unit/typescript/expressions/test_expression.py b/tests/unit/typescript/expressions/test_expression.py index 687c9826c..3e66e8711 100644 --- a/tests/unit/typescript/expressions/test_expression.py +++ b/tests/unit/typescript/expressions/test_expression.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_calls_from_function_call(tmpdir): diff --git a/tests/unit/typescript/expressions/test_list.py b/tests/unit/typescript/expressions/test_list.py index ed40c49af..2b4085158 100644 --- a/tests/unit/typescript/expressions/test_list.py +++ b/tests/unit/typescript/expressions/test_list.py @@ -2,9 +2,9 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.symbol_groups.list import List -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.symbol_groups.list import List +from codegen.sdk.enums import ProgrammingLanguage def test_list_basic(tmpdir) -> None: diff --git a/tests/unit/typescript/expressions/test_number.py b/tests/unit/typescript/expressions/test_number.py index f3e33ff30..a191ff95b 100644 --- a/tests/unit/typescript/expressions/test_number.py +++ b/tests/unit/typescript/expressions/test_number.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.expressions.number import Number -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.expressions.number import Number +from codegen.sdk.enums import ProgrammingLanguage def test_number_parse(tmpdir): diff --git a/tests/unit/typescript/expressions/test_string.py b/tests/unit/typescript/expressions/test_string.py index 6e942bc50..c6237c0a8 100644 --- a/tests/unit/typescript/expressions/test_string.py +++ b/tests/unit/typescript/expressions/test_string.py @@ -1,8 +1,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions import String -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions import String +from codegen.sdk.enums import ProgrammingLanguage def test_string_edit(tmpdir) -> None: diff --git a/tests/unit/typescript/expressions/test_type.py b/tests/unit/typescript/expressions/test_type.py index 0e2a00154..84fb2fc55 100644 --- a/tests/unit/typescript/expressions/test_type.py +++ b/tests/unit/typescript/expressions/test_type.py @@ -1,8 +1,8 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.core.expressions import String -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.expressions.undefined_type import TSUndefinedType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.core.expressions import String +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.expressions.undefined_type import TSUndefinedType def test_type_basic(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_add_symbol.py b/tests/unit/typescript/file/test_file_add_symbol.py index 1d46535c2..6d14d8f86 100644 --- a/tests/unit/typescript/file/test_file_add_symbol.py +++ b/tests/unit/typescript/file/test_file_add_symbol.py @@ -1,7 +1,7 @@ import os -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_add_symbol_to_file_with_comments(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_add_symbol_from_source.py b/tests/unit/typescript/file/test_file_add_symbol_from_source.py index 82dfa61e8..2c9367fa5 100644 --- a/tests/unit/typescript/file/test_file_add_symbol_from_source.py +++ b/tests/unit/typescript/file/test_file_add_symbol_from_source.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.assignment import Assignment -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.function import TSFunction +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.function import TSFunction def test_add_symbol_from_source(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_add_symbol_import.py b/tests/unit/typescript/file/test_file_add_symbol_import.py index f35cf63e8..21532a7c8 100644 --- a/tests/unit/typescript/file/test_file_add_symbol_import.py +++ b/tests/unit/typescript/file/test_file_add_symbol_import.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_file_add_symbol_import_updates_source(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_editable.py b/tests/unit/typescript/file/test_file_editable.py index afad9cf34..bba435880 100644 --- a/tests/unit/typescript/file/test_file_editable.py +++ b/tests/unit/typescript/file/test_file_editable.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_add_symbol_arrow_function(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_export_statements.py b/tests/unit/typescript/file/test_file_export_statements.py index 9150b6e7f..adaa00922 100644 --- a/tests/unit/typescript/file/test_file_export_statements.py +++ b/tests/unit/typescript/file/test_file_export_statements.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage @pytest.fixture diff --git a/tests/unit/typescript/file/test_file_get_config.py b/tests/unit/typescript/file/test_file_get_config.py index 91dbd8675..8801a50c3 100644 --- a/tests/unit/typescript/file/test_file_get_config.py +++ b/tests/unit/typescript/file/test_file_get_config.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_file_get_config(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_get_function.py b/tests/unit/typescript/file/test_file_get_function.py index 13f38e5b5..c339efab0 100644 --- a/tests/unit/typescript/file/test_file_get_function.py +++ b/tests/unit/typescript/file/test_file_get_function.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_file_get_function_async_star_function(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_get_function_calls.py b/tests/unit/typescript/file/test_file_get_function_calls.py index 8b88ba9ed..7ed5e3780 100644 --- a/tests/unit/typescript/file/test_file_get_function_calls.py +++ b/tests/unit/typescript/file/test_file_get_function_calls.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_file_get_function_calls(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_get_global_var.py b/tests/unit/typescript/file/test_file_get_global_var.py index d1a790c53..893d21f39 100644 --- a/tests/unit/typescript/file/test_file_get_global_var.py +++ b/tests/unit/typescript/file/test_file_get_global_var.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_get_global_var_excludes_component_var(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_get_import_string.py b/tests/unit/typescript/file/test_file_get_import_string.py index d1442b6bb..7b36d1e38 100644 --- a/tests/unit/typescript/file/test_file_get_import_string.py +++ b/tests/unit/typescript/file/test_file_get_import_string.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_file_get_import_string_no_params(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_get_symbol.py b/tests/unit/typescript/file/test_file_get_symbol.py index 3d9263a94..4cee2819f 100644 --- a/tests/unit/typescript/file/test_file_get_symbol.py +++ b/tests/unit/typescript/file/test_file_get_symbol.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_get_symbol_in_tsx(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_global_call.py b/tests/unit/typescript/file/test_file_global_call.py index f454ea466..6ce4a8d20 100644 --- a/tests/unit/typescript/file/test_file_global_call.py +++ b/tests/unit/typescript/file/test_file_global_call.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_file_global_call(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_global_vars.py b/tests/unit/typescript/file/test_file_global_vars.py index 8d306f289..7ee0229ba 100644 --- a/tests/unit/typescript/file/test_file_global_vars.py +++ b/tests/unit/typescript/file/test_file_global_vars.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_global_vars_gets_exported_global_vars(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_graph.py b/tests/unit/typescript/file/test_file_graph.py index e40a449d8..0dd436182 100644 --- a/tests/unit/typescript/file/test_file_graph.py +++ b/tests/unit/typescript/file/test_file_graph.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_graph_has_node_for_existing_file(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_import_statemets.py b/tests/unit/typescript/file/test_file_import_statemets.py index ee77594c9..47819cb42 100644 --- a/tests/unit/typescript/file/test_file_import_statemets.py +++ b/tests/unit/typescript/file/test_file_import_statemets.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_file_import_statements_includes_require_statements(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_imports.py b/tests/unit/typescript/file/test_file_imports.py index 1b2e1fae9..cd0f6a436 100644 --- a/tests/unit/typescript/file/test_file_imports.py +++ b/tests/unit/typescript/file/test_file_imports.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_file_imports_default_import(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_remove.py b/tests/unit/typescript/file/test_file_remove.py index 0ee6e98c3..1b8aaba53 100644 --- a/tests/unit/typescript/file/test_file_remove.py +++ b/tests/unit/typescript/file/test_file_remove.py @@ -1,7 +1,7 @@ import os -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def tets_remove_existing_file(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_statements.py b/tests/unit/typescript/file/test_file_statements.py index fc4b442d0..d7039e8df 100644 --- a/tests/unit/typescript/file/test_file_statements.py +++ b/tests/unit/typescript/file/test_file_statements.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage @pytest.mark.skip("TODO: @caroljung fix once file and symbol code block parse is unified") diff --git a/tests/unit/typescript/file/test_file_symbols.py b/tests/unit/typescript/file/test_file_symbols.py index 09f9ea19b..4e709a2f7 100644 --- a/tests/unit/typescript/file/test_file_symbols.py +++ b/tests/unit/typescript/file/test_file_symbols.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.class_definition import TSClass +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.class_definition import TSClass def test_symbols_gets_class_symbol_with_correct_name(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_unicode.py b/tests/unit/typescript/file/test_file_unicode.py index 9ec0a863c..b3fadee28 100644 --- a/tests/unit/typescript/file/test_file_unicode.py +++ b/tests/unit/typescript/file/test_file_unicode.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_unicode_move_symbol(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_file_update_filepath.py b/tests/unit/typescript/file/test_file_update_filepath.py index 50ecd77e2..3a7bc487c 100644 --- a/tests/unit/typescript/file/test_file_update_filepath.py +++ b/tests/unit/typescript/file/test_file_update_filepath.py @@ -2,8 +2,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_rename_file(tmpdir) -> None: diff --git a/tests/unit/typescript/file/test_ts_config_properties.py b/tests/unit/typescript/file/test_ts_config_properties.py index 8e4889151..a2da90b36 100644 --- a/tests/unit/typescript/file/test_ts_config_properties.py +++ b/tests/unit/typescript/file/test_ts_config_properties.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.ts_config import TSConfig +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.ts_config import TSConfig def test_file_get_config(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_arrow.py b/tests/unit/typescript/function/test_function_arrow.py index d9f216210..4fe2a873b 100644 --- a/tests/unit/typescript/function/test_function_arrow.py +++ b/tests/unit/typescript/function/test_function_arrow.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_function_function_calls_gets_calls_in_return_statement(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_assignments.py b/tests/unit/typescript/function/test_function_assignments.py index 89cb59ba9..1d73ce754 100644 --- a/tests/unit/typescript/function/test_function_assignments.py +++ b/tests/unit/typescript/function/test_function_assignments.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_object_unpack(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_async.py b/tests/unit/typescript/function/test_function_async.py index 3538d2aaa..8383b911b 100644 --- a/tests/unit/typescript/function/test_function_async.py +++ b/tests/unit/typescript/function/test_function_async.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_is_async_basic(tmpdir): diff --git a/tests/unit/typescript/function/test_function_call_sites.py b/tests/unit/typescript/function/test_function_call_sites.py index a3061226a..526c5cd91 100644 --- a/tests/unit/typescript/function/test_function_call_sites.py +++ b/tests/unit/typescript/function/test_function_call_sites.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_call_sites_in_symbols(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_calls_from_symbol.py b/tests/unit/typescript/function/test_function_calls_from_symbol.py index 3f97d80e0..871518f3a 100644 --- a/tests/unit/typescript/function/test_function_calls_from_symbol.py +++ b/tests/unit/typescript/function/test_function_calls_from_symbol.py @@ -1,16 +1,16 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.detached_symbols.argument import Argument -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.expressions.parenthesized_expression import ParenthesizedExpression -from graph_sitter.core.statements.expression_statement import ExpressionStatement -from graph_sitter.core.statements.return_statement import ReturnStatement -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.detached_symbols.decorator import TSDecorator -from graph_sitter.typescript.expressions.chained_attribute import TSChainedAttribute -from graph_sitter.typescript.statements.assignment_statement import TSAssignmentStatement +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.detached_symbols.argument import Argument +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.expressions.parenthesized_expression import ParenthesizedExpression +from codegen.sdk.core.statements.expression_statement import ExpressionStatement +from codegen.sdk.core.statements.return_statement import ReturnStatement +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.detached_symbols.decorator import TSDecorator +from codegen.sdk.typescript.expressions.chained_attribute import TSChainedAttribute +from codegen.sdk.typescript.statements.assignment_statement import TSAssignmentStatement def test_function_calls_from_file(tmpdir): diff --git a/tests/unit/typescript/function/test_function_code_block.py b/tests/unit/typescript/function/test_function_code_block.py index 78473ef1d..6d461faf6 100644 --- a/tests/unit/typescript/function/test_function_code_block.py +++ b/tests/unit/typescript/function/test_function_code_block.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage def test_function_code_block_all_statement_types(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_comment.py b/tests/unit/typescript/function/test_function_comment.py index c2b7a2e37..00405680a 100644 --- a/tests/unit/typescript/function/test_function_comment.py +++ b/tests/unit/typescript/function/test_function_comment.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_comment_single_line(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_decorators.py b/tests/unit/typescript/function/test_function_decorators.py index d5246d52f..afa7a1805 100644 --- a/tests/unit/typescript/function/test_function_decorators.py +++ b/tests/unit/typescript/function/test_function_decorators.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_decorators_get_multiple_decorators(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_dependencies.py b/tests/unit/typescript/function/test_function_dependencies.py index 2ee50f60f..5ed0d750c 100644 --- a/tests/unit/typescript/function/test_function_dependencies.py +++ b/tests/unit/typescript/function/test_function_dependencies.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_functions_dependencies_async_function_finds_deps(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_docstring.py b/tests/unit/typescript/function/test_function_docstring.py index 6165089e8..405bd4f5d 100644 --- a/tests/unit/typescript/function/test_function_docstring.py +++ b/tests/unit/typescript/function/test_function_docstring.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_docstring_multiline(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_function_calls.py b/tests/unit/typescript/function/test_function_function_calls.py index 67edd5814..20d7fa7d5 100644 --- a/tests/unit/typescript/function/test_function_function_calls.py +++ b/tests/unit/typescript/function/test_function_function_calls.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_function_calls_gets_calls_in_return_statement(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_function_declaration_docstring.py b/tests/unit/typescript/function/test_function_function_declaration_docstring.py index 5bd41a85f..10b6fea16 100644 --- a/tests/unit/typescript/function/test_function_function_declaration_docstring.py +++ b/tests/unit/typescript/function/test_function_function_declaration_docstring.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_signature(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_get_import_string.py b/tests/unit/typescript/function/test_function_get_import_string.py index cc1f89427..e99d364ce 100644 --- a/tests/unit/typescript/function/test_function_get_import_string.py +++ b/tests/unit/typescript/function/test_function_get_import_string.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_get_import_string_with_alias(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_get_parameter_by_index.py b/tests/unit/typescript/function/test_function_get_parameter_by_index.py index ea51bd925..e5247e7b2 100644 --- a/tests/unit/typescript/function/test_function_get_parameter_by_index.py +++ b/tests/unit/typescript/function/test_function_get_parameter_by_index.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_get_parameter_by_index(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_is_exported.py b/tests/unit/typescript/function/test_function_is_exported.py index e90dcdbb1..afefa32a8 100644 --- a/tests/unit/typescript/function/test_function_is_exported.py +++ b/tests/unit/typescript/function/test_function_is_exported.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_named_function_is_exported_should_return_true(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_is_reexported.py b/tests/unit/typescript/function/test_function_is_reexported.py index f877ac6de..98c30f9bd 100644 --- a/tests/unit/typescript/function/test_function_is_reexported.py +++ b/tests/unit/typescript/function/test_function_is_reexported.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.enums import ProgrammingLanguage def test_function_is_reexported_from_separate_file(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_jsx.py b/tests/unit/typescript/function/test_function_jsx.py index 63e8733a0..313026bfc 100644 --- a/tests/unit/typescript/function/test_function_jsx.py +++ b/tests/unit/typescript/function/test_function_jsx.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_is_jsx(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_level.py b/tests/unit/typescript/function/test_function_level.py index 47a90cbf8..c07eca9e5 100644 --- a/tests/unit/typescript/function/test_function_level.py +++ b/tests/unit/typescript/function/test_function_level.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_level_gets_nested_function(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_move_to_file.py b/tests/unit/typescript/function/test_function_move_to_file.py index ab8d7fdf8..d8849a889 100644 --- a/tests/unit/typescript/function/test_function_move_to_file.py +++ b/tests/unit/typescript/function/test_function_move_to_file.py @@ -3,9 +3,9 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage from tests.utils.normalize import normalize_imports # All the diff types of JS/TS imports: diff --git a/tests/unit/typescript/function/test_function_nested_functions.py b/tests/unit/typescript/function/test_function_nested_functions.py index 3920e88c1..5ec6b6f72 100644 --- a/tests/unit/typescript/function/test_function_nested_functions.py +++ b/tests/unit/typescript/function/test_function_nested_functions.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_nested_base(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_parameter_to_interface.py b/tests/unit/typescript/function/test_function_parameter_to_interface.py index 096b01da5..2db81f15a 100644 --- a/tests/unit/typescript/function/test_function_parameter_to_interface.py +++ b/tests/unit/typescript/function/test_function_parameter_to_interface.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_convert_simple_props_to_interface(tmpdir): diff --git a/tests/unit/typescript/function/test_function_parameters.py b/tests/unit/typescript/function/test_function_parameters.py index 0bc49a291..f27c5cad3 100644 --- a/tests/unit/typescript/function/test_function_parameters.py +++ b/tests/unit/typescript/function/test_function_parameters.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session, get_codebase_session -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session, get_codebase_session +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_function_parameters_multiline(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_prepend_statements.py b/tests/unit/typescript/function/test_function_prepend_statements.py index 42ad880b4..0f33a16d7 100644 --- a/tests/unit/typescript/function/test_function_prepend_statements.py +++ b/tests/unit/typescript/function/test_function_prepend_statements.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_call_without_docstring_prepend_statements(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_rename.py b/tests/unit/typescript/function/test_function_rename.py index 5f36edfd4..f1ed8ca02 100644 --- a/tests/unit/typescript/function/test_function_rename.py +++ b/tests/unit/typescript/function/test_function_rename.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_rename_function_updates_file_content(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_return_statements.py b/tests/unit/typescript/function/test_function_return_statements.py index 90b2cb8fe..548ff22a1 100644 --- a/tests/unit/typescript/function/test_function_return_statements.py +++ b/tests/unit/typescript/function/test_function_return_statements.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_function_return_statements_includes_in_if_else_bocks(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_return_type_annotation.py b/tests/unit/typescript/function/test_function_return_type_annotation.py index fc13f139a..674af10aa 100644 --- a/tests/unit/typescript/function/test_function_return_type_annotation.py +++ b/tests/unit/typescript/function/test_function_return_type_annotation.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.generic_type import GenericType -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.generic_type import GenericType +from codegen.sdk.enums import ProgrammingLanguage def test_return_type_annotation(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_set_docstring.py b/tests/unit/typescript/function/test_function_set_docstring.py index 8e757cb00..b25a5e135 100644 --- a/tests/unit/typescript/function/test_function_set_docstring.py +++ b/tests/unit/typescript/function/test_function_set_docstring.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_set_docstring_multiline(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_set_return_type.py b/tests/unit/typescript/function/test_function_set_return_type.py index f2e8e2692..a7a7a7237 100644 --- a/tests/unit/typescript/function/test_function_set_return_type.py +++ b/tests/unit/typescript/function/test_function_set_return_type.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_set_return_type(tmpdir) -> None: diff --git a/tests/unit/typescript/function/test_function_usages.py b/tests/unit/typescript/function/test_function_usages.py index 1152e697c..ee35da2fc 100644 --- a/tests/unit/typescript/function/test_function_usages.py +++ b/tests/unit/typescript/function/test_function_usages.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_usages_unpack(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_comment.py b/tests/unit/typescript/global_var/test_global_var_comment.py index 17a0f8c2a..ab1aa580b 100644 --- a/tests/unit/typescript/global_var/test_global_var_comment.py +++ b/tests/unit/typescript/global_var/test_global_var_comment.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_comment_single_line(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_dependencies.py b/tests/unit/typescript/global_var/test_global_var_dependencies.py index 2fdd289be..3ac39eb7f 100644 --- a/tests/unit/typescript/global_var/test_global_var_dependencies.py +++ b/tests/unit/typescript/global_var/test_global_var_dependencies.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.core.import_resolution import Import -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.assignment import TSAssignment +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.core.import_resolution import Import +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.assignment import TSAssignment def test_dependencies_gets_symbols_in_same_file(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_get_import_string.py b/tests/unit/typescript/global_var/test_global_var_get_import_string.py index 2b8397d6d..724fda32d 100644 --- a/tests/unit/typescript/global_var/test_global_var_get_import_string.py +++ b/tests/unit/typescript/global_var/test_global_var_get_import_string.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ImportType, ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ImportType, ProgrammingLanguage def test_global_var_get_import_string_import_type_star(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_has_value.py b/tests/unit/typescript/global_var/test_global_var_has_value.py index 5d2a6938b..068e91515 100644 --- a/tests/unit/typescript/global_var/test_global_var_has_value.py +++ b/tests/unit/typescript/global_var/test_global_var_has_value.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_global_var_single_var(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_inline_comment.py b/tests/unit/typescript/global_var/test_global_var_inline_comment.py index 0561b110d..c81c52ee8 100644 --- a/tests/unit/typescript/global_var/test_global_var_inline_comment.py +++ b/tests/unit/typescript/global_var/test_global_var_inline_comment.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_inline_comment_includes_inline(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_is_exported.py b/tests/unit/typescript/global_var/test_global_var_is_exported.py index 329dc207b..edf7eac7b 100644 --- a/tests/unit/typescript/global_var/test_global_var_is_exported.py +++ b/tests/unit/typescript/global_var/test_global_var_is_exported.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_is_exported_should_return_true(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_move_to_file.py b/tests/unit/typescript/global_var/test_global_var_move_to_file.py index 224e494ea..c8f3d2fa8 100644 --- a/tests/unit/typescript/global_var/test_global_var_move_to_file.py +++ b/tests/unit/typescript/global_var/test_global_var_move_to_file.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_global_var_move_to_file_global_var_deps_does_not_throw(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_parsing.py b/tests/unit/typescript/global_var/test_global_var_parsing.py index f0ab5f151..a0063d65b 100644 --- a/tests/unit/typescript/global_var/test_global_var_parsing.py +++ b/tests/unit/typescript/global_var/test_global_var_parsing.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.core.assignment import Assignment -from graph_sitter.core.function import Function -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.core.assignment import Assignment +from codegen.sdk.core.function import Function +from codegen.sdk.enums import ProgrammingLanguage def test_parse_global_vars(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_rename.py b/tests/unit/typescript/global_var/test_global_var_rename.py index 7180ad651..3213b92fc 100644 --- a/tests/unit/typescript/global_var/test_global_var_rename.py +++ b/tests/unit/typescript/global_var/test_global_var_rename.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_rename_global_var_excludes_class_and_function_matches(tmpdir) -> None: diff --git a/tests/unit/typescript/global_var/test_global_var_special.py b/tests/unit/typescript/global_var/test_global_var_special.py index 09f97a0c5..faa8f2814 100644 --- a/tests/unit/typescript/global_var/test_global_var_special.py +++ b/tests/unit/typescript/global_var/test_global_var_special.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_global_var_attribute(tmpdir) -> None: diff --git a/tests/unit/typescript/import_resolution/test_has_import.py b/tests/unit/typescript/import_resolution/test_has_import.py index e9f50f3f4..9c54ed0d0 100644 --- a/tests/unit/typescript/import_resolution/test_has_import.py +++ b/tests/unit/typescript/import_resolution/test_has_import.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_import_rename_usage_with_alias(tmpdir) -> None: diff --git a/tests/unit/typescript/import_resolution/test_import_from_export_statement.py b/tests/unit/typescript/import_resolution/test_import_from_export_statement.py index e1365302d..7cc52d05c 100644 --- a/tests/unit/typescript/import_resolution/test_import_from_export_statement.py +++ b/tests/unit/typescript/import_resolution/test_import_from_export_statement.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ImportType, ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ImportType, ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_from_export_statement_default(tmpdir) -> None: diff --git a/tests/unit/typescript/import_resolution/test_import_resolution_get_import_string.py b/tests/unit/typescript/import_resolution/test_import_resolution_get_import_string.py index 21301c692..dce87217d 100644 --- a/tests/unit/typescript/import_resolution/test_import_resolution_get_import_string.py +++ b/tests/unit/typescript/import_resolution/test_import_resolution_get_import_string.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_remove_import_removes_from_file_imports(tmpdir) -> None: diff --git a/tests/unit/typescript/import_resolution/test_import_resolution_remove.py b/tests/unit/typescript/import_resolution/test_import_resolution_remove.py index af4ca08a6..08f8fd0f3 100644 --- a/tests/unit/typescript/import_resolution/test_import_resolution_remove.py +++ b/tests/unit/typescript/import_resolution/test_import_resolution_remove.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_remove_import_removes_from_file_imports(tmpdir) -> None: diff --git a/tests/unit/typescript/import_resolution/test_import_resolution_rename.py b/tests/unit/typescript/import_resolution/test_import_resolution_rename.py index a59bb4f82..8452a3e4c 100644 --- a/tests/unit/typescript/import_resolution/test_import_resolution_rename.py +++ b/tests/unit/typescript/import_resolution/test_import_resolution_rename.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_rename_import_updates_source(tmpdir) -> None: diff --git a/tests/unit/typescript/import_resolution/test_import_resolution_resolve_import.py b/tests/unit/typescript/import_resolution/test_import_resolution_resolve_import.py index 186b5522c..498666604 100644 --- a/tests/unit/typescript/import_resolution/test_import_resolution_resolve_import.py +++ b/tests/unit/typescript/import_resolution/test_import_resolution_resolve_import.py @@ -1,8 +1,8 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.dataclasses.usage import UsageType -from graph_sitter.core.file import SourceFile -from graph_sitter.core.import_resolution import Import, ImportResolution -from graph_sitter.enums import ImportType, ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.dataclasses.usage import UsageType +from codegen.sdk.core.file import SourceFile +from codegen.sdk.core.import_resolution import Import, ImportResolution +from codegen.sdk.enums import ImportType, ProgrammingLanguage def test_dynamic_import_module_export_const(tmpdir) -> None: diff --git a/tests/unit/typescript/import_resolution/test_import_resolution_set_import_module.py b/tests/unit/typescript/import_resolution/test_import_resolution_set_import_module.py index 2507d4a2e..55451b5b2 100644 --- a/tests/unit/typescript/import_resolution/test_import_resolution_set_import_module.py +++ b/tests/unit/typescript/import_resolution/test_import_resolution_set_import_module.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_set_import_module_updates_source(tmpdir) -> None: diff --git a/tests/unit/typescript/import_resolution/test_import_resolution_set_import_symbol_alias.py b/tests/unit/typescript/import_resolution/test_import_resolution_set_import_symbol_alias.py index c41248b46..6839790d5 100644 --- a/tests/unit/typescript/import_resolution/test_import_resolution_set_import_symbol_alias.py +++ b/tests/unit/typescript/import_resolution/test_import_resolution_set_import_symbol_alias.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_set_import_symbol_alias_updates_source(tmpdir) -> None: diff --git a/tests/unit/typescript/interface/test_interface_attributes.py b/tests/unit/typescript/interface/test_interface_attributes.py index bcdf73761..226b672aa 100644 --- a/tests/unit/typescript/interface/test_interface_attributes.py +++ b/tests/unit/typescript/interface/test_interface_attributes.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session, get_codebase_session -from graph_sitter.core.statements.attribute import Attribute -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session, get_codebase_session +from codegen.sdk.core.statements.attribute import Attribute +from codegen.sdk.enums import ProgrammingLanguage def test_interface_attributes_finds_correct_number(tmpdir) -> None: diff --git a/tests/unit/typescript/interface/test_interface_dependencies.py b/tests/unit/typescript/interface/test_interface_dependencies.py index ab6241328..7527d31ae 100644 --- a/tests/unit/typescript/interface/test_interface_dependencies.py +++ b/tests/unit/typescript/interface/test_interface_dependencies.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_interface_dependencies_includes_extends(tmpdir) -> None: diff --git a/tests/unit/typescript/interface/test_interface_get_attribute.py b/tests/unit/typescript/interface/test_interface_get_attribute.py index fc161cbc0..522ec4630 100644 --- a/tests/unit/typescript/interface/test_interface_get_attribute.py +++ b/tests/unit/typescript/interface/test_interface_get_attribute.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_interface_get_attribute_finds_match(tmpdir) -> None: diff --git a/tests/unit/typescript/interface/test_interface_parents.py b/tests/unit/typescript/interface/test_interface_parents.py index 53c2a2ecf..0fa85d31f 100644 --- a/tests/unit/typescript/interface/test_interface_parents.py +++ b/tests/unit/typescript/interface/test_interface_parents.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_interfaces_inheritance(tmpdir) -> None: diff --git a/tests/unit/typescript/language_engine/test_node_engine.py b/tests/unit/typescript/language_engine/test_node_engine.py index f671a43fd..19c868a45 100644 --- a/tests/unit/typescript/language_engine/test_node_engine.py +++ b/tests/unit/typescript/language_engine/test_node_engine.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_node_engine(tmpdir) -> None: diff --git a/tests/unit/typescript/language_engine/test_v8_engine.py b/tests/unit/typescript/language_engine/test_v8_engine.py index 9fb233dac..9497de1d4 100644 --- a/tests/unit/typescript/language_engine/test_v8_engine.py +++ b/tests/unit/typescript/language_engine/test_v8_engine.py @@ -1,7 +1,7 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage @pytest.mark.skip(reason="V8 engine is not yet fully supported") diff --git a/tests/unit/typescript/move_symbol_to_file/test_move_tsx_to_file.py b/tests/unit/typescript/move_symbol_to_file/test_move_tsx_to_file.py index 0fc1a3719..450b85fa5 100644 --- a/tests/unit/typescript/move_symbol_to_file/test_move_tsx_to_file.py +++ b/tests/unit/typescript/move_symbol_to_file/test_move_tsx_to_file.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_move_component_with_dependencies(tmpdir) -> None: diff --git a/tests/unit/typescript/namespace/test_namespace.py b/tests/unit/typescript/namespace/test_namespace.py index 0140f7ee7..a08142148 100644 --- a/tests/unit/typescript/namespace/test_namespace.py +++ b/tests/unit/typescript/namespace/test_namespace.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.namespace import TSNamespace +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.namespace import TSNamespace def test_empty_namespace(tmpdir) -> None: diff --git a/tests/unit/typescript/namespace/test_namespace_complex_examples.py b/tests/unit/typescript/namespace/test_namespace_complex_examples.py index 0bb3eff6d..93180086b 100644 --- a/tests/unit/typescript/namespace/test_namespace_complex_examples.py +++ b/tests/unit/typescript/namespace/test_namespace_complex_examples.py @@ -1,7 +1,7 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.function import TSFunction -from graph_sitter.typescript.namespace import TSNamespace +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.function import TSFunction +from codegen.sdk.typescript.namespace import TSNamespace def test_namespace_math_operations(tmpdir) -> None: diff --git a/tests/unit/typescript/react/test_interface_props.py b/tests/unit/typescript/react/test_interface_props.py index 46ada9256..7b19f6de5 100644 --- a/tests/unit/typescript/react/test_interface_props.py +++ b/tests/unit/typescript/react/test_interface_props.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_convert_props_to_interface_with_proptypes_oneof(tmpdir): diff --git a/tests/unit/typescript/react/test_ts_declassify.py b/tests/unit/typescript/react/test_ts_declassify.py index 8a8d4c906..790bd7fbb 100644 --- a/tests/unit/typescript/react/test_ts_declassify.py +++ b/tests/unit/typescript/react/test_ts_declassify.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_declassify_class_component(tmpdir): diff --git a/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_get_local_usages.py b/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_get_local_usages.py index 54ef1b8d4..d1d5c3bf8 100644 --- a/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_get_local_usages.py +++ b/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_get_local_usages.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_assignment_statement_get_local_usages(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_get_variable_usages.py b/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_get_variable_usages.py index 3e2ddba2a..73f9f7551 100644 --- a/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_get_variable_usages.py +++ b/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_get_variable_usages.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_get_and_edit_variable_usages(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_type_annotation.py b/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_type_annotation.py index fa88a678f..449cb5efb 100644 --- a/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_type_annotation.py +++ b/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_type_annotation.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_assignment_statement_type_annotation_returns_none_if_missing(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_type_resolution.py b/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_type_resolution.py index 3a95c4f8a..77201c028 100644 --- a/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_type_resolution.py +++ b/tests/unit/typescript/statements/assignment_statement/test_assignment_statement_type_resolution.py @@ -1,8 +1,8 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.expressions.object_type import TSObjectType +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.expressions.object_type import TSObjectType def test_assignment_statement_type_resolution_simple(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/attribute/test_attribute_get_local_usages.py b/tests/unit/typescript/statements/attribute/test_attribute_get_local_usages.py index dc4ae23e7..c5b9d3b25 100644 --- a/tests/unit/typescript/statements/attribute/test_attribute_get_local_usages.py +++ b/tests/unit/typescript/statements/attribute/test_attribute_get_local_usages.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_attribute_get_local_usages(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/attribute/test_attribute_properties.py b/tests/unit/typescript/statements/attribute/test_attribute_properties.py index 5df788305..8a4497fab 100644 --- a/tests/unit/typescript/statements/attribute/test_attribute_properties.py +++ b/tests/unit/typescript/statements/attribute/test_attribute_properties.py @@ -1,9 +1,9 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.detached_symbols.function_call import FunctionCall -from graph_sitter.core.expressions.number import Number -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.statements.assignment_statement import TSAssignmentStatement -from graph_sitter.typescript.statements.attribute import TSAttribute +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.detached_symbols.function_call import FunctionCall +from codegen.sdk.core.expressions.number import Number +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.statements.assignment_statement import TSAssignmentStatement +from codegen.sdk.typescript.statements.attribute import TSAttribute def test_attribute(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/attribute/test_attribute_rename.py b/tests/unit/typescript/statements/attribute/test_attribute_rename.py index aaa7207a8..0e8954a70 100644 --- a/tests/unit/typescript/statements/attribute/test_attribute_rename.py +++ b/tests/unit/typescript/statements/attribute/test_attribute_rename.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_attribute_rename(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/attribute/test_attribute_set_type_annotation.py b/tests/unit/typescript/statements/attribute/test_attribute_set_type_annotation.py index 2cab84109..ac2ef29de 100644 --- a/tests/unit/typescript/statements/attribute/test_attribute_set_type_annotation.py +++ b/tests/unit/typescript/statements/attribute/test_attribute_set_type_annotation.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_attribute_set_type_annotation_from_none(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/for_loop_statement/test_for_loop_statement.py b/tests/unit/typescript/statements/for_loop_statement/test_for_loop_statement.py index bf81c803b..a5219a172 100644 --- a/tests/unit/typescript/statements/for_loop_statement/test_for_loop_statement.py +++ b/tests/unit/typescript/statements/for_loop_statement/test_for_loop_statement.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage def test_for_loop_statement_parse(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/if_block_statement/test_if_block_reduce_block.py b/tests/unit/typescript/statements/if_block_statement/test_if_block_reduce_block.py index b7b2f15e2..760ab23a5 100644 --- a/tests/unit/typescript/statements/if_block_statement/test_if_block_reduce_block.py +++ b/tests/unit/typescript/statements/if_block_statement/test_if_block_reduce_block.py @@ -1,9 +1,9 @@ import pytest -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile def test_reduce_condition_to_true_elif(tmpdir): diff --git a/tests/unit/typescript/statements/if_block_statement/test_if_block_statement_properties.py b/tests/unit/typescript/statements/if_block_statement/test_if_block_statement_properties.py index 9d1735110..5691f8926 100644 --- a/tests/unit/typescript/statements/if_block_statement/test_if_block_statement_properties.py +++ b/tests/unit/typescript/statements/if_block_statement/test_if_block_statement_properties.py @@ -1,8 +1,8 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.file import TSFile -from graph_sitter.typescript.statements.if_block_statement import TSIfBlockStatement +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.file import TSFile +from codegen.sdk.typescript.statements.if_block_statement import TSIfBlockStatement def test_parse_if_else_statement_from_codeblock(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/statement/test_statement_get_variable_usages.py b/tests/unit/typescript/statements/statement/test_statement_get_variable_usages.py index 90ac6cecf..e5268848c 100644 --- a/tests/unit/typescript/statements/statement/test_statement_get_variable_usages.py +++ b/tests/unit/typescript/statements/statement/test_statement_get_variable_usages.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_if_statement_edit_variable_usages(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/switch_statement/test_switch_statement.py b/tests/unit/typescript/statements/switch_statement/test_switch_statement.py index da53822aa..37033ffd9 100644 --- a/tests/unit/typescript/statements/switch_statement/test_switch_statement.py +++ b/tests/unit/typescript/statements/switch_statement/test_switch_statement.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage def test_switch_statement_parse(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/try_catch_statement/test_try_catch_statement.py b/tests/unit/typescript/statements/try_catch_statement/test_try_catch_statement.py index 9f73e570f..48d7a59c1 100644 --- a/tests/unit/typescript/statements/try_catch_statement/test_try_catch_statement.py +++ b/tests/unit/typescript/statements/try_catch_statement/test_try_catch_statement.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.enums import ProgrammingLanguage def test_try_catch_statement_parse(tmpdir) -> None: diff --git a/tests/unit/typescript/statements/while_statement/test_while_statement.py b/tests/unit/typescript/statements/while_statement/test_while_statement.py index 62a4f6176..b1269e24d 100644 --- a/tests/unit/typescript/statements/while_statement/test_while_statement.py +++ b/tests/unit/typescript/statements/while_statement/test_while_statement.py @@ -1,9 +1,9 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.expressions.binary_expression import BinaryExpression -from graph_sitter.core.expressions.boolean import Boolean -from graph_sitter.core.statements.statement import StatementType -from graph_sitter.core.statements.while_statement import WhileStatement -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.expressions.binary_expression import BinaryExpression +from codegen.sdk.core.expressions.boolean import Boolean +from codegen.sdk.core.statements.statement import StatementType +from codegen.sdk.core.statements.while_statement import WhileStatement +from codegen.sdk.enums import ProgrammingLanguage def test_while_statement_parse(tmpdir) -> None: diff --git a/tests/unit/typescript/test_print.py b/tests/unit/typescript/test_print.py index aef85b651..dbe6ff2dc 100644 --- a/tests/unit/typescript/test_print.py +++ b/tests/unit/typescript/test_print.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.core.class_definition import Class -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.core.class_definition import Class +from codegen.sdk.enums import ProgrammingLanguage def test_import_rename_usage_with_alias(tmpdir) -> None: diff --git a/tests/unit/typescript/tsx/test_tsx_edit.py b/tests/unit/typescript/tsx/test_tsx_edit.py index 4837fc922..c67cd486c 100644 --- a/tests/unit/typescript/tsx/test_tsx_edit.py +++ b/tests/unit/typescript/tsx/test_tsx_edit.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_tsx_component_edit(tmpdir) -> None: diff --git a/tests/unit/typescript/tsx/test_tsx_parsing.py b/tests/unit/typescript/tsx/test_tsx_parsing.py index d58739688..3bebf02f5 100644 --- a/tests/unit/typescript/tsx/test_tsx_parsing.py +++ b/tests/unit/typescript/tsx/test_tsx_parsing.py @@ -1,6 +1,6 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage -from graph_sitter.typescript.detached_symbols.jsx.element import JSXElement +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage +from codegen.sdk.typescript.detached_symbols.jsx.element import JSXElement def test_basic_component_parsing(tmpdir) -> None: diff --git a/tests/unit/typescript/type/test_type_attributes.py b/tests/unit/typescript/type/test_type_attributes.py index 9bb3303c6..f4a843fb5 100644 --- a/tests/unit/typescript/type/test_type_attributes.py +++ b/tests/unit/typescript/type/test_type_attributes.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_session +from codegen.sdk.enums import ProgrammingLanguage def test_type_attributes_primitives(tmpdir) -> None: diff --git a/tests/unit/typescript/type/test_type_dependencies.py b/tests/unit/typescript/type/test_type_dependencies.py index d957de72d..2c826650b 100644 --- a/tests/unit/typescript/type/test_type_dependencies.py +++ b/tests/unit/typescript/type/test_type_dependencies.py @@ -1,5 +1,5 @@ -from graph_sitter.codebase.factory.get_session import get_codebase_graph_session -from graph_sitter.enums import ProgrammingLanguage +from codegen.sdk.codebase.factory.get_session import get_codebase_graph_session +from codegen.sdk.enums import ProgrammingLanguage def test_type_dependencies(tmpdir) -> None: diff --git a/tests/utils/codebase_comparison_utils.py b/tests/utils/codebase_comparison_utils.py index 39eccbca7..433998257 100644 --- a/tests/utils/codebase_comparison_utils.py +++ b/tests/utils/codebase_comparison_utils.py @@ -8,9 +8,9 @@ from Levenshtein import distance as levenshtein_distance from pytest_snapshot.plugin import Snapshot +from codegen.sdk.core.codebase import Codebase +from codegen.sdk.testing.constants import DIFF_FILEPATH from codegen_git.schemas.enums import DiffChangeType -from graph_sitter.core.codebase import Codebase -from graph_sitter.testing.constants import DIFF_FILEPATH logger = logging.getLogger(__name__) diff --git a/tests/utils/mocks/mock_ai_helper.py b/tests/utils/mocks/mock_ai_helper.py index b0887721f..b163cfeb3 100644 --- a/tests/utils/mocks/mock_ai_helper.py +++ b/tests/utils/mocks/mock_ai_helper.py @@ -1,4 +1,4 @@ -from graph_sitter.ai.helpers import AbstractAIHelper +from codegen.sdk.ai.helpers import AbstractAIHelper class MockAIHelper(AbstractAIHelper): diff --git a/tests/verified_codemods/codemod_data/N2IyYzIxZW.json b/tests/verified_codemods/codemod_data/N2IyYzIxZW.json index 2f45fd291..3c7c898cb 100644 --- a/tests/verified_codemods/codemod_data/N2IyYzIxZW.json +++ b/tests/verified_codemods/codemod_data/N2IyYzIxZW.json @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:6ce3bf4c779700acec4fe3ca7225dc9258f4a9998897002835b3d0d030c3f229 -size 546309 +oid sha256:4e26881a59ced8505c4456488fe1cf97ae5e8163f92cf56a7bde6d3263533994 +size 546254 diff --git a/tests/verified_codemods/codemod_data/YTY2NWE0NT.json b/tests/verified_codemods/codemod_data/YTY2NWE0NT.json index ad0532bb1..11cad1d92 100644 --- a/tests/verified_codemods/codemod_data/YTY2NWE0NT.json +++ b/tests/verified_codemods/codemod_data/YTY2NWE0NT.json @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c7adb21bd213dfec5311ea1d1e9426b41fd410fbe7a7ba2b3029dcb40b67f7ec -size 15513609 +oid sha256:615e0ddf0aa36fb987c2033fc3db9077d4ca66be8c903b3b20678b58946fe8ee +size 15512725 diff --git a/uv.lock b/uv.lock index f81fc9070..0d9714287 100644 --- a/uv.lock +++ b/uv.lock @@ -4,7 +4,7 @@ requires-python = ">=3.12, <3.14" [manifest] members = [ "codegen-git", - "graph-sitter", + "codegen-sdk", ] [[package]] @@ -380,6 +380,174 @@ requires-dist = [ { name = "termcolor", specifier = ">=2.4.0" }, ] +[[package]] +name = "codegen-sdk" +source = { editable = "." } +dependencies = [ + { name = "anthropic" }, + { name = "astor" }, + { name = "autoflake" }, + { name = "backoff" }, + { name = "black" }, + { name = "click" }, + { name = "codegen-git" }, + { name = "codeowners" }, + { name = "dataclasses-json" }, + { name = "dicttoxml" }, + { name = "docstring-parser" }, + { name = "emoji" }, + { name = "hatch-vcs" }, + { name = "hatchling" }, + { name = "humanize" }, + { name = "inflection" }, + { name = "isort" }, + { name = "jsbeautifier" }, + { name = "lazy-object-proxy" }, + { name = "loguru" }, + { name = "mini-racer" }, + { name = "networkx" }, + { name = "openai" }, + { name = "pip" }, + { name = "plotly" }, + { name = "pydantic" }, + { name = "pydantic-core" }, + { name = "pyinstrument" }, + { name = "pyjson5" }, + { name = "pyright" }, + { name = "pytest-snapshot" }, + { name = "python-levenshtein" }, + { name = "requests" }, + { name = "rich" }, + { name = "rustworkx" }, + { name = "sentry-sdk" }, + { name = "tabulate" }, + { name = "tenacity" }, + { name = "termcolor" }, + { name = "tiktoken" }, + { name = "tree-sitter" }, + { name = "tree-sitter-javascript" }, + { name = "tree-sitter-python" }, + { name = "tree-sitter-typescript" }, + { name = "typing-extensions" }, + { name = "watchfiles" }, + { name = "wrapt" }, + { name = "xmltodict" }, +] + +[package.optional-dependencies] +types = [ + { name = "types-networkx" }, + { name = "types-tabulate" }, +] + +[package.dev-dependencies] +dev = [ + { name = "austin-dist" }, + { name = "austin-python" }, + { name = "autoflake" }, + { name = "braintrust" }, + { name = "cibuildwheel", extra = ["uv"] }, + { name = "coverage" }, + { name = "cython" }, + { name = "deptry" }, + { name = "filelock" }, + { name = "keyrings-codeartifact" }, + { name = "mypy", extra = ["faster-cache", "mypyc"] }, + { name = "pre-commit" }, + { name = "pre-commit-uv" }, + { name = "pytest" }, + { name = "pytest-cov" }, + { name = "pytest-mock" }, + { name = "pytest-timeout" }, + { name = "pytest-xdist" }, + { name = "ruff" }, + { name = "ruff-lsp" }, + { name = "sybil", extra = ["pytest"] }, + { name = "typer" }, + { name = "uv" }, +] + +[package.metadata] +requires-dist = [ + { name = "anthropic", specifier = "==0.23.1" }, + { name = "astor", specifier = ">=0.8.1,<1.0.0" }, + { name = "autoflake", specifier = ">=2.3.1" }, + { name = "backoff", specifier = "==2.2.1" }, + { name = "black", specifier = ">=24.8.0" }, + { name = "click", specifier = ">=8.1.7" }, + { name = "codegen-git", editable = "codegen-git" }, + { name = "codeowners", specifier = ">=0.6.0,<1.0.0" }, + { name = "dataclasses-json", specifier = ">=0.6.4,<1.0.0" }, + { name = "dicttoxml", specifier = ">=1.7.16,<2.0.0" }, + { name = "docstring-parser", specifier = ">=0.16,<1.0" }, + { name = "emoji", specifier = ">=2.14.0" }, + { name = "hatch-vcs", specifier = ">=0.4.0" }, + { name = "hatchling", specifier = ">=1.25.0" }, + { name = "humanize", specifier = ">=4.10.0,<5.0.0" }, + { name = "inflection", specifier = ">=0.5.1,<1.0.0" }, + { name = "isort", specifier = ">=5.13.2" }, + { name = "jsbeautifier", specifier = ">=1.15.1,<2.0.0" }, + { name = "lazy-object-proxy", specifier = ">=0.0.0" }, + { name = "loguru", specifier = ">=0.7.2,<1.0.0" }, + { name = "mini-racer", specifier = ">=0.12.4" }, + { name = "networkx", specifier = ">=3.4.1" }, + { name = "openai", specifier = "==1.59.9" }, + { name = "pip", specifier = ">=24.3.1" }, + { name = "plotly", specifier = ">=5.24.0,<6.0.0" }, + { name = "pydantic", specifier = ">=2.9.2,<3.0.0" }, + { name = "pydantic-core", specifier = ">=2.23.4" }, + { name = "pyinstrument", specifier = ">=5.0.0" }, + { name = "pyjson5", specifier = "==1.6.8" }, + { name = "pyright", specifier = ">=1.1.372,<2.0.0" }, + { name = "pytest-snapshot", specifier = ">=0.9.0" }, + { name = "python-levenshtein", specifier = ">=0.25.1,<1.0.0" }, + { name = "requests", specifier = ">=2.32.3" }, + { name = "rich", specifier = ">=13.7.1,<14.0.0" }, + { name = "rustworkx", specifier = ">=0.15.1" }, + { name = "sentry-sdk", specifier = "==2.20.0" }, + { name = "tabulate", specifier = ">=0.9.0,<1.0.0" }, + { name = "tenacity", specifier = ">=9.0.0" }, + { name = "termcolor", specifier = ">=2.4.0" }, + { name = "tiktoken", specifier = ">=0.5.1,<1.0.0" }, + { name = "tree-sitter", specifier = ">=0.23.1" }, + { name = "tree-sitter-javascript", specifier = ">=0.23.1" }, + { name = "tree-sitter-python", specifier = ">=0.23.4" }, + { name = "tree-sitter-typescript", specifier = ">=0.23.2" }, + { name = "types-networkx", marker = "extra == 'types'", specifier = ">=3.2.1.20240918" }, + { name = "types-tabulate", marker = "extra == 'types'", specifier = ">=0.9.0.20240106" }, + { name = "typing-extensions", specifier = ">=4.12.2" }, + { name = "watchfiles", specifier = ">=1.0.0,<1.1.0" }, + { name = "wrapt", specifier = ">=1.16.0,<2.0.0" }, + { name = "xmltodict", specifier = ">=0.13.0,<1.0.0" }, +] + +[package.metadata.requires-dev] +dev = [ + { name = "austin-dist", specifier = ">=3.7.0" }, + { name = "austin-python", specifier = ">=1.7.1" }, + { name = "autoflake", specifier = ">=2.3.1" }, + { name = "braintrust", specifier = ">=0.0.160" }, + { name = "cibuildwheel", extras = ["uv"], specifier = ">=2.22.0" }, + { name = "coverage", specifier = ">=7.6.1,<8.0.0" }, + { name = "cython", specifier = ">=3.0.11" }, + { name = "deptry", specifier = ">=0.22.0" }, + { name = "filelock", specifier = ">=3.15.4,<4.0.0" }, + { name = "keyrings-codeartifact", specifier = ">=1.3.3" }, + { name = "mypy", extras = ["mypyc", "faster-cache"], specifier = ">=1.13.0" }, + { name = "pre-commit", specifier = ">=4.0.1" }, + { name = "pre-commit-uv", specifier = ">=4.1.4" }, + { name = "pytest", specifier = ">=8.3.3" }, + { name = "pytest-cov", specifier = ">=6.0.0,<6.0.1" }, + { name = "pytest-mock", specifier = ">=3.14.0,<4.0.0" }, + { name = "pytest-timeout", specifier = ">=2.3.1" }, + { name = "pytest-xdist", specifier = ">=3.6.1,<4.0.0" }, + { name = "ruff", specifier = ">=0.6.8" }, + { name = "ruff-lsp", specifier = ">=0.0.55,<1.0.0" }, + { name = "sybil", extras = ["pytest"], specifier = ">=9.0.0" }, + { name = "typer", specifier = ">=0.12.5" }, + { name = "uv", specifier = ">=0.4.25" }, +] + [[package]] name = "codeowners" version = "0.7.0" @@ -668,174 +836,6 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/1d/9a/4114a9057db2f1462d5c8f8390ab7383925fe1ac012eaa42402ad65c2963/GitPython-3.1.44-py3-none-any.whl", hash = "sha256:9e0e10cda9bed1ee64bc9a6de50e7e38a9c9943241cd7f585f6df3ed28011110", size = 207599 }, ] -[[package]] -name = "graph-sitter" -source = { editable = "." } -dependencies = [ - { name = "anthropic" }, - { name = "astor" }, - { name = "autoflake" }, - { name = "backoff" }, - { name = "black" }, - { name = "click" }, - { name = "codegen-git" }, - { name = "codeowners" }, - { name = "dataclasses-json" }, - { name = "dicttoxml" }, - { name = "docstring-parser" }, - { name = "emoji" }, - { name = "hatch-vcs" }, - { name = "hatchling" }, - { name = "humanize" }, - { name = "inflection" }, - { name = "isort" }, - { name = "jsbeautifier" }, - { name = "lazy-object-proxy" }, - { name = "loguru" }, - { name = "mini-racer" }, - { name = "networkx" }, - { name = "openai" }, - { name = "pip" }, - { name = "plotly" }, - { name = "pydantic" }, - { name = "pydantic-core" }, - { name = "pyinstrument" }, - { name = "pyjson5" }, - { name = "pyright" }, - { name = "pytest-snapshot" }, - { name = "python-levenshtein" }, - { name = "requests" }, - { name = "rich" }, - { name = "rustworkx" }, - { name = "sentry-sdk" }, - { name = "tabulate" }, - { name = "tenacity" }, - { name = "termcolor" }, - { name = "tiktoken" }, - { name = "tree-sitter" }, - { name = "tree-sitter-javascript" }, - { name = "tree-sitter-python" }, - { name = "tree-sitter-typescript" }, - { name = "typing-extensions" }, - { name = "watchfiles" }, - { name = "wrapt" }, - { name = "xmltodict" }, -] - -[package.optional-dependencies] -types = [ - { name = "types-networkx" }, - { name = "types-tabulate" }, -] - -[package.dev-dependencies] -dev = [ - { name = "austin-dist" }, - { name = "austin-python" }, - { name = "autoflake" }, - { name = "braintrust" }, - { name = "cibuildwheel", extra = ["uv"] }, - { name = "coverage" }, - { name = "cython" }, - { name = "deptry" }, - { name = "filelock" }, - { name = "keyrings-codeartifact" }, - { name = "mypy", extra = ["faster-cache", "mypyc"] }, - { name = "pre-commit" }, - { name = "pre-commit-uv" }, - { name = "pytest" }, - { name = "pytest-cov" }, - { name = "pytest-mock" }, - { name = "pytest-timeout" }, - { name = "pytest-xdist" }, - { name = "ruff" }, - { name = "ruff-lsp" }, - { name = "sybil", extra = ["pytest"] }, - { name = "typer" }, - { name = "uv" }, -] - -[package.metadata] -requires-dist = [ - { name = "anthropic", specifier = "==0.23.1" }, - { name = "astor", specifier = ">=0.8.1,<1.0.0" }, - { name = "autoflake", specifier = ">=2.3.1" }, - { name = "backoff", specifier = "==2.2.1" }, - { name = "black", specifier = ">=24.8.0" }, - { name = "click", specifier = ">=8.1.7" }, - { name = "codegen-git", editable = "codegen-git" }, - { name = "codeowners", specifier = ">=0.6.0,<1.0.0" }, - { name = "dataclasses-json", specifier = ">=0.6.4,<1.0.0" }, - { name = "dicttoxml", specifier = ">=1.7.16,<2.0.0" }, - { name = "docstring-parser", specifier = ">=0.16,<1.0" }, - { name = "emoji", specifier = ">=2.14.0" }, - { name = "hatch-vcs", specifier = ">=0.4.0" }, - { name = "hatchling", specifier = ">=1.25.0" }, - { name = "humanize", specifier = ">=4.10.0,<5.0.0" }, - { name = "inflection", specifier = ">=0.5.1,<1.0.0" }, - { name = "isort", specifier = ">=5.13.2" }, - { name = "jsbeautifier", specifier = ">=1.15.1,<2.0.0" }, - { name = "lazy-object-proxy", specifier = ">=0.0.0" }, - { name = "loguru", specifier = ">=0.7.2,<1.0.0" }, - { name = "mini-racer", specifier = ">=0.12.4" }, - { name = "networkx", specifier = ">=3.4.1" }, - { name = "openai", specifier = "==1.59.9" }, - { name = "pip", specifier = ">=24.3.1" }, - { name = "plotly", specifier = ">=5.24.0,<6.0.0" }, - { name = "pydantic", specifier = ">=2.9.2,<3.0.0" }, - { name = "pydantic-core", specifier = ">=2.23.4" }, - { name = "pyinstrument", specifier = ">=5.0.0" }, - { name = "pyjson5", specifier = "==1.6.8" }, - { name = "pyright", specifier = ">=1.1.372,<2.0.0" }, - { name = "pytest-snapshot", specifier = ">=0.9.0" }, - { name = "python-levenshtein", specifier = ">=0.25.1,<1.0.0" }, - { name = "requests", specifier = ">=2.32.3" }, - { name = "rich", specifier = ">=13.7.1,<14.0.0" }, - { name = "rustworkx", specifier = ">=0.15.1" }, - { name = "sentry-sdk", specifier = "==2.20.0" }, - { name = "tabulate", specifier = ">=0.9.0,<1.0.0" }, - { name = "tenacity", specifier = ">=9.0.0" }, - { name = "termcolor", specifier = ">=2.4.0" }, - { name = "tiktoken", specifier = ">=0.5.1,<1.0.0" }, - { name = "tree-sitter", specifier = ">=0.23.1" }, - { name = "tree-sitter-javascript", specifier = ">=0.23.1" }, - { name = "tree-sitter-python", specifier = ">=0.23.4" }, - { name = "tree-sitter-typescript", specifier = ">=0.23.2" }, - { name = "types-networkx", marker = "extra == 'types'", specifier = ">=3.2.1.20240918" }, - { name = "types-tabulate", marker = "extra == 'types'", specifier = ">=0.9.0.20240106" }, - { name = "typing-extensions", specifier = ">=4.12.2" }, - { name = "watchfiles", specifier = ">=1.0.0,<1.1.0" }, - { name = "wrapt", specifier = ">=1.16.0,<2.0.0" }, - { name = "xmltodict", specifier = ">=0.13.0,<1.0.0" }, -] - -[package.metadata.requires-dev] -dev = [ - { name = "austin-dist", specifier = ">=3.7.0" }, - { name = "austin-python", specifier = ">=1.7.1" }, - { name = "autoflake", specifier = ">=2.3.1" }, - { name = "braintrust", specifier = ">=0.0.160" }, - { name = "cibuildwheel", extras = ["uv"], specifier = ">=2.22.0" }, - { name = "coverage", specifier = ">=7.6.1,<8.0.0" }, - { name = "cython", specifier = ">=3.0.11" }, - { name = "deptry", specifier = ">=0.22.0" }, - { name = "filelock", specifier = ">=3.15.4,<4.0.0" }, - { name = "keyrings-codeartifact", specifier = ">=1.3.3" }, - { name = "mypy", extras = ["mypyc", "faster-cache"], specifier = ">=1.13.0" }, - { name = "pre-commit", specifier = ">=4.0.1" }, - { name = "pre-commit-uv", specifier = ">=4.1.4" }, - { name = "pytest", specifier = ">=8.3.3" }, - { name = "pytest-cov", specifier = ">=6.0.0,<6.0.1" }, - { name = "pytest-mock", specifier = ">=3.14.0,<4.0.0" }, - { name = "pytest-timeout", specifier = ">=2.3.1" }, - { name = "pytest-xdist", specifier = ">=3.6.1,<4.0.0" }, - { name = "ruff", specifier = ">=0.6.8" }, - { name = "ruff-lsp", specifier = ">=0.0.55,<1.0.0" }, - { name = "sybil", extras = ["pytest"], specifier = ">=9.0.0" }, - { name = "typer", specifier = ">=0.12.5" }, - { name = "uv", specifier = ">=0.4.25" }, -] - [[package]] name = "h11" version = "0.14.0"