chore: Test integration test with main branch

Author: anubhav756

packages/toolbox-core/integration.cloudbuild.yaml

@@ -33,6 +33,7 @@ steps:
     env:
       - TOOLBOX_URL=$_TOOLBOX_URL
       - TOOLBOX_VERSION=$_TOOLBOX_VERSION
+      - TOOLBOX_BUCKET=$_TOOLBOX_BUCKET
       - GOOGLE_CLOUD_PROJECT=$PROJECT_ID
     args:
       - '-c'
@@ -43,4 +44,5 @@ options:
   logging: CLOUD_LOGGING_ONLY
 substitutions:
   _VERSION: '3.13'
-  _TOOLBOX_VERSION: '0.8.0'
+  _TOOLBOX_VERSION: 'main'
+  _TOOLBOX_BUCKET: 'genai-toolbox-dev'

packages/toolbox-core/tests/conftest.py

@@ -101,6 +101,9 @@ def project_id() -> str:
 def toolbox_version() -> str:
     return get_env_var("TOOLBOX_VERSION")
 
+@pytest_asyncio.fixture(scope="session")
+def toolbox_bucket() -> str:
+    return get_env_var("TOOLBOX_BUCKET")
 
 @pytest_asyncio.fixture(scope="session")
 def tools_file_path(project_id: str) -> Generator[str]:
@@ -130,11 +133,11 @@ def auth_token2(project_id: str) -> str:
 
 
 @pytest_asyncio.fixture(scope="session")
-def toolbox_server(toolbox_version: str, tools_file_path: str) -> Generator[None]:
+def toolbox_server(toolbox_version: str, toolbox_bucket: str, tools_file_path: str) -> Generator[None]:
     """Starts the toolbox server as a subprocess."""
     print("Downloading toolbox binary from gcs bucket...")
     source_blob_name = get_toolbox_binary_url(toolbox_version)
-    download_blob("genai-toolbox", source_blob_name, "toolbox")
+    download_blob(toolbox_bucket, source_blob_name, "toolbox")
     print("Toolbox binary downloaded successfully.")
     try:
         print("Opening toolbox server process...")