fix langchain deps (#1740)

benchmarks/struct_indices/spider/generate_sql.py

@@ -7,7 +7,7 @@ import re
 
 from langchain.chat_models import ChatOpenAI
 from langchain.llms import OpenAI
-from langchain.schema import BaseLanguageModel
+from langchain.base_language import BaseLanguageModel
 from sqlalchemy import create_engine, text
 from tqdm import tqdm
 

experimental/cli/configuration.py

@@ -3,7 +3,7 @@ from configparser import ConfigParser, SectionProxy
 from typing import Any, Type
 from gpt_index.embeddings.openai import OpenAIEmbedding
 from langchain import OpenAI
-from langchain.schema import BaseLanguageModel
+from langchain.base_language import BaseLanguageModel
 from gpt_index.indices.base import BaseGPTIndex
 from gpt_index.embeddings.base import BaseEmbedding
 from gpt_index import (

gpt_index/langchain_helpers/agents/agents.py

@@ -3,7 +3,7 @@
 from typing import Any, Optional
 
 from langchain.agents import AgentExecutor, initialize_agent
-from langchain.callbacks import BaseCallbackManager
+from langchain.callbacks.base import BaseCallbackManager
 from langchain.llms.base import BaseLLM
 from langchain.agents.agent_types import AgentType
 

gpt_index/llm_predictor/base.py

@@ -9,7 +9,7 @@ import openai
 from langchain import Cohere, LLMChain, OpenAI
 from langchain.chat_models import ChatOpenAI
 from langchain.llms import AI21
-from langchain.schema import BaseLanguageModel
+from langchain.base_language import BaseLanguageModel
 
 from gpt_index.constants import MAX_CHUNK_SIZE, NUM_OUTPUTS
 from gpt_index.prompts.base import Prompt

gpt_index/llm_predictor/chatgpt.py

@@ -13,7 +13,7 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
 )
 from langchain.prompts.prompt import PromptTemplate
-from langchain.schema import BaseLanguageModel, BaseMessage
+from langchain.base_language import BaseLanguageModel, BaseMessage
 
 from gpt_index.llm_predictor.base import LLMPredictor
 from gpt_index.prompts.base import Prompt

gpt_index/prompts/base.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional, Type, TypeVar
 from langchain import BasePromptTemplate as BaseLangchainPrompt
 from langchain import PromptTemplate as LangchainPrompt
 from langchain.chains.prompt_selector import ConditionalPromptSelector
-from langchain.schema import BaseLanguageModel
+from langchain.base_language import BaseLanguageModel
 
 from gpt_index.output_parsers.base import BaseOutputParser
 from gpt_index.prompts.prompt_type import PromptType