Merge pull request #1 from arbreezy/chore/add-backends

chore: add missing backends
arbreezy · Dec 1, 2023 · d7a315d · d7a315d
2 parents 15229c2 + 766efce
commit d7a315d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/api/v1alpha1/k8sgpt_types.go b/api/v1alpha1/k8sgpt_types.go
@@ -71,7 +71,7 @@ type WebhookRef struct {
 
 type AISpec struct {
 	// +kubebuilder:default:=openai
-	// +kubebuilder:validation:Enum=openai;localai;azureopenai
+	// +kubebuilder:validation:Enum=openai;localai;azureopenai;amazonbedrock;cohere;amazonsagemaker
 	Backend string `json:"backend"`
 	BaseUrl string `json:"baseUrl,omitempty"`
 	// +kubebuilder:default:=gpt-3.5-turbo
@@ -109,9 +109,12 @@ type K8sGPTSpec struct {
 }
 
 const (
-	OpenAI      = "openai"
-	AzureOpenAI = "azureopenai"
-	LocalAI     = "localai"
+	OpenAI          = "openai"
+	AzureOpenAI     = "azureopenai"
+	LocalAI         = "localai"
+	AmazonBedrock   = "amazonbedrock"
+	AmazonSageMaker = "AmazonSageMaker"
+	Cohere          = "cohere"
 )
 
 // K8sGPTStatus defines the observed state of K8sGPT

diff --git a/chart/operator/templates/k8sgpt-crd.yaml b/chart/operator/templates/k8sgpt-crd.yaml
@@ -46,6 +46,9 @@ spec:
                     - openai
                     - localai
                     - azureopenai
+                    - amazonbedrock
+                    - cohere
+                    - amazonsagemaker
                     type: string
                   baseUrl:
                     type: string

diff --git a/config/crd/bases/core.k8sgpt.ai_k8sgpts.yaml b/config/crd/bases/core.k8sgpt.ai_k8sgpts.yaml
@@ -46,6 +46,9 @@ spec:
                     - openai
                     - localai
                     - azureopenai
+                    - amazonbedrock
+                    - cohere
+                    - amazonsagemaker
                     type: string
                   baseUrl:
                     type: string