OpenAIEmbeddingOperator¶

使用 OpenAIEmbeddingOperator 与 Open API 交互，为给定文本创建嵌入。

使用操作符¶

OpenAIEmbeddingOperator 需要 input_text 作为嵌入 API 的输入。使用 conn_id 参数指定要用于连接到您的帐户的 OpenAI 连接。

一个使用操作符的例子如下：

tests/system/openai/example_openai.py[源代码]

    OpenAIEmbeddingOperator(
        task_id="embedding_using_xcom_data",
        conn_id="openai_default",
        input_text=task_to_store_input_text_in_xcom(),
        model="text-embedding-ada-002",
    )

    OpenAIEmbeddingOperator(
        task_id="embedding_using_callable",
        conn_id="openai_default",
        input_text=input_text_callable(
            "input_arg1_value",
            "input2_value",
            input_kwarg1="input_kwarg1_value",
            input_kwarg2="input_kwarg2_value",
        ),
        model="text-embedding-ada-002",
    )
    OpenAIEmbeddingOperator(
        task_id="embedding_using_text",
        conn_id="openai_default",
        input_text=texts,
        model="text-embedding-ada-002",
    )

OpenAITriggerBatchOperator¶

使用 OpenAITriggerBatchOperator 与 Open API 交互以触发批处理作业。此操作符用于触发批处理作业并等待作业完成。

使用操作符¶

OpenAITriggerBatchOperator 需要准备好的批处理文件作为触发批处理作业的输入。提供 file_id 和 endpoint 来触发批处理作业。使用 conn_id 参数指定要使用的 OpenAI 连接。

OpenAITriggerBatchOperator

一个使用操作符的例子如下：

tests/system/openai/example_trigger_batch_operator.py[源代码]

    from airflow.providers.openai.operators.openai import OpenAITriggerBatchOperator

    batch_id = OpenAITriggerBatchOperator(
        task_id="batch_operator_deferred",
        conn_id=OPENAI_CONN_ID,
        file_id=batch_file_id,
        endpoint="/v1/chat/completions",
        deferrable=True,
    )