fix imports

Author: bagel897

.github/codecov.yml

@@ -10,7 +10,7 @@ component_management:
     - 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
@@ -19,7 +19,7 @@ component_management:
     - 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
@@ -28,7 +28,7 @@ component_management:
     - component_id: codegen-sdk-core
       name: codegen-sdk-core
       paths:
-        - src/graph_sitter/**
+        - src/codegen_sdk/**
     - component_id: codegen-git
       name: codegen-git
       paths:

.gitignore

@@ -55,7 +55,7 @@ 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

.pre-commit-config.yaml

@@ -71,7 +71,7 @@ repos:
   #       types: [file]
   #       language: system
   #       pass_filenames: false
-  #       files: "codegen-backend/codegen_tests/graph_sitter/codemod/.*/expected_diff.patch"
+  #       files: "codegen-backend/codegen_tests/codegen_sdk/codemod/.*/expected_diff.patch"
   - repo: https://github.com/fpgmaas/deptry.git
     rev: "0.22.0"
     hooks:

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

conftest.py

@@ -1,8 +1,8 @@
 import logging
 import os
 
-from graph_sitter.core import autocommit
-from graph_sitter.testing.models import Size
+from codegen_sdk.core import autocommit
+from codegen_sdk.testing.models import Size
 
 
 def find_dirs_to_ignore(start_dir, prefix):
@@ -15,7 +15,7 @@ def find_dirs_to_ignore(start_dir, prefix):
     return dirs_to_ignore
 
 
-collect_ignore_glob = ["codegen_tests/graph_sitter/codemod/**/*.py"]
+collect_ignore_glob = ["codegen_tests/codegen_sdk/codemod/**/*.py"]
 if not autocommit.enabled:
     collect_ignore_glob.append("**/test_autocommit.py")
 

docs/building-with-codegen/traversing-the-call-graph.mdx

@@ -34,8 +34,8 @@ 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()

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

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",
 ]
@@ -83,7 +83,7 @@ source_archive = "https://github.com/codegen-sh/codegen-sdk/archive/{commit_hash
 
 [build.targets.wheel.hooks.custom]
 enable-by-default = false
-path = "src/graph_sitter_build/build.py"
+path = "src/codegen_sdk_build/build.py"
 require-runtime-dependencies = true
 [envs.default]
 installer = "uv"

mypy.ini

@@ -46,7 +46,7 @@ exclude = (?x)(
           | scripts/.*
           | .*/codemods/.*/test_.*/
           | ^/expected/
-          | src/graph_sitter/skills/*
+          | src/codegen_sdk/skills/*
     )
 plugins = pydantic.mypy
 

pyproject.toml

@@ -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",