fix: add providerId, maxTokens and topk to ai spec (#545)

Signed-off-by: samir-tahir <[email protected]>
k8sgpt-ai · Dec 2, 2024 · 771acbe · 771acbe
1 parent eaf6456
commit 771acbe
Show file tree

Hide file tree

Showing 4 changed files with 40 additions and 0 deletions.
diff --git a/api/v1alpha1/k8sgpt_types.go b/api/v1alpha1/k8sgpt_types.go
@@ -107,6 +107,11 @@ type AISpec struct {
 	// +kubebuilder:default:=english
 	Language      string `json:"language,omitempty"`
 	ProxyEndpoint string `json:"proxyEndpoint,omitempty"`
+	ProviderId    string `json:"providerId,omitempty"`
+	// +kubebuilder:default:="2048"
+	MaxTokens string `json:"maxTokens,omitempty"`
+	// +kubebuilder:default:="50"
+	Topk string `json:"topk,omitempty"`
 }
 
 type Trivy struct {

diff --git a/chart/operator/templates/k8sgpt-crd.yaml b/chart/operator/templates/k8sgpt-crd.yaml
@@ -75,9 +75,14 @@ spec:
                   language:
                     default: english
                     type: string
+                  maxTokens:
+                    default: "2048"
+                    type: string
                   model:
                     default: gpt-3.5-turbo
                     type: string
+                  providerId:
+                    type: string
                   proxyEndpoint:
                     type: string
                   region:
@@ -89,6 +94,9 @@ spec:
                       name:
                         type: string
                     type: object
+                  topk:
+                    default: "50"
+                    type: string
                 required:
                 - backend
                 type: object

diff --git a/config/crd/bases/core.k8sgpt.ai_k8sgpts.yaml b/config/crd/bases/core.k8sgpt.ai_k8sgpts.yaml
@@ -78,9 +78,14 @@ spec:
                   language:
                     default: english
                     type: string
+                  maxTokens:
+                    default: "2048"
+                    type: string
                   model:
                     default: gpt-3.5-turbo
                     type: string
+                  providerId:
+                    type: string
                   proxyEndpoint:
                     type: string
                   region:
@@ -92,6 +97,9 @@ spec:
                       name:
                         type: string
                     type: object
+                  topk:
+                    default: "50"
+                    type: string
                 required:
                 - backend
                 type: object

diff --git a/pkg/resources/k8sgpt.go b/pkg/resources/k8sgpt.go
@@ -153,6 +153,14 @@ func GetDeployment(config v1alpha1.K8sGPT, outOfClusterMode bool, c client.Clien
 									Name:  "K8SGPT_BACKEND",
 									Value: config.Spec.AI.Backend,
 								},
+								{
+									Name:  "K8SGPT_MAX_TOKENS",
+									Value: config.Spec.AI.MaxTokens,
+								},
+								{
+									Name:  "K8SGPT_TOP_K",
+									Value: config.Spec.AI.Topk,
+								},
 								{
 									Name:  "XDG_CONFIG_HOME",
 									Value: "/k8sgpt-data/.config",
@@ -271,6 +279,17 @@ func GetDeployment(config v1alpha1.K8sGPT, outOfClusterMode bool, c client.Clien
 		}
 	}
 
+	// Add provider ID if in ai spec
+	if config.Spec.AI.ProviderId != "" {
+		providerId := corev1.EnvVar{
+			Name:  "K8SGPT_PROVIDER_ID",
+			Value: config.Spec.AI.ProviderId,
+		}
+		deployment.Spec.Template.Spec.Containers[0].Env = append(
+			deployment.Spec.Template.Spec.Containers[0].Env, providerId,
+		)
+	}
+
 	if config.Spec.AI.BaseUrl != "" {
 		baseUrl := corev1.EnvVar{
 			Name:  "K8SGPT_BASEURL",