openai

Documentation

Index

• Variables
• func NewDiffusionOptionsOpenAI() common.DiffusionOptions
• type DiffusionOpenAI
  • func NewDiffusionOpenAI(token string) *DiffusionOpenAI
  • func (d *DiffusionOpenAI) Generate(message common.DiffusionMessage) (common.DiffusionResult, error)
  • func (d *DiffusionOpenAI) GenerateWithContext(ctx context.Context, message common.DiffusionMessage) (common.DiffusionResult, error)
  • func (d *DiffusionOpenAI) GenerateWithOptions(ctx context.Context, opts common.DiffusionOptions) (common.DiffusionResult, error)
• type DiffusionOptionsOpenAI
  • func (c DiffusionOptionsOpenAI) GetModel() *common.DiffusionModel
  • func (c DiffusionOptionsOpenAI) GetRequest() interface{}
  • func (c DiffusionOptionsOpenAI) GetTimeout() time.Duration
  • func (c DiffusionOptionsOpenAI) WithCount(count int) common.DiffusionOptions
  • func (c DiffusionOptionsOpenAI) WithFormat(format string) common.DiffusionOptions
  • func (c DiffusionOptionsOpenAI) WithModel(model *common.DiffusionModel) common.DiffusionOptions
  • func (c DiffusionOptionsOpenAI) WithPrompt(prompt string) common.DiffusionOptions
  • func (c DiffusionOptionsOpenAI) WithTimeout(timeout time.Duration) common.DiffusionOptions

Variables

var (
	DallE256X256 = &common.DiffusionModel{
		Name:        "dall-e-256x256",
		ContextSize: 1000,
		ContextUnit: schema.ContextUnitChar,
		UsageCost:   &schema.CostObject{Price: 0.016, Unit: schema.CostUnitImage, UnitSize: 1},
	}

	DallE512X512 = &common.DiffusionModel{
		Name:        "dall-e-512x512",
		ContextSize: 1000,
		ContextUnit: schema.ContextUnitChar,
		UsageCost:   &schema.CostObject{Price: 0.018, Unit: schema.CostUnitImage, UnitSize: 1},
	}

	DallE1024X1024 = &common.DiffusionModel{
		Name:        "dall-e-1024x1024",
		ContextSize: 1000,
		ContextUnit: schema.ContextUnitChar,
		UsageCost:   &schema.CostObject{Price: 0.020, Unit: schema.CostUnitImage, UnitSize: 1},
	}
)

Functions

func NewDiffusionOptionsOpenAI

func NewDiffusionOptionsOpenAI() common.DiffusionOptions

Types

type DiffusionOpenAI

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

func NewDiffusionOpenAI

func NewDiffusionOpenAI(token string) *DiffusionOpenAI

func (*DiffusionOpenAI) Generate

func (d *DiffusionOpenAI) Generate(message common.DiffusionMessage) (common.DiffusionResult, error)

func (*DiffusionOpenAI) GenerateWithContext

func (d *DiffusionOpenAI) GenerateWithContext(ctx context.Context, message common.DiffusionMessage) (common.DiffusionResult, error)

func (*DiffusionOpenAI) GenerateWithOptions

func (d *DiffusionOpenAI) GenerateWithOptions(ctx context.Context, opts common.DiffusionOptions) (common.DiffusionResult, error)

type DiffusionOptionsOpenAI

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

func (DiffusionOptionsOpenAI) GetModel

func (c DiffusionOptionsOpenAI) GetModel() *common.DiffusionModel

func (DiffusionOptionsOpenAI) GetRequest

func (c DiffusionOptionsOpenAI) GetRequest() interface{}

func (DiffusionOptionsOpenAI) GetTimeout

func (c DiffusionOptionsOpenAI) GetTimeout() time.Duration

func (DiffusionOptionsOpenAI) WithCount

func (c DiffusionOptionsOpenAI) WithCount(count int) common.DiffusionOptions

func (DiffusionOptionsOpenAI) WithFormat

func (c DiffusionOptionsOpenAI) WithFormat(format string) common.DiffusionOptions

func (DiffusionOptionsOpenAI) WithModel

func (c DiffusionOptionsOpenAI) WithModel(model *common.DiffusionModel) common.DiffusionOptions

func (DiffusionOptionsOpenAI) WithPrompt

func (c DiffusionOptionsOpenAI) WithPrompt(prompt string) common.DiffusionOptions

func (DiffusionOptionsOpenAI) WithTimeout

func (c DiffusionOptionsOpenAI) WithTimeout(timeout time.Duration) common.DiffusionOptions