Adds programming language

examples/dict_to_schema/run.py

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

examples/flask_to_fastapi_migration/run.py

@@ -1,7 +1,7 @@
+import codebase
 from codegen import Codebase
 
 # Initialize codebase
-codebase = Codebase("./")
 
 # Define the target directory
 TARGET_DIR = "repo-before"
@@ -98,7 +98,8 @@ def update_jinja2_syntax(file):
             func_call.set_kwarg("request", "request")
 
 
-def main():
+@codebase.function("flask_to_fastapi_migration")
+def run():
     """Main function to run the Flask to FastAPI migration"""
     print("🚀 Starting Flask to FastAPI migration...\n")
 
@@ -128,4 +129,6 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    codebase = Codebase("./")
+
+    run()

examples/freezegun_to_timemachine_migration/run.py

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

examples/generate_training_data/run.py

@@ -2,6 +2,7 @@
 
 import codegen
 from codegen import Codebase
+from codegen.sdk.enums 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
@@ -94,7 +95,7 @@ def run(codebase: Codebase):
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("fastapi/fastapi")
+    codebase = Codebase.from_repo("fastapi/fastapi", commit="887270ff8a54bb58c406b0651678a27589793d2f", programming_language=ProgrammingLanguage.PYTHON)
 
     print("Generating training data...")
     training_data = run(codebase)

examples/usesuspensequery_to_usesuspensequeries/run.py

@@ -1,5 +1,6 @@
 import codegen
 from codegen import Codebase
+from codegen.sdk.enums import ProgrammingLanguage
 from codegen.sdk.core.detached_symbols.function_call import FunctionCall
 
 
@@ -76,11 +77,11 @@ def run(codebase: Codebase):
     print("\nModification complete:")
     print(f"Files modified: {files_modified}")
     print(f"Functions modified: {functions_modified}")
+    codebase.commit()
 
 
 if __name__ == "__main__":
     print("Initializing codebase...")
-    codebase = Codebase.from_repo("deepfence/ThreatMapper")
-
+    codebase = Codebase.from_repo("deepfence/ThreatMapper", programming_language=ProgrammingLanguage.TYPESCRIPT)
     print("Running codemod...")
     run(codebase)