1 files changed, 5 insertions, 0 deletions
diff --git a/extension/schema/ContinueConfig.d.ts b/extension/schema/ContinueConfig.d.ts
index 64aa5c02..b9eb92ff 100644
--- a/extension/schema/ContinueConfig.d.ts
+++ b/extension/schema/ContinueConfig.d.ts
@@ -56,6 +56,10 @@ export type UniqueId = string;
  */
 export type Model = string;
 /**
+ * The maximum number of tokens to generate.
+ */
+export type MaxTokens = number;
+/**
  * Tokens that will stop the completion.
  */
 export type StopTokens = string[];
@@ -222,6 +226,7 @@ export interface LLM {
   context_length?: ContextLength;
   unique_id?: UniqueId;
   model: Model;
+  max_tokens?: MaxTokens;
   stop_tokens?: StopTokens;
   timeout?: Timeout;
   verify_ssl?: VerifySsl;