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

⏪ fix: fix bring back the test description for func generation

Browse Source
This commit is contained in:
Florian Hönicke
2023-05-26 13:30:56 +02:00
parent a7d5cec0bd
commit 5796244e4a
3 changed files with 32 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
dev_gpt/apis/gpt.py
View File

@@ -16,7 +16,7 @@ from urllib3.exceptions import InvalidChunkLength
from dev_gpt.constants import PRICING_GPT4_PROMPT, PRICING_GPT4_GENERATION, PRICING_GPT3_5_TURBO_PROMPT, \
PRICING_GPT3_5_TURBO_GENERATION, CHARS_PER_TOKEN
from dev_gpt.options.generate.conversation_logger import ConversationLogger
from dev_gpt.options.generate.conversation_logger import ConversationLogger, Timer
from dev_gpt.options.generate.parser import identity_parser
from dev_gpt.options.generate.templates_system import template_system_message_base
from dev_gpt.utils.string_tools import print_colored, get_template_parameters
@@ -128,14 +128,15 @@ class _GPTConversation:
self.conversation_logger = conversation_logger
def print_messages(self, messages):
t = Timer().get_time_since_start()
for i, message in enumerate(messages):
if os.environ['VERBOSE'].lower() == 'true':
if isinstance(message, SystemMessage):
print_colored(f'({i}) system - prompt', message.content, 'magenta')
print_colored(f'{t} - ({i}) system - prompt', message.content, 'magenta')
elif isinstance(message, HumanMessage):
print_colored(f'({i}) user - prompt', message.content, 'blue')
print_colored(f'{t} - ({i}) user - prompt', message.content, 'blue')
elif isinstance(message, AIMessage):
print_colored(f'({i}) assistant - prompt', message.content, 'green')
print_colored(f'{t} - ({i}) assistant - prompt', message.content, 'green')
def chat(self, prompt: str, role: str = 'user'):
MassageClass = HumanMessage if role == 'user' else SystemMessage
@@ -143,7 +144,7 @@ class _GPTConversation:
self.messages.append(chat_message)
self.print_messages(self.messages)
if self.print_stream:
print_colored('assistant', '', 'green', end='')
print_colored(f'{Timer().get_time_since_start()} - assistant', '', 'green', end='')
print('thinking...')
for i in range(10):
try: