aignostics 0.2.36__tar.gz → 0.2.37__tar.gz

{aignostics-0.2.36 → aignostics-0.2.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aignostics
-Version: 0.2.36
+Version: 0.2.37
 Summary: 🔬 Python SDK providing access to the Aignostics Platform. Includes Aignostics Launchpad (Desktop Application), Aignostics CLI (Command-Line Interface), example notebooks, and Aignostics Client Library.
 Project-URL: Homepage, https://aignostics.readthedocs.io/en/latest/
 Project-URL: Documentation, https://aignostics.readthedocs.io/en/latest/

{aignostics-0.2.36 → aignostics-0.2.37}/examples/notebook.py

@@ -2,7 +2,7 @@
 # requires-python = ">=3.13"
 # dependencies = [
 #     "marimo",
-#     "aignostics==0.2.36",
+#     "aignostics==0.2.37",
 # ]
 # ///
 

{aignostics-0.2.36 → aignostics-0.2.37}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aignostics"
-version = "0.2.36"
+version = "0.2.37"
 description = "🔬 Python SDK providing access to the Aignostics Platform. Includes Aignostics Launchpad (Desktop Application), Aignostics CLI (Command-Line Interface), example notebooks, and Aignostics Client Library."
 readme = "README.md"
 authors = [
@@ -356,7 +356,7 @@ source = ["src/"]
 
 
 [tool.bumpversion]
-current_version = "0.2.36"
+current_version = "0.2.37"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 search = "{current_version}"

{aignostics-0.2.36 → aignostics-0.2.37}/src/aignostics/bucket/_service.py

@@ -28,13 +28,13 @@ class Service(BaseService):
         """Initialize service."""
         super().__init__(Settings)
 
-    def info(self) -> dict[str, Any]:  # noqa: PLR6301
+    def info(self) -> dict[str, Any]:
         """Determine info of this service.
 
         Returns:
             dict[str,Any]: The info of this service.
         """
-        return {}
+        return {"settings": self._settings.model_dump()}
 
     def health(self) -> Health:  # noqa: PLR6301
         """Determine health of this service.
@@ -97,7 +97,7 @@ class Service(BaseService):
         url = self._get_s3_client().generate_presigned_url(
             ClientMethod="put_object",
             Params={"Bucket": self._settings.name if bucket_name is None else bucket_name, "Key": object_key},
-            ExpiresIn=3600,
+            ExpiresIn=self._settings.upload_signed_url_expiration_seconds,
         )
         return cast("str", url)
 
@@ -163,7 +163,7 @@ class Service(BaseService):
         url = self._get_s3_client().generate_presigned_url(
             ClientMethod="get_object",
             Params={"Bucket": self._settings.name if bucket_name is None else bucket_name, "Key": object_key},
-            ExpiresIn=3600,
+            ExpiresIn=self._settings.download_signed_url_expiration_seconds,
         )
         return cast("str", url)
 

aignostics-0.2.37/src/aignostics/bucket/_settings.py

@@ -0,0 +1,86 @@
+"""Settings of the bucket module."""
+
+from enum import StrEnum
+from typing import Annotated
+
+from pydantic import Field, PlainSerializer, SecretStr
+from pydantic_settings import SettingsConfigDict
+
+from ..utils import OpaqueSettings, __env_file__, __project_name__  # noqa: TID252
+
+
+class BucketProtocol(StrEnum):
+    GS = "gs"
+    S3 = "s3"
+
+
+class Settings(OpaqueSettings):
+    """Settings."""
+
+    model_config = SettingsConfigDict(
+        env_prefix=f"{__project_name__.upper()}_BUCKET_",
+        extra="ignore",
+        env_file=__env_file__,
+        env_file_encoding="utf-8",
+    )
+
+    protocol: Annotated[
+        BucketProtocol,
+        Field(
+            description=("Protocol to access the cloud bucket, default is gs"),
+            default="gs",
+        ),
+    ]
+
+    name: Annotated[
+        str,
+        Field(description=("Name of the cloud bucket")),
+    ]
+
+    region_name: Annotated[
+        str,
+        Field(description=("Region of the cloud bucket"), default="EUROPE-WEST3"),
+    ]
+
+    hmac_access_key_id: Annotated[
+        SecretStr,
+        PlainSerializer(func=OpaqueSettings.serialize_sensitive_info, return_type=str, when_used="always"),
+        Field(description=("HMAC access key ID of the cloud bucket")),
+    ]
+
+    hmac_secret_access_key: Annotated[
+        SecretStr,
+        PlainSerializer(func=OpaqueSettings.serialize_sensitive_info, return_type=str, when_used="always"),
+        Field(description=("HMAC secret access key of the cloud bucket")),
+    ]
+
+    upload_signed_url_expiration_seconds: Annotated[
+        int,
+        Field(
+            description=("Expiration time for signed URLs created and used by the Python SDK to upload to the bucket."),
+            default=2
+            * 60
+            * 60,  # The Python SDK creates the signed upload URLs immediately before uploading, so 2h is sufficient.
+            ge=60,  # Minimum expiration time is 60 seconds, see https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-presigned-url.html
+            le=7
+            * 24
+            * 60
+            * 60,  # Limit to 7 days as this is the max e.g. at AWS, see https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-presigned-url.html
+        ),
+    ]
+
+    download_signed_url_expiration_seconds: Annotated[
+        int,
+        Field(
+            description=(
+                "Expiration time of the signed URLs provided by the Python SDK to the platform on submitting "
+                "application runs."
+            ),
+            default=7 * 24 * 60 * 60,  # The platform queues application runs, so we set a gracious default of 7d.
+            ge=60,  # Minimum expiration time is 60 seconds, see https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-presigned-url.html
+            le=7
+            * 24
+            * 60
+            * 60,  # Limit to 7 days as this is the max e.g. at AWS, see https://docs.aws.amazon.com/AmazonS3/latest/userguide/using-presigned-url.html
+        ),
+    ]

{aignostics-0.2.36 → aignostics-0.2.37}/src/aignostics/utils/_settings.py

@@ -34,7 +34,7 @@ class OpaqueSettings(BaseSettings):
     def serialize_sensitive_info(input_value: SecretStr, info: FieldSerializationInfo) -> str | None:
         if not input_value:
             return None
-        if info.context.get(UNHIDE_SENSITIVE_INFO, False):  # type: ignore
+        if info.context and info.context.get(UNHIDE_SENSITIVE_INFO, False):
             return input_value.get_secret_value()
         return str(input_value)
 

aignostics-0.2.36/src/aignostics/bucket/_settings.py

@@ -1,53 +0,0 @@
-"""Settings of the bucket module."""
-
-from enum import StrEnum
-from typing import Annotated
-
-from pydantic import Field, SecretStr
-from pydantic_settings import SettingsConfigDict
-
-from ..utils import OpaqueSettings, __env_file__, __project_name__  # noqa: TID252
-
-
-class BucketProtocol(StrEnum):
-    GS = "gs"
-    S3 = "s3"
-
-
-class Settings(OpaqueSettings):
-    """Settings."""
-
-    model_config = SettingsConfigDict(
-        env_prefix=f"{__project_name__.upper()}_BUCKET_",
-        extra="ignore",
-        env_file=__env_file__,
-        env_file_encoding="utf-8",
-    )
-
-    protocol: Annotated[
-        BucketProtocol,
-        Field(
-            description=("Protocol to access the cloud bucket, default is gs"),
-            default="gs",
-        ),
-    ]
-
-    name: Annotated[
-        str,
-        Field(description=("Name of the cloud bucket")),
-    ]
-
-    region_name: Annotated[
-        str,
-        Field(description=("Region of the cloud bucket"), default="EUROPE-WEST3"),
-    ]
-
-    hmac_access_key_id: Annotated[
-        SecretStr,
-        Field(description=("HMAC access key ID of the cloud bucket")),
-    ]
-
-    hmac_secret_access_key: Annotated[
-        SecretStr,
-        Field(description=("HMAC secret access key of the cloud bucket")),
-    ]