stability

Documentation

Overview

Package stability implements engine.Engine for the Stability AI API.

Image generation uses POST /v2beta/stable-image/generate/{model} with multipart/form-data encoding. Auth: Authorization: Bearer {key}, env STABILITY_API_KEY.

Index

• Constants
• Variables
• func ConfigSchema() []engine.ConfigField
• func ModelsByCapability() map[string][]string
• type Config
• type Engine
  • func New(cfg Config) *Engine
  • func (e *Engine) Capabilities() engine.Capability
  • func (e *Engine) Execute(ctx context.Context, g workflow.Graph) (engine.Result, error)

Constants

const (
	ModelSD35Large      = "sd3.5-large"
	ModelSD35LargeTurbo = "sd3.5-large-turbo"
	ModelSD35Medium     = "sd3.5-medium"
	ModelSD3Turbo       = "sd3-turbo"
	ModelImageCore      = "stable-image-core"
	ModelImageUltra     = "stable-image-ultra"
)

Model constants.

Variables

var (
	ErrMissingPrompt = errors.New("stability: missing prompt in workflow graph")
)

Functions

func ConfigSchema

func ConfigSchema() []engine.ConfigField

ConfigSchema returns the configuration fields required by the Stability AI engine.

func ModelsByCapability

func ModelsByCapability() map[string][]string

ModelsByCapability returns all known Stability AI models grouped by capability.

Types

type Config

type Config struct {
	APIKey     string
	BaseURL    string
	Model      string
	HTTPClient *http.Client
}

Config configures the Stability AI engine.

type Engine

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

Engine implements engine.Engine for Stability AI.

func New

func New(cfg Config) *Engine

New creates a Stability AI engine instance.

func (*Engine) Capabilities

func (e *Engine) Capabilities() engine.Capability

Capabilities implements engine.Describer.

func (*Engine) Execute

func (e *Engine) Execute(ctx context.Context, g workflow.Graph) (engine.Result, error)

Execute generates an image via the Stability AI API.