You've already forked adk-python
mirror of
https://github.com/encounter/adk-python.git
synced 2026-03-30 10:57:20 -07:00
Josh Soref
aa1233608a
Merge https://github.com/google/adk-python/pull/2447 This PR corrects misspellings identified by the [check-spelling action](https://github.com/marketplace/actions/check-spelling) The misspellings have been reported at https://github.com/jsoref/adk-python/actions/runs/16840838898/attempts/1#summary-47711379253 The action reports that the changes in this PR would make it happy: https://github.com/jsoref/adk-python/actions/runs/16840839269/attempts/1#summary-47711380479 Note: while I use tooling to identify errors, the tooling doesn't _actually_ provide the corrections, I'm picking them on my own. I'm a human, and I may make mistakes. I've included a couple of changes to make CI happy. Personally, I object to CI being in a state of "random drive by person who adds a blank line in the middle of a file must fix all the preexisting bugs in the file", but that appears to be the state for this repository. COPYBARA_INTEGRATE_REVIEW=https://github.com/google/adk-python/pull/2447 from jsoref:spelling d85398e7fd154d124d477c6af6181481a01f34e0 PiperOrigin-RevId: 827629615
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
# Copyright 2025 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import os
|
|
|
|
from dotenv import load_dotenv
|
|
from google.adk.agents.llm_agent import Agent
|
|
from google.adk.tools.google_api_tool.google_api_toolsets import BigQueryToolset
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
# Access the variable
|
|
oauth_client_id = os.getenv("OAUTH_CLIENT_ID")
|
|
oauth_client_secret = os.getenv("OAUTH_CLIENT_SECRET")
|
|
tools_to_expose = [
|
|
"bigquery_datasets_list",
|
|
"bigquery_datasets_get",
|
|
"bigquery_datasets_insert",
|
|
"bigquery_tables_list",
|
|
"bigquery_tables_get",
|
|
"bigquery_tables_insert",
|
|
]
|
|
bigquery_toolset = BigQueryToolset(
|
|
client_id=oauth_client_id,
|
|
client_secret=oauth_client_secret,
|
|
tool_filter=tools_to_expose,
|
|
)
|
|
|
|
root_agent = Agent(
|
|
model="gemini-2.0-flash",
|
|
name="google_api_bigquery_agent",
|
|
instruction="""
|
|
You are a helpful Google BigQuery agent that help to manage users' data on Google BigQuery.
|
|
Use the provided tools to conduct various operations on users' data in Google BigQuery.
|
|
|
|
Scenario 1:
|
|
The user wants to query their bigquery datasets
|
|
Use bigquery_datasets_list to query user's datasets
|
|
|
|
Scenario 2:
|
|
The user wants to query the details of a specific dataset
|
|
Use bigquery_datasets_get to get a dataset's details
|
|
|
|
Scenario 3:
|
|
The user wants to create a new dataset
|
|
Use bigquery_datasets_insert to create a new dataset
|
|
|
|
Scenario 4:
|
|
The user wants to query their tables in a specific dataset
|
|
Use bigquery_tables_list to list all tables in a dataset
|
|
|
|
Scenario 5:
|
|
The user wants to query the details of a specific table
|
|
Use bigquery_tables_get to get a table's details
|
|
|
|
Scenario 6:
|
|
The user wants to insert a new table into a dataset
|
|
Use bigquery_tables_insert to insert a new table into a dataset
|
|
|
|
Current user:
|
|
<User>
|
|
{userInfo?}
|
|
</User>
|
|
""",
|
|
tools=[bigquery_toolset],
|
|
)
|