Langchain 0.1.0 update (#121)

Author: jexp

api.py

@@ -1,6 +1,6 @@
 import os
 
-from langchain.graphs import Neo4jGraph
+from langchain_community.graphs import Neo4jGraph
 from dotenv import load_dotenv
 from utils import (
     create_vector_index,

bot.py

@@ -3,7 +3,7 @@
 import streamlit as st
 from streamlit.logger import get_logger
 from langchain.callbacks.base import BaseCallbackHandler
-from langchain.graphs import Neo4jGraph
+from langchain_community.graphs import Neo4jGraph
 from dotenv import load_dotenv
 from utils import (
     create_vector_index,

chains.py

@@ -1,18 +1,26 @@
-from langchain.embeddings.openai import OpenAIEmbeddings
-from langchain.embeddings import (
-    OllamaEmbeddings,
-    SentenceTransformerEmbeddings,
-    BedrockEmbeddings,
-)
-from langchain.chat_models import ChatOpenAI, ChatOllama, BedrockChat
-from langchain.vectorstores.neo4j_vector import Neo4jVector
+
+from langchain_openai import OpenAIEmbeddings
+from langchain_community.embeddings import OllamaEmbeddings
+from langchain_community.embeddings import BedrockEmbeddings
+from langchain_community.embeddings.sentence_transformer import SentenceTransformerEmbeddings
+
+from langchain_openai import ChatOpenAI
+from langchain_community.chat_models import ChatOllama
+from langchain_community.chat_models import BedrockChat
+
+from langchain_community.graphs import Neo4jGraph
+
+from langchain_community.vectorstores import Neo4jVector
+
 from langchain.chains import RetrievalQAWithSourcesChain
 from langchain.chains.qa_with_sources import load_qa_with_sources_chain
-from langchain.prompts.chat import (
+
+from langchain.prompts import (
     ChatPromptTemplate,
-    SystemMessagePromptTemplate,
     HumanMessagePromptTemplate,
+    SystemMessagePromptTemplate
 )
+
 from typing import List, Any
 from utils import BaseLogger, extract_title_and_question
 

loader.py

@@ -1,7 +1,7 @@
 import os
 import requests
 from dotenv import load_dotenv
-from langchain.graphs import Neo4jGraph
+from langchain_community.graphs import Neo4jGraph
 import streamlit as st
 from streamlit.logger import get_logger
 from chains import load_embedding_model

pdf_bot.py

@@ -3,9 +3,9 @@
 import streamlit as st
 from langchain.chains import RetrievalQA
 from PyPDF2 import PdfReader
-from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.callbacks.base import BaseCallbackHandler
-from langchain.vectorstores.neo4j_vector import Neo4jVector
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain_community.vectorstores import Neo4jVector
 from streamlit.logger import get_logger
 from chains import (
     load_embedding_model,

requirements.txt

@@ -1,4 +1,4 @@
-openai==0.28.1
+openai
 python-dotenv
 wikipedia
 tiktoken
@@ -13,3 +13,6 @@ pydantic
 uvicorn
 sse-starlette
 boto3
+# missing from the langchain base image?
+langchain-openai
+langchain-community