NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-12-19 11:52:19 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix bigquery credentials and bigquery tool to make it compatible with python 3.9 and make the credential serializable in session

Browse Source 
PiperOrigin-RevId: 763332829
This commit is contained in:
Xiang (Sean) Zhou
2025-05-26 01:57:40 -07:00
committed by Copybara-Service
parent 55cb36edfe
commit 694eca08e5
5 changed files with 233 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unittests/tools/bigquery/test_bigquery_tool.py
View File

@@ -17,7 +17,7 @@ from unittest.mock import Mock
from unittest.mock import patch
from google.adk.tools import ToolContext
from google.adk.tools.bigquery.bigquery_credentials import BigQueryCredentials
from google.adk.tools.bigquery.bigquery_credentials import BigQueryCredentialsConfig
from google.adk.tools.bigquery.bigquery_credentials import BigQueryCredentialsManager
from google.adk.tools.bigquery.bigquery_tool import BigQueryTool
# Mock the Google OAuth and API dependencies
@@ -78,7 +78,7 @@ class TestBigQueryTool:
@pytest.fixture
def credentials_config(self):
"""Create credentials configuration for testing."""
return BigQueryCredentials(
return BigQueryCredentialsConfig(
client_id="test_client_id",
client_secret="test_client_secret",
scopes=["https://www.googleapis.com/auth/bigquery"],