`airflow.providers.cohere.operators.embedding`¶

模块内容¶

类¶

CohereEmbeddingOperator

通过与 Cohere 托管服务交互创建嵌入基础。

class airflow.providers.cohere.operators.embedding.CohereEmbeddingOperator(input_text, conn_id=CohereHook.default_conn_name, timeout=None, max_retries=None, **kwargs)[源代码]¶

基类: airflow.models.BaseOperator

通过与 Cohere 托管服务交互创建嵌入基础。

另请参阅

有关如何使用此操作符的更多信息，请查看以下指南： CohereEmbeddingOperator

参数

input_text (list[str] | str) – 需要嵌入的单个字符串文本或文本项列表。
conn_id (str) – 可选。用于获取 Cohere 连接信息的 Airflow 连接名称。默认为“cohere_default”。
timeout (int | None) – Cohere API 的超时时间（秒）。
max_retries (int | None) – 失败前重试的次数。

template_fields: collections.abc.Sequence[str] = ('input_text',)[源代码]¶

hook()[源代码]¶

返回 CohereHook 的实例。

execute(context)[源代码]¶

使用 Cohere 嵌入服务嵌入文本。

airflow.providers.cohere.operators.embedding¶

模块内容¶

类¶

`airflow.providers.cohere.operators.embedding`¶