fix: type imports

Author: rushilpatel0

src/codegen/extensions/events/app.py

@@ -1,21 +1,9 @@
-from typing import Protocol
 import modal
 import logging
-
 from codegen.extensions.events.linear import Linear
 
-logger = logging.getLogger(__name__)
-
 
-
-class EventHandlerManagerProtocol(Protocol):
-    def subscribe_handler_to_webhook(self, func_name: str, modal_app: modal.App, event_name):
-        pass
-    def unsubscribe_handler_to_webhook(self, func_name: str, modal_app: modal.App, event_name):
-        pass
-
-    def unsubscribe_all_handlers(self):
-        pass
+logger = logging.getLogger(__name__)
 
 
 class CodegenApp(modal.App):

src/codegen/extensions/events/interface.py

@@ -0,0 +1,14 @@
+
+from typing import TYPE_CHECKING, Protocol
+
+if TYPE_CHECKING:
+    from modal import App
+
+class EventHandlerManagerProtocol(Protocol):
+    def subscribe_handler_to_webhook(self, func_name: str, modal_app: App, event_name):
+        pass
+    def unsubscribe_handler_to_webhook(self, func_name: str, modal_app: App, event_name):
+        pass
+
+    def unsubscribe_all_handlers(self):
+        pass

src/codegen/extensions/events/linear.py

@@ -1,16 +1,18 @@
-from typing import Callable
+from typing import TYPE_CHECKING, Callable
 from anthropic import BaseModel
-from codegen.extensions.clients.linear import LinearClient
-
 import functools
 import os
 import modal
 import logging
 
-from codegen.extensions.events.app import CodegenApp, EventHandlerManagerProtocol
+from codegen.extensions.events.interface import EventHandlerManagerProtocol
+from codegen.extensions.clients.linear import LinearClient
+
+if TYPE_CHECKING:
+    from codegen.extensions.events.app import CodegenApp
 
-logger = logging.getLogger(__name__)
 
+logger = logging.getLogger(__name__)
 
 
 class RegisteredWebhookHandler(BaseModel):