Check the init and runtime parameters you can specify for AmazonBedrockDocumentEmbeder.
YAML Init Parameters
These are the parameters you can specify in pipeline YAML:
Parameter | Type | Possible values | Description |
---|---|---|---|
| Literal |
| The embedding model to use. The model must be specified in the format outlined in the Amazon Bedrock documentation. |
| Secret | ` | AWS access key ID. |
| Secret | AWS secret access key. | |
| Secret | AWS session token. | |
| Secret | AWS region name. | |
| Secret | AWS profile name. | |
| Dictionary | Additional parameters to pass for model inference. For example, | |
| Integer | Default: | The number of documents to embed at once. Only Cohere models support batch inference. This parameter is ignored for Amazon Titan models. |
| Boolean |
| Shows progress bar when creating the embeddings. |
| List of strings | Default: | A list of metadata fields you want to embed along with the document text. |
| String | Default: | The separator used to concatenate the metadata fields to the document text. |