From 655f200822e01e64359d80690347ebe4424195c2 Mon Sep 17 00:00:00 2001 From: Twisha Bansal Date: Fri, 17 Jan 2025 12:45:07 +0530 Subject: [PATCH 1/3] tests(langchain-sdk): Update conftest to be able to run tests in local --- tests/conftest.py | 28 +++++++++++++++++++--------- 1 file changed, 19 insertions(+), 9 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 76d1ab5c..e4036ab6 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -80,15 +80,25 @@ def get_toolbox_binary_url(toolbox_version: str) -> str: def get_auth_token(client_id: str) -> str: """Retrieves an authentication token""" - request = google.auth.transport.requests.Request() - credentials = compute_engine.IDTokenCredentials( - request=request, - target_audience=client_id, - use_metadata_identity_endpoint=True, - ) - if not credentials.valid: - credentials.refresh(request) - return credentials.token + try: + # Try getting the token using gcloud (for local development) + result = subprocess.run( + ["gcloud", "auth", "print-identity-token"], + capture_output=True, + text=True, + check=True, + ) + return result.stdout.strip() + except (subprocess.CalledProcessError, FileNotFoundError): + request = google.auth.transport.requests.Request() + credentials = compute_engine.IDTokenCredentials( + request=request, + target_audience=client_id, + use_metadata_identity_endpoint=True, + ) + if not credentials.valid: + credentials.refresh(request) + return credentials.token #### Define Fixtures From 03dbfe21ddd9a0d6881f6bee75560e468e0a0060 Mon Sep 17 00:00:00 2001 From: Twisha Bansal Date: Fri, 17 Jan 2025 12:53:25 +0530 Subject: [PATCH 2/3] Increase sleep time for local runs --- tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index e4036ab6..8adbde4e 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -158,7 +158,7 @@ def toolbox_server(toolbox_version: str, tools_file_path: str) -> Generator[None # Wait for server to start # Retry logic with a timeout for _ in range(5): # retries - time.sleep(2) + time.sleep(4) print("Checking if toolbox is successfully started...") if toolbox_server.poll() is None: print("Toolbox server started successfully.") From d7d17bc982719bda26f33993958865586b953e8b Mon Sep 17 00:00:00 2001 From: Twisha Bansal Date: Fri, 17 Jan 2025 12:57:19 +0530 Subject: [PATCH 3/3] decrease back sleep time for local runs. Local sleep time highly depends on the system --- tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index 8adbde4e..e4036ab6 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -158,7 +158,7 @@ def toolbox_server(toolbox_version: str, tools_file_path: str) -> Generator[None # Wait for server to start # Retry logic with a timeout for _ in range(5): # retries - time.sleep(4) + time.sleep(2) print("Checking if toolbox is successfully started...") if toolbox_server.poll() is None: print("Toolbox server started successfully.")