From 614799ddb3a5cd5773d3fee9a22e35de18c2d6b0 Mon Sep 17 00:00:00 2001 From: Jonghyeon yeo Date: Sun, 15 Jan 2023 11:04:23 +0900 Subject: [PATCH] Fix generate annoying triple backtick --- src/main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/main.py b/src/main.py index 58d38b6..9c938b8 100644 --- a/src/main.py +++ b/src/main.py @@ -1,8 +1,9 @@ import os import sys + +import requests from colorama import Fore from halo import Halo -import requests spinner = Halo(text='Querying GPT-3', spinner='dots') @@ -41,6 +42,9 @@ def model_query(prompt: str) -> str: def process(): prompt = ' '.join(sys.argv[1:]) result = model_query(prompt) + # Monkey patch: if result last line start with ``` remove the last line + if result.splitlines()[-1].startswith('```'): + result = '\n'.join(result.splitlines()[:-1]) print_prompt(result) response = input(Fore.RED + '>> Do you want to run this program? [Y/n] ') if response == '' or response.lower() == 'y':