Deployment Options

# templates/deployment.yaml
apiVersion: apps/v1
kind: Deployment
metadata:
  name: cognee
spec:
  replicas: {{ .Values.replicas }}
  strategy:
    type: {{ .Values.strategy }}
  selector:
    matchLabels:
      app: cognee
  template:
    metadata:
      labels:
        app: cognee
    spec:
      initContainers:
        - name: wait-for-postgres
          image: pgvector/pgvector:pg17
          command:
            - sh
            - -c
            - until pg_isready -h {{ .Values.externalPostgres.host }}; do sleep 2; done
      containers:
        - name: cognee
          image: "{{ .Values.image.repository }}:{{ .Values.image.tag }}"
          ports:
            - containerPort: 8000
          env:
            - name: DB_PROVIDER
              value: "{{ .Values.env.DB_PROVIDER }}"
            - name: DB_HOST
              value: "{{ .Values.externalPostgres.host }}"
            - name: VECTOR_DB_PROVIDER
              value: "{{ .Values.env.VECTOR_DB_PROVIDER }}"
            - name: GRAPH_DATABASE_PROVIDER
              value: "{{ .Values.env.GRAPH_DATABASE_PROVIDER }}"
            - name: LLM_API_KEY
              valueFrom:
                secretKeyRef:
                  name: cognee-secrets
                  key: llmApiKey
          volumeMounts:
            - name: data
              mountPath: /data
          readinessProbe:
            httpGet:
              path: /health
              port: 8000
          resources:
            {{- toYaml .Values.resources | nindent 12 }}
      volumes:
        - name: data
          persistentVolumeClaim:
            claimName: cognee-data
---
# templates/networkpolicy.yaml
apiVersion: networking.k8s.io/v1
kind: NetworkPolicy
metadata:
  name: cognee
spec:
  podSelector:
    matchLabels:
      app: cognee
  policyTypes:
    - Egress
  egress:
    - to:
        - podSelector:
            matchLabels:
              app: postgres
      ports:
        - port: 5432
    - to:
        - ipBlock:
            cidr: 0.0.0.0/0
      ports:
        - port: 443
---
# templates/pdb.yaml
apiVersion: policy/v1
kind: PodDisruptionBudget
metadata:
  name: cognee
spec:
  maxUnavailable: 0
  selector:
    matchLabels:
      app: cognee

apiVersion: v1
kind: Pod
metadata:
  name: agent-with-cognee
spec:
  containers:
    - name: agent
      image: my-agent:1.2.0
      env:
        - name: COGNEE_URL
          value: "http://localhost:8000"
    - name: cognee
      image: cognee/cognee:1.0.6
      ports:
        - containerPort: 8000
      env:
        - name: DATA_ROOT_DIRECTORY
          value: /data/.cognee_data
        - name: SYSTEM_ROOT_DIRECTORY
          value: /data/.cognee_system
      volumeMounts:
        - name: memory
          mountPath: /data
  volumes:
    - name: memory
      persistentVolumeClaim:
        claimName: agent-memory

resource "aws_sqs_queue" "cognee_dlq" {
  name = "cognee-ingest-dlq"
}

resource "aws_sqs_queue" "cognee_ingest" {
  name                       = "cognee-ingest"
  visibility_timeout_seconds = 900
  redrive_policy = jsonencode({
    deadLetterTargetArn = aws_sqs_queue.cognee_dlq.arn
    maxReceiveCount     = 3
  })
}

output "ingest_queue_url" {
  value = aws_sqs_queue.cognee_ingest.url
}

apiVersion: apps/v1
kind: Deployment
metadata:
  name: cognee-writer
spec:
  replicas: 1
  selector:
    matchLabels:
      app: cognee-writer
  template:
    metadata:
      labels:
        app: cognee-writer
    spec:
      containers:
        - name: worker
          image: cognee/cognee:1.0.6
          env:
            - name: INGEST_QUEUE_URL
              valueFrom:
                secretKeyRef:
                  name: cognee-queue
                  key: ingestQueueUrl

resource "aws_efs_file_system" "cognee" {
  encrypted = true
}

resource "aws_efs_access_point" "cognee" {
  file_system_id = aws_efs_file_system.cognee.id

  posix_user {
    gid = 1000
    uid = 1000
  }

  root_directory {
    path = "/cognee"
    creation_info {
      owner_gid   = 1000
      owner_uid   = 1000
      permissions = "0755"
    }
  }
}

resource "aws_lambda_function" "cognee_efs" {
  function_name = "cognee-efs"
  package_type  = "Image"
  image_uri     = "${var.ecr_repo}:1.0.6"
  memory_size   = 3008
  timeout       = 120

  vpc_config {
    subnet_ids         = var.private_subnets
    security_group_ids = [var.lambda_sg]
  }

  file_system_config {
    arn              = aws_efs_access_point.cognee.arn
    local_mount_path = "/mnt/cognee"
  }

  environment {
    variables = {
      DATA_ROOT_DIRECTORY   = "/mnt/cognee/.cognee_data"
      SYSTEM_ROOT_DIRECTORY = "/mnt/cognee/.cognee_system"
    }
  }
}