Swap programming_language argument to just language (#444)

Better user-facing API

Author: EdwardJXLi

codegen-examples/examples/cyclomatic_complexity/run.py

@@ -1,6 +1,5 @@
 import codegen
 from codegen import Codebase
-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
@@ -82,7 +81,7 @@ def calculate_cyclomatic_complexity(code_block):
 
 if __name__ == "__main__":
     print("🔍 Analyzing codebase...")
-    codebase = Codebase.from_repo("fastapi/fastapi", commit="887270ff8a54bb58c406b0651678a27589793d2f", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("fastapi/fastapi", commit="887270ff8a54bb58c406b0651678a27589793d2f", language="python")
 
     print("Running analysis...")
     run(codebase)

codegen-examples/examples/delete_dead_code/README.md

@@ -12,7 +12,7 @@ The script (`run.py`) performs the dead code removal in several key steps:
 1. **Codebase Loading**
 
    ```python
-   codebase = Codebase.from_repo("tox-dev/tox", programming_language=ProgrammingLanguage.PYTHON)
+   codebase = Codebase.from_repo("tox-dev/tox", language="python")
    ```
 
    - Loads a codebase using the `Codebase.from_repo` method

codegen-examples/examples/delete_dead_code/run.py

@@ -1,6 +1,5 @@
 import codegen
 from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 
 
 @codegen.function("delete-dead-code")
@@ -38,7 +37,7 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("🔍 Analyzing codebase...")
-    codebase = Codebase.from_repo("tox-dev/tox", programming_language=ProgrammingLanguage.PYTHON, commit="b588b696e0940c1813014b31b68d7660d8a1914f")
+    codebase = Codebase.from_repo("tox-dev/tox", language="python", commit="b588b696e0940c1813014b31b68d7660d8a1914f")
 
     print("🚮 Deleting dead code...")
     run(codebase)

codegen-examples/examples/dict_to_schema/run.py

@@ -1,5 +1,4 @@
 import codegen
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 from codegen import Codebase
 
 
@@ -97,7 +96,7 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("modal-labs/modal-client", commit="81941c24897889a2ff2f627c693fa734967e693c", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("modal-labs/modal-client", commit="81941c24897889a2ff2f627c693fa734967e693c", language="python")
 
     print("Running codemod...")
     run(codebase)

codegen-examples/examples/fragment_to_shorthand/run.py

@@ -1,6 +1,5 @@
 import codegen
 from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 
 
 @codegen.function("fragment_to_shorthand")
@@ -34,6 +33,6 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("🎯 Starting Fragment to shorthand conversion...")
-    codebase = Codebase.from_repo("RocketChat/Rocket.Chat", commit="a4f2102af1c2e875c60cafebd0163105bdaca678", programming_language=ProgrammingLanguage.TYPESCRIPT)
+    codebase = Codebase.from_repo("RocketChat/Rocket.Chat", commit="a4f2102af1c2e875c60cafebd0163105bdaca678", language="typescript")
     run(codebase)
     print("✅ Done! All Fragments converted to shorthand syntax!")

codegen-examples/examples/freezegun_to_timemachine_migration/run.py

@@ -1,5 +1,4 @@
 import codegen
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 from codegen import Codebase
 
 
@@ -59,5 +58,5 @@ def run(codebase: Codebase):
 
 
 if __name__ == "__main__":
-    codebase = Codebase.from_repo("getmoto/moto", commit="786a8ada7ed0c7f9d8b04d49f24596865e4b7901", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("getmoto/moto", commit="786a8ada7ed0c7f9d8b04d49f24596865e4b7901", language="python")
     run(codebase)

codegen-examples/examples/generate_training_data/run.py

@@ -2,7 +2,6 @@
 
 import codegen
 from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 from codegen.sdk.core.external_module import ExternalModule
 from codegen.sdk.core.import_resolution import Import
 from codegen.sdk.core.symbol import Symbol
@@ -95,7 +94,7 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("fastapi/fastapi", commit="887270ff8a54bb58c406b0651678a27589793d2f", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("fastapi/fastapi", commit="887270ff8a54bb58c406b0651678a27589793d2f", language="python")
 
     print("Generating training data...")
     training_data = run(codebase)

codegen-examples/examples/langchain_agent/run.py

@@ -14,7 +14,6 @@
     SemanticEditTool,
     ViewFileTool,
 )
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 from langchain import hub
 from langchain.agents import AgentExecutor
 from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent
@@ -93,7 +92,7 @@ def create_codebase_agent(
 if __name__ == "__main__":
     # Initialize codebase
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("fastapi/fastapi", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("fastapi/fastapi", language="python")
 
     # Create agent with history
     print("Creating agent...")

codegen-examples/examples/modules_dependencies/run.py

@@ -1,7 +1,6 @@
 import codegen
-from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 import networkx as nx
+from codegen import Codebase
 
 
 @codegen.function("visualize-modules-dependencies")
@@ -35,5 +34,5 @@ def run(codebase: Codebase):
 
 
 if __name__ == "__main__":
-    codebase = Codebase.from_repo("getsentry/sentry", commit="fb0d53b2210cc896fc3e2cf32dae149ea8a8a45a", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("getsentry/sentry", commit="fb0d53b2210cc896fc3e2cf32dae149ea8a8a45a", language="python")
     run(codebase)

codegen-examples/examples/openapi_decorators/run.py

@@ -1,6 +1,5 @@
 import codegen
 from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 
 
 def analyze_model_fields(method) -> dict:
@@ -262,6 +261,6 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("🎯 Starting OpenAPI decorators addition...")
-    codebase = Codebase.from_repo("mindsdb/mindsdb", commit="4b76c44bfaec789289e15fbdff7397e866009f94", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("mindsdb/mindsdb", commit="4b76c44bfaec789289e15fbdff7397e866009f94", language="python")
     run(codebase)
     print("✅ Done! OpenAPI decorators added to all API endpoints!")

codegen-examples/examples/pr_review_bot/run.py

@@ -1,8 +1,8 @@
+import json
+
 import codegen
 from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 from codegen.sdk.codebase.config import CodebaseConfig
-import json
 from codegen.sdk.secrets import Secrets
 from codegen.shared.configs.models.feature_flags import CodebaseFeatureFlags
 
@@ -76,7 +76,7 @@ def run(codebase: Codebase):
     codebase = Codebase.from_repo(
         "getsentry/sentry",
         shallow=False,
-        programming_language=ProgrammingLanguage.PYTHON,
+        language="python",
         config=CodebaseConfig(
             secrets=Secrets(openai_key=open_ai_key, github_api_key=github_token),
             feature_flags=CodebaseFeatureFlags(

codegen-examples/examples/reexport_management/run.py

@@ -1,10 +1,7 @@
 import codegen
 from codegen import Codebase
-
 from codegen.sdk.typescript.file import TSImport
 
-from codegen.shared.enums.programming_language import ProgrammingLanguage
-
 processed_imports = set()
 
 
@@ -125,6 +122,6 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("🎯 Starting reexport organization...")
-    codebase = Codebase("./", programming_language=ProgrammingLanguage.TYPESCRIPT)
+    codebase = Codebase("./", language="typescript")
     run(codebase)
     print("✅ Done! All reexports organized successfully!")

codegen-examples/examples/slack_chatbot/api.py

@@ -6,7 +6,6 @@
 import modal
 from codegen import Codebase
 from codegen.extensions import VectorIndex
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 from fastapi import FastAPI, Request
 from openai import OpenAI
 from slack_bolt import App
@@ -29,7 +28,7 @@ def format_response(answer: str, context: list[tuple[str, int]]) -> str:
 def answer_question(query: str) -> tuple[str, list[tuple[str, int]]]:
     """Use RAG to answer a question about FastAPI."""
     # Initialize codebase. Smart about caching.
-    codebase = Codebase.from_repo("codegen-sh/codegen-sdk", programming_language=ProgrammingLanguage.PYTHON, tmp_dir="/root")
+    codebase = Codebase.from_repo("codegen-sh/codegen-sdk", language="python", tmp_dir="/root")
 
     # Initialize vector index
     index = VectorIndex(codebase)

codegen-examples/examples/sqlalchemy_soft_delete/README.md

@@ -127,7 +127,7 @@ This codemod includes support for running without the graph feature enabled. Thi
 To run in no-graph mode:
 
 ```python
-codebase = Codebase(str(repo_path), programming_language=ProgrammingLanguage.PYTHON, config=CodebaseConfig(feature_flags=GSFeatureFlags(disable_graph=True)))
+codebase = Codebase(str(repo_path), language="python", config=CodebaseConfig(feature_flags=GSFeatureFlags(disable_graph=True)))
 ```
 
 ## Running the Conversion

codegen-examples/examples/sqlalchemy_soft_delete/run.py

@@ -1,11 +1,11 @@
+import shutil
+import subprocess
+from pathlib import Path
+
 import codegen
 from codegen import Codebase
 from codegen.sdk.core.detached_symbols.function_call import FunctionCall
 from codegen.shared.configs.models.feature_flags import CodebaseFeatureFlags
-from codegen.shared.enums.programming_language import ProgrammingLanguage
-import shutil
-import subprocess
-from pathlib import Path
 
 
 def should_process_join_call(call, soft_delete_models, join_methods):
@@ -92,16 +92,16 @@ def process_soft_deletes(codebase):
 
 
 if __name__ == "__main__":
-    from codegen.sdk.core.codebase import Codebase
     from codegen.sdk.codebase.config import CodebaseConfig
+    from codegen.sdk.core.codebase import Codebase
 
     repo_path = Path("/tmp/core")
     repo_url = "https://github.com/hasgeek/funnel.git"
 
     try:
         clone_repo(repo_url, repo_path)
         subprocess.run(["git", "-C", str(repo_path), "checkout", "8454e15"], check=True)
-        codebase = Codebase(str(repo_path), programming_language=ProgrammingLanguage.PYTHON, config=CodebaseConfig(feature_flags=CodebaseFeatureFlags(disable_graph=True)))
+        codebase = Codebase(str(repo_path), language="python", config=CodebaseConfig(feature_flags=CodebaseFeatureFlags(disable_graph=True)))
         process_soft_deletes(codebase)
     finally:
         shutil.rmtree(repo_path)

codegen-examples/examples/usesuspensequery_to_usesuspensequeries/run.py

@@ -1,6 +1,5 @@
 import codegen
 from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 from codegen.sdk.core.detached_symbols.function_call import FunctionCall
 
 
@@ -82,6 +81,6 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("deepfence/ThreatMapper", programming_language=ProgrammingLanguage.TYPESCRIPT)
+    codebase = Codebase.from_repo("deepfence/ThreatMapper", language="typescript")
     print("Running codemod...")
     run(codebase)

codegen-examples/examples/visualize_codebases/blast_radius.py

@@ -1,10 +1,9 @@
 import codegen
-from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 import networkx as nx
-from codegen.sdk.python.symbol import PySymbol
-from codegen.sdk.python.function import PyFunction
+from codegen import Codebase
 from codegen.sdk.core.dataclasses.usage import Usage
+from codegen.sdk.python.function import PyFunction
+from codegen.sdk.python.symbol import PySymbol
 
 # Create a directed graph for visualizing relationships between code elements
 G = nx.DiGraph()
@@ -112,7 +111,7 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("codegen-oss/posthog", commit="b174f2221ea4ae50e715eb6a7e70e9a2b0760800", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("codegen-oss/posthog", commit="b174f2221ea4ae50e715eb6a7e70e9a2b0760800", language="python")
     print(f"Codebase with {len(codebase.files)} files and {len(codebase.functions)} functions.")
     print("Creating graph...")
 

codegen-examples/examples/visualize_codebases/call_trace.py

@@ -1,11 +1,10 @@
 import codegen
-from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 import networkx as nx
+from codegen import Codebase
+from codegen.sdk.core.class_definition import Class
 from codegen.sdk.core.detached_symbols.function_call import FunctionCall
-from codegen.sdk.core.function import Function
 from codegen.sdk.core.external_module import ExternalModule
-from codegen.sdk.core.class_definition import Class
+from codegen.sdk.core.function import Function
 
 G = nx.DiGraph()
 
@@ -114,7 +113,7 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("codegen-oss/posthog", commit="b174f2221ea4ae50e715eb6a7e70e9a2b0760800", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("codegen-oss/posthog", commit="b174f2221ea4ae50e715eb6a7e70e9a2b0760800", language="python")
     print(f"Codebase with {len(codebase.files)} files and {len(codebase.functions)} functions.")
     print("Creating graph...")
 

codegen-examples/examples/visualize_codebases/dependency_trace.py

@@ -1,10 +1,9 @@
 import codegen
-from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 import networkx as nx
+from codegen import Codebase
 from codegen.sdk.core.class_definition import Class
-from codegen.sdk.core.symbol import Symbol
 from codegen.sdk.core.import_resolution import Import
+from codegen.sdk.core.symbol import Symbol
 
 G = nx.DiGraph()
 
@@ -76,7 +75,7 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("codegen-oss/posthog", commit="b174f2221ea4ae50e715eb6a7e70e9a2b0760800", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("codegen-oss/posthog", commit="b174f2221ea4ae50e715eb6a7e70e9a2b0760800", language="python")
     print(f"Codebase with {len(codebase.files)} files and {len(codebase.functions)} functions.")
     print("Creating graph...")
 

codegen-examples/examples/visualize_codebases/method_relationships.py

@@ -1,11 +1,10 @@
 import codegen
-from codegen import Codebase
-from codegen.shared.enums.programming_language import ProgrammingLanguage
 import networkx as nx
+from codegen import Codebase
+from codegen.sdk.core.class_definition import Class
 from codegen.sdk.core.detached_symbols.function_call import FunctionCall
-from codegen.sdk.core.function import Function
 from codegen.sdk.core.external_module import ExternalModule
-from codegen.sdk.core.class_definition import Class
+from codegen.sdk.core.function import Function
 
 G = nx.DiGraph()
 
@@ -100,7 +99,7 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("codegen-oss/modal-client", commit="00bf226a1526f9d775d2d70fc7711406aaf42958", programming_language=ProgrammingLanguage.PYTHON)
+    codebase = Codebase.from_repo("codegen-oss/modal-client", commit="00bf226a1526f9d775d2d70fc7711406aaf42958", language="python")
     print(f"Codebase with {len(codebase.files)} files and {len(codebase.functions)} functions.")
     print("Creating graph...")