add param mapping to claude bedrock

This commit is contained in:
ishaan-jaff 2023-09-29 15:36:57 -07:00
parent acc15f5778
commit 6a81f860f9
2 changed files with 16 additions and 3 deletions

View file

@ -114,7 +114,6 @@ def completion(
if provider == "anthropic":
data = json.dumps({
"prompt": prompt,
"max_tokens_to_sample": 256,
**optional_params
})
elif provider == "ai21":