chore: update tests to be compatible with Python 3.14, set 3.14 testing workflow as required

PiperOrigin-RevId: 825273830

Author: speedstorm1

testing/constraints-3.14.txt

@@ -13,4 +13,4 @@ ipython==8.22.2 # Pinned to unbreak TypeAliasType import error
 google-adk==0.0.2
 google-genai>=1.10.0
 google-vizier==0.1.21
-pyarrow>=18.0.0
+pyarrow>=22.0.0

tests/unit/aiplatform/test_prediction.py

@@ -3287,7 +3287,10 @@ def test_health(self, model_server_env_mock, importlib_import_module_mock_twice)
 
         assert response.status_code == 200
 
-    def test_predict(self, model_server_env_mock, importlib_import_module_mock_twice):
+    @pytest.mark.asyncio
+    async def test_predict(
+        self, model_server_env_mock, importlib_import_module_mock_twice
+    ):
         model_server = CprModelServer()
         client = TestClient(model_server.app)
 

tests/unit/vertex_rag/test_rag_data.py

@@ -419,9 +419,10 @@ def create_transformation_config(
 def rag_corpus_eq(returned_corpus, expected_corpus):
     assert returned_corpus.name == expected_corpus.name
     assert returned_corpus.display_name == expected_corpus.display_name
-    assert returned_corpus.backend_config.__eq__(expected_corpus.backend_config)
-    assert returned_corpus.vertex_ai_search_config.__eq__(
-        expected_corpus.vertex_ai_search_config
+    assert returned_corpus.backend_config == expected_corpus.backend_config
+    assert (
+        returned_corpus.vertex_ai_search_config
+        == expected_corpus.vertex_ai_search_config
     )
 
 
@@ -464,8 +465,8 @@ def import_files_request_eq(returned_request, expected_request):
 
 def rag_engine_config_eq(returned_config, expected_config):
     assert returned_config.name == expected_config.name
-    assert returned_config.rag_managed_db_config.__eq__(
-        expected_config.rag_managed_db_config
+    assert (
+        returned_config.rag_managed_db_config == expected_config.rag_managed_db_config
     )
 
 

tests/unit/vertex_rag/test_rag_data_preview.py

@@ -674,12 +674,13 @@ def create_transformation_config(
 def rag_corpus_eq(returned_corpus, expected_corpus):
     assert returned_corpus.name == expected_corpus.name
     assert returned_corpus.display_name == expected_corpus.display_name
-    assert returned_corpus.vector_db.__eq__(expected_corpus.vector_db)
-    assert returned_corpus.backend_config.__eq__(expected_corpus.backend_config)
-    assert returned_corpus.vertex_ai_search_config.__eq__(
-        expected_corpus.vertex_ai_search_config
+    assert returned_corpus.vector_db == expected_corpus.vector_db
+    assert returned_corpus.backend_config == expected_corpus.backend_config
+    assert (
+        returned_corpus.vertex_ai_search_config
+        == expected_corpus.vertex_ai_search_config
     )
-    assert returned_corpus.corpus_type_config.__eq__(expected_corpus.corpus_type_config)
+    assert returned_corpus.corpus_type_config == expected_corpus.corpus_type_config
 
 
 def rag_file_eq(returned_file, expected_file):
@@ -729,8 +730,8 @@ def import_files_request_eq(returned_request, expected_request):
 
 def rag_engine_config_eq(returned_config, expected_config):
     assert returned_config.name == expected_config.name
-    assert returned_config.rag_managed_db_config.__eq__(
-        expected_config.rag_managed_db_config
+    assert (
+        returned_config.rag_managed_db_config == expected_config.rag_managed_db_config
     )
 
 

vertexai/preview/rag/utils/_gapic_utils.py

@@ -293,6 +293,8 @@ def convert_gapic_to_backend_config(
     gapic_vector_db: GapicRagVectorDbConfig,
 ) -> RagVectorDbConfig:
     """Convert Gapic RagVectorDbConfig to VertexVectorSearch, Pinecone, or RagManagedDb."""
+    if not gapic_vector_db:
+        return None
     vector_config = RagVectorDbConfig()
     if _check_pinecone(gapic_vector_db):
         vector_config.vector_db = Pinecone(
@@ -314,6 +316,8 @@ def convert_gapic_to_backend_config(
                 gapic_vector_db.rag_embedding_model_config
             )
         )
+    if not vector_config.vector_db and not vector_config.rag_embedding_model_config:
+        return None
     return vector_config
 
 
@@ -353,11 +357,12 @@ def convert_gapic_to_rag_corpus(gapic_rag_corpus: GapicRagCorpus) -> RagCorpus:
         vertex_ai_search_config=convert_gapic_to_vertex_ai_search_config(
             gapic_rag_corpus.vertex_ai_search_config
         ),
-        backend_config=convert_gapic_to_backend_config(
-            gapic_rag_corpus.vector_db_config
-        ),
         encryption_spec=gapic_rag_corpus.encryption_spec,
     )
+    if gapic_rag_corpus.vector_db_config:
+        rag_corpus.backend_config = convert_gapic_to_backend_config(
+            gapic_rag_corpus.vector_db_config
+        )
     return rag_corpus
 
 

vertexai/rag/utils/_gapic_utils.py

@@ -175,6 +175,8 @@ def convert_gapic_to_backend_config(
     gapic_vector_db: GapicRagVectorDbConfig,
 ) -> RagVectorDbConfig:
     """Convert Gapic RagVectorDbConfig to VertexVectorSearch, Pinecone, or RagManagedDb."""
+    if not gapic_vector_db:
+        return None
     vector_config = RagVectorDbConfig()
     if _check_pinecone(gapic_vector_db):
         vector_config.vector_db = Pinecone(
@@ -194,6 +196,8 @@ def convert_gapic_to_backend_config(
                 gapic_vector_db.rag_embedding_model_config
             )
         )
+    if not vector_config.vector_db and not vector_config.rag_embedding_model_config:
+        return None
     return vector_config
 
 
@@ -217,11 +221,12 @@ def convert_gapic_to_rag_corpus(gapic_rag_corpus: GapicRagCorpus) -> RagCorpus:
         vertex_ai_search_config=convert_gapic_to_vertex_ai_search_config(
             gapic_rag_corpus.vertex_ai_search_config
         ),
-        backend_config=convert_gapic_to_backend_config(
-            gapic_rag_corpus.vector_db_config
-        ),
         encryption_spec=gapic_rag_corpus.encryption_spec,
     )
+    if gapic_rag_corpus.vector_db_config:
+        rag_corpus.backend_config = convert_gapic_to_backend_config(
+            gapic_rag_corpus.vector_db_config
+        )
     return rag_corpus