Steve Dudenhoeffer
7c9eb08cb4
This update introduces support for `jsonschema.Integer` types and updates the logic to handle nested items in schemas. Added a new default error log for unknown types using `slog.Error`. Also, integrated tool configuration with a `FunctionCallingConfig` when `dontRequireTool` is false.
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package schema
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/openai/openai-go"
|
|
"reflect"
|
|
|
|
"golang.org/x/exp/slices"
|
|
|
|
"github.com/sashabaranov/go-openai/jsonschema"
|
|
)
|
|
|
|
type enum struct {
|
|
basic
|
|
|
|
values []string
|
|
}
|
|
|
|
func (e enum) SchemaType() jsonschema.DataType {
|
|
return jsonschema.String
|
|
}
|
|
|
|
func (e enum) FunctionParameters() openai.FunctionParameters {
|
|
return openai.FunctionParameters{
|
|
"type": "string",
|
|
"description": e.Description(),
|
|
"enum": e.values,
|
|
}
|
|
}
|
|
|
|
func (e enum) Definition() jsonschema.Definition {
|
|
def := e.basic.Definition()
|
|
def.Enum = e.values
|
|
return def
|
|
}
|
|
|
|
func (e enum) FromAny(val any) (reflect.Value, error) {
|
|
v := reflect.ValueOf(val)
|
|
if v.Kind() != reflect.String {
|
|
return reflect.Value{}, errors.New("expected string, got " + v.Kind().String())
|
|
}
|
|
|
|
s := v.String()
|
|
if !slices.Contains(e.values, s) {
|
|
return reflect.Value{}, errors.New("value " + s + " not in enum")
|
|
}
|
|
|
|
return v, nil
|
|
}
|
|
|
|
func (e enum) SetValueOnField(obj reflect.Value, val reflect.Value) {
|
|
if !e.required {
|
|
val = val.Addr()
|
|
}
|
|
obj.Field(e.index).Set(val)
|
|
}
|