cohere-ai / cohere-aws

Important

Cohere python and typescript SDKs now also support bedrock and sagemaker! This is part of an effort to make it as frictionless as possible to get started with and switch between Cohere providers.

As such, package will soon be deprecated. Please see the platform support docs for code snippets and the status of this transition or go straight to the python and typescript repos to start using bedrock/sagemaker straight away

Cohere Python SDK (AWS SageMaker & Bedrock)

This package provides functionality developed to simplify interfacing with the Cohere models via AWS SageMaker Marketplace and the Cohere models via AWS Bedrock in Python >=3.9

Installation

The package can be installed with pip:

pip install --upgrade cohere-aws

Install from source:

python setup.py install

Quick Start

To use this library, you need to configure your AWS credentials. You can do this by running aws configure. Once that's set up, please refer to one of the Jupyter notebooks to get started. Here is the one for our medium command model

Note: by default we assume region configured in AWS CLI (aws configure get region), to override use region_name parameter, e.g.

client = Client(region_name='us-east-1')