Shroominic commited on
Commit
18fc95f
·
1 Parent(s): 474f867

fix circular import

Browse files
Files changed (1) hide show
  1. codeinterpreterapi/callbacks.py +5 -3
codeinterpreterapi/callbacks.py CHANGED
@@ -1,12 +1,14 @@
1
  from uuid import UUID
2
- from typing import Any, Optional, List
3
  from langchain.schema import AgentAction
4
  from langchain.callbacks import AsyncIteratorCallbackHandler
5
- from codeinterpreterapi.session import CodeInterpreterSession
 
 
6
 
7
 
8
  class CodeCallbackHandler(AsyncIteratorCallbackHandler):
9
- def __init__(self, session: CodeInterpreterSession):
10
  self.session = session
11
  super().__init__()
12
 
 
1
  from uuid import UUID
2
+ from typing import Any, Optional, TYPE_CHECKING
3
  from langchain.schema import AgentAction
4
  from langchain.callbacks import AsyncIteratorCallbackHandler
5
+
6
+ if TYPE_CHECKING:
7
+ from codeinterpreterapi.session import CodeInterpreterSession
8
 
9
 
10
  class CodeCallbackHandler(AsyncIteratorCallbackHandler):
11
+ def __init__(self, session: "CodeInterpreterSession"):
12
  self.session = session
13
  super().__init__()
14