diff --git a/src/neo4j_genai/retrievers/hybrid.py b/src/neo4j_genai/retrievers/hybrid.py index a9311d069..0690555a2 100644 --- a/src/neo4j_genai/retrievers/hybrid.py +++ b/src/neo4j_genai/retrievers/hybrid.py @@ -21,6 +21,9 @@ from neo4j_genai.retrievers.base import Retriever from neo4j_genai.types import HybridSearchModel, SearchType, HybridCypherSearchModel from neo4j_genai.neo4j_queries import get_search_query +import logging + +logger = logging.getLogger(__name__) class HybridRetriever(Retriever): @@ -83,6 +86,9 @@ def search( search_query = get_search_query(SearchType.HYBRID, self.return_properties) + logger.debug("HybridRetriever Cypher parameters: %s", parameters) + logger.debug("HybridRetriever Cypher query: %s", search_query) + records, _, _ = self.driver.execute_query(search_query, parameters) return records @@ -157,5 +163,9 @@ def search( search_query = get_search_query( SearchType.HYBRID, retrieval_query=self.retrieval_query ) + + logger.debug("HybridCypherRetriever Cypher parameters: %s", parameters) + logger.debug("HybridCypherRetriever Cypher query: %s", search_query) + records, _, _ = self.driver.execute_query(search_query, parameters) return records diff --git a/src/neo4j_genai/retrievers/vector.py b/src/neo4j_genai/retrievers/vector.py index 771ce3520..954cd04ed 100644 --- a/src/neo4j_genai/retrievers/vector.py +++ b/src/neo4j_genai/retrievers/vector.py @@ -26,6 +26,9 @@ SearchType, ) from neo4j_genai.neo4j_queries import get_search_query +import logging + +logger = logging.getLogger(__name__) class VectorRetriever(Retriever): @@ -92,6 +95,9 @@ def search( search_query = get_search_query(SearchType.VECTOR, self.return_properties) + logger.debug("VectorRetriever Cypher parameters: %s", parameters) + logger.debug("VectorRetriever Cypher query: %s", search_query) + records, _, _ = self.driver.execute_query(search_query, parameters) try: @@ -178,5 +184,9 @@ def search( search_query = get_search_query( SearchType.VECTOR, retrieval_query=self.retrieval_query ) + + logger.debug("VectorCypherRetriever Cypher parameters: %s", parameters) + logger.debug("VectorCypherRetriever Cypher query: %s", search_query) + records, _, _ = self.driver.execute_query(search_query, parameters) return records