diff --git a/GPTtrace.py b/GPTtrace.py index d9d2af2..82d82b9 100755 --- a/GPTtrace.py +++ b/GPTtrace.py @@ -20,9 +20,12 @@ ENV_ACCESS_TOKEN = "GPTTRACE_ACCESS_TOKEN" PROMPTS_DIR = Path("./prompts") -def pretty_print(input, lexer=MarkdownLexer): +def pretty_print(input, lexer=MarkdownLexer, *args, **kwargs): tokens = list(pygments.lex(input, lexer=lexer())) - print_formatted_text(PygmentsTokens(tokens)) + print_formatted_text(PygmentsTokens(tokens), *args, **kwargs) + + +# print = pretty_print def main(): @@ -48,6 +51,9 @@ def main(): group.add_argument( "--train", help="Train ChatGPT with conversions we provided", action="store_true") + parser.add_argument("-v", "--verbose", + help="Show more details", action="store_true") + parser.add_argument( "-u", "--uuid", @@ -120,7 +126,6 @@ No explanation required, no instruction required, don't tell me how to compile a What I want is a eBPF program for: {text}.""" - def construct_running_prompt(text: str) -> str: return f"""You are now a translater from human language to {os.uname()[0]} shell bpftrace command. No explanation required. diff --git a/requirements.txt b/requirements.txt index afdcddb..d8cb519 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,11 +2,9 @@ anyio==3.6.2 astroid==2.14.2 asyncio==3.4.3 autopep8==2.0.1 -black==23.1.0 blobfile==2.0.1 certifi==2022.12.7 charset-normalizer==3.0.1 -click==8.1.3 dill==0.3.6 filelock==3.9.0 flake8==6.0.0 @@ -19,11 +17,7 @@ lazy-object-proxy==1.9.0 lxml==4.9.2 marko==1.3.0 mccabe==0.7.0 -mypy-extensions==1.0.0 OpenAIAuth==0.2.0 -packaging==23.0 -pathspec==0.11.0 -pkg_resources==0.0.0 platformdirs==3.0.0 prompt-toolkit==3.0.36 pycodestyle==2.10.0