Skip to content

GitLab

Commit f3761405 authored by Michael Yang's avatar Michael Yang
Browse files

use image id

parent e49dc9f3
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
server/images.go
View file @ f3761405
...@@ -182,9 +182,10 @@ func (m *Model) ChatPrompts(msgs []api.Message) (*ChatHistory, error) { ...@@ -182,9 +182,10 @@ func (m *Model) ChatPrompts(msgs []api.Message) (*ChatHistory, error) {
currentVars.Prompt = msg.Content currentVars.Prompt = msg.Content
for i := range msg.Images { 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{ currentVars.Images = append(currentVars.Images, llm.ImageData{
ID: len(images) + i, ID: id,
Data: msg.Images[i], Data: msg.Images[i],
}) })
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment