From 8ced7b8f187798516fec820596d833713bcad7cf Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Tue, 2 May 2023 11:16:47 -0400 Subject: [PATCH] Fix for breaking change in langchain (#147) Fix for breaking change in langchain: BaseLanguageModel was moved from ```python from langchain.schema import BaseLanguageModel ``` To: ```python from langchain.base_language import BaseLanguageModel ``` Fix for this issue: https://github.com/eyurtsev/kor/issues/145 --- kor/extraction/api.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/kor/extraction/api.py b/kor/extraction/api.py index 1a3c72f..5b656b4 100644 --- a/kor/extraction/api.py +++ b/kor/extraction/api.py @@ -5,7 +5,11 @@ from langchain import PromptTemplate from langchain.chains import LLMChain from langchain.docstore.document import Document -from langchain.schema import BaseLanguageModel + +try: # Handle breaking change in langchain + from langchain.base_language import BaseLanguageModel +except ImportError: + from langchain.schema import BaseLanguageModel from kor.encoders import Encoder, InputFormatter, initialize_encoder from kor.extraction.typedefs import DocumentExtraction, Extraction