From 818202c6b14d73df130d2a17334d0e5d5024abd5 Mon Sep 17 00:00:00 2001
From: Mike Donnalley <mdonnalley@salesforce.com>
Date: Thu, 12 Dec 2024 12:03:58 -0700
Subject: [PATCH] chore: mark commands as beta

---
 src/commands/agent/generate/definition.ts | 1 +
 src/commands/agent/generate/testset.ts    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/commands/agent/generate/definition.ts b/src/commands/agent/generate/definition.ts
index d70e6ff..dfdd5a4 100644
--- a/src/commands/agent/generate/definition.ts
+++ b/src/commands/agent/generate/definition.ts
@@ -19,6 +19,7 @@ export default class AgentGenerateDefinition extends SfCommand<void> {
   public static readonly description = messages.getMessage('description');
   public static readonly examples = messages.getMessages('examples');
   public static readonly enableJsonFlag = false;
+  public static readonly state = 'beta';
 
   public async run(): Promise<void> {
     const testSetDir = join('force-app', 'main', 'default', 'aiEvaluationTestSets');
diff --git a/src/commands/agent/generate/testset.ts b/src/commands/agent/generate/testset.ts
index faa86ef..74451a2 100644
--- a/src/commands/agent/generate/testset.ts
+++ b/src/commands/agent/generate/testset.ts
@@ -96,6 +96,7 @@ export default class AgentGenerateTestset extends SfCommand<void> {
   public static readonly description = messages.getMessage('description');
   public static readonly examples = messages.getMessages('examples');
   public static readonly enableJsonFlag = false;
+  public static readonly state = 'beta';
 
   public async run(): Promise<void> {
     const testSetName = await input({