Skip to content

Commit fd9cd18

Browse files
authored
Truncate vector in logs (#192)
* Truncate vector in logs * Changed log message for query vector * Modified log * Use correct query to avoid deprecation warning for Hybrid * Modified log condition for query_vector
1 parent 358fafd commit fd9cd18

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

src/neo4j_graphrag/retrievers/hybrid.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
# limitations under the License.
1515
from __future__ import annotations
1616

17+
import copy
1718
import logging
1819
from typing import Any, Callable, Optional
1920

@@ -193,8 +194,10 @@ def get_search_results(
193194
embedding_node_property=self._embedding_node_property,
194195
neo4j_version_is_5_23_or_above=self.neo4j_version_is_5_23_or_above,
195196
)
196-
197-
logger.debug("HybridRetriever Cypher parameters: %s", parameters)
197+
sanitized_parameters = copy.deepcopy(parameters)
198+
if "query_vector" in sanitized_parameters:
199+
sanitized_parameters["query_vector"] = "..."
200+
logger.debug("HybridRetriever Cypher parameters: %s", sanitized_parameters)
198201
logger.debug("HybridRetriever Cypher query: %s", search_query)
199202

200203
records, _, _ = self.driver.execute_query(
@@ -348,9 +351,11 @@ def get_search_results(
348351
retrieval_query=self.retrieval_query,
349352
neo4j_version_is_5_23_or_above=self.neo4j_version_is_5_23_or_above,
350353
)
351-
352-
logger.debug("HybridCypherRetriever Cypher parameters: %s", parameters)
353-
logger.debug("HybridCypherRetriever Cypher query: %s", search_query)
354+
sanitized_parameters = copy.deepcopy(parameters)
355+
if "query_vector" in sanitized_parameters:
356+
sanitized_parameters["query_vector"] = "..."
357+
logger.debug("HybridRetriever Cypher parameters: %s", sanitized_parameters)
358+
logger.debug("HybridRetriever Cypher query: %s", search_query)
354359

355360
records, _, _ = self.driver.execute_query(
356361
search_query, parameters, database_=self.neo4j_database

0 commit comments

Comments
 (0)