From 7af3fe68e4591650858aa9a0a6a98f8f7f7214ff Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 13 Jan 2024 21:12:59 -0800 Subject: [PATCH] (docs) provisioned throughput bedrock models --- docs/my-website/docs/providers/bedrock.md | 26 +++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/docs/my-website/docs/providers/bedrock.md b/docs/my-website/docs/providers/bedrock.md index ae180ff3595..f83e3ed02c6 100644 --- a/docs/my-website/docs/providers/bedrock.md +++ b/docs/my-website/docs/providers/bedrock.md @@ -208,6 +208,32 @@ response = completion( ) ``` +## Provisioned throughput models +To use provisioned throughput Bedrock models pass +- `model=bedrock/`, example `model=bedrock/anthropic.claude-v2`. Set `model` to any of the [Supported AWS models](#supported-aws-bedrock-models) +- `model_id=provisioned-model-arn` + +Completion +```python +import litellm +response = litellm.completion( + model="bedrock/anthropic.claude-instant-v1", + model_id="provisioned-model-arn", + messages=[{"content": "Hello, how are you?", "role": "user"}] +) +``` + +Embedding +```python +import litellm +response = litellm.embedding( + model="bedrock/amazon.titan-embed-text-v1", + model_id="provisioned-model-arn", + input=["hi"], +) +``` + + ## Supported AWS Bedrock Models Here's an example of using a bedrock model with LiteLLM