update params as per docuentation

Portkey-AI · Mar 29, 2024 · f5837ef · f5837ef
1 parent 6a0634d
commit f5837ef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/providers/bedrock/chatComplete.ts b/src/providers/bedrock/chatComplete.ts
@@ -306,13 +306,13 @@ export const BedrockMistralChatCompleteConfig: ProviderConfig = {
     max: 5,
   },
   top_p: {
-    param: 'p',
+    param: 'top_p',
     default: 0.75,
     min: 0,
     max: 1,
   },
   top_k: {
-    param: 'k',
+    param: 'top_k',
     default: 0,
     max: 200,
   },

diff --git a/src/providers/bedrock/complete.ts b/src/providers/bedrock/complete.ts
@@ -142,15 +142,14 @@ export const BedrockMistralCompleteConfig: ProviderConfig = {
     max: 5,
   },
   top_p: {
-    param: 'p',
+    param: 'top_p',
     default: 0.75,
     min: 0,
     max: 1,
   },
   top_k: {
-    param: 'k',
+    param: 'top_k',
     default: 0,
-    max: 200,
   },
   stop: {
     param: 'end_sequences',