server

Documentation

Index

• type DefaultK8sJobClient
  • func NewK8sJobClient(client kubernetes.Interface, jobNamespace string) *DefaultK8sJobClient
  • func (c *DefaultK8sJobClient) CancelJob(ctx context.Context, jobID string) error
• type S
  • func New(store *store.S, fileGetClient fileGetClient, modelClient modelClient, ...) *S
  • func (s *S) CancelJob(ctx context.Context, req *v1.CancelJobRequest) (*v1.Job, error)
  • func (s *S) CreateJob(ctx context.Context, req *v1.CreateJobRequest) (*v1.Job, error)
  • func (s *S) ListJobs(ctx context.Context, req *v1.ListJobsRequest) (*v1.ListJobsResponse, error)
  • func (s *S) Run(ctx context.Context, port int, authConfig config.AuthConfig) error
  • func (s *S) Stop()

Types

type DefaultK8sJobClient

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

DefaultK8sJobClient is a client for Kubernetes Job resources.

func NewK8sJobClient

func NewK8sJobClient(client kubernetes.Interface, jobNamespace string) *DefaultK8sJobClient

NewK8sJobClient creates a new K8sJobClient.

func (*DefaultK8sJobClient) CancelJob

func (c *DefaultK8sJobClient) CancelJob(ctx context.Context, jobID string) error

CancelJob cancels a job.

type S

type S struct {
	v1.UnimplementedFineTuningServiceServer
	// contains filtered or unexported fields
}

S is a server.

func New

func New(
	store *store.S,
	fileGetClient fileGetClient,
	modelClient modelClient,
	k8sJobClient k8sJobClient,
) *S

New creates a server.

func (*S) CancelJob

func (s *S) CancelJob(
	ctx context.Context,
	req *v1.CancelJobRequest,
) (*v1.Job, error)

CancelJob cancels a job.

func (*S) CreateJob

func (s *S) CreateJob(
	ctx context.Context,
	req *v1.CreateJobRequest,
) (*v1.Job, error)

CreateJob creates a new job.

func (*S) ListJobs

func (s *S) ListJobs(
	ctx context.Context,
	req *v1.ListJobsRequest,
) (*v1.ListJobsResponse, error)

ListJobs lists all jobs.

func (*S) Run

func (s *S) Run(ctx context.Context, port int, authConfig config.AuthConfig) error

Run starts the gRPC server.

func (*S) Stop

func (s *S) Stop()

Stop stops the gRPC server.