From a7f3f7804779ff4434742284978551e23343dae3 Mon Sep 17 00:00:00 2001
From: Antoine Jacquemin <ajacquem@gmail.com>
Date: Fri, 7 Jun 2024 11:11:12 +0200
Subject: [PATCH] Update init.lua

---
 kong/llm/schemas/init.lua | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/kong/llm/schemas/init.lua b/kong/llm/schemas/init.lua
index 37b5aaf34761..d1f0e590360d 100644
--- 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.",