use image id

f3761405 · Michael Yang · e49dc9f3 · f3761405
Commit f3761405 authored Feb 01, 2024 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

server/images.go server/images.go +3 -2

No files found.
--- 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],
 				})
 			}