update schema

1 files changed, 42 insertions, 15 deletions

diff --git a/schema/json/Models.json b/schema/json/Models.json
index 19044d93..361ae9eb 100644
--- a/schema/json/Models.json
+++ b/schema/json/Models.json
@@ -11,17 +11,6 @@
           "description": "A title that will identify this model in the model selection dropdown",
           "type": "string"
         },
-        "system_message": {
-          "title": "System Message",
-          "description": "A system message that will always be followed by the LLM",
-          "type": "string"
-        },
-        "context_length": {
-          "title": "Context Length",
-          "description": "The maximum context length of the LLM in tokens, as counted by count_tokens.",
-          "default": 2048,
-          "type": "integer"
-        },
         "unique_id": {
           "title": "Unique Id",
           "description": "The unique ID of the user.",
@@ -32,10 +21,15 @@
           "description": "The name of the model to be used (e.g. gpt-4, codellama)",
           "type": "string"
         },
-        "max_tokens": {
-          "title": "Max Tokens",
-          "description": "The maximum number of tokens to generate.",
-          "default": 1024,
+        "system_message": {
+          "title": "System Message",
+          "description": "A system message that will always be followed by the LLM",
+          "type": "string"
+        },
+        "context_length": {
+          "title": "Context Length",
+          "description": "The maximum context length of the LLM in tokens, as counted by count_tokens.",
+          "default": 2048,
           "type": "integer"
         },
         "stop_tokens": {
@@ -46,6 +40,31 @@
             "type": "string"
           }
         },
+        "temperature": {
+          "title": "Temperature",
+          "description": "The temperature of the completion.",
+          "type": "number"
+        },
+        "top_p": {
+          "title": "Top P",
+          "description": "The top_p of the completion.",
+          "type": "number"
+        },
+        "top_k": {
+          "title": "Top K",
+          "description": "The top_k of the completion.",
+          "type": "integer"
+        },
+        "presence_penalty": {
+          "title": "Presence Penalty",
+          "description": "The presence penalty Aof the completion.",
+          "type": "number"
+        },
+        "frequency_penalty": {
+          "title": "Frequency Penalty",
+          "description": "The frequency penalty of the completion.",
+          "type": "number"
+        },
         "timeout": {
           "title": "Timeout",
           "description": "Set the timeout for each request to the LLM. If you are running a local LLM that takes a while to respond, you might want to set this to avoid timeouts.",
@@ -67,6 +86,14 @@
           "description": "Proxy URL to use when making the HTTP request",
           "type": "string"
         },
+        "headers": {
+          "title": "Headers",
+          "description": "Headers to use when making the HTTP request",
+          "type": "object",
+          "additionalProperties": {
+            "type": "string"
+          }
+        },
         "prompt_templates": {
           "title": "Prompt Templates",
           "description": "A dictionary of prompt templates that can be used to customize the behavior of the LLM in certain situations. For example, set the \"edit\" key in order to change the prompt that is used for the /edit slash command. Each value in the dictionary is a string templated in mustache syntax, and filled in at runtime with the variables specific to the situation. See the documentation for more information.",