From feaa2ba9664b2f8081e5170fc355de18f9cb4ad2 Mon Sep 17 00:00:00 2001 From: Rushil Patel Date: Wed, 12 Feb 2025 10:17:02 -0800 Subject: [PATCH 1/2] fix: update ProgrammingLanguage import path --- codegen-examples/examples/cyclomatic_complexity/run.py | 2 +- codegen-examples/examples/delete_dead_code/run.py | 3 +-- docs/building-with-codegen/parsing-codebases.mdx | 6 +++--- docs/tutorials/slack-bot.mdx | 2 +- src/codegen/extensions/mcp/codebase_agent.py | 2 +- src/codegen/extensions/mcp/codebase_mods.py | 2 +- src/codegen/extensions/mcp/codebase_tools.py | 2 +- src/codegen/sdk/system-prompt.txt | 6 +++--- 8 files changed, 12 insertions(+), 13 deletions(-) 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..35591297d 100644 --- a/codegen-examples/examples/delete_dead_code/run.py +++ b/codegen-examples/examples/delete_dead_code/run.py @@ -1,7 +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") def run(codebase: Codebase): 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 = [ From 017285cc190b0ca7544a7fe57da6e40ecfe0cc57 Mon Sep 17 00:00:00 2001 From: rushilpatel0 <171610820+rushilpatel0@users.noreply.github.com> Date: Wed, 12 Feb 2025 18:18:26 +0000 Subject: [PATCH 2/2] Automated pre-commit update --- codegen-examples/examples/delete_dead_code/run.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codegen-examples/examples/delete_dead_code/run.py b/codegen-examples/examples/delete_dead_code/run.py index 35591297d..cfaf14c0a 100644 --- a/codegen-examples/examples/delete_dead_code/run.py +++ b/codegen-examples/examples/delete_dead_code/run.py @@ -2,6 +2,7 @@ from codegen import Codebase from codegen.shared.enums.programming_language import ProgrammingLanguage + @codegen.function("delete-dead-code") def run(codebase: Codebase): removed_functions_count = 0