feat: add support for configurable prompt mapping

Signed-off-by: hadi2f244 <[email protected]>

Author: hadi2f244

pkg/ai/iai.go

@@ -100,8 +100,9 @@ func NewClient(provider string) IAI {
 }
 
 type AIConfiguration struct {
-	Providers       []AIProvider `mapstructure:"providers"`
-	DefaultProvider string       `mapstructure:"defaultprovider"`
+	Providers       []AIProvider			 `mapstructure:"providers"`
+	DefaultProvider string       			 `mapstructure:"defaultprovider"`
+	PromptMap       map[string]string  `mapstructure:"promptmap"`
 }
 
 type AIProvider struct {

pkg/analysis/analysis.go

@@ -41,6 +41,7 @@ type Analysis struct {
 	Client             *kubernetes.Client
 	Language           string
 	AIClient           ai.IAI
+	PromptMap          map[string]string
 	Results            []common.Result
 	Errors             []string
 	Namespace          string
@@ -159,6 +160,16 @@ func NewAnalysis(
 	if err := aiClient.Configure(&aiProvider); err != nil {
 		return nil, err
 	}
+
+	// Override values from configAI.PromptMap
+	if len(configAI.PromptMap) > 0 {
+		for key, value := range configAI.PromptMap {
+			if key != "raw" {
+				ai.PromptMap[key] = value
+			}
+		}
+	}
+
 	a.AIClient = aiClient
 	a.AnalysisAIProvider = aiProvider.Name
 	return a, nil