From a0c951336e22a6ee14036c3e51527e2b1723f9ed Mon Sep 17 00:00:00 2001 From: Mohammad Date: Wed, 10 Sep 2025 14:20:42 +0200 Subject: [PATCH] feat: add support for AWS session token in S3 configuration --- cognee/infrastructure/files/storage/S3FileStorage.py | 6 +++++- cognee/tasks/ingestion/resolve_data_directories.py | 5 ++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/cognee/infrastructure/files/storage/S3FileStorage.py b/cognee/infrastructure/files/storage/S3FileStorage.py index a0d611241..078d5fe2a 100644 --- a/cognee/infrastructure/files/storage/S3FileStorage.py +++ b/cognee/infrastructure/files/storage/S3FileStorage.py @@ -21,7 +21,11 @@ class S3FileStorage(Storage): def __init__(self, storage_path: str): self.storage_path = storage_path s3_config = get_s3_config() - if s3_config.aws_access_key_id is not None and s3_config.aws_secret_access_key is not None and s3_config.aws_session_token is not None: + if ( + s3_config.aws_access_key_id is not None + and s3_config.aws_secret_access_key is not None + and s3_config.aws_session_token is not None + ): self.s3 = s3fs.S3FileSystem( key=s3_config.aws_access_key_id, secret=s3_config.aws_secret_access_key, diff --git a/cognee/tasks/ingestion/resolve_data_directories.py b/cognee/tasks/ingestion/resolve_data_directories.py index cbd979e16..aa2f95303 100644 --- a/cognee/tasks/ingestion/resolve_data_directories.py +++ b/cognee/tasks/ingestion/resolve_data_directories.py @@ -32,7 +32,10 @@ async def resolve_data_directories( import s3fs fs = s3fs.S3FileSystem( - key=s3_config.aws_access_key_id, secret=s3_config.aws_secret_access_key,token=s3_config.aws_session_token, anon=False + key=s3_config.aws_access_key_id, + secret=s3_config.aws_secret_access_key, + token=s3_config.aws_session_token, + anon=False, ) for item in data: