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

more logging

Browse Source
This commit is contained in:
Nicholas Albion
2023-10-04 15:42:50 +11:00
parent 8914bb438e
commit 11da00605a
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/utils/llm_connection.py
View File

@@ -154,6 +154,7 @@ def retry_on_exception(func):
except Exception as e:
# Convert exception to string
err_str = str(e)
logger.info(f'##### 6, err_str: {err_str}')
# If the specific error "context_length_exceeded" is present, simply return without retry
if isinstance(e, json.JSONDecodeError):
@@ -298,7 +299,7 @@ def stream_gpt_completion(data, req_type, project):
)
# Log the response status code and message
logger.debug(f'Response status code: {response.status_code}')
logger.info(f'Response status code: {response.status_code}')
if response.status_code != 200:
logger.info(f'problem with request: {response.text}')
@@ -307,6 +308,7 @@ def stream_gpt_completion(data, req_type, project):
# function_calls = {'name': '', 'arguments': ''}
for line in response.iter_lines():
logger.info(f'##### 0, line: {line}')
# Ignore keep-alive new lines
if line and line != b': OPENROUTER PROCESSING':
line = line.decode("utf-8") # decode the bytes to string