third-party-mirrors/ollama
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Draft for Multi-Language Modelfile Creation

Browse Source
This commit is contained in:
Roy Han 2024-05-29 11:51:57 -07:00
parent 646371f56d
commit 73fb9ea36e
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
types/model/name.go
View File

@ -10,6 +10,7 @@ import (
"log/slog" "log/slog"
"path/filepath" "path/filepath"
"strings" "strings"
"unicode"
) )
// Errors // Errors
@ -318,14 +319,14 @@ func isValidPart(kind partKind, s string) bool {
if !isValidLen(kind, s) { if !isValidLen(kind, s) {
return false return false
} }
for i := range s { for i, c := range s {
if i == 0 { if i == 0 {
if !isAlphanumericOrUnderscore(s[i]) { if !isLetterorUnderscore(c) {
return false return false
} }
continue continue
} }
switch s[i] { switch c {
case '_', '-': case '_', '-':
case '.': case '.':
if kind == kindNamespace { if kind == kindNamespace {
@ -336,7 +337,7 @@ func isValidPart(kind partKind, s string) bool {
return false return false
} }
default: default:
if !isAlphanumericOrUnderscore(s[i]) { if !isLetterorUnderscore(c) {
return false return false
} }
} }
@ -344,8 +345,8 @@ func isValidPart(kind partKind, s string) bool {
return true return true
} }
func isAlphanumericOrUnderscore(c byte) bool { func isLetterorUnderscore(c rune) bool {
return c >= 'A' && c <= 'Z' || c >= 'a' && c <= 'z' || c >= '0' && c <= '9' || c == '_' return unicode.IsLetter(c) || c == '_'
} }
func cutLast(s, sep string) (before, after string, ok bool) { func cutLast(s, sep string) (before, after string, ok bool) {