mirror of
https://github.com/mudler/LocalAI
synced 2026-04-21 13:27:21 +00:00
26 lines
684 B
Go
26 lines
684 B
Go
|
package functions
|
||
|
|
|
||
|
|
import "encoding/json"
|
||
|
|
|
||
|
|
type Item struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
Properties map[string]interface{} `json:"properties"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type JSONFunctionStructure struct {
|
||
|
|
OneOf []Item `json:"oneOf,omitempty"`
|
||
|
|
AnyOf []Item `json:"anyOf,omitempty"`
|
||
|
|
Defs map[string]interface{} `json:"$defs,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j JSONFunctionStructure) Grammar(options ...func(*GrammarOption)) (string, error) {
|
||
|
|
grammarOpts := &GrammarOption{}
|
||
|
|
grammarOpts.Apply(options...)
|
||
|
|
|
||
|
|
dat, err := json.Marshal(j)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
return NewJSONSchemaConverter(grammarOpts.PropOrder).GrammarFromBytes(dat, options...)
|
||
|
|
}
|