From dda55cd6dc1eaa6cae0ee11f5775f980380c7a9a Mon Sep 17 00:00:00 2001 From: Vivek Haldar Date: Fri, 24 May 2024 15:55:26 -0700 Subject: [PATCH] Add a debug option to show full context. --- installer/client/cli/fabric.py | 2 ++ installer/client/cli/utils.py | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/installer/client/cli/fabric.py b/installer/client/cli/fabric.py index 48f02c5..3ade91a 100755 --- a/installer/client/cli/fabric.py +++ b/installer/client/cli/fabric.py @@ -73,6 +73,8 @@ def main(): help='The URL of the remote ollamaserver to use. ONLY USE THIS if you are using a local ollama server in an non-default location or port') parser.add_argument('--context', '-c', help="Use Context file (context.md) to add context to your pattern", action="store_true") + parser.add_argument('--debug', + help="Print out entire LLM prompt for debugging", action="store_true") args = parser.parse_args() home_holder = os.path.expanduser("~") diff --git a/installer/client/cli/utils.py b/installer/client/cli/utils.py index 9e7cea2..a4ac372 100644 --- a/installer/client/cli/utils.py +++ b/installer/client/cli/utils.py @@ -244,6 +244,8 @@ class Standalone: else: messages = [user_message] try: + if self.args.debug: + print(messages) if self.local: if host: asyncio.run(self.localStream(messages, host=host)) @@ -355,6 +357,8 @@ class Standalone: else: messages = [user_message] try: + if self.args.debug: + print(messages) if self.local: if host: asyncio.run(self.localChat(messages, host=host))