Improve log warning

GoogleCloudPlatform · Sep 4, 2024 · c68d44e · c68d44e
1 parent 21e7def
commit c68d44e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/retrieval_service/datastore/providers/neo4j_graph.py b/retrieval_service/datastore/providers/neo4j_graph.py
@@ -227,7 +227,9 @@ async def amenities_search(
         self, query_embedding: list[float], similarity_threshold: float, top_k: int
     ) -> tuple[list[dict], Optional[str]]:
 
-        logging.warning(f"Overriding top_k value to 2. Provided value was {top_k}.")
+        logging.warning(
+            f"top_k set to {top_k}; for GraphRAG, overriding top_k to 2 to limit search results."
+        )
 
         async with self.__driver.session() as session:
             # OPTIONAL ensures that all similar amenities are returned even if they lack a SIMILAR_TO relationship

diff --git a/retrieval_service/datastore/providers/neo4j_graph_test.py b/retrieval_service/datastore/providers/neo4j_graph_test.py
@@ -240,6 +240,12 @@ async def test_amenities_search(
     top_k: int,
     expected: List[Any],
 ):
+<<<<<<< HEAD
     res, sql = await ds.amenities_search(query_embedding, similarity_threshold, top_k)
+=======
+
+    res, sql = await ds.amenities_search(query_embedding, similarity_threshold, top_k)
+
+>>>>>>> 06e09fa (Improve log warning)
     assert res == expected
     assert sql is None