Move graph_visualization into codebase.visualizations

src/codegen/sdk/core/class_definition.py

@@ -31,7 +31,7 @@
 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
+from codegen.visualizations.enums import VizNode
 
 if TYPE_CHECKING:
     from codegen.sdk.core.detached_symbols.code_block import CodeBlock

src/codegen/sdk/core/codebase.py

@@ -72,7 +72,7 @@
 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 graph_visualization.visualization_manager import VisualizationManager
+from codegen.visualizations.visualization_manager import VisualizationManager
 
 if TYPE_CHECKING:
     from app.codemod.types import Group

src/codegen/sdk/core/detached_symbols/function_call.py

@@ -31,7 +31,7 @@
     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
+    from codegen.visualizations.enums import VizNode
 
 Parent = TypeVar("Parent", bound="Expression | None")
 
@@ -216,7 +216,7 @@ def predecessor(self) -> FunctionCall[Parent] | None:
     @noapidoc
     @override
     def viz(self) -> VizNode:
-        from graph_visualization.enums import VizNode
+        from codegen.visualizations.enums import VizNode
 
         func = self.function_definition
         from codegen.sdk.core.function import Function

src/codegen/sdk/core/external_module.py

@@ -15,7 +15,7 @@
 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
+from codegen.visualizations.enums import VizNode
 
 if TYPE_CHECKING:
     from codegen.sdk.core.detached_symbols.parameter import Parameter

src/codegen/sdk/core/file.py

@@ -38,7 +38,7 @@
 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
+from codegen.visualizations.enums import VizNode
 
 logger = logging.getLogger(__name__)
 

src/codegen/sdk/core/function.py

@@ -23,7 +23,7 @@
 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
+from codegen.visualizations.enums import VizNode
 
 if TYPE_CHECKING:
     from codegen.sdk.core.export import Export

src/codegen/sdk/core/import_resolution.py

@@ -27,7 +27,7 @@
 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
+from codegen.visualizations.enums import VizNode
 
 if TYPE_CHECKING:
     from codegen.sdk.core.file import SourceFile

src/codegen/sdk/core/interfaces/editable.py

@@ -47,7 +47,7 @@
     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
+    from codegen.visualizations.enums import VizNode
 CONTAINER_CHARS = (b"(", b")", b"{", b"}", b"[", b"]", b"<", b">", b"import")
 MAX_REPR_LEN: int = 200
 
@@ -975,7 +975,7 @@ def _log_parse(self, msg: str, *args, **kwargs):
     @noapidoc
     def viz(self) -> VizNode:
         from codegen.sdk.core.interfaces.has_name import HasName
-        from graph_visualization.enums import VizNode
+        from codegen.visualizations.enums import VizNode
 
         if isinstance(self, HasName):
             return VizNode(file_path=self.filepath, start_point=self.start_point, end_point=self.end_point, name=self.name, symbol_name=self.__class__.__name__)

src/graph_visualization/visualization_manager.py → src/codegen/visualizations/visualization_manager.py

@@ -6,7 +6,7 @@
 
 from codegen.git.repo_operator.repo_operator import RepoOperator
 from codegen.sdk.core.interfaces.editable import Editable
-from graph_visualization.viz_utils import graph_to_json
+from codegen.visualizations.viz_utils import graph_to_json
 
 logger = logging.getLogger(__name__)
 

src/graph_visualization/viz_utils.py → src/codegen/visualizations/viz_utils.py

@@ -10,7 +10,7 @@
 from codegen.sdk.core.interfaces.editable import Editable
 from codegen.sdk.core.interfaces.importable import Importable
 from codegen.sdk.output.utils import DeterministicJSONEncoder
-from graph_visualization.enums import GraphJson, GraphType
+from codegen.visualizations.enums import GraphJson, GraphType
 
 if TYPE_CHECKING:
     pass