Configure SQL AI Assistant using the Amazon Bedrock Service

This topic describes how to configure the SQL AI Assistant using the Amazon Bedrock Service.

You must have an AWS account with Bedrock access. For more information on accessing keys, see Amazon Bedrock.

  1. Log in to the Data Warehouse service.
  2. Go to the Virtual Warehouses tab, locate the Virtual Warehouse on which you want to enable this feature, and click > Edit.
  3. Go to Configurations > Hue, select hue-safety-valve from the Configuration files drop-down menu, and add the following lines :
    [aws]
      [[bedrock_account]]
        access_key_id_script='echo $AWS_BEDROCK_ACCESS_KEY_ID'
        secret_access_key_script='echo $AWS_BEDROCK_SECRET_ACCESS_KEY'
        region='us-east-1'
    [desktop]
      [[ai_interface]]
        service='bedrock'
        model='claude'
    AWS_BEDROCK_ACCESS_KEY_ID and AWS_BEDROCK_SECRET_ACCESS_KEY must be added as encoded values under hue-secret. For more information, see Secure approach for passing a token.
  4. Click Apply Changes.
You see Assistant on the Hue SQL editor, and the SQL AI Assistant will connect to the specified model in the Amazon Bedrock service.