From c811d0544ea4d61530aa2506a387d0a322991521 Mon Sep 17 00:00:00 2001 From: ncoop57 Date: Mon, 18 Nov 2024 18:10:13 -0600 Subject: [PATCH] Update core.py with increased output lines and code theme option --- bash_buddy/core.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/bash_buddy/core.py b/bash_buddy/core.py index 5eee05b..7b3bb86 100755 --- a/bash_buddy/core.py +++ b/bash_buddy/core.py @@ -54,7 +54,8 @@ def main( query: Param('The query to send to the LLM', str, nargs="+"), NH: bool = False, # Don't include terminal history n: int = 200, # Number of history lines - o: int = 30, # Number of output lines before piping to less + o: int = 70, # Number of output lines before piping to less + code_theme: str = 'monokai', # The code theme to use when rendering BashBuddy's responses ): query = ' '.join(query) ctxt = "" @@ -70,4 +71,4 @@ def main( if len(r.splitlines()) > o: p = subprocess.Popen(['less'], stdin=subprocess.PIPE) p.communicate(input=r.encode()) - else: print(Markdown(r)) + else: print(Markdown(r, code_theme=code_theme)) \ No newline at end of file