weaviate remove deprecations (#39707)

apache · May 20, 2024 · 79cf7e0 · 79cf7e0
1 parent 655bb60
commit 79cf7e0
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 9 deletions.
diff --git a/tests/always/test_example_dags.py b/tests/always/test_example_dags.py
@@ -63,9 +63,6 @@
     "tests/system/providers/google/cloud/kubernetes_engine/example_kubernetes_engine_resource.py",
     "tests/system/providers/google/cloud/life_sciences/example_life_sciences.py",
     "tests/system/providers/google/marketing_platform/example_analytics.py",
-    "tests/system/providers/weaviate/example_weaviate_cohere.py",
-    "tests/system/providers/weaviate/example_weaviate_openai.py",
-    "tests/system/providers/weaviate/example_weaviate_operator.py",
     # Deprecated Operators/Hooks, which replaced by common.sql Operators/Hooks
     "tests/system/providers/apache/drill/example_drill_dag.py",
     "tests/system/providers/jdbc/example_jdbc_queries.py",

diff --git a/tests/system/providers/weaviate/example_weaviate_cohere.py b/tests/system/providers/weaviate/example_weaviate_cohere.py
@@ -82,7 +82,7 @@ def update_vector_data_in_json(**kwargs):
         task_id="perform_ingestion",
         conn_id="weaviate_default",
         class_name="Weaviate_example_class",
-        input_json=update_vector_data_in_json["return_value"],
+        input_data=update_vector_data_in_json["return_value"],
     )
 
     embed_query = CohereEmbeddingOperator(

diff --git a/tests/system/providers/weaviate/example_weaviate_openai.py b/tests/system/providers/weaviate/example_weaviate_openai.py
@@ -80,7 +80,7 @@ def update_vector_data_in_json(**kwargs):
         task_id="perform_ingestion",
         conn_id="weaviate_default",
         class_name="Weaviate_example_class",
-        input_json=update_vector_data_in_json["return_value"],
+        input_data=update_vector_data_in_json["return_value"],
     )
 
     embed_query = OpenAIEmbeddingOperator(

diff --git a/tests/system/providers/weaviate/example_weaviate_operator.py b/tests/system/providers/weaviate/example_weaviate_operator.py
@@ -121,7 +121,7 @@ def store_data_with_vectors_in_xcom():
         task_id="batch_data_with_vectors_xcom_data",
         conn_id="weaviate_default",
         class_name="QuestionWithoutVectorizerUsingOperator",
-        input_json=store_data_with_vectors_in_xcom(),
+        input_data=store_data_with_vectors_in_xcom(),
         trigger_rule="all_done",
     )
     # [END howto_operator_weaviate_embedding_and_ingest_xcom_data_with_vectors]
@@ -131,7 +131,7 @@ def store_data_with_vectors_in_xcom():
         task_id="batch_data_with_vectors_callable_data",
         conn_id="weaviate_default",
         class_name="QuestionWithoutVectorizerUsingOperator",
-        input_json=get_data_with_vectors(),
+        input_data=get_data_with_vectors(),
         trigger_rule="all_done",
     )
     # [END howto_operator_weaviate_embedding_and_ingest_callable_data_with_vectors]
@@ -231,7 +231,7 @@ def store_doc_data_without_vectors_in_xcom():
         task_id="batch_data_without_vectors_xcom_data",
         conn_id="weaviate_default",
         class_name="QuestionWithOpenAIVectorizerUsingOperator",
-        input_json=xcom_data_without_vectors["return_value"],
+        input_data=xcom_data_without_vectors["return_value"],
         trigger_rule="all_done",
     )
     # [END howto_operator_weaviate_ingest_xcom_data_without_vectors]
@@ -241,7 +241,7 @@ def store_doc_data_without_vectors_in_xcom():
         task_id="batch_data_without_vectors_callable_data",
         conn_id="weaviate_default",
         class_name="QuestionWithOpenAIVectorizerUsingOperator",
-        input_json=get_data_without_vectors(),
+        input_data=get_data_without_vectors(),
         trigger_rule="all_done",
     )
     # [END howto_operator_weaviate_ingest_callable_data_without_vectors]