aryn-ai · bsowell · Oct 23, 2024 · Oct 22, 2024 · Oct 22, 2024
diff --git a/lib/sycamore/sycamore/connectors/opensearch/opensearch_reader.py b/lib/sycamore/sycamore/connectors/opensearch/opensearch_reader.py
@@ -89,6 +89,7 @@ def to_docs(self, query_params: "BaseDBReader.QueryParams") -> list[Document]:
                     }
                 )
                 doc.properties[DocumentPropertyTypes.SOURCE] = DocumentSource.DB_QUERY
+                doc.properties["score"] = data["_score"]
                 result.append(doc)
         else:
             assert (

diff --git a/lib/sycamore/sycamore/tests/unit/connectors/opensearch/test_opensearch.py b/lib/sycamore/sycamore/tests/unit/connectors/opensearch/test_opensearch.py
@@ -1,4 +1,5 @@
 from unittest.mock import Mock
+import random
 
 from opensearchpy import OpenSearch, RequestError, ConnectionError
 import pytest
@@ -176,7 +177,7 @@ def test_to_docs(self):
             },
             {"text_representation": "this is a parent doc", "parent_id": None, "doc_id": "doc_1"},
         ]
-        hits = [{"_source": record} for record in records]
+        hits = [{"_source": record, "_score": random.random()} for record in records]
         query_response = OpenSearchReaderQueryResponse(hits)
         query_params = OpenSearchReaderQueryParams(index_name="some index")
         docs = query_response.to_docs(query_params)
@@ -186,6 +187,7 @@ def test_to_docs(self):
         for i in range(len(docs)):
             assert docs[i].parent_id == records[i]["parent_id"]
             assert docs[i].text_representation == records[i]["text_representation"]
+            assert "score" in docs[i].properties
 
     def test_to_docs_reconstruct_require_client(self):
         query_response = OpenSearchReaderQueryResponse([])