From 4ac80ad73dd37e2bfb46aeefbf3de3731eaf24cb Mon Sep 17 00:00:00 2001 From: boladeE Date: Fri, 18 Apr 2025 20:10:18 +0100 Subject: [PATCH] Enable reranking of search results in BrandStyleManager for improved content retrieval accuracy. --- backend/brand_style.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/backend/brand_style.py b/backend/brand_style.py index 58ba16a..0d819fd 100644 --- a/backend/brand_style.py +++ b/backend/brand_style.py @@ -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 []