tool

Documentation

Index

• type JSONSchema
• type Registry
  • func NewRegistry() *Registry
  • func (r *Registry) Get(name string) (Tool, bool)
  • func (r *Registry) Register(tool Tool)
  • func (r *Registry) Specs() []Spec
• type Spec
• type Tool
  • func Func[I any, O any](name string, description string, fn func(context.Context, I) (O, error)) (Tool, error)

Types

type JSONSchema

type JSONSchema struct {
	Type        string                `json:"type,omitempty"`
	Description string                `json:"description,omitempty"`
	Properties  map[string]JSONSchema `json:"properties,omitempty"`
	Required    []string              `json:"required,omitempty"`
	Items       *JSONSchema           `json:"items,omitempty"`
}

type Registry

type Registry struct {
	// contains filtered or unexported fields
}

func NewRegistry

func NewRegistry() *Registry

func (*Registry) Get

func (r *Registry) Get(name string) (Tool, bool)

func (*Registry) Register

func (r *Registry) Register(tool Tool)

func (*Registry) Specs

func (r *Registry) Specs() []Spec

type Spec

type Spec struct {
	Name        string     `json:"name"`
	Description string     `json:"description"`
	Parameters  JSONSchema `json:"parameters"`
}

type Tool

type Tool interface {
	Name() string
	Description() string
	Spec() Spec

	Call(ctx context.Context, input json.RawMessage) (any, error)
}

func Func

func Func[I any, O any](
	name string,
	description string,
	fn func(context.Context, I) (O, error),
) (Tool, error)