Automated pre-commit update

Author: rushilpatel0

codegen-examples/examples/snapshot_event_handler/event_handlers.py

@@ -3,6 +3,7 @@
 from fastapi import FastAPI, Request
 
 from codegen.extensions.github.types.pull_request import PullRequestLabeledEvent
+
 load_dotenv(".env")
 
 import modal
@@ -40,22 +41,21 @@
         "fastapi[standard]",
         "slack_sdk",
         "fastapi_utils[all]",
-        "classy-fastapi>=0.6.1"
+        "classy-fastapi>=0.6.1",
     )
-
-)   
+)
 
 
 event_handlers_app = modal.App("codegen-event-handlers")
 
 
 @event_handlers_app.cls(image=base_image, secrets=[modal.Secret.from_dotenv(".env")], enable_memory_snapshot=True, container_idle_timeout=300)
 class CustomEventHandlersAPI(CodebaseEventsApp):
-    commit: str = modal.parameter(default="79114f67ccfe8700416cd541d1c7c43659a95342") 
+    commit: str = modal.parameter(default="79114f67ccfe8700416cd541d1c7c43659a95342")
     repo_org: str = modal.parameter(default="codegen-sh")
     repo_name: str = modal.parameter(default="Kevin-s-Adventure-Game")
     snapshot_index_id: str = SNAPSHOT_DICT_ID
-    
+
     def setup_handlers(self, cg: CodegenApp):
         @cg.slack.event("app_mention")
         async def handle_mention(event: SlackEvent):
@@ -76,7 +76,6 @@ async def handle_mention(event: SlackEvent):
 
             return {"message": "Mentioned", "received_text": event.text, "response": response}
 
-
         @cg.github.event("pull_request:labeled")
         def handle_pr(event: PullRequestLabeledEvent):
             logger.info("PR labeled")
@@ -90,37 +89,30 @@ def handle_pr(event: PullRequestLabeledEvent):
             logger.info("> Checking out commit")
             codebase.checkout(commit=event.pull_request.head.sha)
 
-
             logger.info("> Running PR Lints")
             # LINT CODEMOD
             lint_for_dev_import_violations(codebase, event)
 
             # REVIEW CODEMOD
             review_with_codegen_agent(codebase, event)
 
-            
-
             return {"message": "PR event handled", "num_files": len(codebase.files), "num_functions": len(codebase.functions)}
 
-
         @cg.linear.event("Issue")
         def handle_issue(event: LinearEvent):
             logger.info(f"Issue created: {event}")
             codebase = cg.get_codebase()
             return {"message": "Linear Issue event", "num_files": len(codebase.files), "num_functions": len(codebase.functions)}
 
 
-
-@codegen_events_app.cls(image=base_image, secrets=[modal.Secret.from_dotenv('.env')])
+@codegen_events_app.cls(image=base_image, secrets=[modal.Secret.from_dotenv(".env")])
 class WebhookEventRouterAPI(EventRouterMixin, Routable):
-
     snapshot_index_id: str = SNAPSHOT_DICT_ID
-    
 
     def get_event_handler_cls(self):
         modal_cls = modal.Cls.from_name(app_name="Events", name="CustomEventHandlersAPI")
         return modal_cls
-    
+
     @post("/{org}/{repo}/{provider}/events")
     async def handle_event(self, org: str, repo: str, provider: Literal["slack", "github", "linear"], request: Request):
         # Define the route for the webhook url sink, it will need to indicate the repo repo org, and the provider
@@ -133,14 +125,14 @@ def api(self):
         route_view = WebhookEventRouterAPI()
         event_api.include_router(route_view.router)
         return event_api
-    
 
-# Setup a cron job to trigger updates to the codebase snapshots. 
+
+# Setup a cron job to trigger updates to the codebase snapshots.
 @codegen_events_app.function(schedule=modal.Cron("*/10 * * * *"), image=base_image, secrets=[modal.Secret.from_dotenv(".env")])
 def refresh_repository_snapshots():
     WebhookEventRouterAPI().refresh_repository_snapshots(snapshot_index_id=SNAPSHOT_DICT_ID)
 
 
-app = modal.App("Events", secrets=[modal.Secret.from_dotenv('.env')])
+app = modal.App("Events", secrets=[modal.Secret.from_dotenv(".env")])
 app.include(event_handlers_app)
 app.include(codegen_events_app)

codegen-examples/examples/snapshot_event_handler/pr_tasks.py

@@ -1,9 +1,5 @@
-
-
-
 import logging
 from codegen.agents.code_agent import CodeAgent
-from codegen.configs.models.secrets import SecretsConfig
 from codegen.extensions.github.types.pull_request import PullRequestLabeledEvent
 
 from codegen.extensions.langchain.tools import GithubCreatePRCommentTool, GithubCreatePRReviewCommentTool, GithubViewPRTool
@@ -12,61 +8,61 @@
 logging.basicConfig(level=logging.INFO, force=True)
 logger = logging.getLogger(__name__)
 
+
 def lint_for_dev_import_violations(codebase: Codebase, event: PullRequestLabeledEvent):
-         
     patch, commit_shas, modified_symbols = codebase.get_modified_symbols_in_pr(event.pull_request.number)
     modified_files = set(commit_shas.keys())
 
-    DIR_NAME = 'packages/next/src/client/components/react-dev-overlay'
+    DIR_NAME = "packages/next/src/client/components/react-dev-overlay"
     directory = codebase.get_directory(DIR_NAME)
 
     # Initialize a list to store all violations
     violations = []
 
-    print('modified_files', modified_files)
-    
+    print("modified_files", modified_files)
+
     # Check if directory exists before proceeding
-    if directory is not None and hasattr(directory, 'files'):
+    if directory is not None and hasattr(directory, "files"):
         for file in directory.files:
-            print('checking file', file.filepath)
+            print("checking file", file.filepath)
             for imp in file.inbound_imports:
-                print('file', imp.file.filepath)
-                print('checking import', imp.import_statement)
+                print("file", imp.file.filepath)
+                print("checking import", imp.import_statement)
                 # Check if the import is from outside the directory and is in the modified files
                 if imp.file not in directory and imp.file.filepath in modified_files:
                     # Skip require statements
-                    if 'require' in imp.import_statement:
+                    if "require" in imp.import_statement:
                         continue
-                    violation = f'- Violation in `{file.filepath}`: Importing from `{imp.file.filepath}` ([link]({imp.github_url}))'
+                    violation = f"- Violation in `{file.filepath}`: Importing from `{imp.file.filepath}` ([link]({imp.github_url}))"
                     violations.append(violation)
                     logger.info(f"Found violation: {violation}")
-        
+
         # Only create a PR comment if violations are found
         if violations:
             review_attention_message = "## Dev Import Violations Found\n\n"
             review_attention_message += "The following files have imports that violate development overlay rules:\n\n"
             review_attention_message += "\n".join(violations)
             review_attention_message += "\n\nPlease ensure that development imports are not imported in production code."
-            
+
             # Create PR comment with the formatted message
             codebase._op.create_pr_comment(event.pull_request.number, review_attention_message)
 
 
 def review_with_codegen_agent(codebase: Codebase, event: PullRequestLabeledEvent):
-        review_initial_message = f"CodegenBot is starting to review the PR please wait..."    
-        comment = codebase._op.create_pr_comment(event.number, review_initial_message)
-        # Define tools first
-        pr_tools = [
-            GithubViewPRTool(codebase),
-            GithubCreatePRCommentTool(codebase),
-            GithubCreatePRReviewCommentTool(codebase),
-        ]
-
-        # Create agent with the defined tools
-        agent = CodeAgent(codebase=codebase, tools=pr_tools)
-
-        # Using a prompt from SWE Bench
-        prompt = f"""
+    review_initial_message = "CodegenBot is starting to review the PR please wait..."
+    comment = codebase._op.create_pr_comment(event.number, review_initial_message)
+    # Define tools first
+    pr_tools = [
+        GithubViewPRTool(codebase),
+        GithubCreatePRCommentTool(codebase),
+        GithubCreatePRReviewCommentTool(codebase),
+    ]
+
+    # Create agent with the defined tools
+    agent = CodeAgent(codebase=codebase, tools=pr_tools)
+
+    # Using a prompt from SWE Bench
+    prompt = f"""
     Hey CodegenBot!
 
     Here's a SWE task for you. Please Review this pull request!
@@ -77,6 +73,6 @@ def review_with_codegen_agent(codebase: Codebase, event: PullRequestLabeledEvent
     be explicit about the changes, produce a short summary, and point out possible improvements where pressent dont be self congratulatory stick to the facts
     use the tools at your disposal to create propper pr reviews include code snippets if needed, and suggest improvements if feel its necesary
     """
-        # Run the agent
-        result = agent.run(prompt)
-        comment.delete()
+    # Run the agent
+    agent.run(prompt)
+    comment.delete()

codegen-examples/examples/snapshot_event_handler/pyproject.toml

@@ -4,24 +4,24 @@ version = "0.0.0"
 readme = "README.md"
 requires-python = ">=3.12, <3.14"
 dependencies = [
-    "fastapi-utils>=0.8.0",
-    "fastapi[standard]>=0.115.9",
-    "modal>=0.73.72",
-    "fastapi-restful[all]>=0.6.0",
-    "classy-fastapi>=0.6.1",
-    "codegen@git+https://github.com/codegen-sh/codegen-sdk.git@f398107d31bbd53fc77bc9c5f8763d2dc8fcae5b",
+  "fastapi-utils>=0.8.0",
+  "fastapi[standard]>=0.115.9",
+  "modal>=0.73.72",
+  "fastapi-restful[all]>=0.6.0",
+  "classy-fastapi>=0.6.1",
+  "codegen@git+https://github.com/codegen-sh/codegen-sdk.git@f398107d31bbd53fc77bc9c5f8763d2dc8fcae5b",
 ]
 license = { file = "LICENSE" }
 classifiers = [
-    "License :: OSI Approved :: Apache Software License",
-    "Intended Audience :: Developers",
-    "Programming Language :: Python :: 3.12",
-    "Programming Language :: Python :: 3.13",
-    "Topic :: Software Development",
-    "Development Status :: 4 - Beta",
-    "Environment :: MacOS X",
-    "Programming Language :: Python :: 3",
-    "Programming Language :: Python",
+  "License :: OSI Approved :: Apache Software License",
+  "Intended Audience :: Developers",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Topic :: Software Development",
+  "Development Status :: 4 - Beta",
+  "Environment :: MacOS X",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python",
 ]
 
 [tool.ruff]

src/codegen/extensions/events/codegen_app.py

@@ -15,6 +15,7 @@
 
 logger = get_logger(__name__)
 
+
 class CodegenApp:
     """A FastAPI-based application for handling various code-related events."""
 
@@ -38,11 +39,9 @@ def __init__(self, name: str, repo: Optional[str] = None, tmp_dir: str = "/tmp/c
         # Initialize codebase cache
         self.codebase: Codebase | None = None
 
-
         # Register routes
         self._setup_routes()
 
-
     def parse_repo(self) -> None:
         # Parse initial repos if provided
         if self.repo:
@@ -57,10 +56,7 @@ def _parse_repo(self, repo_name: str, commit: str | None = None) -> None:
         try:
             logger.info(f"[CODEBASE] Parsing repository: {repo_name}")
             config = CodebaseConfig(sync_enabled=True)
-            secrets = SecretsConfig(
-                github_token=os.environ.get("GITHUB_ACCESS_TOKEN"),
-                linear_api_key=os.environ.get("LINEAR_ACCESS_TOKEN")
-            )
+            secrets = SecretsConfig(github_token=os.environ.get("GITHUB_ACCESS_TOKEN"), linear_api_key=os.environ.get("LINEAR_ACCESS_TOKEN"))
             self.codebase = Codebase.from_repo(repo_full_name=repo_name, tmp_dir=self.tmp_dir, commit=commit, config=config, secrets=secrets)
             logger.info(f"[CODEBASE] Successfully parsed and cached: {repo_name}")
         except Exception as e:
@@ -80,7 +76,7 @@ def get_codebase(self) -> Codebase:
             KeyError: If the repository hasn't been parsed
         """
         if not self.codebase:
-            msg = f"Repository has not been parsed"
+            msg = "Repository has not been parsed"
             raise KeyError(msg)
         return self.codebase
 
@@ -111,10 +107,10 @@ async def simulate_event(self, provider: str, event_type: str, payload: dict) ->
 
         handler = provider_map[provider]
         return await handler.handle(payload)
-    
+
     async def root(self):
-            """Render the main page."""
-            return """
+        """Render the main page."""
+        return """
             <!DOCTYPE html>
             <html>
                 <head>
@@ -142,17 +138,17 @@ async def root(self):
                 </body>
             </html>
             """
-        
+
     async def handle_slack_event(self, request: Request):
-            """Handle incoming Slack events."""
-            payload = await request.json()
-            return await self.slack.handle(payload)
+        """Handle incoming Slack events."""
+        payload = await request.json()
+        return await self.slack.handle(payload)
 
     async def handle_github_event(self, request: Request):
         """Handle incoming GitHub events."""
         payload = await request.json()
         return await self.github.handle(payload, request)
-    
+
     async def handle_linear_event(self, request: Request):
         """Handle incoming Linear events."""
         payload = await request.json()
@@ -176,8 +172,6 @@ async def _handle_github_event(request: Request):
         @self.app.post("/{org}/{repo}/linear/events")
         async def handle_linear_event(request: Request):
             return await self.handle_linear_event(request)
-        
-
 
     def run(self, host: str = "0.0.0.0", port: int = 8000, **kwargs):
         """Run the FastAPI application."""