jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

types: add any type and validation for ToolFunction enum (#10166)

Browse Source
This commit is contained in:
Parth Sareen
2025-04-08 15:05:38 -07:00
committed by GitHub
parent ccc8c6777b
commit 6747099d71
4 changed files with 71 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
server/routes_generate_test.go
View File

@@ -374,7 +374,7 @@ func TestGenerateChat(t *testing.T) {
Properties map[string]struct {
Type api.PropertyType `json:"type"`
Description string `json:"description"`
Enum []string `json:"enum,omitempty"`
Enum []any `json:"enum,omitempty"`
} `json:"properties"`
}{
Type: "object",
@@ -382,7 +382,7 @@ func TestGenerateChat(t *testing.T) {
Properties: map[string]struct {
Type api.PropertyType `json:"type"`
Description string `json:"description"`
Enum []string `json:"enum,omitempty"`
Enum []any `json:"enum,omitempty"`
}{
"location": {
Type: api.PropertyType{"string"},
@@ -390,7 +390,7 @@ func TestGenerateChat(t *testing.T) {
},
"unit": {
Type: api.PropertyType{"string"},
Enum: []string{"celsius", "fahrenheit"},
Enum: []any{"celsius", "fahrenheit"},
},
},
},
@@ -471,7 +471,7 @@ func TestGenerateChat(t *testing.T) {
Properties map[string]struct {
Type api.PropertyType `json:"type"`
Description string `json:"description"`
Enum []string `json:"enum,omitempty"`
Enum []any `json:"enum,omitempty"`
} `json:"properties"`
}{
Type: "object",
@@ -479,7 +479,7 @@ func TestGenerateChat(t *testing.T) {
Properties: map[string]struct {
Type api.PropertyType `json:"type"`
Description string `json:"description"`
Enum []string `json:"enum,omitempty"`
Enum []any `json:"enum,omitempty"`
}{
"location": {
Type: api.PropertyType{"string"},
@@ -487,7 +487,7 @@ func TestGenerateChat(t *testing.T) {
},
"unit": {
Type: api.PropertyType{"string"},
Enum: []string{"celsius", "fahrenheit"},
Enum: []any{"celsius", "fahrenheit"},
},
},
},