model

package
v0.10.1 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Nov 11, 2024 License: Apache-2.0 Imports: 15 Imported by: 0

Documentation

Index

Constants

View Source 
const (
	ImageStorePVCName         = "ollama-models-store-pvc"
	ImageStoreStatefulSetName = "ollama-models-store"
)
View Source 
const (
	OllamaBaseImage = "ollama/ollama"
)

Variables

This section is empty.

Functions

func AppendIfNotFound added in v0.10.0 

func AppendIfNotFound[T any](source []T, predicate func(item T) bool, newFn func() T) []T

func AssignOllamaPullerContainer added in v0.10.0 

func AssignOllamaPullerContainer(name string, image string, serverLocatedNamespace string, resources corev1.ResourceRequirements, extraEnvFrom []corev1.EnvFromSource, extraEnv []corev1.EnvVar) func(container corev1.Container, _ int) corev1.Container

func AssignOllamaServerContainer added in v0.10.0 

func AssignOllamaServerContainer(readOnly bool, resources corev1.ResourceRequirements, extraEnvFrom []corev1.EnvFromSource, extraEnv []corev1.EnvVar) func(container corev1.Container, _ int) corev1.Container

func AssignOrAppend added in v0.10.0 

func AssignOrAppend[T any](source []T, predicate func(item T) bool, modifier func(item T, index int) T, newFn func() T) []T

func ClientFromContext added in v0.10.0 

func ClientFromContext(ctx context.Context, key ...baseClientContextKey) client.Client

func EnsureDeploymentCreated 

func EnsureDeploymentCreated(
	ctx context.Context,
	namespace string,
	name string,
	image string,
	replicas *int32,
	model *ollamav1.Model,
) (*appsv1.Deployment, error)

func EnsureImageStorePVCCreated 

func EnsureImageStorePVCCreated(
	ctx context.Context,
	namespace string,
	storageClassName *string,
	pvcSource *corev1.PersistentVolumeClaimVolumeSource,
	pvSpec *ollamav1.ModelPersistentVolumeSpec,
) (*corev1.PersistentVolumeClaim, error)

func EnsureImageStoreServiceCreated 

func EnsureImageStoreServiceCreated(
	ctx context.Context,
	namespace string,
	statefulSet *appsv1.StatefulSet,
) (*corev1.Service, error)

func EnsureImageStoreStatefulSetCreated 

func EnsureImageStoreStatefulSetCreated(
	ctx context.Context,
	namespace string,
	model *ollamav1.Model,
) (*appsv1.StatefulSet, error)

func EnsureServiceCreated 

func EnsureServiceCreated(
	ctx context.Context,
	namespace string,
	name string,
	deployment *appsv1.Deployment,
) (*corev1.Service, error)

func FindOllamaPullerContainer added in v0.10.0 

func FindOllamaPullerContainer(container corev1.Container) bool

func FindOllamaServerContainer added in v0.10.0 

func FindOllamaServerContainer(container corev1.Container) bool

func GetImageStorePVByPVC 

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

func ImageStoreLabels added in v0.5.2 

func ImageStoreLabels() map[string]string

func IsAvailable added in v0.2.0 

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

func IsDeploymentReady 

func IsDeploymentReady(
	ctx context.Context,
	namespace string,
	name string,
) (bool, error)

func IsImageStoreServiceReady 

func IsImageStoreServiceReady(
	ctx context.Context,
	namespace string,
) (bool, error)

func IsImageStoreStatefulSetReady 

func IsImageStoreStatefulSetReady(
	ctx context.Context,
	namespace string,
) (bool, error)

func IsProgressing added in v0.2.0 

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

func IsServiceReady 

func IsServiceReady(
	ctx context.Context,
	namespace string,
	name string,
) (bool, error)

func MergePodTemplate added in v0.10.0 

func MergePodTemplate(
	ctx context.Context,
	namespace string,
	name string,
	image string,
	replicas *int32,
	model *ollamav1.Model,
) corev1.PodTemplateSpec

func ModelAnnotations added in v0.5.0 

func ModelAnnotations(name string, imageStore bool) map[string]string

func ModelAppName 

func ModelAppName(name string) string

func ModelLabels added in v0.5.0 

func ModelLabels(name string) map[string]string

func NewClientContextKey added in v0.10.0 

func NewClientContextKey(key string) baseClientContextKey

func NewOllamaPullerContainer 

func NewOllamaPullerContainer(name string, image string, serverLocatedNamespace string, resources corev1.ResourceRequirements, extraEnvFrom []corev1.EnvFromSource, extraEnv []corev1.EnvVar) corev1.Container

func NewOllamaServerContainer 

func NewOllamaServerContainer(readOnly bool, resources corev1.ResourceRequirements, extraEnvFrom []corev1.EnvFromSource, extraEnv []corev1.EnvVar) corev1.Container

func NewServiceForModel added in v0.4.0 

func NewServiceForModel(namespace, name string, deployment *appsv1.Deployment, serviceType corev1.ServiceType) *corev1.Service

func NewWrappedRecorderContextKey added in v0.10.0 

func NewWrappedRecorderContextKey(key string) baseWrapperRecorderContextKey

func SetAvailable added in v0.2.0 

func SetAvailable(
	ctx context.Context,
	ollamaModelResource *ollamav1.Model,
) (bool, error)

func SetProgressing added in v0.2.0 

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

func SetReplicas added in v0.2.0 

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

func ShouldSetReplicas added in v0.2.0 

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

func UniqEnvVar added in v0.10.0 

func UniqEnvVar(env []corev1.EnvVar) []corev1.EnvVar

func UpdateDeployment 

func UpdateDeployment(
	ctx context.Context,
	model *ollamav1.Model,
) (bool, error)

func WithClient added in v0.10.0 

func WithClient(ctx context.Context, client client.Client, key ...baseClientContextKey) context.Context

func WithWrappedRecorder added in v0.10.0 

func WithWrappedRecorder[T runtime.Object](ctx context.Context, recorder *WrappedRecorder[T], key ...baseWrapperRecorderContextKey) context.Context

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 WrappedRecorderFromContext added in v0.10.0 

func WrappedRecorderFromContext[T runtime.Object](ctx context.Context, key ...baseWrapperRecorderContextKey) *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.

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.