model

Documentation

Index

• Constants
• func EnsureDeploymentCreated(ctx context.Context, c client.Client, namespace string, name string, ...) (*appsv1.Deployment, error)
• func EnsureImageStorePVCCreated(ctx context.Context, client client.Client, namespace string, ...) (*corev1.PersistentVolumeClaim, error)
• func EnsureImageStoreServiceCreated(ctx context.Context, client client.Client, namespace string, ...) (*corev1.Service, error)
• func EnsureImageStoreStatefulSetCreated(ctx context.Context, client client.Client, namespace string, ...) (*appsv1.StatefulSet, error)
• func EnsureServiceCreated(ctx context.Context, c client.Client, namespace string, name string, ...) (*corev1.Service, error)
• func GetImageStorePVByPVC(ctx context.Context, c client.Client, pvc *corev1.PersistentVolumeClaim) (*corev1.PersistentVolume, error)
• func IsAvailable(ctx context.Context, ollamaModelResource ollamav1.Model) bool
• func IsDeploymentReady(ctx context.Context, c client.Client, namespace string, name string, ...) (bool, error)
• func IsImageStoreServiceReady(ctx context.Context, client client.Client, namespace string, ...) (bool, error)
• func IsImageStoreStatefulSetReady(ctx context.Context, client client.Client, namespace string, ...) (bool, error)
• func IsProgressing(ctx context.Context, ollamaModelResource ollamav1.Model) bool
• func IsServiceReady(ctx context.Context, c client.Client, namespace string, name string, ...) (bool, error)
• func ModelAppName(name string) string
• func NewOllamaPullerContainer(image string, serverLocatedNamespace string) corev1.Container
• func NewOllamaServerContainer(readOnly bool) corev1.Container
• func SetAvailable(ctx context.Context, c client.Client, ollamaModelResource ollamav1.Model) (bool, error)
• func SetProgressing(ctx context.Context, c client.Client, ollamaModelResource ollamav1.Model) (bool, error)
• func SetReplicas(ctx context.Context, c client.Client, ollamaModelResource ollamav1.Model, ...) (bool, error)
• func ShouldSetReplicas(ctx context.Context, ollamaModelResource ollamav1.Model, replicas int32, ...) bool
• func UpdateDeployment(ctx context.Context, c client.Client, model *ollamav1.Model, ...) (bool, error)
• type WrappedRecorder
  • func NewWrappedRecorder[T runtime.Object](recorder record.EventRecorder, object T) *WrappedRecorder[T]
  • func (r *WrappedRecorder[T]) AnnotatedEventf(annotations map[string]string, eventType, reason, messageFmt string, ...)
  • func (r *WrappedRecorder[T]) Event(eventType, reason, message string)
  • func (r *WrappedRecorder[T]) Eventf(eventType, reason, messageFmt string, args ...any)

Constants

const (
	OllamaBaseImage = "ollama/ollama"
)

Functions

func EnsureDeploymentCreated

func EnsureDeploymentCreated(
	ctx context.Context,
	c client.Client,
	namespace string,
	name string,
	image string,
	replicas *int32,
	model *ollamav1.Model,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (*appsv1.Deployment, error)

func EnsureImageStorePVCCreated

func EnsureImageStorePVCCreated(
	ctx context.Context,
	client client.Client,
	namespace string,
	storageClassName *string,
	pvcSource *corev1.PersistentVolumeClaimVolumeSource,
	pvSpec *ollamav1.ModelPersistentVolumeSpec,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (*corev1.PersistentVolumeClaim, error)

func EnsureImageStoreServiceCreated

func EnsureImageStoreServiceCreated(
	ctx context.Context,
	client client.Client,
	namespace string,
	statefulSet *appsv1.StatefulSet,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (*corev1.Service, error)

func EnsureImageStoreStatefulSetCreated

func EnsureImageStoreStatefulSetCreated(
	ctx context.Context,
	client client.Client,
	namespace string,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (*appsv1.StatefulSet, error)

func EnsureServiceCreated

func EnsureServiceCreated(
	ctx context.Context,
	c client.Client,
	namespace string,
	name string,
	deployment *appsv1.Deployment,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (*corev1.Service, error)

func GetImageStorePVByPVC

func GetImageStorePVByPVC(ctx context.Context, c client.Client, pvc *corev1.PersistentVolumeClaim) (*corev1.PersistentVolume, error)

func IsAvailable

func IsAvailable(ctx context.Context, ollamaModelResource ollamav1.Model) bool

func IsDeploymentReady

func IsDeploymentReady(
	ctx context.Context,
	c client.Client,
	namespace string,
	name string,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (bool, error)

func IsImageStoreServiceReady

func IsImageStoreServiceReady(
	ctx context.Context,
	client client.Client,
	namespace string,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (bool, error)

func IsImageStoreStatefulSetReady

func IsImageStoreStatefulSetReady(
	ctx context.Context,
	client client.Client,
	namespace string,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (bool, error)

func IsProgressing

func IsProgressing(ctx context.Context, ollamaModelResource ollamav1.Model) bool

func IsServiceReady

func IsServiceReady(
	ctx context.Context,
	c client.Client,
	namespace string,
	name string,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (bool, error)

func ModelAppName

func ModelAppName(name string) string

func NewOllamaPullerContainer

func NewOllamaPullerContainer(image string, serverLocatedNamespace string) corev1.Container

func NewOllamaServerContainer

func NewOllamaServerContainer(readOnly bool) corev1.Container

func SetAvailable

func SetAvailable(
	ctx context.Context,
	c client.Client,
	ollamaModelResource ollamav1.Model,
) (bool, error)

func SetProgressing

func SetProgressing(
	ctx context.Context,
	c client.Client,
	ollamaModelResource ollamav1.Model,
) (bool, error)

func SetReplicas

func SetReplicas(
	ctx context.Context,
	c client.Client,
	ollamaModelResource ollamav1.Model,
	replicas int32,
	readyReplicas int32,
	availableReplicas int32,
	unavailableReplicas int32,
) (bool, error)

func ShouldSetReplicas

func ShouldSetReplicas(
	ctx context.Context,
	ollamaModelResource ollamav1.Model,
	replicas int32,
	readyReplicas int32,
	availableReplicas int32,
	unavailableReplicas int32,
) bool

func UpdateDeployment

func UpdateDeployment(
	ctx context.Context,
	c client.Client,
	model *ollamav1.Model,
	modelRecorder *WrappedRecorder[*ollamav1.Model],
) (bool, error)

Types

type WrappedRecorder

type WrappedRecorder[T runtime.Object] struct {
	// contains filtered or unexported fields
}

func NewWrappedRecorder

func NewWrappedRecorder[T runtime.Object](recorder record.EventRecorder, object T) *WrappedRecorder[T]

func (*WrappedRecorder[T]) AnnotatedEventf

func (r *WrappedRecorder[T]) AnnotatedEventf(annotations map[string]string, eventType, reason, messageFmt string, args ...any)

AnnotatedEventf is just like eventf, but with annotations attached

func (*WrappedRecorder[T]) Event

func (r *WrappedRecorder[T]) Event(eventType, reason, message string)

func (*WrappedRecorder[T]) Eventf

func (r *WrappedRecorder[T]) Eventf(eventType, reason, messageFmt string, args ...any)

Eventf is just like Event, but with Sprintf for the message field.