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

Merge pull request #50 from jmorganca/go-prompts

Browse source 
fix prompt templates
This commit is contained in:
Michael Yang 2023-07-06 17:07:00 -07:00 committed by GitHub
commit 7d500692f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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" "fmt"
"io" "io"
"log" "log"
"math"
"net" "net"
"net/http" "net/http"
"path" "path"
@ -48,7 +49,7 @@ func generate(c *gin.Context) {
templateNames = append(templateNames, template.Name()) 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 { if template := templates.Lookup(match); template != nil {
var sb strings.Builder var sb strings.Builder
if err := template.Execute(&sb, req); err != nil { 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) { func matchRankOne(source string, targets []string) (bestMatch string, bestRank int) {
bestRank = math.MaxInt
for _, target := range targets { for _, target := range targets {
if rank := fuzzy.LevenshteinDistance(source, target); bestRank < rank { if rank := fuzzy.LevenshteinDistance(source, target); bestRank > rank {
bestRank = rank bestRank = rank
bestMatch = target bestMatch = target
} }