From 2f4accff3ecb6b5cf696ec403411fa30cc200832 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Mon, 5 Aug 2024 14:28:28 +0200
Subject: Refactor

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

(limited to 'rag')

diff --git a/rag/cli.py b/rag/cli.py
index b047255..9d54549 100644
--- a/rag/cli.py
+++ b/rag/cli.py
@@ -71,8 +71,8 @@ def rag(client: str, verbose: int):
 
         rag.add_message(rag.bot, response)
 
-        show_context = input("Display context? [y/n] ").lower() == "y"
-        print("\n\n")
+        show_context = input("\nDisplay context? [y/n] ").lower() == "y"
+        print("\n")
         if show_context:
             for i, doc in enumerate(prompt.documents):
                 print(f"### Document {i}")
-- 
cgit v1.2.3-70-g09d2