feat: expose expiry time as environment variable (#2131)

* feat: expose expiry time as environment variable

* fix: ruff

Author: desaxce

backend/chainlit/data/storage_clients/azure_blob.py

@@ -4,7 +4,7 @@
 from azure.storage.blob import BlobSasPermissions, ContentSettings, generate_blob_sas
 from azure.storage.blob.aio import BlobServiceClient as AsyncBlobServiceClient
 
-from chainlit.data.storage_clients.base import EXPIRY_TIME, BaseStorageClient
+from chainlit.data.storage_clients.base import BaseStorageClient, storage_expiry_time
 from chainlit.logger import logger
 
 
@@ -33,7 +33,7 @@ async def get_read_url(self, object_key: str) -> str:
 
         sas_permissions = BlobSasPermissions(read=True)
         start_time = datetime.now(tz=timezone.utc)
-        expiry_time = start_time + timedelta(seconds=EXPIRY_TIME)
+        expiry_time = start_time + timedelta(seconds=storage_expiry_time)
 
         sas_token = generate_blob_sas(
             account_name=self.storage_account,

backend/chainlit/data/storage_clients/base.py

@@ -1,7 +1,8 @@
+import os
 from abc import ABC, abstractmethod
 from typing import Any, Dict, Union
 
-EXPIRY_TIME = 3600
+storage_expiry_time = int(os.getenv("STORAGE_EXPIRY_TIME", 3600))
 
 
 class BaseStorageClient(ABC):

backend/chainlit/data/storage_clients/gcs.py

@@ -4,7 +4,7 @@
 from google.oauth2 import service_account
 
 from chainlit import make_async
-from chainlit.data.storage_clients.base import EXPIRY_TIME, BaseStorageClient
+from chainlit.data.storage_clients.base import BaseStorageClient, storage_expiry_time
 from chainlit.logger import logger
 
 
@@ -29,7 +29,7 @@ def __init__(
 
     def sync_get_read_url(self, object_key: str) -> str:
         return self.bucket.blob(object_key).generate_signed_url(
-            version="v4", expiration=EXPIRY_TIME, method="GET"
+            version="v4", expiration=storage_expiry_time, method="GET"
         )
 
     async def get_read_url(self, object_key: str) -> str:

backend/chainlit/data/storage_clients/s3.py

@@ -3,7 +3,7 @@
 import boto3  # type: ignore
 
 from chainlit import make_async
-from chainlit.data.storage_clients.base import EXPIRY_TIME, BaseStorageClient
+from chainlit.data.storage_clients.base import BaseStorageClient, storage_expiry_time
 from chainlit.logger import logger
 
 
@@ -25,7 +25,7 @@ def sync_get_read_url(self, object_key: str) -> str:
             url = self.client.generate_presigned_url(
                 "get_object",
                 Params={"Bucket": self.bucket, "Key": object_key},
-                ExpiresIn=EXPIRY_TIME,
+                ExpiresIn=storage_expiry_time,
             )
             return url
         except Exception as e: