From ca210ba48059c8d7f274fe5b7c77e603b371fdf2 Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Wed, 19 Jul 2023 19:43:00 -0700
Subject: [PATCH] handle vnd.ollama.image.prompt for compat

---
 server/images.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/server/images.go b/server/images.go
index 16b32f83..9be2c68b 100644
--- a/server/images.go
+++ b/server/images.go
@@ -39,6 +39,9 @@ func (m *Model) Prompt(request api.GenerateRequest) (string, error) {
 	var vars struct {
 		System string
 		Prompt string
+
+		// deprecated: versions <= 0.0.7 used this to omit the system prompt
+		Context []int
 	}
 
 	vars.System = m.System
@@ -149,6 +152,14 @@ func GetModel(name string) (*Model, error) {
 			}
 
 			model.System = string(bts)
+		case "application/vnd.ollama.image.prompt":
+			log.Printf("PROMPT is deprecated. Please use TEMPLATE and SYSTEM instead.")
+			bts, err := os.ReadFile(filename)
+			if err != nil {
+				return nil, err
+			}
+
+			model.Template = string(bts)
 		case "application/vnd.ollama.image.params":
 			params, err := os.Open(filename)
 			if err != nil {