mock

type MockTool ¶

type MockTool struct {
	tool.DefaultTool
	Name_           string
	Description_    string
	InputSchema_    *jsonschema.Schema
	InputSchemaErr_ error
	Result_         any
	RunFn           func(ctx context.Context, input json.RawMessage) (any, error)
}

MockTool is a configurable implementation of llm.Tool for use in tests. Set Name_, Description_, and Result_ before registering it on a server. RunFn, if set, overrides Result_ and is called with the raw JSON input. InputSchemaErr_, if set, makes InputSchema() return that error.

func (m *MockTool) Description() string

func (m *MockTool) InputSchema() (*jsonschema.Schema, error)

func (m *MockTool) Name() string

func (m *MockTool) Run(ctx context.Context, input json.RawMessage) (any, error)