feat(s3): add support for region_name via client_kwargs in S3Reader (#19530) (#19546)

Author: fahadAbulSudo

llama-index-integrations/readers/llama-index-readers-s3/llama_index/readers/s3/base.py

@@ -53,6 +53,8 @@ class S3Reader(BasePydanticReader, ResourcesReaderMixin, FileSystemReaderMixin):
         Default is None.
     aws_access_id (Optional[str]): provide AWS access key directly.
     aws_access_secret (Optional[str]): provide AWS access key directly.
+    region_name (Optional[str]): AWS region for the S3 bucket. If not provided,
+    the default environment region or AWS config will be used.
     s3_endpoint_url (Optional[str]): provide S3 endpoint URL directly.
 
     """
@@ -73,6 +75,7 @@ class S3Reader(BasePydanticReader, ResourcesReaderMixin, FileSystemReaderMixin):
     aws_access_id: Optional[str] = None
     aws_access_secret: Optional[str] = None
     aws_session_token: Optional[str] = None
+    region_name: Optional[str] = None
     s3_endpoint_url: Optional[str] = None
     custom_reader_path: Optional[str] = None
     invalidate_s3fs_cache: bool = True
@@ -84,11 +87,16 @@ def class_name(cls) -> str:
     def _get_s3fs(self):
         from s3fs import S3FileSystem
 
+        client_kwargs = {}
+        if isinstance(self.region_name, str) and self.region_name.strip():
+            client_kwargs["region_name"] = self.region_name.strip()
+
         s3fs = S3FileSystem(
             key=self.aws_access_id,
             endpoint_url=self.s3_endpoint_url,
             secret=self.aws_access_secret,
             token=self.aws_session_token,
+            client_kwargs=client_kwargs or None,
         )
         if self.invalidate_s3fs_cache:
             s3fs.invalidate_cache()

llama-index-integrations/readers/llama-index-readers-s3/pyproject.toml

@@ -27,7 +27,7 @@ dev = [
 
 [project]
 name = "llama-index-readers-s3"
-version = "0.4.0"
+version = "0.4.1"
 description = "llama-index readers s3 integration"
 authors = [{name = "Your Name", email = "[email protected]"}]
 requires-python = ">=3.9,<4.0"