From 2bcddedca5a24df78a17c2ee86a268d03332e3cd Mon Sep 17 00:00:00 2001 From: Showdown76py Date: Mon, 19 May 2025 13:13:28 +0200 Subject: [PATCH] fix: Adjust message handling in process method; ensure correct image assignment and add next steps output --- ai/processor.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/ai/processor.py b/ai/processor.py index 86c5480..3df2440 100644 --- a/ai/processor.py +++ b/ai/processor.py @@ -67,12 +67,19 @@ class AIProcessor: ) outputs.append(r) if r else None # Make sure the two last messages from user has an image, but set disable_image to True for the others - for msg in self.session.messages[-2:]: + for msg in self.session.messages[:-2]: msg.image = None logger.debug( self.session.messages ) if reexec: + self.session.messages.append( + aic.Message( + role="user", + content="Tool Output: Next Steps: " + nextsteps, + ) + ) + img = ai.compute.screenshot_to_base64( ai.compute.take_screenshot() )