Skip to content

Commit 69a08b0

Browse files
Nir Diamantclaude
authored andcommitted
Update deprecated LangChain imports to modern package structure
Migrate imports from monolithic langchain package to the split packages: - langchain.prompts -> langchain_core.prompts - langchain.docstore.document -> langchain_core.documents - langchain.vectorstores -> langchain_community.vectorstores - langchain.embeddings -> langchain_openai - langchain.schema -> langchain_core.messages - langchain.text_splitter -> langchain_text_splitters - langchain.callbacks -> langchain_community.callbacks - langchain.tools -> langchain_community.tools Also fix deprecated model name gpt-4-turbo-preview -> gpt-4-turbo Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 3613702 commit 69a08b0

11 files changed

Lines changed: 22 additions & 22 deletions

all_rag_techniques_runnable_scripts/context_enrichment_window_around_chunk.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22
import sys
33
from dotenv import load_dotenv
4-
from langchain.docstore.document import Document
4+
from langchain_core.documents import Document
55
from helper_functions import *
66
from evaluation.evalute_rag import *
77
from typing import List

all_rag_techniques_runnable_scripts/crag.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22
import sys
33
import argparse
44
from dotenv import load_dotenv
5-
from langchain.prompts import PromptTemplate
5+
from langchain_core.prompts import PromptTemplate
66
from langchain_openai import ChatOpenAI
77
from langchain_core.pydantic_v1 import BaseModel, Field
8-
from langchain.tools import DuckDuckGoSearchResults
8+
from langchain_community.tools import DuckDuckGoSearchResults
99
from helper_functions import encode_pdf
1010
import json
1111

all_rag_techniques_runnable_scripts/document_augmentation.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
import sys
22
import os
33
import re
4-
from langchain.docstore.document import Document
5-
from langchain.vectorstores import FAISS
4+
from langchain_core.documents import Document
5+
from langchain_community.vectorstores import FAISS
66
from enum import Enum
7-
from langchain.embeddings.openai import OpenAIEmbeddings
7+
from langchain_openai import OpenAIEmbeddings
88
from langchain_openai import ChatOpenAI
99
from typing import Any, Dict, List, Tuple
1010
from pydantic import BaseModel, Field

all_rag_techniques_runnable_scripts/fusion_retrieval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22
import sys
33
from dotenv import load_dotenv
4-
from langchain.docstore.document import Document
4+
from langchain_core.documents import Document
55
from typing import List
66
from rank_bm25 import BM25Okapi
77
import numpy as np

all_rag_techniques_runnable_scripts/graph_rag.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
import networkx as nx
2-
from langchain.vectorstores import FAISS
3-
from langchain.text_splitter import RecursiveCharacterTextSplitter
4-
from langchain.prompts import PromptTemplate
2+
from langchain_community.vectorstores import FAISS
3+
from langchain_text_splitters import RecursiveCharacterTextSplitter
4+
from langchain_core.prompts import PromptTemplate
55
from langchain.retrievers import ContextualCompressionRetriever
66
from langchain.retrievers.document_compressors import LLMChainExtractor
7-
from langchain.callbacks import get_openai_callback
7+
from langchain_community.callbacks import get_openai_callback
88

99
from sklearn.metrics.pairwise import cosine_similarity
1010
import matplotlib.pyplot as plt

all_rag_techniques_runnable_scripts/query_transformations.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22
from dotenv import load_dotenv
33
from langchain_openai import ChatOpenAI
4-
from langchain.prompts import PromptTemplate
4+
from langchain_core.prompts import PromptTemplate
55

66
# Load environment variables from a .env file
77
load_dotenv()

all_rag_techniques_runnable_scripts/raptor.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,16 @@
22
import pandas as pd
33
from typing import List, Dict, Any
44

5-
from langchain.chains.llm import LLMChain
5+
from langchain.chains import LLMChain
66
from sklearn.mixture import GaussianMixture
7-
from langchain.embeddings import OpenAIEmbeddings
8-
from langchain.vectorstores import FAISS
7+
from langchain_openai import OpenAIEmbeddings
8+
from langchain_community.vectorstores import FAISS
99
from langchain_openai import ChatOpenAI
10-
from langchain.prompts import ChatPromptTemplate
10+
from langchain_core.prompts import ChatPromptTemplate
1111
from langchain.retrievers import ContextualCompressionRetriever
1212
from langchain.retrievers.document_compressors import LLMChainExtractor
13-
from langchain.schema import AIMessage
14-
from langchain.docstore.document import Document
13+
from langchain_core.messages import AIMessage
14+
from langchain_core.documents import Document
1515
import matplotlib.pyplot as plt
1616
import logging
1717
import os

all_rag_techniques_runnable_scripts/reranking.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22
import sys
33
from dotenv import load_dotenv
4-
from langchain.docstore.document import Document
4+
from langchain_core.documents import Document
55
from typing import List, Any
66
from langchain_openai import ChatOpenAI
77
from langchain.chains import RetrievalQA

all_rag_techniques_runnable_scripts/retrieval_with_feedback_loop.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from langchain_text_splitters import RecursiveCharacterTextSplitter
99
from langchain_openai import ChatOpenAI
1010
from langchain.chains import RetrievalQA
11-
from langchain.prompts import PromptTemplate
11+
from langchain_core.prompts import PromptTemplate
1212

1313
sys.path.append(os.path.abspath(os.path.join(os.getcwd(), '..'))) # Add the parent directory to the path
1414
from helper_functions import *

all_rag_techniques_runnable_scripts/self_rag.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22
import sys
33
from dotenv import load_dotenv
4-
from langchain.prompts import PromptTemplate
4+
from langchain_core.prompts import PromptTemplate
55
from langchain_openai import ChatOpenAI
66
from langchain_core.pydantic_v1 import BaseModel, Field
77

0 commit comments

Comments
 (0)