feat: AI action draft mode

api/v1/playbook_actions.go

@@ -374,6 +374,9 @@ type AIAction struct {
 	AIActionClient  `json:",inline" yaml:",inline"`
 	AIActionContext `json:",inline" yaml:",inline" template:"true"`
 
+	// When enabled, the prompt is simply saved without passing it on to the LLM.
+	DryRun bool `json:"dryRun,omitempty"`
+
 	// Use an AI agent that can autonomously drive the diagnosis using tools that interface directly with the database.
 	// NOTE: Not exposed for now
 	UseAgent bool `json:"-"`

config/crds/mission-control.flanksource.com_playbooks.yaml

@@ -124,6 +124,10 @@ spec:
                         connection:
                           description: Connection to setup the llm backend connection
                           type: string
+                        dryRun:
+                          description: When enabled, the prompt is simply saved without
+                            passing it on to the LLM.
+                          type: boolean
                         formats:
                           description: |-
                             Output format of the prompt.

config/schemas/playbook-spec.schema.json

@@ -35,6 +35,9 @@
           },
           "type": "array"
         },
+        "dryRun": {
+          "type": "boolean"
+        },
         "systemPrompt": {
           "type": "string"
         },

config/schemas/playbook.schema.json

@@ -35,6 +35,9 @@
           },
           "type": "array"
         },
+        "dryRun": {
+          "type": "boolean"
+        },
         "systemPrompt": {
           "type": "string"
         },

playbook/actions/ai.go

@@ -3,6 +3,7 @@ package actions
 import (
 	"encoding/json"
 	"fmt"
+	"strings"
 	"time"
 
 	"github.com/flanksource/commons/duration"
@@ -39,6 +40,10 @@ func (t *AIAction) Run(ctx context.Context, spec v1.AIAction) (*AIActionResult,
 		return nil, fmt.Errorf("faield to populate llm client connection: %w", err)
 	}
 
+	if spec.DryRun {
+		return &AIActionResult{Markdown: strings.Join(prompt, "\n")}, nil
+	}
+
 	llmConf := llm.Config{AIActionClient: spec.AIActionClient, UseAgent: spec.UseAgent}
 	response, err := llm.Prompt(ctx, llmConf, spec.SystemPrompt, prompt...)
 	if err != nil {
@@ -54,7 +59,7 @@ func buildPrompt(ctx context.Context, prompt string, spec v1.AIActionContext) ([
 		return nil, fmt.Errorf("failed to get prompt context: %w", err)
 	}
 
-	knowledgeJSON, err := json.Marshal(knowledge)
+	knowledgeJSON, err := json.MarshalIndent(knowledge, "", "\t")
 	if err != nil {
 		return nil, fmt.Errorf("failed to get prompt context: %w", err)
 	}