WeaviateIngestOperator¶

使用 WeaviateIngestOperator 与 Weaviate API 交互，为给定文本创建嵌入并将其摄取到数据库中。或者，您也可以为您的文本提供自定义向量，这些向量可以被摄取到数据库中。

使用操作符¶

WeaviateIngestOperator 需要将 input_data 作为操作符的输入。使用 conn_id 参数指定要使用的 Weaviate 连接以连接到您的帐户。

一个使用操作符摄取从 XCOM 检索到的自定义向量数据的示例

tests/system/providers/weaviate/example_weaviate_operator.py[源代码]

    batch_data_with_vectors_xcom_data = WeaviateIngestOperator(
        task_id="batch_data_with_vectors_xcom_data",
        conn_id="weaviate_default",
        class_name="QuestionWithoutVectorizerUsingOperator",
        input_data=store_data_with_vectors_in_xcom(),
        trigger_rule="all_done",
    )

一个使用操作符摄取从 Python 可调用对象检索到的自定义向量数据的示例

tests/system/providers/weaviate/example_weaviate_operator.py[源代码]

    batch_data_with_vectors_callable_data = WeaviateIngestOperator(
        task_id="batch_data_with_vectors_callable_data",
        conn_id="weaviate_default",
        class_name="QuestionWithoutVectorizerUsingOperator",
        input_data=get_data_with_vectors(),
        trigger_rule="all_done",
    )

一个使用操作符摄取没有从 XCOM 检索到的向量数据的示例，操作符将为其生成嵌入向量

tests/system/providers/weaviate/example_weaviate_operator.py[源代码]

    batch_data_without_vectors_xcom_data = WeaviateIngestOperator(
        task_id="batch_data_without_vectors_xcom_data",
        conn_id="weaviate_default",
        class_name="QuestionWithOpenAIVectorizerUsingOperator",
        input_data=xcom_data_without_vectors["return_value"],
        trigger_rule="all_done",
    )

一个使用操作符摄取没有从 Python 可调用对象检索到的向量数据的示例，操作符将为其生成嵌入向量

tests/system/providers/weaviate/example_weaviate_operator.py[源代码]

    batch_data_without_vectors_callable_data = WeaviateIngestOperator(
        task_id="batch_data_without_vectors_callable_data",
        conn_id="weaviate_default",
        class_name="QuestionWithOpenAIVectorizerUsingOperator",
        input_data=get_data_without_vectors(),
        trigger_rule="all_done",
    )