handle langchain.schema

src/retrieval/context.py

@@ -1,15 +1,22 @@
 """Context retrieval with reranking capabilities."""
 
-import os
+
+import traceback
 from typing import List, Optional, Tuple, Dict, Any
-from langchain.schema import Document
+
+from sentence_transformers import CrossEncoder
 from langchain_community.vectorstores import Qdrant
 from langchain_community.embeddings import HuggingFaceEmbeddings
-from sentence_transformers import CrossEncoder
-import numpy as np
+
 import torch
+import numpy as np
 from qdrant_client.http import models as rest
-import traceback
+
+
+try:
+    from langchain.docstore.document import Document
+except:
+    from langchain_core.documents import Document
 
 from .filter import create_filter
 

src/retrieval/hybrid.py

@@ -1,15 +1,21 @@
 """Hybrid search implementation combining vector and sparse retrieval."""
 
-import json
-import numpy as np
-from typing import List, Dict, Any, Tuple
+import os
+import pickle
 from pathlib import Path
-from langchain.docstore.document import Document
+from typing import List, Dict, Any, Tuple
+
+
+import numpy as np
 from langchain_qdrant import QdrantVectorStore
 from langchain_community.retrievers import BM25Retriever
+try:
+    from langchain.docstore.document import Document
+except:
+    from langchain_core.documents import Document
+
+
 from .filter import create_filter
-import pickle
-import os
 
 
 class HybridRetriever: