| Arguments | Category | Description | Default Values | Permitted Values | Data Type | Case Sensitive |
|---|---|---|---|---|---|---|
| API_TYPE | Required | Specifies the API to be used. | None | When connecting to Bedrock: aws-bedrock | String | - |
| AUTHORIZATION | Required | Specifies a JSON string representing key to access Amazon Bedrock. | None | JSON String of the following format {"Access_ID": "id", "Region": "region", "Access_Key": "key", "Session_Token": "token"}' |
String | No, JSON keys are not case sensitive.MULTIDRAG |
| NUM_EMBEDDINGS | Optional | 1536 | Positive Integer | - | - | - |
| MODEL_NAME | Required | Name of Bedrock model to be used for computing embeddings. | “amazon.titan-embed-text-v1” |
Alphanumeric string with dashes | String | Yes |
| OUTPUT_FORMAT | Optional | Output format for embeddings | varbyte |
varchar, varbyte |
String | No |
| DATA_COLUMNS | Required | Columns whose data will be concatenated and sent as input for computing embeddings. | None | String of comma-separated column names | VARCHAR | Argument is not case sensitive. |
| INITIAL_DELAY_MS | Optional | Millisecond delay after each input table row is sent for embeddings | 0 | Nonnegative integer | Int | - |
| DELAY_MAX_RETRIES | Optional | Maximum number of attempts after a failed input table row embedding request | 0 | Nonnegative integer | Int | - |
| DELAY_EXP_BASE | Optional | Exponential base of delay time increase | 1 | Nonnegative integer | Int | - |
| DELAY_JITTER | Optional | Random sum term in exponent | 0 (no jitter) | 0, 1 | Int | - |