OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-01-05 02:40:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improved logging

Browse Source
This commit is contained in:
Nicholas Albion
2023-09-29 18:18:48 +10:00
parent d0ff649707
commit 7bf5de8861
6 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/helpers/AgentConvo.py
View File

@@ -80,6 +80,7 @@ class AgentConvo:
# TODO handle errors from OpenAI
if response == {}:
logger.error(f'Aborting with "OpenAI API error happened": {response}')
raise Exception("OpenAI API error happened.")
response = parse_agent_response(response, function_calls)
@@ -187,7 +188,7 @@ class AgentConvo:
if self.agent.project.checkpoints['last_development_step'] is not None:
print(yellow("\nDev step ") + yellow_bold(str(self.agent.project.checkpoints['last_development_step'])) + '\n', end='')
print(f"\n{content}\n")
logger.info(f"{print_msg}: {content}\n")
logger.info(f"{print_msg}: {content}\n", {'password': 'secret', 'user': 'nick'})
def to_playground(self):
with open('const/convert_to_playground_convo.js', 'r', encoding='utf-8') as file:

2
pilot/helpers/Project.py
View File

@@ -19,6 +19,7 @@ from helpers.agents.ProductOwner import ProductOwner
from database.models.development_steps import DevelopmentSteps
from database.models.file_snapshot import FileSnapshot
from database.models.files import File
from logger.logger import logger
class Project:
@@ -119,6 +120,7 @@ class Project:
ignore_user_input_count=True
)
logger.info('should_overwrite_files: %s', should_overwrite_files)
if should_overwrite_files == 'n':
break
elif should_overwrite_files == 'y':

2
pilot/helpers/agents/Developer.py
View File

@@ -162,6 +162,7 @@ class Developer(Agent):
'Type y/n'
)
logger.info(f"Retry step implementation? %s", answer)
if answer == 'n':
return self.dev_help_needed(step)
@@ -198,6 +199,7 @@ class Developer(Agent):
self.project,
'Once you\'re done, type "continue"?'
)
logger.info(f"help needed: %s", answer)
return { "success": True, "user_input": answer }

3
pilot/prompts/prompts.py
View File

@@ -55,6 +55,9 @@ def ask_user(project, question: str, require_some_input=True, hint: str = None):
print(hint, type='hint')
answer = styled_text(project, question)
logger.info('Q: %s', question)
logger.info('A: %s', answer)
if answer is None:
print("Exiting application.")
exit(0)

17
pilot/utils/llm_connection.py
View File

@@ -155,6 +155,7 @@ def retry_on_exception(func):
if isinstance(e, json.JSONDecodeError):
# codellama-34b-instruct seems to send incomplete JSON responses
if e.msg == 'Expecting value':
logger.info('Received incomplete JSON response from LLM. Asking for the rest...')
args[0]['function_buffer'] = e.doc
continue
if "context_length_exceeded" in err_str:
@@ -165,6 +166,7 @@ def retry_on_exception(func):
match = re.search(r"Please try again in (\d+)ms.", err_str)
if match:
# spinner = spinner_start(colored("Rate limited. Waiting...", 'yellow'))
logger.debug('Rate limited. Waiting...')
wait_duration = int(match.group(1)) / 1000
time.sleep(wait_duration)
continue
@@ -172,6 +174,7 @@ def retry_on_exception(func):
print(red(f'There was a problem with request to openai API:'))
# spinner_stop(spinner)
print(err_str)
logger.error(f'There was a problem with request to openai API: {err_str}')
user_message = questionary.text(
"Do you want to try make the same request again? If yes, just press ENTER. Otherwise, type 'no'.",
@@ -217,7 +220,7 @@ def stream_gpt_completion(data, req_type):
def return_result(result_data, lines_printed):
if buffer:
lines_printed += count_lines_based_on_width(buffer, terminal_width)
logger.info(f'lines printed: {lines_printed} - {terminal_width}')
logger.debug(f'lines printed: {lines_printed} - {terminal_width}')
delete_last_n_lines(lines_printed)
return result_data
@@ -225,12 +228,13 @@ def stream_gpt_completion(data, req_type):
# spinner = spinner_start(yellow("Waiting for OpenAI API response..."))
# print(yellow("Stream response from OpenAI:"))
logger.info(f'Request data: {data}')
# Configure for the selected ENDPOINT
model = os.getenv('MODEL_NAME')
endpoint = os.getenv('ENDPOINT')
logger.info(f'> Request model: {model} ({data["model"]}) messages: {data["messages"]}')
if endpoint == 'AZURE':
# If yes, get the AZURE_ENDPOINT from .ENV file
endpoint_url = os.getenv('AZURE_ENDPOINT') + '/openai/deployments/' + model + '/chat/completions?api-version=2023-05-15'
@@ -263,10 +267,10 @@ def stream_gpt_completion(data, req_type):
)
# Log the response status code and message
logger.info(f'Response status code: {response.status_code}')
logger.debug(f'Response status code: {response.status_code}')
if response.status_code != 200:
logger.debug(f'problem with request: {response.text}')
logger.info(f'problem with request: {response.text}')
raise Exception(f"API responded with status code: {response.status_code}. Response text: {response.text}")
# function_calls = {'name': '', 'arguments': ''}
@@ -338,7 +342,7 @@ def stream_gpt_completion(data, req_type):
# logger.info(f'Response via function call: {function_calls["arguments"]}')
# function_calls['arguments'] = load_data_to_json(function_calls['arguments'])
# return return_result({'function_calls': function_calls}, lines_printed)
logger.info(f'Response message: {gpt_response}')
logger.info(f'< Response message: {gpt_response}')
if expecting_json:
gpt_response = clean_json_response(gpt_response)
@@ -352,6 +356,7 @@ def assert_json_response(response: str, or_fail=True) -> bool:
if re.match(r'.*(```(json)?|{|\[)', response):
return True
elif or_fail:
logger.error(f'LLM did not respond with JSON: {response}')
raise ValueError('LLM did not respond with JSON')
else:
return False

1
pilot/utils/questionary.py
View File

@@ -43,7 +43,6 @@ def styled_text(project, question, ignore_user_input_count=False):
user_input = save_user_input(project, question, response)
print('\n\n', end='')
return response