1 files changed, 31 insertions, 0 deletions
diff --git a/src/Agent.Planner/TokenBudgetClass.cs b/src/Agent.Planner/TokenBudgetClass.cs
new file mode 100644
index 0000000..35894c7
--- /dev/null
+++ b/src/Agent.Planner/TokenBudgetClass.cs
@@ -0,0 +1,31 @@
+namespace Agent.Planner;
+
+public enum TokenBudgetClass
+{
+    K1 = 1_024,
+    K2 = 2_048,
+    K4 = 4_096,
+    K8 = 8_192,
+    K16 = 16_384,
+    K32 = 32_768,
+    K64 = 65_536,
+    K128 = 131_072
+}
+
+public static class TokenBudgetClassExtensions
+{
+    public static long MaxTokens(this TokenBudgetClass value) => (long)value;
+
+    public static TokenBudgetClass Parse(string value) => value switch
+    {
+        "1K" => TokenBudgetClass.K1,
+        "2K" => TokenBudgetClass.K2,
+        "4K" => TokenBudgetClass.K4,
+        "8K" => TokenBudgetClass.K8,
+        "16K" => TokenBudgetClass.K16,
+        "32K" => TokenBudgetClass.K32,
+        "64K" => TokenBudgetClass.K64,
+        "128K" => TokenBudgetClass.K128,
+        _ => throw new ArgumentOutOfRangeException(nameof(value), value, "Unknown token budget class.")
+    };
+}