Cohere Python SDK
This package provides functionality developed to simplify interfacing with the cohere.ai natural language API in Python 3.
Documentation
See the API's documentation.
Also see some code examples for the SDK here.
Installation
If you want the package, you can install it through pip
:
pip install --upgrade cohere
Install from source:
python setup.py install
Requirements
- Python 3.6+
Usage
import cohere
# initialize the Cohere Client with an API Key
co = cohere.CohereClient('YOUR_API_KEY')
# generate a prediction for a prompt
prediction = co.generate(
model="baseline-shrimp",
prompt="co:here",
max_tokens=10)
# print the predicted text
print('prediction: {}'.format(prediction.text))
More usage examples can be found here.
Endpoints
For a full breakdown of endpoints and arguments, please consult the Cohere Docs.
Cohere Endpoint | Function |
---|---|
/generate | co.generate() |
/similarity | co.similarity() |
/choose-best | co.choose_best() |
/embed | co.embed() |
/likelihood | co.likelihood() |
Models
To view an up-to-date list of available models please consult the Cohere CLI. To get started try out baseline-shrimp
or baseline-seal
.