fix: langchain integration attributes

bclavie · bclavie · commit a7b8fa996588 · 2024-03-19T22:00:38.000+01:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ packages = [
 name = "rerankers" 
 
 
-version = "0.1.0post1"
+version = "0.1.1"
 
 description = "A unified API for various document re-ranking models."
 
diff --git a/rerankers/__init__.py b/rerankers/__init__.py
@@ -1,4 +1,4 @@
 from rerankers.reranker import Reranker
 
 __all__ = ["Reranker"]
-__version__ = "0.1.0post1"
+__version__ = "0.1.1"
diff --git a/rerankers/integrations/langchain.py b/rerankers/integrations/langchain.py
@@ -32,7 +32,7 @@ def compress_documents(
         )
         final_results = []
         for r in results.top_k(kwargs.get("k", self.k)):
-            doc = doc_list[r["doc_id"]]
-            doc.metadata["relevance_score"] = r["score"]
+            doc = doc_list[r.doc_id]
+            doc.metadata["relevance_score"] = r.score
             final_results.append(doc)
         return final_results
