mirrors/ollama-for-amd
1
0
Fork 0
mirror of https://github.com/likelovewant/ollama-for-amd.git synced 2025-12-21 14:26:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

templates: fix crash in improperly defined templates (#12483)

Browse Source
This commit is contained in:
Patrick Devine
2025-10-02 17:25:55 -07:00
committed by GitHub
parent 0bda72892c
commit 1ed2881ef0
3 changed files with 65 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server/images.go
View File

@@ -105,12 +105,16 @@ func (m *Model) Capabilities() []model.Capability {
builtinParser := parsers.ParserForName(m.Config.Parser)
// Check for tools capability
if slices.Contains(m.Template.Vars(), "tools") || (builtinParser != nil && builtinParser.HasToolSupport()) {
v, err := m.Template.Vars()
if err != nil {
slog.Warn("model template contains errors", "error", err)
}
if slices.Contains(v, "tools") || (builtinParser != nil && builtinParser.HasToolSupport()) {
capabilities = append(capabilities, model.CapabilityTools)
}
// Check for insert capability
if slices.Contains(m.Template.Vars(), "suffix") {
if slices.Contains(v, "suffix") {
capabilities = append(capabilities, model.CapabilityInsert)
}