autogen.oai.bedrock.BedrockClient

BedrockClient

BedrockClient(**kwargs: Any)

Client for Amazon’s Bedrock Converse API.
Initialises BedrockClient for Amazon’s Bedrock Converse API

Parameters:

Name	Description
**kwargs	Type: Any

Instance Methods

cost

cost(self, response: ChatCompletion) -> float

Calculate the cost of the response.

Parameters:

Name	Description
response	Type: ChatCompletion

create

create(self, params) -> ChatCompletionExtended

Run Amazon Bedrock inference and return AG2 response

Parameters:

Name	Description
params

get_usage

get_usage(response) -> dict

Get the usage of tokens and their cost information.

Parameters:

Name	Description
response

message_retrieval

message_retrieval(self, response) -> 

Retrieve the messages from the response.

Parameters:

Name	Description
response

parse_custom_params

parse_custom_params(self, params: dict[str, Any]) -> 

Parses custom parameters for logic in this client class

Parameters:

Name	Description
params	Type: dict[str, Any]

parse_params

parse_params(self, params: dict[str, Any]) -> tuple[dict[str, Any], dict[str, Any]]

Loads the valid parameters required to invoke Bedrock Converse Returns a tuple of (base_params, additional_params)

Parameters:

Name	Description
params	Type: dict[str, Any]