-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Closed
Labels
bot triaged[Bot] This issue is triaged by ADK bot[Bot] This issue is triaged by ADK botquestion[Component] This issue is asking a question or clarification[Component] This issue is asking a question or clarification
Description
Hi,
I’m building an agent that queries a BigQuery table using the following function:
def run_bigquery_sql(sql_query: str) -> str:
client = bigquery.Client()
query_job = client.query(sql_query)
results = query_job.result()
return json.dumps(results, indent=2, default=str)
This works fine locally. However, when deployed to Vertex AI, the function appears to run (i.e., no errors), but no data is returned. The model’s response confirms the function is invoked, but there’s no result.
Example call
User prompt:
Show me the type of partners and their count?
Model function call:
{
"function_call": {
"name": "run_bigquery_sql",
"args": {
"sql_query": "SELECT partner_type__c, COUNT(*) FROM `modified-talon-451917-8.ff_df_sf_partner.Partner_c_mini` GROUP BY partner_type__c"
}
}
}
Suspected Cause
Likely an authentication issue with the BigQuery client when running in Vertex AI (possibly missing service account permissions or not properly configured auth context).
Question
What is the correct way to authenticate a BigQuery client in a Vertex AI function/tool call so that it returns data as expected?
Metadata
Metadata
Assignees
Labels
bot triaged[Bot] This issue is triaged by ADK bot[Bot] This issue is triaged by ADK botquestion[Component] This issue is asking a question or clarification[Component] This issue is asking a question or clarification