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.",