From 2d91c118d71a8dd7fbd7f9cf21f86e92da33827e Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 8 May 2024 21:14:33 +0200
Subject: Fix cli

---
 rag/cli.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/rag/cli.py b/rag/cli.py
index e8fdd86..6c4d3e0 100644
--- a/rag/cli.py
+++ b/rag/cli.py
@@ -70,7 +70,7 @@ def rag(query: str, model: str, verbose: int):
     generator = get_generator(model)
     reranker = get_reranker(model)
     documents = retriever.retrieve(query)
-    prompt = reranker.rerank(Prompt(query, documents))
+    prompt = reranker.rank(Prompt(query, documents))
     print("Answer: ")
     for chunk in generator.generate(prompt):
         print(chunk, end="", flush=True)
@@ -94,7 +94,6 @@ def drop():
     vec_db.delete_collection()
 
 
-cli.add_command(configure_logging)
 cli.add_command(rag)
 cli.add_command(upload)
 cli.add_command(drop)
-- 
cgit v1.2.3-70-g09d2