add bedrock format test scaffolding

Kong · Jul 9, 2024 · ad1c67d · ad1c67d
1 parent da42ff6
commit ad1c67d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/spec/03-plugins/38-ai-proxy/01-unit_spec.lua b/spec/03-plugins/38-ai-proxy/01-unit_spec.lua
@@ -237,6 +237,20 @@ local FORMATS = {
       },
     },
   },
+  bedrock = {
+    ["llm/v1/chat"] = {
+      config = {
+        name = "bedrock",
+        provider = "bedrock",
+        options = {
+          max_tokens = 8192,
+          temperature = 0.8,
+          top_k = 1,
+          top_p = 0.6,
+        },
+      },
+    },
+  },
 }
 
 local STREAMS = {