From 85f948fb5f904d59b2b4b250934b3b482aa76aa4 Mon Sep 17 00:00:00 2001
From: german <gmartin.py@gmail.com>
Date: Thu, 1 Feb 2024 23:53:10 -0300
Subject: [PATCH] Include metadata if present in retrieved documents.

---
 ragatouille/integrations/_langchain.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/ragatouille/integrations/_langchain.py b/ragatouille/integrations/_langchain.py
index e7e5555..0130e20 100644
--- a/ragatouille/integrations/_langchain.py
+++ b/ragatouille/integrations/_langchain.py
@@ -18,7 +18,12 @@ def _get_relevant_documents(
     ) -> List[Document]:
         """Get documents relevant to a query."""
         docs = self.model.search(query, **self.kwargs)
-        return [Document(page_content=doc["content"]) for doc in docs]
+        return [
+            Document(
+                page_content=doc["content"], metadata=doc.get("document_metadata", {})
+            )
+            for doc in docs
+        ]
 
 
 class RAGatouilleLangChainCompressor(BaseDocumentCompressor):