Enable reranking of search results in BrandStyleManager for improved content retrieval accuracy.

This commit is contained in:
boladeE
2025-04-18 20:10:18 +01:00
parent 942255661b
commit 4ac80ad73d
+5 -5
View File
@@ -91,11 +91,11 @@ class BrandStyleManager:
results = self.vector_store.search(prompt_embedding, k=k)
# Optionally rerank results
# if results:
# texts = [result["text"] for result in results]
# reranked = self.embeddings.rerank_results(prompt, texts, top_n=k)
# # Convert reranked results to the expected format
# return [{"text": text} for text in reranked]
if results:
texts = [result["text"] for result in results]
reranked = self.embeddings.rerank_results(prompt, texts, top_n=k)
# Convert reranked results to the expected format
return [{"text": text} for text in reranked]
# If no results, return empty list
return []