formmatting

Author: davidsbatista

integrations/opensearch/src/haystack_integrations/document_stores/opensearch/document_store.py

@@ -343,25 +343,25 @@ def _deserialize_search_hits(hits: list[dict[str, Any]]) -> list[Document]:
         out = []
         # Fields that are not metadata (should stay at top level)
         non_meta_fields = {"id", "content", "embedding", "blob", "sparse_embedding", "score"}
-        
+
         for hit in hits:
             data = hit["_source"].copy()
-            
+
             # Reconstruct metadata dict from flattened fields
             meta = {}
             fields_to_remove = []
             for key, value in data.items():
                 if key not in non_meta_fields:
                     meta[key] = value
                     fields_to_remove.append(key)
-            
+
             # Remove metadata fields from top level and add them to meta
             for key in fields_to_remove:
                 data.pop(key, None)
-            
+
             if meta:
                 data["meta"] = meta
-            
+
             if "highlight" in hit:
                 if "meta" not in data:
                     data["meta"] = {}

integrations/opensearch/tests/test_document_store.py

@@ -704,7 +704,7 @@ def test_get_field_min_max(self, document_store: OpenSearchDocumentStore):
         min_max_single = document_store.get_field_min_max("meta.single_value")
         assert min_max_single["min"] == 42
         assert min_max_single["max"] == 42
-                
+
         # Test with float values
         min_max_score = document_store.get_field_min_max("meta.rating")
         assert min_max_score["min"] == pytest.approx(5.2)
@@ -767,6 +767,8 @@ def test_get_field_unique_values(self, document_store: OpenSearchDocumentStore):
         assert priority_count == 3
 
         # Test with search term on integer field
-        unique_priorities_filtered, priority_count = document_store.get_field_unique_values("meta.priority", "Doc 1", 0, 10)
+        unique_priorities_filtered, priority_count = document_store.get_field_unique_values(
+            "meta.priority", "Doc 1", 0, 10
+        )
         assert set(unique_priorities_filtered) == {"1"}
-        assert priority_count == 1
+        assert priority_count == 1

integrations/opensearch/tests/test_document_store_async.py

@@ -523,24 +523,32 @@ async def test_get_field_unique_values(self, document_store: OpenSearchDocumentS
         assert lang_count == 3
 
         # Test pagination - first page
-        unique_values_page1, total_count = await document_store.get_field_unique_values_async("meta.category", None, 0, 2)
+        unique_values_page1, total_count = await document_store.get_field_unique_values_async(
+            "meta.category", None, 0, 2
+        )
         assert len(unique_values_page1) == 2
         assert total_count == 3
         assert all(val in ["A", "B", "C"] for val in unique_values_page1)
 
         # Test pagination - second page
-        unique_values_page2, total_count = await document_store.get_field_unique_values_async("meta.category", None, 2, 2)
+        unique_values_page2, total_count = await document_store.get_field_unique_values_async(
+            "meta.category", None, 2, 2
+        )
         assert len(unique_values_page2) == 1
         assert total_count == 3
         assert unique_values_page2[0] in ["A", "B", "C"]
 
         # Test with search term - filter by content matching "Python"
-        unique_values_filtered, total_count = await document_store.get_field_unique_values_async("meta.category", "Python", 0, 10)
+        unique_values_filtered, total_count = await document_store.get_field_unique_values_async(
+            "meta.category", "Python", 0, 10
+        )
         assert set(unique_values_filtered) == {"A"}  # Only category A has documents with "Python" in content
         assert total_count == 1
 
         # Test with search term - filter by content matching "Java"
-        unique_values_java, total_count = await document_store.get_field_unique_values_async("meta.category", "Java", 0, 10)
+        unique_values_java, total_count = await document_store.get_field_unique_values_async(
+            "meta.category", "Java", 0, 10
+        )
         assert set(unique_values_java) == {"B"}  # Only category B has documents with "Java" in content
         assert total_count == 1
 
@@ -552,13 +560,15 @@ async def test_get_field_unique_values(self, document_store: OpenSearchDocumentS
             Document(content="Doc 4", meta={"priority": 3}),
         ]
         await document_store.write_documents_async(int_docs)
-        unique_priorities, priority_count = await document_store.get_field_unique_values_async("meta.priority", None, 0, 10)
+        unique_priorities, priority_count = await document_store.get_field_unique_values_async(
+            "meta.priority", None, 0, 10
+        )
         assert set(unique_priorities) == {"1", "2", "3"}
         assert priority_count == 3
 
         # Test with search term on integer field
-        unique_priorities_filtered, priority_count = await document_store.get_field_unique_values_async("meta.priority", "Doc 1", 0, 10)
+        unique_priorities_filtered, priority_count = await document_store.get_field_unique_values_async(
+            "meta.priority", "Doc 1", 0, 10
+        )
         assert set(unique_priorities_filtered) == {"1"}
         assert priority_count == 1
-
-