diff --git a/benchmarks/struct_indices/spider/generate_sql.py b/benchmarks/struct_indices/spider/generate_sql.py index 0cd99f9b17ee649648c7fd7b76f59ec95f89176c..6024998fa57e8f89d66dfd0dcbf7b4ab151d5978 100644 --- a/benchmarks/struct_indices/spider/generate_sql.py +++ b/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 diff --git a/experimental/cli/configuration.py b/experimental/cli/configuration.py index c7b914c5cad324fe5a93710cdc8967911f680f73..3f3b5e40f0b24df1fb06da7dcf9236100ad99106 100644 --- a/experimental/cli/configuration.py +++ b/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 ( diff --git a/gpt_index/langchain_helpers/agents/agents.py b/gpt_index/langchain_helpers/agents/agents.py index 396a7e8c3a344ec9d0fa18d62ab2962cf8c12ec7..b0775119787c44e66896ffbfd9d074c77cb7411f 100644 --- a/gpt_index/langchain_helpers/agents/agents.py +++ b/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 diff --git a/gpt_index/llm_predictor/base.py b/gpt_index/llm_predictor/base.py index a22cd43860bce7240a58813ffadaea32e6b045fc..803d53151670fbf842ef1696c353cfcaa544658e 100644 --- a/gpt_index/llm_predictor/base.py +++ b/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 diff --git a/gpt_index/llm_predictor/chatgpt.py b/gpt_index/llm_predictor/chatgpt.py index 5b2e1c638d05e25d1d762dad18471f4481d6db98..63eed37c613ba7f1b2430b4b52c74c3c57a12bda 100644 --- a/gpt_index/llm_predictor/chatgpt.py +++ b/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 diff --git a/gpt_index/prompts/base.py b/gpt_index/prompts/base.py index da45d4f3cc5be292e65dca886cf2ff318ae076d4..13b6154ace26240d7355cce23a706a9cf0ba5a36 100644 --- a/gpt_index/prompts/base.py +++ b/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