loader

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

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

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

type ModelClient interface {
	CreateBaseModel(ctx context.Context, in *mv1.CreateBaseModelRequest, opts ...grpc.CallOption) (*mv1.BaseModel, error)
	GetBaseModelPath(ctx context.Context, in *mv1.GetBaseModelPathRequest, opts ...grpc.CallOption) (*mv1.GetBaseModelPathResponse, error)
	GetModelPath(ctx context.Context, in *mv1.GetModelPathRequest, opts ...grpc.CallOption) (*mv1.GetModelPathResponse, error)
	RegisterModel(ctx context.Context, in *mv1.RegisterModelRequest, opts ...grpc.CallOption) (*mv1.RegisterModelResponse, error)
	PublishModel(ctx context.Context, in *mv1.PublishModelRequest, opts ...grpc.CallOption) (*mv1.PublishModelResponse, error)

	CreateHFModelRepo(ctx context.Context, in *mv1.CreateHFModelRepoRequest, opts ...grpc.CallOption) (*mv1.HFModelRepo, error)
	GetHFModelRepo(ctx context.Context, in *mv1.GetHFModelRepoRequest, opts ...grpc.CallOption) (*mv1.HFModelRepo, error)
}

type ModelDownloader interface {
	// contains filtered or unexported methods
}

type NoopModelDownloader struct {
}

type NoopS3Client struct {
}

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

type S3Client interface {
	Upload(ctx context.Context, r io.Reader, key string) error
}

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