From 310b4c5f12851b4466a7e0c20fe556a7fd2736bf Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 23 Apr 2024 00:50:37 +0200
Subject: Update cli

---
 rag/cli.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/rag/cli.py b/rag/cli.py
index c6ea21c..690563e 100644
--- a/rag/cli.py
+++ b/rag/cli.py
@@ -79,16 +79,26 @@ def rag(generator: str, query: str, limit):
     type=click.Path(exists=True),
     default=None,
 )
+@click.option(
+    "-q",
+    "--query",
+    prompt_required=False,
+    help="The query for rag",
+    prompt="Enter your query",
+)
 @click.option("-v", "--verbose", count=True)
 def main(
-    query: Optional[str], generator: str, limit: int, directory: Optional[str], verbose
+    query: Optional[str],
+    generator: str,
+    limit: int,
+    directory: Optional[str],
+    verbose: int,
 ):
     configure_logging(verbose)
+    if directory:
+        upload(directory)
     if query:
         rag(generator, query, limit)
-    elif directory:
-        upload(directory)
-    # TODO: truncate databases
     # TODO: maybe add override for models
 
 
-- 
cgit v1.2.3-70-g09d2