google-cloud-aiplatform 2.6.0__tar.gz → 2.8.0__tar.gz

{google_cloud_aiplatform-2.6.0/google_cloud_aiplatform.egg-info → google_cloud_aiplatform-2.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-aiplatform
-Version: 2.6.0
+Version: 2.8.0
 Summary: Vertex AI API client library
 Home-page: https://github.com/googleapis/python-aiplatform
 Author: Google LLC
@@ -17,6 +17,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
 Classifier: Topic :: Internet
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
@@ -40,44 +41,43 @@ 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: uvicorn[standard]>=0.16.0; extra == "full"
+Requires-Dist: requests>=2.28.1; extra == "full"
 Requires-Dist: google-cloud-bigquery-storage; extra == "full"
-Requires-Dist: litellm<1.97.0,>=1.83.7; python_version < "3.14" and extra == "full"
-Requires-Dist: starlette>=0.17.1; extra == "full"
-Requires-Dist: scikit-learn; python_version > "3.10" and extra == "full"
-Requires-Dist: litellm<1.97.0,>=1.93.0; python_version >= "3.14" and extra == "full"
-Requires-Dist: pandas>=1.0.0; extra == "full"
 Requires-Dist: jsonschema; extra == "full"
-Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "full"
-Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "full"
-Requires-Dist: pyarrow>=6.0.1; extra == "full"
+Requires-Dist: scikit-learn; python_version > "3.10" and extra == "full"
+Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "full"
+Requires-Dist: mlflow>=1.27.0; (python_version >= "3.13" and python_version < "3.15") and extra == "full"
+Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "full"
+Requires-Dist: pyyaml; extra == "full"
+Requires-Dist: ruamel.yaml; extra == "full"
 Requires-Dist: explainable-ai-sdk>=1.0.0; python_version < "3.13" and extra == "full"
-Requires-Dist: fastapi<=0.124.4,>=0.71.0; extra == "full"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and extra == "full"
-Requires-Dist: urllib3<1.27,>=1.21.1; extra == "full"
 Requires-Dist: immutabledict; extra == "full"
-Requires-Dist: pyyaml<7,>=5.3.1; extra == "full"
-Requires-Dist: numpy>=1.15.0; extra == "full"
-Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "full"
-Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "full"
-Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "full"
-Requires-Dist: tqdm>=4.23.0; extra == "full"
-Requires-Dist: google-vizier>=0.1.6; extra == "full"
+Requires-Dist: docker>=5.0.3; extra == "full"
 Requires-Dist: pyarrow<8.0.0,>=3.0.0; python_version < "3.10" 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: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "full"
-Requires-Dist: google-cloud-bigquery; extra == "full"
-Requires-Dist: docker>=5.0.3; extra == "full"
-Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "full"
-Requires-Dist: pyyaml; extra == "full"
-Requires-Dist: requests>=2.28.1; extra == "full"
-Requires-Dist: uvicorn[standard]>=0.16.0; extra == "full"
-Requires-Dist: requests-toolbelt<=1.0.0; extra == "full"
+Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "full"
 Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "full"
-Requires-Dist: ruamel.yaml; extra == "full"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "full"
-Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "full"
 Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "full"
+Requires-Dist: google-vizier>=0.1.6; extra == "full"
+Requires-Dist: pyyaml<7,>=5.3.1; extra == "full"
+Requires-Dist: urllib3<1.27,>=1.21.1; extra == "full"
 Requires-Dist: scikit-learn<1.6.0; python_version <= "3.10" and extra == "full"
+Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "full"
+Requires-Dist: starlette>=0.17.1; extra == "full"
+Requires-Dist: pyarrow>=6.0.1; extra == "full"
+Requires-Dist: werkzeug<4.0.0,>=2.0.0; 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: lit-nlp==0.4.0; python_version < "3.13" and extra == "full"
+Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "full"
+Requires-Dist: litellm<1.107.0,>=1.106.5; python_version < "3.15" and extra == "full"
+Requires-Dist: google-cloud-bigquery; extra == "full"
+Requires-Dist: requests-toolbelt<=1.0.0; extra == "full"
+Requires-Dist: pandas>=1.0.0; extra == "full"
+Requires-Dist: fastapi<=0.124.4,>=0.71.0; 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"
 Provides-Extra: metadata
 Requires-Dist: pandas>=1.0.0; extra == "metadata"
 Requires-Dist: numpy>=1.15.0; extra == "metadata"
@@ -85,47 +85,46 @@ 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: uvicorn[standard]>=0.16.0; extra == "testing"
+Requires-Dist: requests>=2.28.1; extra == "testing"
 Requires-Dist: google-cloud-bigquery-storage; extra == "testing"
-Requires-Dist: litellm<1.97.0,>=1.83.7; python_version < "3.14" and extra == "testing"
-Requires-Dist: starlette>=0.17.1; extra == "testing"
-Requires-Dist: scikit-learn; python_version > "3.10" and extra == "testing"
-Requires-Dist: litellm<1.97.0,>=1.93.0; python_version >= "3.14" and extra == "testing"
-Requires-Dist: pandas>=1.0.0; extra == "testing"
 Requires-Dist: jsonschema; extra == "testing"
-Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "testing"
-Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: pyarrow>=6.0.1; extra == "testing"
+Requires-Dist: scikit-learn; python_version > "3.10" and extra == "testing"
+Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "testing"
+Requires-Dist: mlflow>=1.27.0; (python_version >= "3.13" and python_version < "3.15") and extra == "testing"
+Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "testing"
+Requires-Dist: pyyaml; extra == "testing"
+Requires-Dist: ruamel.yaml; extra == "testing"
 Requires-Dist: explainable-ai-sdk>=1.0.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: fastapi<=0.124.4,>=0.71.0; extra == "testing"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and extra == "testing"
-Requires-Dist: urllib3<1.27,>=1.21.1; extra == "testing"
 Requires-Dist: immutabledict; extra == "testing"
-Requires-Dist: pyyaml<7,>=5.3.1; extra == "testing"
-Requires-Dist: numpy>=1.15.0; extra == "testing"
-Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "testing"
-Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "testing"
-Requires-Dist: tqdm>=4.23.0; extra == "testing"
-Requires-Dist: google-vizier>=0.1.6; extra == "testing"
+Requires-Dist: docker>=5.0.3; extra == "testing"
 Requires-Dist: pyarrow<8.0.0,>=3.0.0; python_version < "3.10" 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: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: google-cloud-bigquery; extra == "testing"
-Requires-Dist: docker>=5.0.3; extra == "testing"
-Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "testing"
-Requires-Dist: pyyaml; extra == "testing"
-Requires-Dist: requests>=2.28.1; extra == "testing"
-Requires-Dist: uvicorn[standard]>=0.16.0; extra == "testing"
-Requires-Dist: requests-toolbelt<=1.0.0; extra == "testing"
+Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "testing"
 Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: ruamel.yaml; extra == "testing"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "testing"
-Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "testing"
 Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "testing"
+Requires-Dist: google-vizier>=0.1.6; extra == "testing"
+Requires-Dist: pyyaml<7,>=5.3.1; extra == "testing"
+Requires-Dist: urllib3<1.27,>=1.21.1; extra == "testing"
 Requires-Dist: scikit-learn<1.6.0; python_version <= "3.10" and extra == "testing"
+Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "testing"
+Requires-Dist: starlette>=0.17.1; extra == "testing"
+Requires-Dist: pyarrow>=6.0.1; extra == "testing"
+Requires-Dist: werkzeug<4.0.0,>=2.0.0; 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: lit-nlp==0.4.0; python_version < "3.13" and extra == "testing"
+Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "testing"
+Requires-Dist: litellm<1.107.0,>=1.106.5; python_version < "3.15" and extra == "testing"
+Requires-Dist: google-cloud-bigquery; extra == "testing"
+Requires-Dist: requests-toolbelt<=1.0.0; extra == "testing"
+Requires-Dist: pandas>=1.0.0; extra == "testing"
+Requires-Dist: fastapi<=0.124.4,>=0.71.0; 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: 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"
+Requires-Dist: sentencepiece<0.2.3,>=0.2.0; python_version < "3.15" and extra == "testing"
 Requires-Dist: nltk; extra == "testing"
 Requires-Dist: google-vizier>=0.1.6; extra == "testing"
 Requires-Dist: aiohttp; extra == "testing"
@@ -134,6 +133,7 @@ Requires-Dist: google-api-core<3.0.0,>=2.27.0; extra == "testing"
 Requires-Dist: grpcio-testing; extra == "testing"
 Requires-Dist: grpcio-tools>=1.63.0; python_version >= "3.13" and extra == "testing"
 Requires-Dist: ipython; extra == "testing"
+Requires-Dist: jinja2; extra == "testing"
 Requires-Dist: kfp<3.0.0,>=2.6.0; python_version < "3.13" and extra == "testing"
 Requires-Dist: pytest-asyncio; extra == "testing"
 Requires-Dist: pytest-cov; extra == "testing"
@@ -181,7 +181,7 @@ Requires-Dist: urllib3<1.27,>=1.21.1; extra == "private-endpoints"
 Requires-Dist: requests>=2.28.1; extra == "private-endpoints"
 Provides-Extra: autologging
 Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "autologging"
-Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "autologging"
+Requires-Dist: mlflow>=1.27.0; (python_version >= "3.13" and python_version < "3.15") and extra == "autologging"
 Provides-Extra: preview
 Provides-Extra: ray
 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 == "ray"
@@ -210,17 +210,17 @@ 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-exporter-otlp-proto-http<2; extra == "adk-testing"
-Requires-Dist: pytest-xdist; extra == "adk-testing"
-Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "adk-testing"
-Requires-Dist: aiohttp; extra == "adk-testing"
-Requires-Dist: google-adk<3.0.0,>=1.27.0; extra == "adk-testing"
-Requires-Dist: typing_extensions; extra == "adk-testing"
-Requires-Dist: pydantic<3,>=2.11.1; extra == "adk-testing"
-Requires-Dist: absl-py; extra == "adk-testing"
 Requires-Dist: cloudpickle<4.0,>=3.0; extra == "adk-testing"
+Requires-Dist: aiohttp; extra == "adk-testing"
 Requires-Dist: opentelemetry-sdk<2; extra == "adk-testing"
 Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "adk-testing"
+Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "adk-testing"
+Requires-Dist: typing_extensions; extra == "adk-testing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "adk-testing"
+Requires-Dist: absl-py; extra == "adk-testing"
+Requires-Dist: pydantic<3,>=2.11.1; extra == "adk-testing"
+Requires-Dist: pytest-xdist; extra == "adk-testing"
+Requires-Dist: google-adk<3.0.0,>=1.27.0; extra == "adk-testing"
 Provides-Extra: reasoningengine
 Requires-Dist: cloudpickle<4.0,>=3.0; extra == "reasoningengine"
 Requires-Dist: opentelemetry-sdk<2; extra == "reasoningengine"
@@ -246,8 +246,7 @@ Requires-Dist: scikit-learn; python_version > "3.10" and extra == "evaluation"
 Requires-Dist: jsonschema; extra == "evaluation"
 Requires-Dist: ruamel.yaml; extra == "evaluation"
 Requires-Dist: pyyaml; extra == "evaluation"
-Requires-Dist: litellm<1.97.0,>=1.83.7; python_version < "3.14" and extra == "evaluation"
-Requires-Dist: litellm<1.97.0,>=1.93.0; python_version >= "3.14" and extra == "evaluation"
+Requires-Dist: litellm<1.107.0,>=1.106.5; python_version < "3.15" and extra == "evaluation"
 Provides-Extra: langchain
 Requires-Dist: langchain<2.0.0,>=1.0.0; extra == "langchain"
 Requires-Dist: langchain-classic; extra == "langchain"
@@ -256,58 +255,58 @@ 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: cloudpickle<4.0,>=3.0; extra == "langchain-testing"
+Requires-Dist: aiohttp; extra == "langchain-testing"
+Requires-Dist: opentelemetry-sdk<2; 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: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "langchain-testing"
-Requires-Dist: aiohttp; extra == "langchain-testing"
 Requires-Dist: typing_extensions; extra == "langchain-testing"
-Requires-Dist: pydantic<3,>=2.11.1; extra == "langchain-testing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "langchain-testing"
 Requires-Dist: langchain-google-genai<4.2.3,>=4.0.0; extra == "langchain-testing"
-Requires-Dist: langchain-classic; extra == "langchain-testing"
+Requires-Dist: absl-py; extra == "langchain-testing"
+Requires-Dist: langgraph<2.0.0,>=1.0.0; extra == "langchain-testing"
+Requires-Dist: pydantic<3,>=2.11.1; extra == "langchain-testing"
+Requires-Dist: pytest-xdist; extra == "langchain-testing"
 Requires-Dist: langchain<2.0.0,>=1.0.0; extra == "langchain-testing"
+Requires-Dist: langchain-classic; extra == "langchain-testing"
 Requires-Dist: openinference-instrumentation-langchain<0.2,>=0.1.19; extra == "langchain-testing"
 Requires-Dist: langchain-core<2.0.0,>=1.0.0; extra == "langchain-testing"
-Requires-Dist: cloudpickle<4.0,>=3.0; extra == "langchain-testing"
-Requires-Dist: absl-py; extra == "langchain-testing"
-Requires-Dist: opentelemetry-sdk<2; extra == "langchain-testing"
-Requires-Dist: langgraph<2.0.0,>=1.0.0; extra == "langchain-testing"
 Provides-Extra: tokenization
-Requires-Dist: sentencepiece>=0.2.0; extra == "tokenization"
+Requires-Dist: sentencepiece<0.2.3,>=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: openinference-instrumentation-autogen<0.2,>=0.1.6; extra == "ag2-testing"
-Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "ag2-testing"
+Requires-Dist: cloudpickle<4.0,>=3.0; extra == "ag2-testing"
+Requires-Dist: aiohttp; extra == "ag2-testing"
+Requires-Dist: opentelemetry-sdk<2; extra == "ag2-testing"
+Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; 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: aiohttp; extra == "ag2-testing"
-Requires-Dist: autogen[gemini]<1,>=0.14.1; extra == "ag2-testing"
 Requires-Dist: typing_extensions; 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: openinference-instrumentation-autogen<0.2,>=0.1.6; extra == "ag2-testing"
 Requires-Dist: absl-py; extra == "ag2-testing"
-Requires-Dist: cloudpickle<4.0,>=3.0; extra == "ag2-testing"
-Requires-Dist: opentelemetry-sdk<2; extra == "ag2-testing"
-Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "ag2-testing"
+Requires-Dist: pydantic<3,>=2.11.1; extra == "ag2-testing"
+Requires-Dist: autogen[gemini]<1,>=0.14.1; 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-exporter-otlp-proto-http<2; extra == "llama-index-testing"
-Requires-Dist: pytest-xdist; extra == "llama-index-testing"
-Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "llama-index-testing"
-Requires-Dist: openinference-instrumentation-llama-index<4.0,>=3.0; extra == "llama-index-testing"
+Requires-Dist: cloudpickle<4.0,>=3.0; 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: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "llama-index-testing"
+Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "llama-index-testing"
 Requires-Dist: typing_extensions; extra == "llama-index-testing"
-Requires-Dist: pydantic<3,>=2.11.1; extra == "llama-index-testing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "llama-index-testing"
+Requires-Dist: llama-index; extra == "llama-index-testing"
 Requires-Dist: absl-py; extra == "llama-index-testing"
-Requires-Dist: cloudpickle<4.0,>=3.0; extra == "llama-index-testing"
+Requires-Dist: pydantic<3,>=2.11.1; 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: llama-index-llms-google-genai; extra == "llama-index-testing"
-Requires-Dist: opentelemetry-sdk<2; extra == "llama-index-testing"
-Requires-Dist: opentelemetry-resourcedetector-gcp>=1.5.0.dev0; extra == "llama-index-testing"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{google_cloud_aiplatform-2.6.0 → google_cloud_aiplatform-2.8.0}/agentplatform/_genai/_evals_common.py

@@ -3238,6 +3238,31 @@ def _process_multi_turn_agent_response(
     ).model_dump(exclude_unset=True)
 
 
+def _extract_final_response_text(
+    events: list[dict[str, Any]],
+) -> tuple[Optional[int], str]:
+    """Returns the index and text of the last final response in the events.
+
+    As in ADK's `Event.is_final_response()`, events with function calls or
+    responses are not final responses, so they stay in the intermediate events.
+    Thought parts are not part of the response text.
+    """
+    for index in range(len(events) - 1, -1, -1):
+        parts = (events[index].get("content") or {}).get("parts") or []
+        if any(
+            part.get("function_call") or part.get("function_response") for part in parts
+        ):
+            continue
+        texts = [
+            part["text"]
+            for part in parts
+            if part.get("text") and not part.get("thought")
+        ]
+        if texts:
+            return index, "".join(texts)
+    return None, ""
+
+
 def _process_single_turn_agent_response(
     resp_item: Any,
     agent_data_agents: Optional[dict[str, Any]],
@@ -3253,8 +3278,10 @@ def _process_single_turn_agent_response(
 
     if isinstance(resp_item, list):
         try:
-            response_row = resp_item[-1]["content"]["parts"][0]["text"]
-            for intermediate_event in resp_item[:-1]:
+            response_index, response_row = _extract_final_response_text(resp_item)
+            for index, intermediate_event in enumerate(resp_item):
+                if index == response_index:
+                    continue
                 intermediate_events_row.append(
                     {
                         "event_id": intermediate_event.get("id"),
@@ -3263,7 +3290,7 @@ def _process_single_turn_agent_response(
                         "author": intermediate_event.get("author"),
                     }
                 )
-            # Construct AgentData natively for single-turn runs
+                # Construct AgentData natively for single-turn runs
             agent_events = []
             for event_dict in resp_item:
                 content_dict = event_dict.get("content")
@@ -3288,6 +3315,7 @@ def _process_single_turn_agent_response(
                 agents=agent_data_agents,
             ).model_dump(exclude_unset=True)
         except Exception as e:  # pylint: disable=broad-exception-caught
+            intermediate_events_row = []
             error_payload = {
                 "error": (
                     f"Failed to parse agent run response {str(resp_item)} to "

{google_cloud_aiplatform-2.6.0 → google_cloud_aiplatform-2.8.0}/agentplatform/_genai/_evals_metric_handlers.py

@@ -846,6 +846,66 @@ class LLMMetricHandler(MetricHandler[types.LLMMetric]):
             return _default_aggregate_scores(self.metric.name, eval_case_metric_results)
 
 
+class StepBasedLLMMetricHandler(LLMMetricHandler):
+    """Metric handler for step-based LLM metrics."""
+
+    @override
+    def get_metric_result(
+        self, eval_case: types.EvalCase, response_index: int
+    ) -> types.EvalCaseMetricResult:
+        """Evaluates one case with the step-based LLM metric."""
+        try:
+            response_content = _get_response_from_eval_case(
+                eval_case, response_index, self.metric_name
+            )
+            if not response_content and not getattr(eval_case, "agent_data", None):
+                raise ValueError(
+                    f"Response content missing for candidate {response_index}."
+                )
+
+            # Joined so dataset columns referenced by any step reach other_data.
+            prompt_templates = "\n".join(
+                step.prompt_template or "" for step in self.metric.steps or []
+            )
+            instance = _build_evaluation_instance(
+                eval_case, response_content, prompt_template=prompt_templates
+            )
+            api_response = _call_with_retry(
+                lambda: self.module._evaluate_instances(
+                    metrics=[self.metric],
+                    instance=instance,
+                    allow_cross_region_model=self.allow_cross_region_model,
+                ),
+                self.metric_name,
+            )
+
+            if api_response and api_response.metric_results:
+                result = api_response.metric_results[0]
+                error_msg = None
+                if result.error and getattr(result.error, "code"):
+                    error_msg = f"Error in metric result: {result.error}"
+                return types.EvalCaseMetricResult(
+                    metric_name=self.metric_name,
+                    score=result.score,
+                    explanation=result.explanation,
+                    error_message=error_msg,
+                )
+            return types.EvalCaseMetricResult(
+                metric_name=self.metric_name,
+                error_message="Metric results missing in API response.",
+            )
+        except Exception as e:  # pylint: disable=broad-exception-caught
+            logger.error(
+                "Error processing metric %s for case %s.",
+                self.metric_name,
+                eval_case.eval_case_id,
+                exc_info=True,
+            )
+            return types.EvalCaseMetricResult(
+                metric_name=self.metric_name, error_message=str(e)
+            )
+
+
 class CustomMetricHandler(MetricHandler[types.Metric]):
     """Metric handler for custom metrics."""
 
@@ -1375,6 +1435,7 @@ _METRIC_HANDLER_MAPPING = [
         lambda m: getattr(m, "metric_resource_name", None) is not None,
         RegisteredMetricHandler,
     ),
+    (lambda m: getattr(m, "steps", None), StepBasedLLMMetricHandler),
     (
         lambda m: m.name in ComputationMetricHandler.SUPPORTED_COMPUTATION_METRICS,
         ComputationMetricHandler,
@@ -1395,6 +1456,7 @@ MetricHandlerType = TypeVar(
     ComputationMetricHandler,
     TranslationMetricHandler,
     LLMMetricHandler,
+    StepBasedLLMMetricHandler,
     CustomMetricHandler,
     CustomCodeExecutionMetricHandler,
     PredefinedMetricHandler,

{google_cloud_aiplatform-2.6.0 → google_cloud_aiplatform-2.8.0}/agentplatform/_genai/_runtimes_utils.py

@@ -1294,11 +1294,6 @@ def _upload_runtime(
                     "`from a2a import types`."
                 )
             raise TypeError(error_msg) from e
-    with blob.open("rb") as f:
-        try:
-            _ = cloudpickle.load(f)
-        except Exception as e:
-            raise TypeError("agent runtime serialized to an invalid format") from e
     dir_name = f"gs://{gcs_bucket.name}/{gcs_dir_name}"
     logger.info(f"Wrote to {dir_name}/{_BLOB_FILENAME}")
 

{google_cloud_aiplatform-2.6.0 → google_cloud_aiplatform-2.8.0}/agentplatform/_genai/_transformers.py

@@ -89,6 +89,11 @@ def t_metrics(
             metric_payload_item["custom_code_execution_spec"] = {
                 "evaluation_function": metric.custom_function
             }
+        # Step-based LLM metrics
+        elif getv(metric, ["steps"]):
+            metric_payload_item["step_based_llm_metric_spec"] = (
+                _step_based_llm_metric_spec(metric)
+            )
         # LLM-based metrics
         elif hasattr(metric, "prompt_template") and metric.prompt_template:
             llm_based_spec: dict[str, Any] = {
@@ -145,6 +150,47 @@ def t_metrics(
     return metrics_payload
 
 
+def _dump(value: Any) -> Any:
+    if hasattr(value, "model_dump"):
+        return value.model_dump(exclude_none=True)
+    return value
+
+
+def _step_based_llm_metric_spec(metric: Any) -> dict[str, Any]:
+    """Builds a StepBasedLlmMetricSpec payload.
+
+    autorater_config fields a step leaves unset fall back to the metric's judge
+    model settings.
+    """
+    default_autorater_config: dict[str, Any] = {}
+    if getv(metric, ["judge_model"]):
+        default_autorater_config["autorater_model"] = getv(metric, ["judge_model"])
+    if getv(metric, ["judge_model_generation_config"]):
+        default_autorater_config["generation_config"] = _dump(
+            getv(metric, ["judge_model_generation_config"])
+        )
+    if getv(metric, ["judge_model_sampling_count"]):
+        default_autorater_config["sampling_count"] = getv(
+            metric, ["judge_model_sampling_count"]
+        )
+
+    steps = []
+    for step in getv(metric, ["steps"]):
+        step_payload = _dump(step)
+        if default_autorater_config:
+            step_payload["autorater_config"] = {
+                **default_autorater_config,
+                **step_payload.get("autorater_config", {}),
+            }
+        steps.append(step_payload)
+
+    spec: dict[str, Any] = {"steps": steps}
+    preprocessor = getv(metric, ["preprocessor"])
+    if preprocessor:
+        spec["preprocessor"] = _dump(preprocessor)
+    return spec
+
+
 def t_metric_sources(metrics: list[Any]) -> list[dict[str, Any]]:
     """Prepares the MetricSource payload."""
     sources_payload = []
@@ -217,6 +263,12 @@ def t_metric_for_registry(
             "evaluation_function": metric.custom_function
         }
 
+    # Step-based LLM metric
+    elif getv(metric, ["steps"]):
+        metric_payload_item["step_based_llm_metric_spec"] = _step_based_llm_metric_spec(
+            metric
+        )
+
     # LLM-based metric
     elif (hasattr(metric, "prompt_template") and metric.prompt_template) or (
         hasattr(metric, "rubric_group_name") and metric.rubric_group_name

{google_cloud_aiplatform-2.6.0 → google_cloud_aiplatform-2.8.0}/agentplatform/_genai/evals.py

@@ -1172,6 +1172,13 @@ def _UnifiedMetric_from_vertex(
             getv(from_object, ["computationBasedMetricSpec"]),
         )
 
+    if getv(from_object, ["stepBasedLlmMetricSpec"]) is not None:
+        setv(
+            to_object,
+            ["step_based_llm_metric_spec"],
+            getv(from_object, ["stepBasedLlmMetricSpec"]),
+        )
+
     return to_object
 
 
@@ -1223,6 +1230,13 @@ def _UnifiedMetric_to_vertex(
             getv(from_object, ["computation_based_metric_spec"]),
         )
 
+    if getv(from_object, ["step_based_llm_metric_spec"]) is not None:
+        setv(
+            to_object,
+            ["stepBasedLlmMetricSpec"],
+            getv(from_object, ["step_based_llm_metric_spec"]),
+        )
+
     return to_object