OpenAIEmbeddingOperator¶

使用 OpenAIEmbeddingOperator 與 OpenAI API 互動，為給定的文字建立嵌入 (embeddings)。

使用 Operator¶

OpenAIEmbeddingOperator 需要 input_text 作為 embedding 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 與 OpenAI API 互動以觸發批次任務。此 operator 用於觸發批次任務並等待任務完成。

使用 Operator¶

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