diff --git a/packages/toolbox-core/integration.cloudbuild.yaml b/packages/toolbox-core/integration.cloudbuild.yaml index 89132be9..20a56ead 100644 --- a/packages/toolbox-core/integration.cloudbuild.yaml +++ b/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: 'latest' + _TOOLBOX_BUCKET: 'genai-toolbox-dev' diff --git a/packages/toolbox-core/tests/conftest.py b/packages/toolbox-core/tests/conftest.py index 14493ad2..b8cfa6e5 100644 --- a/packages/toolbox-core/tests/conftest.py +++ b/packages/toolbox-core/tests/conftest.py @@ -75,7 +75,7 @@ def get_toolbox_binary_url(toolbox_version: str) -> str: arch = ( "arm64" if os_system == "darwin" and platform.machine() == "arm64" else "amd64" ) - return f"v{toolbox_version}/{os_system}/{arch}/toolbox" + return f"{toolbox_version}/{os_system}/{arch}/toolbox" def get_auth_token(client_id: str) -> str: @@ -102,6 +102,11 @@ 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]: """Provides a temporary file path containing the tools manifest.""" @@ -130,11 +135,13 @@ 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...")