From 69084cb5c75828ea158e75112561e64db39dc307 Mon Sep 17 00:00:00 2001
From: jayhack <jayhack.0@gmail.com>
Date: Mon, 3 Mar 2025 11:12:15 -0800
Subject: [PATCH] .

---
 pyproject.toml                               | 1 +
 src/codegen/cli/commands/serve/main.py       | 2 +-
 src/codegen/extensions/events/codegen_app.py | 9 ++++++---
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 4a092d1a6..a30bc5219 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -77,6 +77,7 @@ dependencies = [
   "urllib3>=2.0.0",
   "datasets",
   "colorlog>=6.9.0",
+  "langsmith>=0.3.11",
 ]
 
 license = { text = "Apache-2.0" }
diff --git a/src/codegen/cli/commands/serve/main.py b/src/codegen/cli/commands/serve/main.py
index c8d0868fd..4c2dbc12e 100644
--- a/src/codegen/cli/commands/serve/main.py
+++ b/src/codegen/cli/commands/serve/main.py
@@ -81,7 +81,7 @@ def create_app_module(file_path: Path) -> str:
     module_name = f"codegen_app_{file_path.stem}"
     module_code = f"""
 from {file_path.stem} import app
-app = app
+app = app.app  # Get the FastAPI instance from the CodegenApp
 """
     module_path = file_path.parent / f"{module_name}.py"
     module_path.write_text(module_code)
diff --git a/src/codegen/extensions/events/codegen_app.py b/src/codegen/extensions/events/codegen_app.py
index 8034b3619..024730168 100644
--- a/src/codegen/extensions/events/codegen_app.py
+++ b/src/codegen/extensions/events/codegen_app.py
@@ -161,15 +161,18 @@ def _setup_routes(self):
         async def _root():
             return await self.root()
 
-        @self.app.post("/{org}/{repo}/slack/events")
+        # @self.app.post("/{org}/{repo}/slack/events")
+        @self.app.post("/slack/events")
         async def _handle_slack_event(request: Request):
             return await self.handle_slack_event(request)
 
-        @self.app.post("/{org}/{repo}/github/events")
+        # @self.app.post("/{org}/{repo}/github/events")
+        @self.app.post("/github/events")
         async def _handle_github_event(request: Request):
             return await self.handle_github_event(request)
 
-        @self.app.post("/{org}/{repo}/linear/events")
+        # @self.app.post("/{org}/{repo}/linear/events")
+        @self.app.post("/linear/events")
         async def handle_linear_event(request: Request):
             return await self.handle_linear_event(request)