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.
Cohere documentation and API reference is available here.
pip install cohere
import cohere
co = cohere.Client(
api_key="YOUR_API_KEY",
)
chat = co.chat(
message="hello world!",
model="command"
)
print(chat)
[!TIP] You can set a system environment variable
CO_API_KEY
to avoid writing your api key within your code, e.g. addexport CO_API_KEY=theapikeyforyouraccount
in your ~/.zshrc or ~/.bashrc, open a new terminal, then code callingcohere.Client()
will read this key.
The SDK supports streaming endpoints. To take advantage of this feature for chat,
use chat_stream
.
import cohere
co = cohere.Client(
api_key="YOUR_API_KEY",
)
stream = co.chat_stream(
message="Tell me a short story"
)
for event in stream:
if event.event_type == "text-generation":
print(event.text, end='')
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!