add .First

60b4db63 · Michael Yang · 7c6ea2a9 · 60b4db63
Commit 60b4db63 authored Jul 19, 2023 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

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

No files found.
--- a/server/images.go
+++ b/server/images.go
@@ -37,6 +37,7 @@ func (m *Model) Prompt(request api.GenerateRequest) (string, error) {
 	}

 	var vars struct {
+		First  bool
 		System string
 		Prompt string

@@ -44,8 +45,10 @@ func (m *Model) Prompt(request api.GenerateRequest) (string, error) {
 		Context []int
 	}

+	vars.First = len(vars.Context) == 0
 	vars.System = m.System
 	vars.Prompt = request.Prompt
+	vars.Context = request.Context

 	var sb strings.Builder
 	if err := tmpl.Execute(&sb, vars); err != nil {