diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-24 01:10:43 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-24 01:10:43 +0200 |
| commit | 95305f59df84caded50286b1a57b6075e48725a8 (patch) | |
| tree | c0f0157a99da6332a3c96462b0aba2bd02dfcb33 /rag/retriever/retriever.py | |
| parent | 75be0914f6bd2cdeda1539f83b38fcbc854d5cfa (diff) | |
Rerank working
llama3 sucks at rag
Diffstat (limited to 'rag/retriever/retriever.py')
| -rw-r--r-- | rag/retriever/retriever.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/rag/retriever/retriever.py b/rag/retriever/retriever.py index deffae5..351cfb0 100644 --- a/rag/retriever/retriever.py +++ b/rag/retriever/retriever.py @@ -45,7 +45,7 @@ class Retriever: else: log.error("Invalid input!") - def retrieve(self, query: str, limit: int = 5) -> List[Document]: + def retrieve(self, query: str) -> List[Document]: log.debug(f"Finding documents matching query: {query}") query_emb = self.encoder.encode_query(query) - return self.vec_db.search(query_emb, limit) + return self.vec_db.search(query_emb) |