OpenAIEmbeddingOperator¶

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

使用 Operator¶

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

使用该 Operator 的示例如下：

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 与 OpenAI API 交互以触发批处理作业。此 Operator 用于触发批处理作业并等待作业完成。

使用 Operator¶

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

OpenAITriggerBatchOperator

使用该 Operator 的示例如下：

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,
    )