badalsahani's picture
feat: chroma initial deploy
287a0bc
variable "project_id" {
type = string
description = "The project id to deploy to"
}
variable "chroma_release" {
description = "The chroma release to deploy"
type = string
default = "0.4.9"
}
variable "zone" {
type = string
default = "us-central1-a"
}
variable "image" {
default = "debian-cloud/debian-11"
description = "The image to use for the instance"
type = string
}
variable "vm_user" {
default = "debian"
description = "The user to use for connecting to the instance. This is usually the default image user"
type = string
}
variable "machine_type" {
type = string
default = "e2-small"
}
variable "public_access" {
description = "Enable public ingress on port 8000"
type = bool
default = true // or true depending on your needs
}
variable "enable_auth" {
description = "Enable authentication"
type = bool
default = true // or false depending on your needs
}
variable "auth_type" {
description = "Authentication type"
type = string
default = "token" // or token depending on your needs
validation {
condition = contains(["basic", "token"], var.auth_type)
error_message = "The auth type must be either basic or token"
}
}
resource "random_password" "chroma_password" {
length = 16
special = true
lower = true
upper = true
}
resource "random_password" "chroma_token" {
length = 32
special = false
lower = true
upper = true
}
locals {
basic_auth_credentials = {
username = "chroma"
password = random_password.chroma_password.result
}
token_auth_credentials = {
token = random_password.chroma_token.result
}
tags = [
"chroma",
"release-${replace(var.chroma_release, ".", "")}",
]
}
variable "ssh_public_key" {
description = "SSH Public Key"
type = string
default = "./chroma-aws.pub"
}
variable "ssh_private_key" {
description = "SSH Private Key"
type = string
default = "./chroma-aws"
}
variable "chroma_instance_volume_size" {
description = "The size of the instance volume - the root volume"
type = number
default = 30
}
variable "chroma_data_volume_size" {
description = "Volume Size of the attached data volume where your chroma data is stored"
type = number
default = 20
}
variable "chroma_data_volume_device_name" {
default = "chroma-disk-0"
description = "The device name of the chroma data volume"
type = string
}
variable "prevent_chroma_data_volume_delete" {
description = "Prevent the chroma data volume from being deleted when the instance is terminated"
type = bool
default = false
}
variable "disk_type" {
default = "pd-ssd"
description = "The type of disk to use for the instance. Can be either pd-standard or pd-ssd"
}
variable "labels" {
default = {
environment = "dev"
}
description = "Labels to apply to all resources in this example"
type = map(string)
}
variable "chroma_port" {
default = "8000"
description = "The port that chroma listens on"
type = string
}
variable "source_ranges" {
default = ["0.0.0.0/0"]
type = list(string)
description = "List of CIDR ranges to allow through the firewall"
}