Skip to content

Seamlessly invoke Amazon Bedrock or your custom models, enabling a smooth experience with AWS GenAI services.

License

Notifications You must be signed in to change notification settings

aws-samples/sample-connector-for-bedrock

Sample Connector for Bedrock

This is a Bedrock (and other generative AI tools from AWS) API forwarding tool that can issue virtual keys, log chats, and manage costs.

It is compatible with any OPENAI client that can define Host and API Key.

Quick references

Quick start

Quick Deployment

Providers

Docker Image: DockerHub, Public ECR

Docker Image for Lambda: DockerHub, Public ECR

Main Features

Supported Models and Platforms

  • Supports all current and future large language models from Bedrock (supported through bedrock-converse).
  • Supports models deployed through Sagemaker LMI (partial models).
  • Supports other forms of custom models, including Ollama, etc.
  • More AI workflow applications, such as internet search, AWS command executors, etc.

API Key and Cost Management

  • Create API keys. Keys can be created for regular users and administrators. Regular users can chat, while administrators can manage API keys and costs.
  • Record the cost of each call and use it as a basis for cost control.
  • Cost control. You can set monthly quotas and account balances for each API key. When the monthly quota or account balance is insufficient, it cannot be used.
  • Calculate the overall cost.

Important

The cost calculation of this project cannot be used as the basis for AWS billing. Please refer to the AWS bill for actual expenses. Please refer to the official website for the Bedrock pricing.

Model Management

Models and their parameters can be defined from the backend.

Once defined, models can be bound to groups or API Keys.

Changelogs

0.0.16

  1. Added a new configuration option for bedrock-converse: maxTokens

0.0.15

  1. Added haiku 3.5 model in NoDB mode.

  2. Removed some unnecessary log outputs.

Disclaimer

This connector is an open-source software aimed at providing proxy services for using Bedrock services. We make our best efforts to ensure the security and legality of the software, but we are not responsible for the users' behavior.

The connector is intended solely for personal learning and research purposes. Users shall not use it for any illegal activities, including but not limited to hacking, spreading illegal information, etc. Otherwise, users shall bear the corresponding legal responsibilities themselves. Users are responsible for complying with the laws and regulations in their respective jurisdictions and shall not use the connector for any illegal or non-compliant purposes. The developers and maintainers of this software shall not be liable for any disputes, losses, or legal liabilities arising from the use of this connector.

We reserve the right to modify or terminate the connector's code at any time without further notice. Users are expected to understand and comply with the relevant local laws and regulations.

If you have any questions regarding this disclaimer, please feel free to contact us through the open-source channels.

Security

See CONTRIBUTING for more information.

License

This library is licensed under the MIT-0 License. See the LICENSE file.