From 1b73e1d794df4405139c8a3e8ea3782bc286516a Mon Sep 17 00:00:00 2001
From: Jerry Liu <jerryjliu98@gmail.com>
Date: Sun, 30 Apr 2023 20:34:29 -0700
Subject: [PATCH] fix langchain deps (#1740)
---
benchmarks/struct_indices/spider/generate_sql.py | 2 +-
experimental/cli/configuration.py | 2 +-
gpt_index/langchain_helpers/agents/agents.py | 2 +-
gpt_index/llm_predictor/base.py | 2 +-
gpt_index/llm_predictor/chatgpt.py | 2 +-
gpt_index/prompts/base.py | 2 +-
6 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/benchmarks/struct_indices/spider/generate_sql.py b/benchmarks/struct_indices/spider/generate_sql.py
index 0cd99f9b17..6024998fa5 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 c7b914c5ca..3f3b5e40f0 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 396a7e8c3a..b077511978 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 a22cd43860..803d531516 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 5b2e1c638d..63eed37c61 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 da45d4f3cc..13b6154ace 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
--
GitLab