fix: Update image handling in process method; ensure only the last two messages retain images and improve debugging output
This commit is contained in:
@@ -75,16 +75,17 @@ class AIProcessor:
|
||||
processor=self,
|
||||
)
|
||||
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]:
|
||||
msg.image = None
|
||||
# Make sure every images except the two last are removed
|
||||
for msg in self.session.messages[:-3]:
|
||||
if msg.image and not msg.disable_image:
|
||||
msg.image = None
|
||||
# copy of self.session.messages, but shorten the image data for better debugging
|
||||
cps = [
|
||||
aic.Message(
|
||||
role=msg.role,
|
||||
content=msg.content,
|
||||
image=msg.image[:20] if isinstance(msg.image, str) else None, # type: ignore
|
||||
disable_image=True,
|
||||
disable_image=msg.disable_image,
|
||||
name=msg.name,
|
||||
)
|
||||
for msg in self.session.messages
|
||||
|
Reference in New Issue
Block a user