service

Documentation

Index

• func ConvertPipelineToResourceName(pipelineName string) string
• func GenOwnerPermalink(owner *mgmtPB.User) string
• func InjectOwnerToContext(ctx context.Context, owner *mgmtPB.User) context.Context
• type ImageInput
• type Service
  • func NewService(r repository.Repository, u mgmtPB.MgmtPrivateServiceClient, ...) Service
• type TextGenerationInput
• type TextToImageInput

Functions

func ConvertPipelineToResourceName

func ConvertPipelineToResourceName(pipelineName string) string

func GenOwnerPermalink

func GenOwnerPermalink(owner *mgmtPB.User) string

func InjectOwnerToContext

func InjectOwnerToContext(ctx context.Context, owner *mgmtPB.User) context.Context

Types

type ImageInput

type ImageInput struct {
	Content     []byte
	FileNames   []string
	FileLengths []uint64
}

type Service

type Service interface {
	GetMgmtPrivateServiceClient() mgmtPB.MgmtPrivateServiceClient
	GetRedisClient() *redis.Client

	CreatePipeline(owner *mgmtPB.User, pipeline *datamodel.Pipeline) (*datamodel.Pipeline, error)
	ListPipelines(owner *mgmtPB.User, pageSize int64, pageToken string, isBasicView bool, filter filtering.Filter) ([]datamodel.Pipeline, int64, string, error)
	GetPipelineByID(id string, owner *mgmtPB.User, isBasicView bool) (*datamodel.Pipeline, error)
	GetPipelineByUID(uid uuid.UUID, owner *mgmtPB.User, isBasicView bool) (*datamodel.Pipeline, error)
	UpdatePipeline(id string, owner *mgmtPB.User, updatedPipeline *datamodel.Pipeline) (*datamodel.Pipeline, error)
	DeletePipeline(id string, owner *mgmtPB.User) error
	UpdatePipelineState(id string, owner *mgmtPB.User, state datamodel.PipelineState) (*datamodel.Pipeline, error)
	UpdatePipelineID(id string, owner *mgmtPB.User, newID string) (*datamodel.Pipeline, error)
	TriggerPipeline(req *pipelinePB.TriggerPipelineRequest, owner *mgmtPB.User, pipeline *datamodel.Pipeline) (*pipelinePB.TriggerPipelineResponse, error)
	TriggerPipelineBinaryFileUpload(owner *mgmtPB.User, pipeline *datamodel.Pipeline, task modelPB.Model_Task, input interface{}) (*pipelinePB.TriggerPipelineBinaryFileUploadResponse, error)
	GetModelByName(owner *mgmtPB.User, modelName string) (*modelPB.Model, error)

	ListPipelinesAdmin(pageSize int64, pageToken string, isBasicView bool, filter filtering.Filter) ([]datamodel.Pipeline, int64, string, error)
	GetPipelineByIDAdmin(id string, isBasicView bool) (*datamodel.Pipeline, error)
	GetPipelineByUIDAdmin(uid uuid.UUID, isBasicView bool) (*datamodel.Pipeline, error)
	// Controller APIs
	GetResourceState(pipelineName string) (*pipelinePB.Pipeline_State, error)
	UpdateResourceState(pipelineName string, state pipelinePB.Pipeline_State, progress *int32) error
	DeleteResourceState(pipelineName string) error
}

Service interface

func NewService

func NewService(r repository.Repository,
	u mgmtPB.MgmtPrivateServiceClient,
	c connectorPB.ConnectorPublicServiceClient,
	cPrivate connectorPB.ConnectorPrivateServiceClient,
	m modelPB.ModelPublicServiceClient,
	mPrivate modelPB.ModelPrivateServiceClient,
	ct controllerPB.ControllerPrivateServiceClient,
	rc *redis.Client,
) Service

NewService initiates a service instance

type TextGenerationInput

type TextGenerationInput struct {
	Prompt        string
	OutputLen     int64
	BadWordsList  string
	StopWordsList string
	TopK          int64
	Seed          int64
}

type TextToImageInput

type TextToImageInput struct {
	Prompt   string
	Steps    int64
	CfgScale float32
	Seed     int64
	Samples  int64
}