mirror of
https://github.com/warmbly/warmbly.git
synced 2026-09-06 08:01:24 +00:00
35 lines
901 B
Go
35 lines
901 B
Go
package generation
|
|
|
|
import (
|
|
"github.com/invopop/jsonschema"
|
|
"github.com/openai/openai-go/v2"
|
|
"github.com/openai/openai-go/v2/option"
|
|
)
|
|
|
|
// GenerateSchema reflects T into the JSON Schema sent as an OpenAI
|
|
// structured-output response_format. The reflector defaults emit a root
|
|
// {"$id", "$ref": "#/$defs/T", "$defs": {...}}, which OpenAI rejects with
|
|
// "$ref cannot have keywords {'$id'}", so the type is inlined anonymously.
|
|
func GenerateSchema[T any]() *jsonschema.Schema {
|
|
reflector := jsonschema.Reflector{
|
|
AllowAdditionalProperties: false,
|
|
DoNotReference: true,
|
|
Anonymous: true,
|
|
}
|
|
var v T
|
|
schema := reflector.Reflect(v)
|
|
schema.Version = ""
|
|
return schema
|
|
}
|
|
|
|
type GenerationClient struct {
|
|
client openai.Client
|
|
}
|
|
|
|
func NewClient(apiKey string) *GenerationClient {
|
|
return &GenerationClient{
|
|
client: openai.NewClient(option.WithAPIKey(apiKey)),
|
|
}
|
|
|
|
}
|