소스 검색

use image id

Michael Yang 1 년 전
부모
커밋
f3761405c8
1개의 변경된 파일3개의 추가작업 그리고 2개의 파일을 삭제
  1. 3 2
      server/images.go

+ 3 - 2
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],
 				})
 			}