google-cloud-aiplatform 1.158.0__tar.gz → 2.0.0__tar.gz

{google_cloud_aiplatform-1.158.0/google_cloud_aiplatform.egg-info → google_cloud_aiplatform-2.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-aiplatform
-Version: 1.158.0
+Version: 2.0.0
 Summary: Vertex AI API client library
 Home-page: https://github.com/googleapis/python-aiplatform
 Author: Google LLC
@@ -40,43 +40,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: tqdm>=4.23.0; extra == "full"
-Requires-Dist: pyyaml; extra == "full"
-Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "full"
+Requires-Dist: fastapi<=0.124.4,>=0.71.0; extra == "full"
+Requires-Dist: ruamel.yaml; extra == "full"
+Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "full"
+Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "full"
 Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "full"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "full"
 Requires-Dist: google-cloud-bigquery; extra == "full"
-Requires-Dist: ruamel.yaml; extra == "full"
-Requires-Dist: starlette>=0.17.1; extra == "full"
-Requires-Dist: pyarrow<8.0.0,>=3.0.0; python_version < "3.10" and extra == "full"
+Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "full"
+Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "full"
+Requires-Dist: immutabledict; extra == "full"
+Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "full"
+Requires-Dist: jsonschema; extra == "full"
+Requires-Dist: pyyaml<7,>=5.3.1; extra == "full"
+Requires-Dist: docker>=5.0.3; extra == "full"
+Requires-Dist: scikit-learn<1.6.0; python_version <= "3.10" and extra == "full"
+Requires-Dist: numpy>=1.15.0; extra == "full"
+Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; extra == "full"
+Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "full"
+Requires-Dist: google-cloud-bigquery-storage; extra == "full"
+Requires-Dist: pyarrow>=6.0.1; extra == "full"
+Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and 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: uvicorn[standard]>=0.16.0; extra == "full"
 Requires-Dist: pandas>=1.0.0; extra == "full"
-Requires-Dist: pyarrow>=6.0.1; extra == "full"
+Requires-Dist: pyyaml; extra == "full"
+Requires-Dist: requests>=2.28.1; extra == "full"
+Requires-Dist: requests-toolbelt<=1.0.0; extra == "full"
+Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "full"
 Requires-Dist: google-vizier>=0.1.6; extra == "full"
-Requires-Dist: urllib3<1.27,>=1.21.1; extra == "full"
-Requires-Dist: uvicorn[standard]>=0.16.0; extra == "full"
-Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "full"
-Requires-Dist: scikit-learn; python_version > "3.10" and extra == "full"
 Requires-Dist: litellm<1.86.0,>=1.83.7; extra == "full"
-Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "full"
-Requires-Dist: requests-toolbelt<=1.0.0; extra == "full"
-Requires-Dist: requests>=2.28.1; extra == "full"
-Requires-Dist: numpy>=1.15.0; extra == "full"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and extra == "full"
-Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "full"
-Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "full"
-Requires-Dist: docker>=5.0.3; extra == "full"
+Requires-Dist: scikit-learn; python_version > "3.10" and 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: google-cloud-bigquery-storage; extra == "full"
-Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; 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: werkzeug<4.0.0,>=2.0.0; extra == "full"
-Requires-Dist: immutabledict; extra == "full"
-Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "full"
-Requires-Dist: pyyaml<7,>=5.3.1; extra == "full"
-Requires-Dist: scikit-learn<1.6.0; python_version <= "3.10" and extra == "full"
-Requires-Dist: jsonschema; extra == "full"
+Requires-Dist: tqdm>=4.23.0; 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: starlette>=0.17.1; extra == "full"
 Provides-Extra: metadata
 Requires-Dist: pandas>=1.0.0; extra == "metadata"
 Requires-Dist: numpy>=1.15.0; extra == "metadata"
@@ -84,43 +84,43 @@ 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: tqdm>=4.23.0; extra == "testing"
-Requires-Dist: pyyaml; extra == "testing"
-Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "testing"
+Requires-Dist: fastapi<=0.124.4,>=0.71.0; extra == "testing"
+Requires-Dist: ruamel.yaml; extra == "testing"
+Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "testing"
+Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "testing"
 Requires-Dist: tensorflow<3.0.0,>=2.3.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.11" and extra == "testing"
 Requires-Dist: google-cloud-bigquery; extra == "testing"
-Requires-Dist: ruamel.yaml; extra == "testing"
-Requires-Dist: starlette>=0.17.1; extra == "testing"
-Requires-Dist: pyarrow<8.0.0,>=3.0.0; python_version < "3.10" and extra == "testing"
+Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "testing"
+Requires-Dist: werkzeug<4.0.0,>=2.0.0; extra == "testing"
+Requires-Dist: immutabledict; extra == "testing"
+Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "testing"
+Requires-Dist: jsonschema; extra == "testing"
+Requires-Dist: pyyaml<7,>=5.3.1; extra == "testing"
+Requires-Dist: docker>=5.0.3; extra == "testing"
+Requires-Dist: scikit-learn<1.6.0; python_version <= "3.10" 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: mlflow>=1.27.0; python_version >= "3.13" and extra == "testing"
+Requires-Dist: google-cloud-bigquery-storage; extra == "testing"
+Requires-Dist: pyarrow>=6.0.1; extra == "testing"
+Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and 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: uvicorn[standard]>=0.16.0; extra == "testing"
 Requires-Dist: pandas>=1.0.0; extra == "testing"
-Requires-Dist: pyarrow>=6.0.1; extra == "testing"
+Requires-Dist: pyyaml; extra == "testing"
+Requires-Dist: requests>=2.28.1; extra == "testing"
+Requires-Dist: requests-toolbelt<=1.0.0; extra == "testing"
+Requires-Dist: httpx<=0.28.1,>=0.23.0; extra == "testing"
 Requires-Dist: google-vizier>=0.1.6; extra == "testing"
-Requires-Dist: urllib3<1.27,>=1.21.1; extra == "testing"
-Requires-Dist: uvicorn[standard]>=0.16.0; extra == "testing"
-Requires-Dist: lit-nlp==0.4.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: scikit-learn; python_version > "3.10" and extra == "testing"
 Requires-Dist: litellm<1.86.0,>=1.83.7; extra == "testing"
-Requires-Dist: pyarrow>=14.0.0; python_version >= "3.12" and extra == "testing"
-Requires-Dist: requests-toolbelt<=1.0.0; extra == "testing"
-Requires-Dist: requests>=2.28.1; extra == "testing"
-Requires-Dist: numpy>=1.15.0; extra == "testing"
-Requires-Dist: pyarrow>=10.0.1; python_version == "3.10" and extra == "testing"
-Requires-Dist: mlflow>=1.27.0; python_version >= "3.13" and extra == "testing"
-Requires-Dist: ray[default]<=2.47.1,>=2.5; python_version == "3.11" and extra == "testing"
-Requires-Dist: docker>=5.0.3; extra == "testing"
+Requires-Dist: scikit-learn; python_version > "3.10" and 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: google-cloud-bigquery-storage; extra == "testing"
-Requires-Dist: tensorboard-plugin-profile<2.18.0,>=2.4.0; 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: werkzeug<4.0.0,>=2.0.0; extra == "testing"
-Requires-Dist: immutabledict; extra == "testing"
-Requires-Dist: mlflow<=2.16.0,>=1.27.0; python_version < "3.13" and extra == "testing"
-Requires-Dist: pyyaml<7,>=5.3.1; extra == "testing"
-Requires-Dist: scikit-learn<1.6.0; python_version <= "3.10" and extra == "testing"
-Requires-Dist: jsonschema; extra == "testing"
+Requires-Dist: tqdm>=4.23.0; 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: starlette>=0.17.1; 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"
@@ -208,19 +208,19 @@ Requires-Dist: xgboost_ray; extra == "ray-testing"
 Provides-Extra: adk
 Requires-Dist: google-adk<3.0.0,>=1.5.0; extra == "adk"
 Provides-Extra: adk-testing
-Requires-Dist: pytest-xdist; extra == "adk-testing"
-Requires-Dist: opentelemetry-sdk<2; extra == "adk-testing"
-Requires-Dist: aiohttp; extra == "adk-testing"
-Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "adk-testing"
 Requires-Dist: typing_extensions; extra == "adk-testing"
-Requires-Dist: google-adk<3.0.0,>=1.5.0; extra == "adk-testing"
-Requires-Dist: opentelemetry-exporter-gcp-trace<2; extra == "adk-testing"
 Requires-Dist: absl-py; extra == "adk-testing"
+Requires-Dist: opentelemetry-sdk<2; extra == "adk-testing"
+Requires-Dist: pytest-xdist; extra == "adk-testing"
 Requires-Dist: pydantic<3,>=2.11.1; extra == "adk-testing"
-Requires-Dist: google-cloud-trace<2; extra == "adk-testing"
-Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "adk-testing"
 Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "adk-testing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; 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: aiohttp; extra == "adk-testing"
+Requires-Dist: google-adk<3.0.0,>=1.5.0; extra == "adk-testing"
+Requires-Dist: opentelemetry-exporter-gcp-trace<2; extra == "adk-testing"
+Requires-Dist: google-cloud-trace<2; extra == "adk-testing"
 Provides-Extra: reasoningengine
 Requires-Dist: cloudpickle<4.0,>=3.0; extra == "reasoningengine"
 Requires-Dist: google-cloud-trace<2; extra == "reasoningengine"
@@ -262,64 +262,64 @@ 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: pytest-xdist; extra == "langchain-testing"
+Requires-Dist: typing_extensions; extra == "langchain-testing"
+Requires-Dist: absl-py; extra == "langchain-testing"
 Requires-Dist: opentelemetry-sdk<2; extra == "langchain-testing"
-Requires-Dist: aiohttp; extra == "langchain-testing"
+Requires-Dist: langchain-core<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: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "langchain-testing"
+Requires-Dist: langgraph<2.0.0,>=1.0.0; extra == "langchain-testing"
 Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "langchain-testing"
-Requires-Dist: typing_extensions; extra == "langchain-testing"
+Requires-Dist: cloudpickle<4.0,>=3.0; extra == "langchain-testing"
 Requires-Dist: langchain-classic; extra == "langchain-testing"
+Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "langchain-testing"
+Requires-Dist: aiohttp; extra == "langchain-testing"
 Requires-Dist: langchain-google-genai<4.2.3,>=4.0.0; extra == "langchain-testing"
 Requires-Dist: openinference-instrumentation-langchain<0.2,>=0.1.19; extra == "langchain-testing"
-Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "langchain-testing"
 Requires-Dist: opentelemetry-exporter-gcp-trace<2; 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: pydantic<3,>=2.11.1; extra == "langchain-testing"
-Requires-Dist: langchain<2.0.0,>=1.0.0; extra == "langchain-testing"
 Requires-Dist: google-cloud-trace<2; extra == "langchain-testing"
-Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "langchain-testing"
-Requires-Dist: cloudpickle<4.0,>=3.0; 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"
 Provides-Extra: ag2
 Requires-Dist: ag2[gemini]; extra == "ag2"
 Requires-Dist: openinference-instrumentation-autogen<0.2,>=0.1.6; extra == "ag2"
 Provides-Extra: ag2-testing
-Requires-Dist: pytest-xdist; extra == "ag2-testing"
-Requires-Dist: opentelemetry-sdk<2; extra == "ag2-testing"
-Requires-Dist: aiohttp; extra == "ag2-testing"
-Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "ag2-testing"
 Requires-Dist: typing_extensions; extra == "ag2-testing"
-Requires-Dist: ag2[gemini]; extra == "ag2-testing"
-Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "ag2-testing"
-Requires-Dist: opentelemetry-exporter-gcp-trace<2; extra == "ag2-testing"
 Requires-Dist: absl-py; extra == "ag2-testing"
+Requires-Dist: opentelemetry-sdk<2; extra == "ag2-testing"
+Requires-Dist: pytest-xdist; extra == "ag2-testing"
 Requires-Dist: pydantic<3,>=2.11.1; extra == "ag2-testing"
-Requires-Dist: google-cloud-trace<2; extra == "ag2-testing"
 Requires-Dist: openinference-instrumentation-autogen<0.2,>=0.1.6; extra == "ag2-testing"
-Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "ag2-testing"
+Requires-Dist: opentelemetry-instrumentation-google-genai<1.0.0,>=0.3b0; extra == "ag2-testing"
+Requires-Dist: ag2[gemini]; 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: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "ag2-testing"
+Requires-Dist: aiohttp; extra == "ag2-testing"
+Requires-Dist: opentelemetry-exporter-gcp-trace<2; extra == "ag2-testing"
+Requires-Dist: google-cloud-trace<2; 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: pytest-xdist; extra == "llama-index-testing"
-Requires-Dist: opentelemetry-sdk<2; extra == "llama-index-testing"
-Requires-Dist: aiohttp; extra == "llama-index-testing"
-Requires-Dist: opentelemetry-exporter-otlp-proto-http<2; extra == "llama-index-testing"
 Requires-Dist: typing_extensions; extra == "llama-index-testing"
+Requires-Dist: absl-py; extra == "llama-index-testing"
 Requires-Dist: openinference-instrumentation-llama-index<4.0,>=3.0; extra == "llama-index-testing"
+Requires-Dist: llama-index; 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: pydantic<3,>=2.11.1; 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: opentelemetry-exporter-otlp-proto-http<2; 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: aiohttp; extra == "llama-index-testing"
 Requires-Dist: opentelemetry-exporter-gcp-trace<2; 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-llms-google-genai; extra == "llama-index-testing"
 Requires-Dist: google-cloud-trace<2; extra == "llama-index-testing"
-Requires-Dist: llama-index; extra == "llama-index-testing"
-Requires-Dist: opentelemetry-exporter-gcp-logging<2.0.0,>=1.11.0a0; extra == "llama-index-testing"
-Requires-Dist: cloudpickle<4.0,>=3.0; extra == "llama-index-testing"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{google_cloud_aiplatform-1.158.0 → google_cloud_aiplatform-2.0.0}/agentplatform/_genai/_datasets_utils.py

@@ -280,6 +280,63 @@ async def save_dataframe_to_bigquery_async(
     await asyncio.to_thread(bq_client.delete_table, temp_table_id)
 
 
+def load_dataframe_from_bigquery(
+    *,
+    bigquery_uri: str,
+    project: str,
+    location: str,
+    credentials: google.auth.credentials.Credentials,
+) -> "bigframes.pandas.DataFrame":  # type: ignore # noqa: F821
+    """Loads a BigQuery table into a BigFrames DataFrame.
+
+    Args:
+      bigquery_uri: The URI of the BigQuery table, with or without the `bq://`
+        prefix.
+      project: The project to use for the BigFrames session.
+      location: The location to use for the BigFrames session.
+      credentials: The credentials to use for the BigFrames session.
+
+    Returns:
+      A BigFrames DataFrame backed by the BigQuery table.
+    """
+    bigframes = _try_import_bigframes()
+    session_options = bigframes.BigQueryOptions(
+        credentials=credentials,
+        project=project,
+        location=location,
+    )
+    with bigframes.connect(session_options) as session:
+        return session.read_gbq(bigquery_uri.removeprefix("bq://"))
+
+
+async def load_dataframe_from_bigquery_async(
+    *,
+    bigquery_uri: str,
+    project: str,
+    location: str,
+    credentials: google.auth.credentials.Credentials,
+) -> "bigframes.pandas.DataFrame":  # type: ignore # noqa: F821
+    """Loads a BigQuery table into a BigFrames DataFrame.
+
+    Args:
+      bigquery_uri: The URI of the BigQuery table, with or without the `bq://`
+        prefix.
+      project: The project to use for the BigFrames session.
+      location: The location to use for the BigFrames session.
+      credentials: The credentials to use for the BigFrames session.
+
+    Returns:
+      A BigFrames DataFrame backed by the BigQuery table.
+    """
+    return await asyncio.to_thread(
+        load_dataframe_from_bigquery,
+        bigquery_uri=bigquery_uri,
+        project=project,
+        location=location,
+        credentials=credentials,
+    )
+
+
 def resolve_dataset_name(resource_name_or_id: str, project: str, location: str) -> str:
     """Resolves a dataset name or ID to a full resource name."""
     if "/" not in resource_name_or_id:

{google_cloud_aiplatform-1.158.0 → google_cloud_aiplatform-2.0.0}/agentplatform/_genai/client.py

@@ -39,6 +39,9 @@ if TYPE_CHECKING:
     )
     from agentplatform._genai import prompts as prompts_module
     from agentplatform._genai import skills as skills_module
+    from agentplatform._genai import (
+        model_garden as model_garden_module,
+    )
     from agentplatform._genai import live as live_module
     from agentplatform._genai import rag as rag_module
 
@@ -85,6 +88,7 @@ class AsyncClient:
         self._datasets: Optional[ModuleType] = None
         self._skills: Optional[ModuleType] = None
         self._rag: Optional[ModuleType] = None
+        self._model_garden: Optional[ModuleType] = None
 
     @property
     @_common.experimental_warning(
@@ -179,6 +183,18 @@ class AsyncClient:
             )
         return self._rag.AsyncRag(self._api_client)  # type: ignore[no-any-return]
 
+    @property
+    @_common.experimental_warning(
+        "The Model Garden module is experimental, and may change in future " "versions."
+    )
+    def model_garden(self) -> "model_garden_module.AsyncModelGarden":
+        if self._model_garden is None:
+            self._model_garden = importlib.import_module(
+                ".model_garden",
+                __package__,
+            )
+        return self._model_garden.AsyncModelGarden(self._api_client)  # type: ignore[no-any-return]
+
     async def aclose(self) -> None:
         """Closes the async client explicitly.
 
@@ -284,6 +300,7 @@ class Client:
         self._datasets: Optional[ModuleType] = None
         self._skills: Optional[ModuleType] = None
         self._rag: Optional[ModuleType] = None
+        self._model_garden: Optional[ModuleType] = None
 
     @property
     def evals(self) -> "evals_module.Evals":
@@ -402,3 +419,15 @@ class Client:
                 __package__,
             )
         return self._rag.Rag(self._api_client)  # type: ignore[no-any-return]
+
+    @property
+    @_common.experimental_warning(
+        "The Model Garden module is experimental, and may change in future " "versions."
+    )
+    def model_garden(self) -> "model_garden_module.ModelGarden":
+        if self._model_garden is None:
+            self._model_garden = importlib.import_module(
+                ".model_garden",
+                __package__,
+            )
+        return self._model_garden.ModelGarden(self._api_client)  # type: ignore[no-any-return]

{google_cloud_aiplatform-1.158.0 → google_cloud_aiplatform-2.0.0}/agentplatform/_genai/datasets.py

@@ -1324,8 +1324,9 @@ class Datasets(_api_module.BaseModule):
         gemini_request_read_config: Optional[
             types.GeminiRequestReadConfigOrDict
         ] = None,
+        load_dataframe: bool = False,
         config: Optional[types.AssembleDatasetConfigOrDict] = None,
-    ) -> str:
+    ) -> tuple[str, Optional["bigframes.pandas.DataFrame"]]:  # type: ignore # noqa: F821
         """Assemble the dataset into a BigQuery table.
 
         Waits for the assemble operation to complete before returning.
@@ -1338,12 +1339,20 @@ class Datasets(_api_module.BaseModule):
             Optional. The read config to use to assemble the dataset. If
             not provided, the read config attached to the dataset will be
             used.
+          load_dataframe:
+            Optional. Whether to load the assembled BigQuery table into a
+            BigFrames DataFrame and return it. If False, the returned
+            DataFrame is None, no BigQuery read is performed, and `bigframes`
+            is not required. Defaults to False.
           config:
             Optional. A configuration for assembling the dataset. If not
             provided, the default configuration will be used.
 
         Returns:
-            The URI of the bigquery table of the assembled dataset.
+            A tuple `(table_id, dataframe)`, where `table_id` is the BigQuery
+            table id of the assembled dataset (without the `bq://` prefix) and
+            `dataframe` is the assembled table loaded as a BigFrames DataFrame.
+            `dataframe` is None if `load_dataframe` is False.
         """
         if isinstance(config, dict):
             config = types.AssembleDatasetConfig(**config)
@@ -1363,7 +1372,17 @@ class Datasets(_api_module.BaseModule):
             operation=operation,
             timeout_seconds=config.timeout,
         )
-        return response["bigqueryDestination"]  # type: ignore[no-any-return]
+        bigquery_uri = response["bigqueryDestination"]
+        table_id = bigquery_uri.removeprefix("bq://")
+        dataframe = None
+        if load_dataframe:
+            dataframe = _datasets_utils.load_dataframe_from_bigquery(
+                bigquery_uri=bigquery_uri,
+                project=self._api_client.project,
+                location=self._api_client.location,
+                credentials=self._api_client._credentials,
+            )
+        return (table_id, dataframe)
 
     def assess_tuning_resources(
         self,
@@ -2713,8 +2732,9 @@ class AsyncDatasets(_api_module.BaseModule):
         gemini_request_read_config: Optional[
             types.GeminiRequestReadConfigOrDict
         ] = None,
+        load_dataframe: bool = False,
         config: Optional[types.AssembleDatasetConfigOrDict] = None,
-    ) -> str:
+    ) -> tuple[str, Optional["bigframes.pandas.DataFrame"]]:  # type: ignore # noqa: F821
         """Assemble the dataset into a BigQuery table.
 
         Waits for the assemble operation to complete before returning.
@@ -2727,12 +2747,20 @@ class AsyncDatasets(_api_module.BaseModule):
             Optional. The read config to use to assemble the dataset. If
             not provided, the read config attached to the dataset will be
             used.
+          load_dataframe:
+            Optional. Whether to load the assembled BigQuery table into a
+            BigFrames DataFrame and return it. If False, the returned
+            DataFrame is None, no BigQuery read is performed, and `bigframes`
+            is not required. Defaults to False.
           config:
             Optional. A configuration for assembling the dataset. If not
             provided, the default configuration will be used.
 
         Returns:
-            The URI of the bigquery table of the assembled dataset.
+            A tuple `(table_id, dataframe)`, where `table_id` is the BigQuery
+            table id of the assembled dataset (without the `bq://` prefix) and
+            `dataframe` is the assembled table loaded as a BigFrames DataFrame.
+            `dataframe` is None if `load_dataframe` is False.
         """
         if isinstance(config, dict):
             config = types.AssembleDatasetConfig(**config)
@@ -2752,7 +2780,17 @@ class AsyncDatasets(_api_module.BaseModule):
             operation=operation,
             timeout_seconds=config.timeout,
         )
-        return response["bigqueryDestination"]  # type: ignore[no-any-return]
+        bigquery_uri = response["bigqueryDestination"]
+        table_id = bigquery_uri.removeprefix("bq://")
+        dataframe = None
+        if load_dataframe:
+            dataframe = await _datasets_utils.load_dataframe_from_bigquery_async(
+                bigquery_uri=bigquery_uri,
+                project=self._api_client.project,
+                location=self._api_client.location,
+                credentials=self._api_client._credentials,
+            )
+        return (table_id, dataframe)
 
     async def assess_tuning_resources(
         self,