Spaces:
Running
Running
Update rag_engine.py
Browse files- rag_engine.py +2 -2
rag_engine.py
CHANGED
|
@@ -139,7 +139,7 @@ class KnowledgeBase:
|
|
| 139 |
except Exception as e:
|
| 140 |
return f"Error vectorizing: {e}"
|
| 141 |
|
| 142 |
-
def search(self, query: str, top_k: int =
|
| 143 |
"""
|
| 144 |
Retrieves docs using FAISS -> Reranks using FlashRank.
|
| 145 |
"""
|
|
@@ -164,7 +164,7 @@ class KnowledgeBase:
|
|
| 164 |
results = self.ranker.rerank(rerank_request)
|
| 165 |
|
| 166 |
# Return Top 5 re-ranked chunks
|
| 167 |
-
top_results = results[:
|
| 168 |
return "\n\n".join([f"[Local RAG] {r['text']}" for r in top_results])
|
| 169 |
|
| 170 |
local_kb = KnowledgeBase()
|
|
|
|
| 139 |
except Exception as e:
|
| 140 |
return f"Error vectorizing: {e}"
|
| 141 |
|
| 142 |
+
def search(self, query: str, top_k: int = 10) -> str:
|
| 143 |
"""
|
| 144 |
Retrieves docs using FAISS -> Reranks using FlashRank.
|
| 145 |
"""
|
|
|
|
| 164 |
results = self.ranker.rerank(rerank_request)
|
| 165 |
|
| 166 |
# Return Top 5 re-ranked chunks
|
| 167 |
+
top_results = results[:1]
|
| 168 |
return "\n\n".join([f"[Local RAG] {r['text']}" for r in top_results])
|
| 169 |
|
| 170 |
local_kb = KnowledgeBase()
|