langchain.docstore Document

src/gemini/file_search.py

@@ -19,6 +19,12 @@ except ImportError:
     GEMINI_AVAILABLE = False
 
 
+try:
+    from langchain.docstore.document import Document
+except:
+    from langchain_core.documents import Document
+
+
 @dataclass
 class GeminiFileSearchResult:
     """Result from Gemini File Search query"""
@@ -369,7 +375,6 @@ IMPORTANT: Only use information from the retrieved documents. Do not use informa
         
         Returns list of document-like objects compatible with existing display code.
         """
-        from langchain.docstore.document import Document
         
         formatted_sources = []