nebu 0.1.45__tar.gz → 0.1.46__tar.gz

{nebu-0.1.45/src/nebu.egg-info → nebu-0.1.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nebu
-Version: 0.1.45
+Version: 0.1.46
 Summary: A globally distributed container runtime
 Requires-Python: >=3.10.14
 Description-Content-Type: text/markdown

{nebu-0.1.45 → nebu-0.1.46}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nebu"
-version = "0.1.45"
+version = "0.1.46"
 description = "A globally distributed container runtime"
 readme = "README.md"
 requires-python = ">=3.10.14"

{nebu-0.1.45 → nebu-0.1.46}/src/nebu/__init__.py

@@ -3,7 +3,6 @@
 # ruff: noqa: F401
 # ruff: noqa: F403
 
-from .adapter import *
 from .auth import is_allowed
 from .cache import Cache, OwnedValue
 from .chatx.convert import *

{nebu-0.1.45 → nebu-0.1.46}/src/nebu/data.py

@@ -116,14 +116,35 @@ def _parse_s3_path(path: str) -> Tuple[Optional[str], Optional[str]]:
 class Bucket:
     """Handles interactions with AWS S3."""
 
-    def __init__(self, verbose: bool = True):
+    def __init__(
+        self,
+        verbose: bool = True,
+        aws_access_key_id: Optional[str] = None,
+        aws_secret_access_key: Optional[str] = None,
+        aws_session_token: Optional[str] = None,
+    ):
         """
-        Initializes the S3 handler.
+        Initializes the S3 handler. Can use default credentials or provided temporary ones.
 
         Args:
             verbose (bool): If True, prints status messages. Defaults to True.
+            aws_access_key_id (Optional[str]): Temporary AWS Access Key ID.
+            aws_secret_access_key (Optional[str]): Temporary AWS Secret Access Key.
+            aws_session_token (Optional[str]): Temporary AWS Session Token (required if keys are temporary).
         """
-        self.client = boto3.client("s3")
+        if aws_access_key_id and aws_secret_access_key:
+            if verbose:
+                print("Initializing S3 client with provided temporary credentials.")
+            self.client = boto3.client(
+                "s3",
+                aws_access_key_id=aws_access_key_id,
+                aws_secret_access_key=aws_secret_access_key,
+                aws_session_token=aws_session_token,  # Pass session token if provided
+            )
+        else:
+            if verbose:
+                print("Initializing S3 client with default credentials.")
+            self.client = boto3.client("s3")
         self.verbose = verbose
 
     def _parse_path(self, path: str) -> Tuple[Optional[str], Optional[str]]: