From f3761405c88d36becaba7589362aa976a39aa59c Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Thu, 1 Feb 2024 11:52:42 -0800 Subject: [PATCH] use image id --- server/images.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/server/images.go b/server/images.go index 68dae0fe..6f59d72d 100644 --- a/server/images.go +++ b/server/images.go @@ -182,9 +182,10 @@ func (m *Model) ChatPrompts(msgs []api.Message) (*ChatHistory, error) { currentVars.Prompt = msg.Content for i := range msg.Images { - currentVars.Prompt += fmt.Sprintf(" [img-%d]", len(images)+i) + id := len(images) + i + currentVars.Prompt += fmt.Sprintf(" [img-%d]", id) currentVars.Images = append(currentVars.Images, llm.ImageData{ - ID: len(images) + i, + ID: id, Data: msg.Images[i], }) }