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,6 +419,11 @@ 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 == expected_corpus.backend_config
+    # assert (
+    #     returned_corpus.vertex_ai_search_config
+    #     == expected_corpus.vertex_ai_search_config
+    # )
     assert returned_corpus.backend_config.__eq__(expected_corpus.backend_config)
     assert returned_corpus.vertex_ai_search_config.__eq__(
         expected_corpus.vertex_ai_search_config
@@ -464,6 +469,9 @@ 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 == expected_config.rag_managed_db_config
+    # )
     assert returned_config.rag_managed_db_config.__eq__(
         expected_config.rag_managed_db_config
     )

tests/unit/vertex_rag/test_rag_data_preview.py

@@ -674,6 +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 == 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 == expected_corpus.corpus_type_config
     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__(
@@ -729,6 +736,9 @@ 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 == expected_config.rag_managed_db_config
+    # )
     assert returned_config.rag_managed_db_config.__eq__(
         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
 
 

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