Fix langchain base chain import

PiperOrigin-RevId: 643323466
google · Jun 14, 2024 · b102d5f · b102d5f
1 parent 8d61619
commit b102d5f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/genc/python/interop/langchain/custom_chain.py b/genc/python/interop/langchain/custom_chain.py
@@ -17,11 +17,12 @@
 
 from langchain import chains
 from langchain.callbacks import manager
+from langchain.chains import base as chains_base
 
 from genc.proto.v0 import computation_pb2 as pb
 
 ChainedOpType: TypeAlias = Union[
-    int, pb.Value, chains.base.Chain, chains.LLMChain, Any
+    int, pb.Value, chains_base.Chain, chains.LLMChain, Any
 ]