fix circular import

id-2 · Jul 14, 2023 · 18fc95f · 18fc95f
1 parent 474f867
commit 18fc95f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/codeinterpreterapi/callbacks.py b/codeinterpreterapi/callbacks.py
@@ -1,12 +1,14 @@
 from uuid import UUID
-from typing import Any, Optional, List
+from typing import Any, Optional, TYPE_CHECKING
 from langchain.schema import AgentAction
 from langchain.callbacks import AsyncIteratorCallbackHandler
-from codeinterpreterapi.session import CodeInterpreterSession
+
+if TYPE_CHECKING:
+    from codeinterpreterapi.session import CodeInterpreterSession
 
 
 class CodeCallbackHandler(AsyncIteratorCallbackHandler):
-    def __init__(self, session: CodeInterpreterSession):
+    def __init__(self, session: "CodeInterpreterSession"):
         self.session = session
         super().__init__()