set consistency level of ANN queries

Author: bjchambers

libs/knowledge-store/ragstack_knowledge_store/knowledge_store.py

@@ -11,7 +11,7 @@
  Union,
 )
 
-from cassandra.cluster import ResponseFuture, Session
+from cassandra.cluster import ResponseFuture, Session, ConsistencyLevel
 from cassio.config import check_resolve_keyspace, check_resolve_session
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
@@ -178,6 +178,7 @@ def __init__(
  LIMIT ?
  """
  )
+ self._query_by_embedding.consistency_level = ConsistencyLevel.QUORUM
 
  self._query_ids_by_embedding = session.prepare(
  f"""
@@ -187,6 +188,7 @@ def __init__(
  LIMIT ?
  """
  )
+ self._query_ids_by_embedding = ConsistencyLevel.QUORUM
 
  self._query_ids_and_embedding_by_embedding = session.prepare(
  f"""
@@ -196,6 +198,7 @@ def __init__(
  LIMIT ?
  """
  )
+ self._query_ids_by_and_embedding_by_embedding = ConsistencyLevel.QUORUM
 
  self._query_linked_ids = session.prepare(
  f"""