Update utils/retriever.py

utils/retriever.py

@@ -131,8 +131,8 @@ def rerank_documents(query: str, documents: List[Dict[str, Any]]) -> List[Dict[s
         langchain_docs = []
         for doc in documents:
             # Use correct keys from the data storage test module
-            content = doc.get('answer', '')
-            metadata = doc.get('answer_metadata', {})
+            content = doc.get("answer", "")
+            metadata = doc.get("answer_metadata", {})
             
             if not content:
                 logging.warning(f"Document missing content: {doc}")
@@ -156,8 +156,8 @@ def rerank_documents(query: str, documents: List[Dict[str, Any]]) -> List[Dict[s
         result = []
         for doc in reranked_docs:
             result.append({
-                'answer': doc.page_content,
-                'answer_metadata': doc.metadata,
+                "answer": doc.page_content,
+                "answer_metadata": doc.metadata,
             })
         
         logging.info(f"Successfully reranked {len(documents)} documents to top {len(result)}")