diff --git a/codegen-examples/examples/cyclomatic_complexity/run.py b/codegen-examples/examples/cyclomatic_complexity/run.py index 0b5943ad1..50b8c5614 100644 --- a/codegen-examples/examples/cyclomatic_complexity/run.py +++ b/codegen-examples/examples/cyclomatic_complexity/run.py @@ -1,6 +1,6 @@ import codegen from codegen import Codebase -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage from codegen.sdk.core.statements.for_loop_statement import ForLoopStatement from codegen.sdk.core.statements.if_block_statement import IfBlockStatement from codegen.sdk.core.statements.try_catch_statement import TryCatchStatement diff --git a/codegen-examples/examples/delete_dead_code/run.py b/codegen-examples/examples/delete_dead_code/run.py index 888cc4012..cfaf14c0a 100644 --- a/codegen-examples/examples/delete_dead_code/run.py +++ b/codegen-examples/examples/delete_dead_code/run.py @@ -1,6 +1,6 @@ import codegen from codegen import Codebase -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage @codegen.function("delete-dead-code") diff --git a/docs/building-with-codegen/parsing-codebases.mdx b/docs/building-with-codegen/parsing-codebases.mdx index 61abd3778..7d10717d8 100644 --- a/docs/building-with-codegen/parsing-codebases.mdx +++ b/docs/building-with-codegen/parsing-codebases.mdx @@ -13,7 +13,7 @@ Construct a Codebase by passing in a path to a local `git` repository or any sub ```python from codegen import Codebase -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage # Parse from a git repository root codebase = Codebase("path/to/repository") @@ -46,7 +46,7 @@ To fetch and parse a repository directly from GitHub, use the `from_repo` functi ```python import codegen -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage # Fetch and parse a repository (defaults to /tmp/codegen/{repo_name}) codebase = codegen.from_repo('fastapi/fastapi') @@ -109,7 +109,7 @@ from codegen import Codebase from codegen.git.repo_operator.local_repo_operator import LocalRepoOperator from codegen.git.schemas.repo_config import BaseRepoConfig from codegen.sdk.codebase.config import ProjectConfig -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage codebase = Codebase( projects = [ diff --git a/docs/tutorials/slack-bot.mdx b/docs/tutorials/slack-bot.mdx index 179f8d768..c75199c13 100644 --- a/docs/tutorials/slack-bot.mdx +++ b/docs/tutorials/slack-bot.mdx @@ -28,7 +28,7 @@ First, we initialize the codebase and create a vector index for semantic search: ```python from codegen import Codebase from codegen.extensions import VectorIndex -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage def initialize_codebase(): """Initialize and index the codebase.""" diff --git a/src/codegen/extensions/mcp/codebase_agent.py b/src/codegen/extensions/mcp/codebase_agent.py index e458017ca..bf15004d3 100644 --- a/src/codegen/extensions/mcp/codebase_agent.py +++ b/src/codegen/extensions/mcp/codebase_agent.py @@ -5,7 +5,7 @@ from codegen.extensions.langchain.agent import create_codebase_inspector_agent from codegen.sdk.core.codebase import Codebase -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage # Initialize FastMCP server diff --git a/src/codegen/extensions/mcp/codebase_mods.py b/src/codegen/extensions/mcp/codebase_mods.py index cefb41ac6..92aa3de41 100644 --- a/src/codegen/extensions/mcp/codebase_mods.py +++ b/src/codegen/extensions/mcp/codebase_mods.py @@ -5,7 +5,7 @@ from mcp.server.fastmcp import FastMCP from codegen.sdk.core.codebase import Codebase -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage mcp = FastMCP( "codebase-mods-mcp", diff --git a/src/codegen/extensions/mcp/codebase_tools.py b/src/codegen/extensions/mcp/codebase_tools.py index f87282207..bf3f4d2ff 100644 --- a/src/codegen/extensions/mcp/codebase_tools.py +++ b/src/codegen/extensions/mcp/codebase_tools.py @@ -6,7 +6,7 @@ from codegen.extensions.tools import reveal_symbol from codegen.extensions.tools.search import search from codegen.sdk.core.codebase import Codebase -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage mcp = FastMCP( "codebase-tools-mcp", diff --git a/src/codegen/sdk/system-prompt.txt b/src/codegen/sdk/system-prompt.txt index 942b9b7c3..da053db21 100644 --- a/src/codegen/sdk/system-prompt.txt +++ b/src/codegen/sdk/system-prompt.txt @@ -1325,7 +1325,7 @@ Construct a Codebase by passing in a path to a local `git` repository or any sub ```python from codegen import Codebase -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage # Parse from a git repository root codebase = Codebase("path/to/repository") @@ -1358,7 +1358,7 @@ To fetch and parse a repository directly from GitHub, use the `from_repo` functi ```python import codegen -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage # Fetch and parse a repository (defaults to /tmp/codegen/{repo_name}) codebase = codegen.from_repo('fastapi/fastapi') @@ -1421,7 +1421,7 @@ from codegen import Codebase from codegen.git.repo_operator.local_repo_operator import LocalRepoOperator from codegen.git.schemas.repo_config import BaseRepoConfig from codegen.sdk.codebase.config import ProjectConfig -from codegen.sdk.enums import ProgrammingLanguage +from codegen.shared.enums.programming_language import ProgrammingLanguage codebase = Codebase( projects = [