ArtisanCloud

Documentation

Index

• type Driver
  • func NewDriver(config *rcconfig.ArtBot) *Driver
  • func (d *Driver) DetectControlNet(ctx context.Context, info *controlNet.DetectInfo) (interface{}, error)
  • func (d *Driver) GetConfig() *rcconfig.ArtBot
  • func (d *Driver) GetControlNetControlTypesList(ctx context.Context) (*controlNet.ControlNetTypes, error)
  • func (d *Driver) GetControlNetModelList(ctx context.Context) (*controlNet.ControlNetModels, error)
  • func (d *Driver) GetControlNetModuleList(ctx context.Context) (*controlNet.Modules, error)
  • func (d *Driver) GetControlNetSettings(ctx context.Context) (*controlNet.ControlNetSettings, error)
  • func (d *Driver) GetControlNetVersion(ctx context.Context) (*controlNet.ControlNetVersion, error)
  • func (d *Driver) GetLoras(ctx context.Context) ([]*model.Lora, error)
  • func (d *Driver) GetModels(ctx context.Context) ([]*model2.ArtBotModel, error)
  • func (d *Driver) GetOptions(ctx context.Context) (*model2.OptionsResponse, error)
  • func (d *Driver) GetSamplers(ctx context.Context) ([]*model2.Sampler, error)
  • func (d *Driver) GetUrlFromEndpoint(endpoint string) (string, error)
  • func (d *Driver) Image2Image(ctx context.Context, message *model.Message) (*model.Message, error)
  • func (d *Driver) OverrideGetMiddlewareOfLog()
  • func (d *Driver) OverrideGetMiddlewares()
  • func (d *Driver) Progress(ctx context.Context) (*model2.ProgressResponse, error)
  • func (d *Driver) Query(ctx context.Context, endpoint string) (*model.Message, error)
  • func (d *Driver) RefreshLoras(ctx context.Context) error
  • func (d *Driver) RegisterHttpMiddlewares()
  • func (d *Driver) Send(ctx context.Context, endpoint string, message *model.Message) (*model.Message, error)
  • func (d *Driver) SetConfig(config *rcconfig.ArtBot)
  • func (d *Driver) SetOptions(ctx context.Context, options *model2.OptionsRequest) error
  • func (d *Driver) Text2Image(ctx context.Context, message *model.Message) (*model.Message, error)

Types

type Driver

type Driver struct {
	HttpClient httphelper.Helper
	Logger     contract.LoggerInterface

	GetMiddlewareOfLog func(logger contract.LoggerInterface) dataflow.RequestMiddleware
	// contains filtered or unexported fields
}

func NewDriver

func NewDriver(config *rcconfig.ArtBot) *Driver

func (*Driver) DetectControlNet

func (d *Driver) DetectControlNet(ctx context.Context, info *controlNet.DetectInfo) (interface{}, error)

func (*Driver) GetConfig

func (d *Driver) GetConfig() *rcconfig.ArtBot

GetConfig 获取基本配置

func (*Driver) GetControlNetControlTypesList

func (d *Driver) GetControlNetControlTypesList(ctx context.Context) (*controlNet.ControlNetTypes, error)

func (*Driver) GetControlNetModelList

func (d *Driver) GetControlNetModelList(ctx context.Context) (*controlNet.ControlNetModels, error)

func (*Driver) GetControlNetModuleList

func (d *Driver) GetControlNetModuleList(ctx context.Context) (*controlNet.Modules, error)

func (*Driver) GetControlNetSettings

func (d *Driver) GetControlNetSettings(ctx context.Context) (*controlNet.ControlNetSettings, error)

func (*Driver) GetControlNetVersion

func (d *Driver) GetControlNetVersion(ctx context.Context) (*controlNet.ControlNetVersion, error)

func (*Driver) GetLoras

func (d *Driver) GetLoras(ctx context.Context) ([]*model.Lora, error)

func (*Driver) GetModels

func (d *Driver) GetModels(ctx context.Context) ([]*model2.ArtBotModel, error)

func (*Driver) GetOptions

func (d *Driver) GetOptions(ctx context.Context) (*model2.OptionsResponse, error)

func (*Driver) GetSamplers

func (d *Driver) GetSamplers(ctx context.Context) ([]*model2.Sampler, error)

func (*Driver) GetUrlFromEndpoint

func (d *Driver) GetUrlFromEndpoint(endpoint string) (string, error)

func (*Driver) Image2Image

func (d *Driver) Image2Image(ctx context.Context, message *model.Message) (*model.Message, error)

func (*Driver) OverrideGetMiddlewareOfLog

func (d *Driver) OverrideGetMiddlewareOfLog()

func (*Driver) OverrideGetMiddlewares

func (d *Driver) OverrideGetMiddlewares()

func (*Driver) Progress

func (d *Driver) Progress(ctx context.Context) (*model2.ProgressResponse, error)

func (*Driver) Query

func (d *Driver) Query(ctx context.Context, endpoint string) (*model.Message, error)

func (*Driver) RefreshLoras

func (d *Driver) RefreshLoras(ctx context.Context) error

func (*Driver) RegisterHttpMiddlewares

func (d *Driver) RegisterHttpMiddlewares()

func (*Driver) Send

func (d *Driver) Send(ctx context.Context, endpoint string, message *model.Message) (*model.Message, error)

func (*Driver) SetConfig

func (d *Driver) SetConfig(config *rcconfig.ArtBot)

SetConfig 设置基本配置

func (*Driver) SetOptions

func (d *Driver) SetOptions(ctx context.Context, options *model2.OptionsRequest) error

func (*Driver) Text2Image

func (d *Driver) Text2Image(ctx context.Context, message *model.Message) (*model.Message, error)