google-cloud-aiplatform 2.1.3__tar.gz → 2.2.0__tar.gz

{google_cloud_aiplatform-2.1.3/google_cloud_aiplatform.egg-info → google_cloud_aiplatform-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-aiplatform
-Version: 2.1.3
+Version: 2.2.0
 Summary: Vertex AI API client library
 Home-page: https://github.com/googleapis/python-aiplatform
 Author: Google LLC
@@ -40,44 +40,44 @@ Provides-Extra: endpoint
 Requires-Dist: requests>=2.28.1; extra == "endpoint"
 Requires-Dist: requests-toolbelt<=1.0.0; extra == "endpoint"
 Provides-Extra: full
-Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "full"
+Requires-Dist: pyarrow>=6.0.1; extra == "full"
 Requires-Dist: fastapi<=0.124.4,>=0.71.0; extra == "full"
-Requires-Dist: google-cloud-bigquery; extra == "full"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "full"
-Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "full"
-Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "full"
-Requires-Dist: tqdm>=4.23.0; extra == "full"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and extra == "full"
-Requires-Dist: ruamel.yaml; extra == "full"
-Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "full"
-Requires-Dist: requests-toolbelt<=1.0.0; extra == "full"
-Requires-Dist: litellm<1.86.0,>=1.83.7; python_version < "3.14" and extra == "full"
-Requires-Dist: litellm<1.97.0,>=1.93.0; python_version >= "3.14" and extra == "full"
-Requires-Dist: immutabledict; extra == "full"
-Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "full"
-Requires-Dist: docker>=5.0.3; extra == "full"
-Requires-Dist: uvicorn[standard]>=0.16.0; extra == "full"
-Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "full"
-Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "full"
-Requires-Dist: starlette>=0.17.1; extra == "full"
 Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "full"
-Requires-Dist: jsonschema; extra == "full"
-Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "full"
-Requires-Dist: numpy>=1.15.0; extra == "full"
-Requires-Dist: scikit-learn; python_version > "3.10" and extra == "full"
-Requires-Dist: requests>=2.28.1; extra == "full"
 Requires-Dist: explainable-ai-sdk>=1.0.0; python_version < "3.13" and extra == "full"
-Requires-Dist: google-cloud-bigquery-storage; extra == "full"
-Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "full"
 Requires-Dist: pyyaml<7,>=5.3.1; extra == "full"
-Requires-Dist: pyarrow<8.0.0,>=3.0.0; python_version < "3.10" and extra == "full"
-Requires-Dist: urllib3<1.27,>=1.21.1; extra == "full"
+Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "full"
 Requires-Dist: scikit-learn<1.6.0; python_version <= "3.10" and extra == "full"
+Requires-Dist: scikit-learn; python_version > "3.10" and extra == "full"
+Requires-Dist: jsonschema; extra == "full"
+Requires-Dist: litellm<1.97.0,>=1.93.0; python_version >= "3.14" and extra == "full"
+Requires-Dist: requests>=2.28.1; extra == "full"
+Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and extra == "full"
+Requires-Dist: google-cloud-bigquery; extra == "full"
 Requires-Dist: pyyaml; extra == "full"
-Requires-Dist: google-vizier>=0.1.6; extra == "full"
-Requires-Dist: pyarrow>=6.0.1; extra == "full"
+Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "full"
+Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "full"
 Requires-Dist: ray[default]!=2.10.*,!=2.11.*,!=2.12.*,!=2.13.*,!=2.14.*,!=2.15.*,!=2.16.*,!=2.17.*,!=2.18.*,!=2.19.*,!=2.20.*,!=2.21.*,!=2.22.*,!=2.23.*,!=2.24.*,!=2.25.*,!=2.26.*,!=2.27.*,!=2.28.*,!=2.29.*,!=2.30.*,!=2.31.*,!=2.32.*,!=2.34.*,!=2.35.*,!=2.36.*,!=2.37.*,!=2.38.*,!=2.39.*,!=2.40.*,!=2.41.*,!=2.5.*,!=2.6.*,!=2.7.*,!=2.8.*,!=2.9.0,!=2.9.1,!=2.9.2,<=2.42.0,>=2.4; python_version < "3.11" and extra == "full"
+Requires-Dist: urllib3<1.27,>=1.21.1; extra == "full"
+Requires-Dist: pyarrow<8.0.0,>=3.0.0; python_version < "3.10" and extra == "full"
+Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "full"
+Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "full"
+Requires-Dist: google-cloud-bigquery-storage; extra == "full"
+Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "full"
+Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "full"
+Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "full"
 Requires-Dist: pandas>=1.0.0; extra == "full"
+Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "full"
+Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "full"
+Requires-Dist: starlette>=0.17.1; extra == "full"
+Requires-Dist: google-vizier>=0.1.6; extra == "full"
+Requires-Dist: tqdm>=4.23.0; extra == "full"
+Requires-Dist: immutabledict; extra == "full"
+Requires-Dist: requests-toolbelt<=1.0.0; extra == "full"
+Requires-Dist: uvicorn[standard]>=0.16.0; extra == "full"
+Requires-Dist: numpy>=1.15.0; extra == "full"
+Requires-Dist: litellm<1.86.0,>=1.83.7; python_version < "3.14" and extra == "full"
+Requires-Dist: docker>=5.0.3; extra == "full"
+Requires-Dist: ruamel.yaml; extra == "full"
 Provides-Extra: metadata
 Requires-Dist: pandas>=1.0.0; extra == "metadata"
 Requires-Dist: numpy>=1.15.0; extra == "metadata"
@@ -85,44 +85,44 @@ Provides-Extra: tensorboard
 Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "tensorboard"
 Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "tensorboard"
 Provides-Extra: testing
-Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "testing"
+Requires-Dist: pyarrow>=6.0.1; extra == "testing"
 Requires-Dist: fastapi<=0.124.4,>=0.71.0; extra == "testing"
-Requires-Dist: google-cloud-bigquery; extra == "testing"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "testing"
 Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "testing"
-Requires-Dist: tqdm>=4.23.0; extra == "testing"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and extra == "testing"
-Requires-Dist: ruamel.yaml; extra == "testing"
-Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "testing"
-Requires-Dist: requests-toolbelt<=1.0.0; extra == "testing"
-Requires-Dist: litellm<1.86.0,>=1.83.7; python_version < "3.14" and extra == "testing"
-Requires-Dist: litellm<1.97.0,>=1.93.0; python_version >= "3.14" and extra == "testing"
-Requires-Dist: immutabledict; extra == "testing"
-Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "testing"
-Requires-Dist: docker>=5.0.3; extra == "testing"
-Requires-Dist: uvicorn[standard]>=0.16.0; extra == "testing"
-Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "testing"
-Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "testing"
-Requires-Dist: starlette>=0.17.1; extra == "testing"
-Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: jsonschema; extra == "testing"
-Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: numpy>=1.15.0; extra == "testing"
-Requires-Dist: scikit-learn; python_version > "3.10" and extra == "testing"
-Requires-Dist: requests>=2.28.1; extra == "testing"
 Requires-Dist: explainable-ai-sdk>=1.0.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: google-cloud-bigquery-storage; extra == "testing"
-Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "testing"
 Requires-Dist: pyyaml<7,>=5.3.1; extra == "testing"
-Requires-Dist: pyarrow<8.0.0,>=3.0.0; python_version < "3.10" and extra == "testing"
-Requires-Dist: urllib3<1.27,>=1.21.1; extra == "testing"
+Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "testing"
 Requires-Dist: scikit-learn<1.6.0; python_version <= "3.10" and extra == "testing"
+Requires-Dist: scikit-learn; python_version > "3.10" and extra == "testing"
+Requires-Dist: jsonschema; extra == "testing"
+Requires-Dist: litellm<1.97.0,>=1.93.0; python_version >= "3.14" and extra == "testing"
+Requires-Dist: requests>=2.28.1; extra == "testing"
+Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and extra == "testing"
+Requires-Dist: google-cloud-bigquery; extra == "testing"
 Requires-Dist: pyyaml; extra == "testing"
-Requires-Dist: google-vizier>=0.1.6; extra == "testing"
-Requires-Dist: pyarrow>=6.0.1; extra == "testing"
+Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "testing"
+Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "testing"
 Requires-Dist: ray[default]!=2.10.*,!=2.11.*,!=2.12.*,!=2.13.*,!=2.14.*,!=2.15.*,!=2.16.*,!=2.17.*,!=2.18.*,!=2.19.*,!=2.20.*,!=2.21.*,!=2.22.*,!=2.23.*,!=2.24.*,!=2.25.*,!=2.26.*,!=2.27.*,!=2.28.*,!=2.29.*,!=2.30.*,!=2.31.*,!=2.32.*,!=2.34.*,!=2.35.*,!=2.36.*,!=2.37.*,!=2.38.*,!=2.39.*,!=2.40.*,!=2.41.*,!=2.5.*,!=2.6.*,!=2.7.*,!=2.8.*,!=2.9.0,!=2.9.1,!=2.9.2,<=2.42.0,>=2.4; python_version < "3.11" and extra == "testing"
+Requires-Dist: urllib3<1.27,>=1.21.1; extra == "testing"
+Requires-Dist: pyarrow<8.0.0,>=3.0.0; python_version < "3.10" and extra == "testing"
+Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "testing"
+Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "testing"
+Requires-Dist: google-cloud-bigquery-storage; extra == "testing"
+Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "testing"
+Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "testing"
+Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "testing"
 Requires-Dist: pandas>=1.0.0; extra == "testing"
+Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "testing"
+Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "testing"
+Requires-Dist: starlette>=0.17.1; extra == "testing"
+Requires-Dist: google-vizier>=0.1.6; extra == "testing"
+Requires-Dist: tqdm>=4.23.0; extra == "testing"
+Requires-Dist: immutabledict; extra == "testing"
+Requires-Dist: requests-toolbelt<=1.0.0; extra == "testing"
+Requires-Dist: uvicorn[standard]>=0.16.0; extra == "testing"
+Requires-Dist: numpy>=1.15.0; extra == "testing"
+Requires-Dist: litellm<1.86.0,>=1.83.7; python_version < "3.14" and extra == "testing"
+Requires-Dist: docker>=5.0.3; extra == "testing"
+Requires-Dist: ruamel.yaml; extra == "testing"
 Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "testing"
 Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "testing"
 Requires-Dist: sentencepiece>=0.2.0; extra == "testing"
@@ -210,18 +210,18 @@ Provides-Extra: adk
 Requires-Dist: google-adk<3.0.0,>=1.27.0; extra == "adk"
 Requires-Dist: aiohttp; extra == "adk"
 Provides-Extra: adk-testing
-Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "adk-testing"
 Requires-Dist: cloudpickle<4.0,>=3.0; extra == "adk-testing"
-Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "adk-testing"
-Requires-Dist: pytest-xdist; extra == "adk-testing"
-Requires-Dist: pydantic<3,>=2.11.1; extra == "adk-testing"
 Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "adk-testing"
+Requires-Dist: pydantic<3,>=2.11.1; extra == "adk-testing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "adk-testing"
+Requires-Dist: pytest-xdist; extra == "adk-testing"
+Requires-Dist: google-adk<3.0.0,>=1.27.0; extra == "adk-testing"
+Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "adk-testing"
 Requires-Dist: absl-py; extra == "adk-testing"
 Requires-Dist: typing_extensions; extra == "adk-testing"
-Requires-Dist: google-adk<3.0.0,>=1.27.0; extra == "adk-testing"
-Requires-Dist: aiohttp; extra == "adk-testing"
 Requires-Dist: opentelemetry-sdk<2; extra == "adk-testing"
-Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "adk-testing"
+Requires-Dist: aiohttp; extra == "adk-testing"
+Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "adk-testing"
 Provides-Extra: reasoningengine
 Requires-Dist: cloudpickle<4.0,>=3.0; extra == "reasoningengine"
 Requires-Dist: opentelemetry-sdk<2; extra == "reasoningengine"
@@ -260,61 +260,61 @@ Requires-Dist: langchain-google-genai<4.2.3,>=4.0.0; extra == "langchain"
 Requires-Dist: langgraph<2.0.0,>=1.0.0; extra == "langchain"
 Requires-Dist: openinference-instrumentation-langchain<0.2,>=0.1.19; extra == "langchain"
 Provides-Extra: langchain-testing
-Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "langchain-testing"
+Requires-Dist: langchain-classic; extra == "langchain-testing"
 Requires-Dist: cloudpickle<4.0,>=3.0; extra == "langchain-testing"
-Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "langchain-testing"
-Requires-Dist: langchain-google-genai<4.2.3,>=4.0.0; extra == "langchain-testing"
-Requires-Dist: pytest-xdist; extra == "langchain-testing"
+Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "langchain-testing"
 Requires-Dist: openinference-instrumentation-langchain<0.2,>=0.1.19; extra == "langchain-testing"
 Requires-Dist: pydantic<3,>=2.11.1; extra == "langchain-testing"
-Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "langchain-testing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "langchain-testing"
+Requires-Dist: pytest-xdist; extra == "langchain-testing"
+Requires-Dist: langchain-core<2.0.0,>=1.0.0; extra == "langchain-testing"
 Requires-Dist: langgraph<2.0.0,>=1.0.0; extra == "langchain-testing"
-Requires-Dist: langchain-classic; extra == "langchain-testing"
-Requires-Dist: typing_extensions; extra == "langchain-testing"
-Requires-Dist: aiohttp; extra == "langchain-testing"
+Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "langchain-testing"
+Requires-Dist: langchain-google-genai<4.2.3,>=4.0.0; extra == "langchain-testing"
 Requires-Dist: absl-py; extra == "langchain-testing"
-Requires-Dist: langchain-core<2.0.0,>=1.0.0; extra == "langchain-testing"
+Requires-Dist: typing_extensions; extra == "langchain-testing"
 Requires-Dist: opentelemetry-sdk<2; extra == "langchain-testing"
+Requires-Dist: aiohttp; extra == "langchain-testing"
 Requires-Dist: langchain<2.0.0,>=1.0.0; extra == "langchain-testing"
-Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "langchain-testing"
+Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "langchain-testing"
 Provides-Extra: tokenization
 Requires-Dist: sentencepiece>=0.2.0; extra == "tokenization"
 Provides-Extra: ag2
 Requires-Dist: autogen[gemini]<1,>=0.14.1; extra == "ag2"
 Requires-Dist: openinference-instrumentation-autogen<0.2,>=0.1.6; extra == "ag2"
 Provides-Extra: ag2-testing
-Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "ag2-testing"
 Requires-Dist: cloudpickle<4.0,>=3.0; extra == "ag2-testing"
-Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "ag2-testing"
-Requires-Dist: pytest-xdist; extra == "ag2-testing"
-Requires-Dist: pydantic<3,>=2.11.1; extra == "ag2-testing"
 Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "ag2-testing"
-Requires-Dist: openinference-instrumentation-autogen<0.2,>=0.1.6; extra == "ag2-testing"
+Requires-Dist: pydantic<3,>=2.11.1; extra == "ag2-testing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "ag2-testing"
+Requires-Dist: pytest-xdist; extra == "ag2-testing"
+Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "ag2-testing"
+Requires-Dist: autogen[gemini]<1,>=0.14.1; extra == "ag2-testing"
 Requires-Dist: absl-py; extra == "ag2-testing"
+Requires-Dist: openinference-instrumentation-autogen<0.2,>=0.1.6; extra == "ag2-testing"
 Requires-Dist: typing_extensions; extra == "ag2-testing"
-Requires-Dist: autogen[gemini]<1,>=0.14.1; extra == "ag2-testing"
-Requires-Dist: aiohttp; extra == "ag2-testing"
 Requires-Dist: opentelemetry-sdk<2; extra == "ag2-testing"
-Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "ag2-testing"
+Requires-Dist: aiohttp; extra == "ag2-testing"
+Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "ag2-testing"
 Provides-Extra: llama-index
 Requires-Dist: llama-index; extra == "llama-index"
 Requires-Dist: llama-index-llms-google-genai; extra == "llama-index"
 Requires-Dist: openinference-instrumentation-llama-index<4.0,>=3.0; extra == "llama-index"
 Provides-Extra: llama-index-testing
-Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "llama-index-testing"
 Requires-Dist: cloudpickle<4.0,>=3.0; extra == "llama-index-testing"
-Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "llama-index-testing"
-Requires-Dist: pytest-xdist; extra == "llama-index-testing"
-Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "llama-index-testing"
-Requires-Dist: pydantic<3,>=2.11.1; extra == "llama-index-testing"
+Requires-Dist: llama-index-llms-google-genai; extra == "llama-index-testing"
 Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "llama-index-testing"
-Requires-Dist: absl-py; extra == "llama-index-testing"
+Requires-Dist: pydantic<3,>=2.11.1; extra == "llama-index-testing"
+Requires-Dist: llama-index; extra == "llama-index-testing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "llama-index-testing"
+Requires-Dist: pytest-xdist; extra == "llama-index-testing"
 Requires-Dist: openinference-instrumentation-llama-index<4.0,>=3.0; extra == "llama-index-testing"
+Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "llama-index-testing"
+Requires-Dist: absl-py; extra == "llama-index-testing"
 Requires-Dist: typing_extensions; extra == "llama-index-testing"
-Requires-Dist: aiohttp; extra == "llama-index-testing"
-Requires-Dist: llama-index; extra == "llama-index-testing"
 Requires-Dist: opentelemetry-sdk<2; extra == "llama-index-testing"
-Requires-Dist: llama-index-llms-google-genai; extra == "llama-index-testing"
+Requires-Dist: aiohttp; extra == "llama-index-testing"
+Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "llama-index-testing"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{google_cloud_aiplatform-2.1.3 → google_cloud_aiplatform-2.2.0}/agentplatform/__init__.py

@@ -54,8 +54,17 @@ def __getattr__(name):  # type: ignore[no-untyped-def]
         global _genai_types
         if _genai_types is None:
             _genai_types = importlib.import_module("._genai.types", __name__)
-        if "vertexai.types" not in sys.modules:
-            sys.modules["vertexai.types"] = _genai_types
+        # `types` is an alias for `._genai.types` rather than a real submodule,
+        # so register it to keep
+        # `from agentplatform.types import TypeName`
+        # working without a prior attribute access. Spell it the google3 way:
+        # Copybara rewrites that prefix in both directions, and the external
+        # spelling fails its reversibility check. This key was `vertexai.types`,
+        # which both misnamed this package's own alias and shadowed the real
+        # `vertexai.types` module.
+        types_module_name = f"{__name__}.types"
+        if types_module_name not in sys.modules:
+            sys.modules[types_module_name] = _genai_types
         return _genai_types
 
     raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

{google_cloud_aiplatform-2.1.3 → google_cloud_aiplatform-2.2.0}/agentplatform/_genai/_evals_common.py

@@ -37,6 +37,11 @@ from google.genai._gaos.types.interactions import functioncallstep
 from google.genai._gaos.types.interactions import functionresultstep
 from google.genai._gaos.types.interactions import modeloutputstep
 from google.genai._gaos.types.interactions import userinputstep
+
+try:
+    from google.genai._gaos.utils.serializers import ALLOW_UNKNOWN_UNION_VARIANTS
+except (ImportError, AttributeError):
+    ALLOW_UNKNOWN_UNION_VARIANTS = "speakeasy_allow_unknown_union_variants"
 from google.genai.models import Models
 import pandas as pd
 from tqdm import tqdm
@@ -775,7 +780,16 @@ def _interaction_dict_to_agent_data(
     Returns:
         An AgentData object with one or more ConversationTurns.
     """
-    typed_interaction = interaction_types.Interaction.model_validate(interaction)
+    # A server response, so unknown step types must degrade rather than raise.
+    # `Step` is an open discriminated union: parse_open_union only falls back to
+    # the Unknown variant when the validation context carries
+    # ALLOW_UNKNOWN_UNION_VARIANTS, and raises without it so that a
+    # user-constructed request payload surfaces its mistakes locally. This
+    # payload comes off the Interactions API, so it is on the tolerant side of
+    # that line. _interaction_steps_to_events already drops steps it cannot map.
+    typed_interaction = interaction_types.Interaction.model_validate(
+        interaction, context={ALLOW_UNKNOWN_UNION_VARIANTS: True}
+    )
     all_events = _interaction_steps_to_events(typed_interaction.steps or [])
 
     # Group events into turns. Each UserInputStep starts a new turn.
@@ -1689,8 +1703,12 @@ def _resolve_interactions_to_eval_cases(
                 break
             interaction_dict = json.loads(response.body)
             try:
+                # Also a server response -- see _interaction_dict_to_agent_data.
+                # Without the context a single step type this client does not
+                # know about raised, and the except below turned that into a
+                # `break`, silently truncating the interaction history.
                 typed_interaction = interaction_types.Interaction.model_validate(
-                    interaction_dict
+                    interaction_dict, context={ALLOW_UNKNOWN_UNION_VARIANTS: True}
                 )
             except Exception as e:
                 logger.warning("Failed to validate interaction model: %s", e)

{google_cloud_aiplatform-2.1.3 → google_cloud_aiplatform-2.2.0}/agentplatform/_genai/evals.py

@@ -215,6 +215,13 @@ def _CustomCodeExecutionSpec_from_vertex(
             getv(from_object, ["evaluation_function"]),
         )
 
+    if getv(from_object, ["codeExecutionRegion"]) is not None:
+        setv(
+            to_object,
+            ["code_execution_region"],
+            getv(from_object, ["codeExecutionRegion"]),
+        )
+
     return to_object
 
 
@@ -237,6 +244,13 @@ def _CustomCodeExecutionSpec_to_vertex(
             getv(from_object, ["remote_custom_function"]),
         )
 
+    if getv(from_object, ["code_execution_region"]) is not None:
+        setv(
+            to_object,
+            ["codeExecutionRegion"],
+            getv(from_object, ["code_execution_region"]),
+        )
+
     return to_object
 
 

{google_cloud_aiplatform-2.1.3 → google_cloud_aiplatform-2.2.0}/agentplatform/_genai/memory_banks.py

@@ -504,6 +504,13 @@ def _ReasoningEngine_from_vertex(
     if getv(from_object, ["trafficConfig"]) is not None:
         setv(to_object, ["traffic_config"], getv(from_object, ["trafficConfig"]))
 
+    if getv(from_object, ["revisionGarbageCollectionStrategy"]) is not None:
+        setv(
+            to_object,
+            ["revision_garbage_collection_strategy"],
+            getv(from_object, ["revisionGarbageCollectionStrategy"]),
+        )
+
     return to_object
 
 

{google_cloud_aiplatform-2.1.3 → google_cloud_aiplatform-2.2.0}/agentplatform/_genai/runtimes.py

@@ -484,6 +484,13 @@ def _ReasoningEngine_from_vertex(
     if getv(from_object, ["trafficConfig"]) is not None:
         setv(to_object, ["traffic_config"], getv(from_object, ["trafficConfig"]))
 
+    if getv(from_object, ["revisionGarbageCollectionStrategy"]) is not None:
+        setv(
+            to_object,
+            ["revision_garbage_collection_strategy"],
+            getv(from_object, ["revisionGarbageCollectionStrategy"]),
+        )
+
     return to_object
 
 

{google_cloud_aiplatform-2.1.3 → google_cloud_aiplatform-2.2.0}/agentplatform/_genai/types/__init__.py

@@ -1682,6 +1682,12 @@ from .common import ReasoningEngineContextSpecMemoryBankConfigTtlConfigOrDict
 from .common import ReasoningEngineContextSpecOrDict
 from .common import ReasoningEngineDict
 from .common import ReasoningEngineOrDict
+from .common import ReasoningEngineRevisionGarbageCollectionStrategy
+from .common import ReasoningEngineRevisionGarbageCollectionStrategyDict
+from .common import ReasoningEngineRevisionGarbageCollectionStrategyKeepNLatest
+from .common import ReasoningEngineRevisionGarbageCollectionStrategyKeepNLatestDict
+from .common import ReasoningEngineRevisionGarbageCollectionStrategyKeepNLatestOrDict
+from .common import ReasoningEngineRevisionGarbageCollectionStrategyOrDict
 from .common import ReasoningEngineRuntimeRevision
 from .common import ReasoningEngineRuntimeRevisionDict
 from .common import ReasoningEngineRuntimeRevisionOrDict
@@ -2888,6 +2894,12 @@ __all__ = [
     "ReasoningEngineTrafficConfig",
     "ReasoningEngineTrafficConfigDict",
     "ReasoningEngineTrafficConfigOrDict",
+    "ReasoningEngineRevisionGarbageCollectionStrategyKeepNLatest",
+    "ReasoningEngineRevisionGarbageCollectionStrategyKeepNLatestDict",
+    "ReasoningEngineRevisionGarbageCollectionStrategyKeepNLatestOrDict",
+    "ReasoningEngineRevisionGarbageCollectionStrategy",
+    "ReasoningEngineRevisionGarbageCollectionStrategyDict",
+    "ReasoningEngineRevisionGarbageCollectionStrategyOrDict",
     "ReasoningEngine",
     "ReasoningEngineDict",
     "ReasoningEngineOrDict",