aljaz/dev-gpt
1
0
Fork 0
mirror of https://github.com/aljazceru/dev-gpt.git synced 2025-12-29 11:24:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

🧪 test: level 2

Browse Source
This commit is contained in:
Florian Hönicke
2023-04-30 18:10:05 +02:00
parent e4fc2e6785
commit cb738c9332
2 changed files with 7 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/apis/gpt.py
View File

@@ -108,6 +108,9 @@ class _GPTConversation:
self.messages = messages
self.print_stream = print_stream
self.print_costs = print_costs
def print_messages(self, messages):
for i, message in enumerate(messages):
if os.environ['VERBOSE'].lower() == 'true':
if isinstance(message, SystemMessage):
@@ -121,9 +124,7 @@ class _GPTConversation:
MassageClass = HumanMessage if role == 'user' else SystemMessage
chat_message = MassageClass(content=prompt)
self.messages.append(chat_message)
if os.environ['VERBOSE'].lower() == 'true':
color = 'blue' if role == 'user' else 'magenta'
print_colored(role, prompt, color)
self.print_messages(self.messages)
if self.print_stream:
print_colored('assistant', '', 'green', end='')
print('thinking...')