Skip to content

Commit 29dace0

Browse files
committed
add aws-vault + fix method call
1 parent 7603fa3 commit 29dace0

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

mcp-server.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@ export VAULT_ADDR=https://vault.us1.staging.dog
22
vault login -method=oidc
33
export VAULT_TOKEN=$(vault print token)
44

5-
uv run -m spark_history_mcp.core.main
5+
aws-vault exec sso-staging-engineering -- uv run -m spark_history_mcp.core.main

src/spark_history_mcp/common/s3_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def is_spark_event_logs_already_indexed(self, spark_app_id: str) -> bool:
2525
def copy_spark_events_logs(self, spark_app_id: str):
2626
# get spark events logs file to copy/index
2727
src_prefix = f"spark_logs/{spark_app_id}"
28-
base_logs = self.client.list_contents_by_prefix(src_prefix, self.bucket_name)
28+
base_logs = self.list_contents_by_prefix(src_prefix, self.bucket_name)
2929
if not base_logs:
3030
raise Exception(f"Logs for {spark_app_id} not found. Is the job older than one month?", 404)
3131

0 commit comments

Comments
 (0)