3 files changed, 15 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;
diff --git a/extension/schema/LLM.d.ts b/extension/schema/LLM.d.ts
index 2c1ced29..fbd5aa41 100644
--- a/extension/schema/LLM.d.ts
+++ b/extension/schema/LLM.d.ts
@@ -27,6 +27,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[];
@@ -57,6 +61,7 @@ export interface LLM1 {
   context_length?: ContextLength;
   unique_id?: UniqueId;
   model: Model;
+  max_tokens?: MaxTokens;
   stop_tokens?: StopTokens;
   timeout?: Timeout;
   verify_ssl?: VerifySsl;
diff --git a/extension/schema/Models.d.ts b/extension/schema/Models.d.ts
index 67d73cfc..9ce79b60 100644
--- a/extension/schema/Models.d.ts
+++ b/extension/schema/Models.d.ts
@@ -27,6 +27,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[];
@@ -70,6 +74,7 @@ export interface LLM {
   context_length?: ContextLength;
   unique_id?: UniqueId;
   model: Model;
+  max_tokens?: MaxTokens;
   stop_tokens?: StopTokens;
   timeout?: Timeout;
   verify_ssl?: VerifySsl;