baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix prompt templates

Browse source
This commit is contained in:
Michael Yang 2023-07-06 17:03:00 -07:00
parent 963c20254e
commit 15c114decb
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
server/routes.go
View file

@ -6,6 +6,7 @@ import (
"fmt"
"io"
"log"
"math"
"net"
"net/http"
"path"
@ -48,7 +49,7 @@ func generate(c *gin.Context) {
templateNames = append(templateNames, template.Name())
}
match, _ := matchRankOne(path.Base(req.Prompt), templateNames)
match, _ := matchRankOne(path.Base(req.Model), templateNames)
if template := templates.Lookup(match); template != nil {
var sb strings.Builder
if err := template.Execute(&sb, req); err != nil {
@ -146,8 +147,9 @@ func Serve(ln net.Listener) error {
}
func matchRankOne(source string, targets []string) (bestMatch string, bestRank int) {
bestRank = math.MaxInt
for _, target := range targets {
if rank := fuzzy.LevenshteinDistance(source, target); bestRank < rank {
if rank := fuzzy.LevenshteinDistance(source, target); bestRank > rank {
bestRank = rank
bestMatch = target
}