airflow.providers.cohere.hooks.cohere¶

Attributes¶

logger

Classes¶

CohereHook

Use Cohere Python SDK to interact with Cohere platform using API v2.

Module Contents¶

airflow.providers.cohere.hooks.cohere.logger[source]¶

class airflow.providers.cohere.hooks.cohere.CohereHook(conn_id=default_conn_name, timeout=None, max_retries=None, request_options=None)[source]¶

Bases: airflow.hooks.base.BaseHook

Use Cohere Python SDK to interact with Cohere platform using API v2.

See also

https://docs.cohere.com/docs

Parameters:

conn_id (str) – Cohere connection id
timeout (int | None) – Request timeout in seconds. Optional.
max_retries (int | None) – Maximal number of retries for requests. Deprecated, use request_options instead. Optional.
request_options (cohere.core.request_options.RequestOptions | None) – Dictionary for function-specific request configuration. Optional.

conn_name_attr = 'conn_id'[source]¶

default_conn_name = 'cohere_default'[source]¶

conn_type = 'cohere'[source]¶

hook_name = 'Cohere'[source]¶

conn_id = 'cohere_default'[source]¶

timeout = None[source]¶

max_retries = None[source]¶

request_options = None[source]¶

property get_conn: cohere.ClientV2[source]¶

Return connection for the hook.

create_embeddings(texts, model='embed-multilingual-v3.0')[source]¶

classmethod get_ui_field_behaviour()[source]¶

test_connection(model='command-r-plus-08-2024', messages=None)[source]¶