Rajhuggingface4253 commited on
Commit
6ae06e3
·
verified ·
1 Parent(s): 9a92b90

Update rag_engine.py

Browse files
Files changed (1) hide show
  1. 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 = 15) -> str:
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[:5]
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()