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