Update init.lua

Kong · Jun 7, 2024 · a7f3f78 · a7f3f78
1 parent 81d6814
commit a7f3f78
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/kong/llm/schemas/init.lua b/kong/llm/schemas/init.lua
@@ -49,6 +49,16 @@ local model_options_schema = {
         description = "Defines the max_tokens, if using chat or completion models.",
         required = false,
         default = 256 }},
+    { input_cost = {
+        type = "number",
+        description = "Defines the cost per 1000 tokens in your prompt.",
+        required = false,
+        between = { 0.0, 0.1 }}},
+    { output_cost = {
+        type = "number",
+        description = "Defines the cost per 1000 tokens in the output of the AI.",
+        required = false,
+        between = { 0.0, 0.1 }}},
     { temperature = {
         type = "number",
         description = "Defines the matching temperature, if using chat or completion models.",