Spaces:
Sleeping
Sleeping
Update backend/semantic_search.py
Browse files
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).
|
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:
|