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",