From 1f55f899b5b7e2d7ba879aef71017c1d3a5d90c6 Mon Sep 17 00:00:00 2001 From: Zvonimir Sabljic Date: Tue, 1 Aug 2023 08:47:19 +0200 Subject: [PATCH] 2 fixes --- euclid/helpers/AgentConvo.py | 3 +-- euclid/prompts/prompts.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/euclid/helpers/AgentConvo.py b/euclid/helpers/AgentConvo.py index 2c87577..781b63d 100644 --- a/euclid/helpers/AgentConvo.py +++ b/euclid/helpers/AgentConvo.py @@ -8,9 +8,8 @@ from termcolor import colored class AgentConvo: - messages = [] - def __init__(self, high_level_step): + self.messages = [] self.high_level_step = high_level_step self.agent = find_role_from_step(high_level_step) diff --git a/euclid/prompts/prompts.py b/euclid/prompts/prompts.py index f98257b..40926a2 100644 --- a/euclid/prompts/prompts.py +++ b/euclid/prompts/prompts.py @@ -110,7 +110,7 @@ def get_additional_info_from_user(messages, role): for message in messages: while True: - if 'text' in message: + if isinstance(message, dict) and 'text' in message: message = message['text'] print(colored( f"Please check this message and say what needs to be changed. If everything is ok just press ENTER",