OpenAIEmbeddingOperator¶

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

使用操作符¶

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

使用该操作符的示例如下：

tests/system/providers/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",
    )