embed templates

This commit is contained in:
Michael Yang 2023-07-06 11:33:29 -07:00 committed by Jeffrey Morgan
parent 39f4d8edaa
commit 9b8a456c7d
14 changed files with 4 additions and 1 deletions

View file

@ -1,6 +1,7 @@
package server
import (
"embed"
"encoding/json"
"fmt"
"io"
@ -19,7 +20,9 @@ import (
"github.com/jmorganca/ollama/llama"
)
var templates = template.Must(template.ParseGlob("templates/*.prompt"))
//go:embed templates/*
var templatesFS embed.FS
var templates = template.Must(template.ParseFS(templatesFS, "templates/*.prompt"))
func generate(c *gin.Context) {
// TODO: these should be request parameters