Cohere Python SDK
The Cohere Python SDK allows access to Cohere models across many different platforms: the cohere platform, AWS (Bedrock, Sagemaker), Azure, GCP and Oracle OCI. For a full list of support and snippets, please take a look at the SDK support docs page.
Documentation
Cohere documentation and API reference is available here.
Installation
pip install cohere
Usage
import cohere
co = cohere.ClientV2()
response = co.chat(
model="command-r-plus-08-2024",
messages=[{"role": "user", "content": "hello world!"}],
)
print(response)
Streaming
The SDK supports streaming endpoints. To take advantage of this feature for chat,
use chat_stream.
import cohere
co = cohere.ClientV2()
response = co.chat_stream(
model="command-r-plus-08-2024",
messages=[{"role": "user", "content": "hello world!"}],
)
for event in response:
if event.type == "content-delta":
print(event.delta.message.content.text, end="")
Oracle Cloud Infrastructure (OCI)
The SDK supports Oracle Cloud Infrastructure (OCI) Generative AI service. First, install the OCI SDK:
pip install 'cohere[oci]'
Then use the OciClient or OciClientV2:
import cohere
# Using OCI config file authentication (default: ~/.oci/config)
co = cohere.OciClient(
oci_region="us-chicago-1",
oci_compartment_id="ocid1.compartment.oc1...",
)
response = co.embed(
model="embed-english-v3.0",
texts=["Hello world"],
input_type="search_document",
)
print(response.embeddings)
OCI Authentication Methods
1. Config File (Default)
co = cohere.OciClient(
oci_region="us-chicago-1",
oci_compartment_id="ocid1.compartment.oc1...",
# Uses ~/.oci/config with DEFAULT profile
)
2. Custom Profile
co = cohere.OciClient(
oci_profile="MY_PROFILE",
oci_region="us-chicago-1",
oci_compartment_id="ocid1.compartment.oc1...",
)
3. Session-based Authentication (Security Token)
# Works with OCI CLI session tokens
co = cohere.OciClient(
oci_profile="MY_SESSION_PROFILE", # Profile with security_token_file
oci_region="us-chicago-1",
oci_compartment_id="ocid1.compartment.oc1...",
)
4. Direct Credentials
co = cohere.OciClient(
oci_user_id="ocid1.user.oc1...",
oci_fingerprint="xx:xx:xx:...",
oci_tenancy_id="ocid1.tenancy.oc1...",
oci_private_key_path="~/.oci/key.pem",
oci_region="us-chicago-1",
oci_compartment_id="ocid1.compartment.oc1...",
)
5. Instance Principal (for OCI Compute instances)
co = cohere.OciClient(
auth_type="instance_principal",
oci_region="us-chicago-1",
oci_compartment_id="ocid1.compartment.oc1...",
)
Supported OCI APIs
The OCI client supports the following Cohere APIs:
- Embed: Full support for all embedding models
- Chat: Full support with both V1 (
OciClient) and V2 (OciClientV2) APIs- Streaming available via
chat_stream() - Supports Command-R and Command-A model families
- Streaming available via
OCI Model Availability and Limitations
Available on OCI On-Demand Inference:
- ✅ Embed models: available on OCI Generative AI
- ✅ Chat models: available via
OciClient(V1) andOciClientV2(V2)
Not Available on OCI On-Demand Inference:
- ❌ Generate API: OCI TEXT_GENERATION models are base models that require fine-tuning before deployment
- ❌ Rerank API: OCI TEXT_RERANK models are base models that require fine-tuning before deployment
- ❌ Multiple Embedding Types: OCI on-demand models only support single embedding type per request (cannot request both
floatandint8simultaneously)
Note: To use Generate or Rerank models on OCI, you need to:
- Fine-tune the base model using OCI's fine-tuning service
- Deploy the fine-tuned model to a dedicated endpoint
- Update your code to use the deployed model endpoint
For the latest model availability, see the OCI Generative AI documentation.
Contributing
While we value open-source contributions to this SDK, the code is generated programmatically. Additions made directly would have to be moved over to our generation code, otherwise they would be overwritten upon the next generated release. Feel free to open a PR as a proof of concept, but know that we will not be able to merge it as-is. We suggest opening an issue first to discuss with us!
On the other hand, contributions to the README are always very welcome!
4 maintainers
Release files for cohere 7.2.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| cohere-7.2.0.tar.gz | 219.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| cohere-7.2.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 589.7 kB
Release files / cohere-7.2.0.tar.gz
| Download URL | cohere-7.2.0.tar.gz |
|---|---|
| Size | 219.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
eb74b46882dcff913f826945ae9630cee6b67237ca05d763005c4ad9c4ad5bba
|
|
BLAKE2b-256 checksum How to use checksums |
6949631a5fac290b704ba0be751d0370f8fb931ad7cb2de3c2f8e4b6a4ffb141
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.5.1 CPython/3.10.21 Linux/6.17.0-1022-azure
|
Release files / cohere-7.2.0-py3-none-any.whl
| Download URL | cohere-7.2.0-py3-none-any.whl |
|---|---|
| Size | 370.7 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
d35cb91814d13eac6587e62bf78302b7683cdd036e7697a8e173ec76f2fb4e37
|
|
BLAKE2b-256 checksum How to use checksums |
4ea7823e5bed159b0df507757743db3e0fb8de2deb39271b83bf6e97897e7536
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.5.1 CPython/3.10.21 Linux/6.17.0-1022-azure
|
