contracts

type CallOpts struct {
	Messages Messages
}

type ChatRole string

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

type HTTPOptions struct {
	Host   string `json:"host,omitempty" yaml:"host,omitempty"`
	Port   string `json:"port,omitempty" yaml:"port,omitempty"`
	Path   string `json:"path,omitempty" yaml:"path,omitempty"`
	Method string `json:"method,omitempty" yaml:"method,omitempty"`
	Schema string `json:"schema,omitempty" yaml:"schema,omitempty"`
}

type Handler struct {
	Name        string            `json:"name"`                  // Name is the name of the handler.
	Description map[string]string `json:"description,omitempty"` // Description is an optional description of the handler.
	Args        map[string]any    `json:"args"`                  // Args are the arguments for the handler.

	Do HandlerFunc `json:"-"` // Do is the handler function itself, which is not serialized to JSON.
}

type HandlerContext interface {

	// Context returns the underlying context.
	Context() context.Context

	// Get retrieves the value of the specified input.
	Get(inputName string) any

	// GetString retrieves the string representation of the specified input.
	GetString(inputName string) (string, bool)

	// GetInt retrieves the integer representation of the specified input.
	GetInt(inputName string) (int, bool)

	// GetFloat retrieves the float representation of the specified input.
	GetFloat(inputName string) (float64, bool)

	// GetBool retrieves the boolean representation of the specified input.
	GetBool(inputName string) (bool, bool)

	// GetInput retrieves an Input by its name, if present.
	GetInput(name string) (Input, bool)

	// Inputs returns all input mappings.
	Inputs() map[string]Input

	// Set assigns a value to the specified output.
	Set(outputName string, value any)

	// GetOutput retrieves an Output by its name, if present.
	GetOutput(name string) (Output, bool)

	// Outputs returns all output mappings.
	Outputs() map[string]Output

	// JSON populates the outputs by marshaling and unmarshaling the provided data.
	JSON(data any) error

	// Messages returns all messages in the context.
	Messages() Messages

	// SetMessages sets the messages in the context.
	SetMessages(messages Messages)
}

type HandlerContextOpts func(handlerContext HandlerContext)

type HandlerFunc func(ctx HandlerContext) error

type IOType string

type Input struct {
	Name         string            `json:"name" yaml:"name"`                                   // Name of the input.
	Type         IOType            `json:"type" yaml:"type"`                                   // Type of the input (e.g., text, audio, image, etc.).
	Description  map[string]string `json:"description,omitempty" yaml:"description,omitempty"` // Optional description of the input.
	DefaultValue any               `json:"default,omitempty" yaml:"default,omitempty"`         // Optional default value for the input.
	IsRequired   bool              `json:"is_required" yaml:"is_required"`                     // Indicates if the input is required.
	// contains filtered or unexported fields
}

type InputFilter func(input Input) bool

type InputPublic struct {
	Name         string            `json:"name" yaml:"name"`                                   // Name of the input.
	Type         IOType            `json:"type" yaml:"type"`                                   // Type of the input (e.g., text, audio, image, etc.).
	Description  map[string]string `json:"description,omitempty" yaml:"description,omitempty"` // Optional description of the input.
	DefaultValue any               `json:"default,omitempty" yaml:"default,omitempty"`         // Optional default value for the input.
	IsRequired   bool              `json:"is_required" yaml:"is_required"`                     // Indicates if the input is required.
}

type Messages []map[ChatRole]string

type Method struct {
	Name        string            `json:"name" yaml:"name"`
	Description map[string]string `json:"description,omitempty" yaml:"description,omitempty"`

	Inputs  []Input  `json:"inputs" yaml:"inputs"`
	Outputs []Output `json:"outputs" yaml:"outputs"`

	IsDefault bool `json:"is_default" yaml:"is_default"`

	Handler *Handler `json:"handler" yaml:"handler"`
}

type MethodPublic struct {
	Name        string            `json:"name" yaml:"name"`
	Description map[string]string `json:"description,omitempty" yaml:"description,omitempty"`
	Inputs      []InputPublic     `json:"inputs" yaml:"inputs"`
	Outputs     []OutputPublic    `json:"outputs" yaml:"outputs"`
	IsDefault   bool              `json:"is_default" yaml:"is_default"`
}

type MethodResponse struct {
	Outputs     map[string]Output // Outputs contains the output data of the method.
	ServiceName string            // ServiceName is the name of the service that the method belongs to.
	Err         error             // Err contains any error encountered during method execution.
	ErrCode     int               // ErrCode is the error code of the method.
}

type Output struct {
	Name         string            `json:"name" yaml:"name"`                                   // Name of the output.
	Type         IOType            `json:"type" yaml:"type"`                                   // Type of the output (e.g., text, audio, image, etc.).
	Description  map[string]string `json:"description,omitempty" yaml:"description,omitempty"` // Optional description of the output.
	DefaultValue any               `json:"default,omitempty" yaml:"default,omitempty"`         // Optional default value for the output.
	IsPrivate    bool              `json:"is_private" yaml:"is_private"`                       // Indicates if the output is private (for users). TODO: rename.
	// contains filtered or unexported fields
}

type OutputPublic struct {
	Name         string            `json:"name" yaml:"name"`                                   // Name of the output.
	Type         IOType            `json:"type" yaml:"type"`                                   // Type of the output (e.g., text, audio, image, etc.).
	Description  map[string]string `json:"description,omitempty" yaml:"description,omitempty"` // Optional description of the output.
	DefaultValue any               `json:"default,omitempty" yaml:"default,omitempty"`         // Optional default value for the output.
	IsPrivate    bool              `json:"is_private" yaml:"is_private"`                       // Indicates if the output is private (for users). TODO: rename.
}

type Ping struct {
	// Name of the ping operation
	Name string `json:"name" yaml:"name"`

	// Args - Arguments for the ping operation, can vary by implementation
	Args map[string]any `json:"args,omitempty" yaml:"args,omitempty"`
	// contains filtered or unexported fields
}

type PingFunc func(ctx context.Context) error

type Service struct {
	Name        string            `json:"name" yaml:"name"`
	Description map[string]string `json:"description,omitempty" yaml:"description,omitempty"`

	Methods map[string]*Method `json:"methods" yaml:"methods"`

	Pinger *Ping `json:"pinger,omitempty" yaml:"pinger,omitempty"`
}

type ServicePublic struct {
	Name        string                  `json:"name" yaml:"name"`
	Description map[string]string       `json:"description,omitempty" yaml:"description,omitempty"`
	Methods     map[string]MethodPublic `json:"methods" yaml:"methods"`
}

type WSOptions struct {
	Host   string `json:"host,omitempty" yaml:"host,omitempty"`
	Port   string `json:"port,omitempty" yaml:"port,omitempty"`
	Path   string `json:"path,omitempty" yaml:"path,omitempty"`
	Schema string `json:"schema,omitempty" yaml:"schema,omitempty"`
}