Aragoner commited on
Commit
4f4776c
·
verified ·
1 Parent(s): 73883f2

Update backend/semantic_search.py

Browse files
Files changed (1) hide show
  1. backend/semantic_search.py +1 -1
backend/semantic_search.py CHANGED
@@ -47,7 +47,7 @@ def rerank(query, documents, k):
47
  def retrieve(query, top_k_retriever=25, use_reranking=True, top_k_reranker=5):
48
  query_vec = retriever.encode(query)
49
  try:
50
- documents = TABLE.search(query_vec, vector_column_name=VECTOR_COLUMN).limit(top_k_retriever).to_pylist()
51
  documents = [doc[TEXT_COLUMN] for doc in documents]
52
 
53
  if use_reranking:
 
47
  def retrieve(query, top_k_retriever=25, use_reranking=True, top_k_reranker=5):
48
  query_vec = retriever.encode(query)
49
  try:
50
+ documents = TABLE.search(query_vec, vector_column_name=VECTOR_COLUMN).limit(top_k_retriever).to_list()
51
  documents = [doc[TEXT_COLUMN] for doc in documents]
52
 
53
  if use_reranking: