Skip to main content

​
module pixeltable.functions.bedrock

View source on GitHub Pixeltable UDFs for AWS Bedrock AI models. Provides integration with AWS Bedrock for accessing various foundation models including Anthropic Claude, Amazon Titan, and other providers.

​
func invoke_tools()

invoke_tools(
    tools: pixeltable.func.tools.Tools,
    response: pixeltable.exprs.expr.Expr
) -> pixeltable.exprs.inline_expr.InlineDict
Converts an Anthropic response dict to Pixeltable tool invocation format and calls tools._invoke().

​
udf converse()

converse(
    messages: Json,
    *,
    model_id: String,
    system: Json | None = None,
    inference_config: Json | None = None,
    additional_model_request_fields: Json | None = None,
    tool_config: Json | None = None
) -> Json
Generate a conversation response. Equivalent to the AWS Bedrock converse API endpoint. For additional details, see: https://docs.aws.amazon.com/bedrock/latest/userguide/conversation-inference.html Requirements:
  • pip install boto3
Parameters:
  • messages (Json): Input messages.
  • model_id (String): The model that will complete your prompt.
  • system (Json | None): An optional system prompt.
  • inference_config (Json | None): Base inference parameters to use.
  • additional_model_request_fields (Json | None): Additional inference parameters to use.
Returns:
  • Json: A dictionary containing the response and other metadata.
Examples: Add a computed column that applies the model anthropic.claude-3-haiku-20240307-v1:0 to an existing Pixeltable column tbl.prompt of the table tbl: 
msgs = [{'role': 'user', 'content': [{'text': tbl.prompt}]}]
tbl.add_computed_column(response=messages(msgs, model_id='anthropic.claude-3-haiku-20240307-v1:0'))