3.7 KiB
GCS Tools Sample
Introduction
This sample agent demonstrates the Google Cloud Storage (GCS) first-party tools in ADK,
distributed via the google.adk.integrations.gcs module. These tools include:
gcs_get_bucket
Get metadata information about a GCS bucket.
gcs_list_objects
List object names in a GCS bucket.
gcs_get_object_metadata
Get metadata information about a GCS object (blob).
How to use
Set up environment variables in your .env file for using
Google AI Studio
or
Google Cloud Vertex AI
for the LLM service for your agent. For example, for using Google AI Studio you
would set:
- GOOGLE_GENAI_USE_ENTERPRISE=FALSE
- GOOGLE_API_KEY={your api key}
With Application Default Credentials (gcloud)
This is the easiest way to use your own Google Cloud identity for both the tools AND the LLM.
- Install the Google Cloud CLI.
- Run
gcloud auth application-default loginin your terminal. - Configure your environment to use Vertex AI (which supports ADC) instead of AI Studio:
export GOOGLE_GENAI_USE_ENTERPRISE=TRUEexport GOOGLE_CLOUD_PROJECT={your-project-id}
- Ensure the Vertex AI API is enabled and you have the correct permissions:
- Enable API:
gcloud services enable aiplatform.googleapis.com - Grant Role:
gcloud projects add-iam-policy-binding {your-project-id} --member="user:{your-email}" --role="roles/aiplatform.user"
- Enable API:
- Set
CREDENTIALS_TYPE = Noneinagent.py. - Run the agent.
With Service Account Keys
This mode is useful for quick development when the agent builder wants to run the agent with service account credentials. The tools are run with these credentials.
-
Create service account key by following https://cloud.google.com/iam/docs/service-account-creds#user-managed-keys.
-
Set
CREDENTIALS_TYPE=AuthCredentialTypes.SERVICE_ACCOUNTinagent.py -
Download the key file and replace
"service_account_key.json"with the path -
Run the agent
With Interactive OAuth
-
Follow https://developers.google.com/identity/protocols/oauth2#1.-obtain-oauth-2.0-credentials-from-the-dynamic_data.setvar.console_name. to get your client id and client secret. Be sure to choose "web" as your client type.
-
Follow https://developers.google.com/workspace/guides/configure-oauth-consent to add scope "https://www.googleapis.com/auth/cloud-platform" and "https://www.googleapis.com/auth/devstorage.full_control" as a declaration, this is used for review purpose.
-
Follow https://developers.google.com/identity/protocols/oauth2/web-server#creatingcred to add http://localhost/dev-ui/ to "Authorized redirect URIs".
Note: localhost here is just a hostname that you use to access the dev ui, replace it with the actual hostname you use to access the dev ui.
-
For 1st run, allow popup for localhost in Chrome.
-
Configure your
.envfile to add two more variables before running the agent:- OAUTH_CLIENT_ID={your client id}
- OAUTH_CLIENT_SECRET={your client secret}
Note: don't create a separate .env, instead put it to the same .env file that stores your Vertex AI or Dev ML credentials
-
Set
CREDENTIALS_TYPE=AuthCredentialTypes.OAUTH2inagent.pyand run the agent
Sample prompts
- Show me metadata for the my-bucket bucket.
- List all objects in the my-bucket bucket.
- Get metadata for the my-object.txt object in my-bucket.
- Download the GCS object my-object.txt in my-bucket to a local file ~/Downloads/downloaded.txt.
- Upload my local file /tmp/local_report.pdf to my-bucket as report.pdf.