delete vectara

document_processor.py

@@ -5,6 +5,9 @@
 
 try:
     from langchain_text_splitters import RecursiveCharacterTextSplitter
+except ImportError:
+    try:
+    from langchain_text_splitters import RecursiveCharacterTextSplitter
 except ImportError:
     from langchain.text_splitter import RecursiveCharacterTextSplitter
 

entity_extractor.py

@@ -10,6 +10,9 @@ import aiohttp
 import json
 try:
     from langchain_core.prompts import PromptTemplate
+except ImportError:
+    try:
+    from langchain_core.prompts import PromptTemplate
 except ImportError:
     from langchain.prompts import PromptTemplate
 

evaluate_retrieval.py

@@ -18,14 +18,31 @@ load_dotenv()
 from main import AdaptiveRAGSystem
 from document_processor import DocumentProcessor
 from retrieval_evaluation import RetrievalEvaluator, RetrievalResult, RetrievalTestSet
-from langchain.schema import Document
+try:
+    from langchain_core.documents import Document
+except ImportError:
+    try:
+    from langchain_core.documents import Document
+except ImportError:
+    from langchain.schema import Document
 
 # 导入LangChain相关模块
 from langchain_community.vectorstores import FAISS, Chroma
 from langchain_community.retrievers import BM25Retriever
-from langchain.retrievers import EnsembleRetriever
-from langchain.retrievers import ContextualCompressionRetriever
-from langchain.retrievers.document_compressors import LLMChainExtractor
+try:
+    from langchain.retrievers import EnsembleRetriever
+    from langchain.retrievers import ContextualCompressionRetriever
+    from langchain.retrievers.document_compressors import LLMChainExtractor
+except ImportError:
+    try:
+        from langchain_core.retrievers import EnsembleRetriever
+        from langchain_core.retrievers import ContextualCompressionRetriever
+        from langchain.retrievers.document_compressors import LLMChainExtractor
+    except ImportError:
+        print("Warning: Could not import advanced retriever components. Some features may be limited.")
+        EnsembleRetriever = None
+        ContextualCompressionRetriever = None
+        LLMChainExtractor = None
 
 
 class AdaptiveRAGRetriever:

graph_indexer.py

@@ -7,9 +7,20 @@ from typing import List, Dict, Optional
 import asyncio
 try:
     from langchain_core.documents import Document
+except ImportError:
+    try:
+    from langchain_core.documents import Document
+except ImportError:
+    try:
+    from langchain_core.documents import Document
 except ImportError:
     from langchain.schema import Document
 
+try:
+    from langchain_text_splitters import RecursiveCharacterTextSplitter
+except ImportError:
+    from langchain.text_splitter import RecursiveCharacterTextSplitter
+
 from entity_extractor import EntityExtractor, EntityDeduplicator
 from knowledge_graph import KnowledgeGraph, CommunitySummarizer
 

graph_retriever.py

@@ -6,6 +6,9 @@ GraphRAG检索器
 from typing import List, Dict, Set, Tuple
 try:
     from langchain_core.prompts import PromptTemplate
+except ImportError:
+    try:
+    from langchain_core.prompts import PromptTemplate
 except ImportError:
     from langchain.prompts import PromptTemplate
 

knowledge_graph.py

@@ -17,6 +17,9 @@ except ImportError:
 
 try:
     from langchain_core.prompts import PromptTemplate
+except ImportError:
+    try:
+    from langchain_core.prompts import PromptTemplate
 except ImportError:
     from langchain.prompts import PromptTemplate
 from langchain_community.chat_models import ChatOllama

local_llm_rag.py

@@ -19,7 +19,10 @@ _set_env("NOMIC_API_KEY")
 # Ollama model name
 local_llm = "mistral"
 
-from langchain.text_splitter import RecursiveCharacterTextSplitter
+try:
+    from langchain_text_splitters import RecursiveCharacterTextSplitter
+except ImportError:
+    from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.document_loaders import WebBaseLoader
 from langchain_community.vectorstores import Chroma
 from langchain_nomic.embeddings import NomicEmbeddings
@@ -48,7 +51,10 @@ retriever = vectorstore.as_retriever()
 
 ### Router
 
-from langchain.prompts import PromptTemplate
+try:
+    from langchain_core.prompts import PromptTemplate
+except ImportError:
+    from langchain.prompts import PromptTemplate
 from langchain_community.chat_models import ChatOllama
 from langchain_core.output_parsers import JsonOutputParser
 
@@ -161,6 +167,12 @@ class GraphState(TypedDict):
     generation: str
     documents: List[str]
 
+    try:
+    from langchain_core.documents import Document
+except ImportError:
+    try:
+    from langchain_core.documents import Document
+except ImportError:
     from langchain.schema import Document
 
 

retrieval_evaluation.py

@@ -8,7 +8,13 @@ import json
 import numpy as np
 from typing import List, Dict, Tuple, Any, Optional, Union
 from dataclasses import dataclass, asdict
-from langchain.schema import Document
+try:
+    from langchain_core.documents import Document
+except ImportError:
+    try:
+    from langchain_core.documents import Document
+except ImportError:
+    from langchain.schema import Document
 from sklearn.metrics import ndcg_score, precision_score, recall_score, f1_score
 from sentence_transformers import SentenceTransformer, util
 import matplotlib.pyplot as plt

routers_and_graders.py

@@ -5,6 +5,9 @@
 
 try:
     from langchain_core.prompts import PromptTemplate
+except ImportError:
+    try:
+    from langchain_core.prompts import PromptTemplate
 except ImportError:
     from langchain.prompts import PromptTemplate
 

test_reranking.py

@@ -10,7 +10,13 @@ sys.path.append(os.path.dirname(__file__))
 
 from document_processor import DocumentProcessor
 from reranker import *
-from langchain.schema import Document
+try:
+    from langchain_core.documents import Document
+except ImportError:
+    try:
+    from langchain_core.documents import Document
+except ImportError:
+    from langchain.schema import Document
 import time
 
 

workflow_nodes.py

@@ -8,6 +8,12 @@ from typing import List
 from typing_extensions import TypedDict
 try:
     from langchain_core.documents import Document
+except ImportError:
+    try:
+    from langchain_core.documents import Document
+except ImportError:
+    try:
+    from langchain_core.documents import Document
 except ImportError:
     from langchain.schema import Document
 from langchain_community.chat_models import ChatOllama
@@ -15,6 +21,9 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_community.tools.tavily_search import TavilySearchResults
 try:
     from langchain_core.prompts import PromptTemplate
+except ImportError:
+    try:
+    from langchain_core.prompts import PromptTemplate
 except ImportError:
     from langchain.prompts import PromptTemplate