From b7a08a7a8961e43a88deb7fe24549d08abf3794b 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 --- sdks/langchain/tests/conftest.py | 28 +++++++++++++++++++--------- 1 file changed, 19 insertions(+), 9 deletions(-) diff --git a/sdks/langchain/tests/conftest.py b/sdks/langchain/tests/conftest.py index 079883d91..442b95e1e 100644 --- a/sdks/langchain/tests/conftest.py +++ b/sdks/langchain/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 2021fa760bc17168e495d79c54b78810035b55ba 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 --- sdks/langchain/tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdks/langchain/tests/conftest.py b/sdks/langchain/tests/conftest.py index 442b95e1e..8abc4f820 100644 --- a/sdks/langchain/tests/conftest.py +++ b/sdks/langchain/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 fe230940357e942d7b31548ad0dd46d806c4d5cb 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 --- sdks/langchain/tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdks/langchain/tests/conftest.py b/sdks/langchain/tests/conftest.py index 8abc4f820..442b95e1e 100644 --- a/sdks/langchain/tests/conftest.py +++ b/sdks/langchain/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.")