WeaviateIngestOperator¶

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

使用操作符¶

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

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

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

    batch_data_with_vectors_xcom_data = WeaviateIngestOperator(
        task_id="batch_data_with_vectors_xcom_data",
        conn_id="weaviate_default",
        collection_name=COLLECTION_NAME,
        input_data=store_data_with_vectors_in_xcom(),
        trigger_rule="all_done",
    )

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

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

    batch_data_with_vectors_callable_data = WeaviateIngestOperator(
        task_id="batch_data_with_vectors_callable_data",
        conn_id="weaviate_default",
        collection_name=COLLECTION_NAME,
        input_data=get_data_with_vectors(),
        trigger_rule="all_done",
    )

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

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

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

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

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

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