mirror of https://github.com/ollama/ollama.git
Merge pull request #12393 from ollama/drifkin/fix-built-ins
harmony: don't sanitize built-ins
This commit is contained in:
commit
bbb195a6ff
|
@ -463,6 +463,10 @@ func (h *HarmonyMessageHandler) HasThinkingSupport() bool {
|
||||||
|
|
||||||
func (m *FunctionNameMap) ConvertAndAdd(userFunctionName string) string {
|
func (m *FunctionNameMap) ConvertAndAdd(userFunctionName string) string {
|
||||||
harmonyFunctionName := m.deriveName(userFunctionName)
|
harmonyFunctionName := m.deriveName(userFunctionName)
|
||||||
|
// built-in functions should not be renamed
|
||||||
|
if userFunctionName == "browser.open" || userFunctionName == "browser.search" || userFunctionName == "browser.find" || userFunctionName == "python" {
|
||||||
|
harmonyFunctionName = userFunctionName
|
||||||
|
}
|
||||||
m.userToHarmony[userFunctionName] = harmonyFunctionName
|
m.userToHarmony[userFunctionName] = harmonyFunctionName
|
||||||
m.harmonyToUser[harmonyFunctionName] = userFunctionName
|
m.harmonyToUser[harmonyFunctionName] = userFunctionName
|
||||||
return harmonyFunctionName
|
return harmonyFunctionName
|
||||||
|
|
|
@ -513,6 +513,7 @@ func TestFunctionConvertAndAdd(t *testing.T) {
|
||||||
{name: "dupes from different user-specified names", in: []string{"get weather", "get_weather", "get-weather"}, want: []string{"get_weather", "get_weather_2", "get_weather_3"}},
|
{name: "dupes from different user-specified names", in: []string{"get weather", "get_weather", "get-weather"}, want: []string{"get_weather", "get_weather_2", "get_weather_3"}},
|
||||||
{name: "non dupes after dupes", in: []string{"get weather", "get_weather", "get-weather", "something-different"}, want: []string{"get_weather", "get_weather_2", "get_weather_3", "something_different"}},
|
{name: "non dupes after dupes", in: []string{"get weather", "get_weather", "get-weather", "something-different"}, want: []string{"get_weather", "get_weather_2", "get_weather_3", "something_different"}},
|
||||||
{name: "multiple sets of dupes", in: []string{"a", "a", "b", "a", "a", "b", "a"}, want: []string{"a", "a_2", "b", "a_3", "a_4", "b_2", "a_5"}},
|
{name: "multiple sets of dupes", in: []string{"a", "a", "b", "a", "a", "b", "a"}, want: []string{"a", "a_2", "b", "a_3", "a_4", "b_2", "a_5"}},
|
||||||
|
{name: "built-in functions should not be renamed", in: []string{"browser.open", "python", "not.a.built-in.function", "browser.not_a_real_built_in"}, want: []string{"browser.open", "python", "not_a_built_in_function", "browser_not_a_real_built_in"}},
|
||||||
}
|
}
|
||||||
|
|
||||||
for i, tt := range tests {
|
for i, tt := range tests {
|
||||||
|
|
Loading…
Reference in New Issue