aljaz/gpt-engineer
1
0
Fork 0
mirror of https://github.com/aljazceru/gpt-engineer.git synced 2025-12-17 12:45:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add --verbose option for logging request/response

Browse Source
This commit is contained in:
Carl Thomé
2023-06-18 16:21:16 +02:00
parent 16a3278ee3
commit 38401b2f35
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
gpt_engineer/ai.py
View File

@@ -1,5 +1,9 @@
import logging
import openai
logger = logging.getLogger(__name__)
class AI:
def __init__(self, **kwargs):
@@ -36,6 +40,7 @@ class AI:
if prompt:
messages = messages + [{"role": "user", "content": prompt}]
logger.debug(f"Creating a new chat completion: {messages}")
response = openai.ChatCompletion.create(
messages=messages, stream=True, **self.kwargs
)
@@ -46,4 +51,6 @@ class AI:
msg = delta.get("content", "")
print(msg, end="")
chat.append(msg)
return messages + [{"role": "assistant", "content": "".join(chat)}]
messages = messages + [{"role": "assistant", "content": "".join(chat)}]
logger.debug(f"Chat completion finished: {messages}")
return messages