NeMo-Retriever/helm/templates/nemoretriever-table-structure-v1.yaml at df750a03a34f5c4f6ba24777cd8860284e1fe5db · NVIDIA/NeMo-Retriever · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
{{ if and (.Capabilities.APIVersions.Has "apps.nvidia.com/v1alpha1") (eq .Values.nimOperator.table_structure.enabled true) -}}
apiVersion: apps.nvidia.com/v1alpha1
kind: NIMCache
metadata:
  name: nemoretriever-table-structure-v1
  annotations:
    helm.sh/resource-policy: keep
spec:
  source:
    ngc:
      modelPuller: "{{ include "nv-ingest.release.nimRef" (dict "root" . "key" "table_structure") }}"
      pullSecret: "{{ index .Values.nimOperator.table_structure.image.pullSecrets 0 }}"
      authSecret: {{ .Values.nimOperator.table_structure.authSecret }}
  storage:
    pvc:
      create: {{ .Values.nimOperator.table_structure.storage.pvc.create }}
      storageClass: {{ .Values.nimOperator.table_structure.storage.pvc.storageClass }}
      size: {{ .Values.nimOperator.table_structure.storage.pvc.size }}
      volumeAccessMode: {{ .Values.nimOperator.table_structure.storage.pvc.volumeAccessMode }}
---
apiVersion: apps.nvidia.com/v1alpha1
kind: NIMService
metadata:
  name: nemoretriever-table-structure-v1
spec:
  image:
    repository: {{ include "nv-ingest.release.nimRepository" (dict "root" . "key" "table_structure") }}
    tag: {{ include "nv-ingest.release.nimTag" (dict "root" . "key" "table_structure") }}
    pullPolicy: {{ .Values.nimOperator.table_structure.image.pullPolicy }}
    pullSecrets:
{{ toYaml .Values.nimOperator.table_structure.image.pullSecrets | nindent 6 }}
  authSecret: {{ .Values.nimOperator.table_structure.authSecret }}
  storage:
    nimCache:
      name: nemoretriever-table-structure-v1
  replicas: {{ .Values.nimOperator.table_structure.replicas }}
  nodeSelector:
{{ toYaml .Values.nimOperator.table_structure.nodeSelector | nindent 4 }}
  resources:
{{ toYaml .Values.nimOperator.table_structure.resources | nindent 4 }}
  tolerations:
{{ toYaml .Values.nimOperator.table_structure.tolerations | nindent 4 }}
  expose:
{{ toYaml .Values.nimOperator.table_structure.expose | nindent 4 }}
  env:
{{ toYaml .Values.nimOperator.table_structure.env | nindent 4 }}
{{- end }}