This documentation page is also available as an interactive notebook. You can launch the notebook in
Kaggle or Colab, or download it for use with an IDE or local Jupyter installation, by clicking one of the
above links.
Prerequisites
- Activate Bedrock in your AWS account.
- Request access to your desired models (e.g. Claude Sonnet 3.7,
Amazon Nova Pro).
- Obtain a Bedrock API Key from the AWS console (under Bedrock >
API keys), or configure standard AWS IAM credentials.
Important notes
- Bedrock usage may incur costs based on your Bedrock plan.
- Be mindful of sensitive data and consider security measures when
integrating with external services.
First you’ll need to install required libraries and configure your
Bedrock credentials.
%pip install -qU pixeltable boto3
import getpass
import os
if 'BEDROCK_API_KEY' not in os.environ:
os.environ['BEDROCK_API_KEY'] = getpass.getpass(
'Enter your Bedrock API Key: '
)
# Optional: set the region if your Bedrock endpoint is not in us-east-1
# os.environ['BEDROCK_REGION_NAME'] = 'us-west-2'
import pixeltable as pxt
# Remove the `bedrock_demo` directory and its contents, if it exists
pxt.drop_dir('bedrock_demo', force=True)
pxt.create_dir('bedrock_demo')
Created directory ‘bedrock_demo’.
<pixeltable.catalog.dir.Dir at 0x1512adb10>
Messages
Create a Table: In Pixeltable, create a table with columns to represent
your input data and the columns where you want to store the results from
Bedrock.
from pixeltable.functions import bedrock
# Create a table in Pixeltable and add a computed column that calls Bedrock
t = pxt.create_table('bedrock_demo/chat', {'input': pxt.String})
t.add_computed_column(
output=bedrock.converse(
model_id='amazon.nova-pro-v1:0',
messages=[{'role': 'user', 'content': [{'text': t.input}]}],
)
)
Created table ‘chat’.
Added 0 column values with 0 errors in 0.01 s
No rows affected.
# Parse the response into a new column
t.add_computed_column(response=t.output.output.message.content[0].text)
Added 0 column values with 0 errors in 0.01 s
No rows affected.
# Start a conversation
t.insert(
[
{
'input': 'What was the outcome of the 1904 US Presidential election?'
}
]
)
t.select(t.input, t.response).show()
Inserted 1 row with 0 errors in 2.75 s (0.36 rows/s)
Learn more
To learn more about advanced techniques like RAG operations in
Pixeltable, check out the RAG Operations in
Pixeltable
tutorial.
If you have any questions, don’t hesitate to reach out.
