fal 1.5.1__tar.gz → 1.5.2__tar.gz

{fal-1.5.1 → fal-1.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fal
-Version: 1.5.1
+Version: 1.5.2
 Summary: fal is an easy-to-use Serverless Python Framework
 Author: Features & Labels <support@fal.ai>
 Requires-Python: >=3.8

{fal-1.5.1 → fal-1.5.2}/fal.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fal
-Version: 1.5.1
+Version: 1.5.2
 Summary: fal is an easy-to-use Serverless Python Framework
 Author: Features & Labels <support@fal.ai>
 Requires-Python: >=3.8

{fal-1.5.1 → fal-1.5.2}/fal.egg-info/SOURCES.txt

@@ -144,6 +144,7 @@ tests/integration_test.py
 tests/mainify_target.py
 tests/test_apps.py
 tests/test_stability.py
+tests/assets/cat.png
 tests/cli/__init__.py
 tests/cli/test_apps.py
 tests/cli/test_auth.py

{fal-1.5.1 → fal-1.5.2}/src/fal/_fal_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.5.1'
-__version_tuple__ = version_tuple = (1, 5, 1)
+__version__ = version = '1.5.2'
+__version_tuple__ = version_tuple = (1, 5, 2)

{fal-1.5.1 → fal-1.5.2}/src/fal/app.py

@@ -402,7 +402,10 @@ def _fal_websocket_template(
                 batch.append(next_input)
 
             t0 = loop.time()
-            output = await loop.run_in_executor(None, func, self, *batch)  # type: ignore
+            if inspect.iscoroutinefunction(func):
+                output = await func(self, *batch)
+            else:
+                output = await loop.run_in_executor(None, func, self, *batch)  # type: ignore
             total_time = loop.time() - t0
             if not isinstance(output, dict):
                 # Handle pydantic output modal

{fal-1.5.1 → fal-1.5.2}/src/fal/toolkit/file/file.py

@@ -8,6 +8,7 @@ from urllib.parse import urlparse
 from zipfile import ZipFile
 
 import pydantic
+from fastapi import Request
 
 # https://github.com/pydantic/pydantic/pull/2573
 if not hasattr(pydantic, "__version__") or pydantic.__version__.startswith("1."):
@@ -24,6 +25,7 @@ from fal.toolkit.file.providers.fal import (
     FalCDNFileRepository,
     FalFileRepository,
     FalFileRepositoryV2,
+    FalFileRepositoryV3,
     InMemoryRepository,
 )
 from fal.toolkit.file.providers.gcp import GoogleStorageRepository
@@ -36,6 +38,7 @@ FileRepositoryFactory = Callable[[], FileRepository]
 BUILT_IN_REPOSITORIES: dict[RepositoryId, FileRepositoryFactory] = {
     "fal": lambda: FalFileRepository(),
     "fal_v2": lambda: FalFileRepositoryV2(),
+    "fal_v3": lambda: FalFileRepositoryV3(),
     "in_memory": lambda: InMemoryRepository(),
     "gcp_storage": lambda: GoogleStorageRepository(),
     "r2": lambda: R2Repository(),
@@ -53,6 +56,7 @@ get_builtin_repository.__module__ = "__main__"
 
 DEFAULT_REPOSITORY: FileRepository | RepositoryId = "fal_v2"
 FALLBACK_REPOSITORY: FileRepository | RepositoryId = "cdn"
+OBJECT_LIFECYCLE_PREFERENCE_KEY = "x-fal-object-lifecycle-preference"
 
 
 class File(BaseModel):
@@ -130,6 +134,7 @@ class File(BaseModel):
         fallback_repository: Optional[
             FileRepository | RepositoryId
         ] = FALLBACK_REPOSITORY,
+        request: Optional[Request] = None,
     ) -> File:
         repo = (
             repository
@@ -139,8 +144,10 @@ class File(BaseModel):
 
         fdata = FileData(data, content_type, file_name)
 
+        object_lifecycle_preference = _get_lifecycle_preference(request)
+
         try:
-            url = repo.save(fdata)
+            url = repo.save(fdata, object_lifecycle_preference)
         except Exception:
             if not fallback_repository:
                 raise
@@ -151,7 +158,7 @@ class File(BaseModel):
                 else get_builtin_repository(fallback_repository)
             )
 
-            url = fallback_repo.save(fdata)
+            url = fallback_repo.save(fdata, object_lifecycle_preference)
 
         return cls(
             url=url,
@@ -171,6 +178,7 @@ class File(BaseModel):
         fallback_repository: Optional[
             FileRepository | RepositoryId
         ] = FALLBACK_REPOSITORY,
+        request: Optional[Request] = None,
     ) -> File:
         file_path = Path(path)
         if not file_path.exists():
@@ -183,12 +191,14 @@ class File(BaseModel):
         )
 
         content_type = content_type or "application/octet-stream"
+        object_lifecycle_preference = _get_lifecycle_preference(request)
 
         try:
             url, data = repo.save_file(
                 file_path,
                 content_type=content_type,
                 multipart=multipart,
+                object_lifecycle_preference=object_lifecycle_preference,
             )
         except Exception:
             if not fallback_repository:
@@ -204,6 +214,7 @@ class File(BaseModel):
                 file_path,
                 content_type=content_type,
                 multipart=multipart,
+                object_lifecycle_preference=object_lifecycle_preference,
             )
 
         return cls(
@@ -261,3 +272,23 @@ class CompressedFile(File):
     def __del__(self):
         if self.extract_dir:
             shutil.rmtree(self.extract_dir)
+
+
+def _get_lifecycle_preference(request: Request) -> dict[str, str] | None:
+    import json
+
+    preference_str = (
+        request.headers.get(OBJECT_LIFECYCLE_PREFERENCE_KEY)
+        if request is not None
+        else None
+    )
+    if preference_str is None:
+        return None
+
+    object_lifecycle_preference = {}
+    try:
+        object_lifecycle_preference = json.loads(preference_str)
+        return object_lifecycle_preference
+    except Exception as e:
+        print(f"Failed to parse object lifecycle preference: {e}")
+        return None

{fal-1.5.1 → fal-1.5.2}/src/fal/toolkit/file/providers/fal.py

@@ -19,6 +19,7 @@ from fal.toolkit.file.types import FileData, FileRepository
 from fal.toolkit.utils.retry import retry
 
 _FAL_CDN = "https://fal.media"
+_FAL_CDN_V3 = "https://v3.fal.media"
 
 
 @dataclass
@@ -91,11 +92,11 @@ fal_v2_token_manager = FalV2TokenManager()
 
 @dataclass
 class ObjectLifecyclePreference:
-    expriation_duration_seconds: int
+    expiration_duration_seconds: int
 
 
 GLOBAL_LIFECYCLE_PREFERENCE = ObjectLifecyclePreference(
-    expriation_duration_seconds=86400
+    expiration_duration_seconds=86400
 )
 
 
@@ -158,7 +159,9 @@ class FalFileRepositoryBase(FileRepository):
 
 @dataclass
 class FalFileRepository(FalFileRepositoryBase):
-    def save(self, file: FileData) -> str:
+    def save(
+        self, file: FileData, object_lifecycle_preference: dict[str, str] | None = None
+    ) -> str:
         return self._save(file, "gcs")
 
 
@@ -275,7 +278,9 @@ class MultipartUpload:
 @dataclass
 class FalFileRepositoryV2(FalFileRepositoryBase):
     @retry(max_retries=3, base_delay=1, backoff_type="exponential", jitter=True)
-    def save(self, file: FileData) -> str:
+    def save(
+        self, file: FileData, object_lifecycle_preference: dict[str, str] | None = None
+    ) -> str:
         token = fal_v2_token_manager.get_token()
         headers = {
             "Authorization": f"{token.token_type} {token.token}",
@@ -327,6 +332,7 @@ class FalFileRepositoryV2(FalFileRepositoryBase):
         multipart_threshold: int | None = None,
         multipart_chunk_size: int | None = None,
         multipart_max_concurrency: int | None = None,
+        object_lifecycle_preference: dict[str, str] | None = None,
     ) -> tuple[str, FileData | None]:
         if multipart is None:
             threshold = multipart_threshold or MultipartUpload.MULTIPART_THRESHOLD
@@ -347,7 +353,7 @@ class FalFileRepositoryV2(FalFileRepositoryBase):
                     content_type=content_type,
                     file_name=os.path.basename(file_path),
                 )
-            url = self.save(data)
+            url = self.save(data, object_lifecycle_preference)
 
         return url, data
 
@@ -357,6 +363,7 @@ class InMemoryRepository(FileRepository):
     def save(
         self,
         file: FileData,
+        object_lifecycle_preference: dict[str, str] | None = None,
     ) -> str:
         return f'data:{file.content_type};base64,{b64encode(file.data).decode("utf-8")}'
 
@@ -367,6 +374,7 @@ class FalCDNFileRepository(FileRepository):
     def save(
         self,
         file: FileData,
+        object_lifecycle_preference: dict[str, str] | None = None,
     ) -> str:
         headers = {
             **self.auth_headers,
@@ -401,3 +409,50 @@ class FalCDNFileRepository(FileRepository):
             "Authorization": f"Bearer {key_id}:{key_secret}",
             "User-Agent": "fal/0.1.0",
         }
+
+
+@dataclass
+class FalFileRepositoryV3(FileRepository):
+    @retry(max_retries=3, base_delay=1, backoff_type="exponential", jitter=True)
+    def save(
+        self, file: FileData, user_lifecycle_preference: dict[str, str] | None
+    ) -> str:
+        object_lifecycle_preference = dataclasses.asdict(GLOBAL_LIFECYCLE_PREFERENCE)
+
+        if user_lifecycle_preference is not None:
+            object_lifecycle_preference = {
+                key: user_lifecycle_preference[key]
+                if key in user_lifecycle_preference
+                else value
+                for key, value in object_lifecycle_preference.items()
+            }
+
+        headers = {
+            **self.auth_headers,
+            "Accept": "application/json",
+            "Content-Type": file.content_type,
+            "X-Fal-File-Name": file.file_name,
+            "X-Fal-Object-Lifecycle-Preference": json.dumps(
+                object_lifecycle_preference
+            ),
+        }
+        url = os.getenv("FAL_CDN_V3_HOST", _FAL_CDN_V3) + "/files/upload"
+        request = Request(url, headers=headers, method="POST", data=file.data)
+        try:
+            with urlopen(request) as response:
+                result = json.load(response)
+        except HTTPError as e:
+            raise FileUploadException(
+                f"Error initiating upload. Status {e.status}: {e.reason}"
+            )
+
+        access_url = result["access_url"]
+        return access_url
+
+    @property
+    def auth_headers(self) -> dict[str, str]:
+        token = fal_v2_token_manager.get_token()
+        return {
+            "Authorization": f"{token.token_type} {token.token}",
+            "User-Agent": "fal/0.1.0",
+        }

{fal-1.5.1 → fal-1.5.2}/src/fal/toolkit/file/types.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from dataclasses import dataclass
 from mimetypes import guess_extension, guess_type
 from pathlib import Path
-from typing import Literal
+from typing import Literal, Optional
 from uuid import uuid4
 
 
@@ -29,12 +29,18 @@ class FileData:
             self.file_name = file_name
 
 
-RepositoryId = Literal["fal", "fal_v2", "in_memory", "gcp_storage", "r2", "cdn"]
+RepositoryId = Literal[
+    "fal", "fal_v2", "fal_v3", "in_memory", "gcp_storage", "r2", "cdn"
+]
 
 
 @dataclass
 class FileRepository:
-    def save(self, data: FileData) -> str:
+    def save(
+        self,
+        data: FileData,
+        object_lifecycle_preference: Optional[dict[str, str]] = None,
+    ) -> str:
         raise NotImplementedError()
 
     def save_file(
@@ -45,6 +51,7 @@ class FileRepository:
         multipart_threshold: int | None = None,
         multipart_chunk_size: int | None = None,
         multipart_max_concurrency: int | None = None,
+        object_lifecycle_preference: Optional[dict[str, str]] = None,
     ) -> tuple[str, FileData | None]:
         if multipart:
             raise NotImplementedError()
@@ -52,4 +59,4 @@ class FileRepository:
         with open(file_path, "rb") as fobj:
             data = FileData(fobj.read(), content_type, Path(file_path).name)
 
-        return self.save(data), data
+        return self.save(data, object_lifecycle_preference), data

{fal-1.5.1 → fal-1.5.2}/src/fal/toolkit/image/image.py

@@ -4,6 +4,7 @@ import io
 from tempfile import NamedTemporaryFile
 from typing import TYPE_CHECKING, Literal, Optional, Union
 
+from fastapi import Request
 from pydantic import BaseModel, Field
 
 from fal.toolkit.file.file import DEFAULT_REPOSITORY, FALLBACK_REPOSITORY, File
@@ -82,6 +83,7 @@ class Image(File):
         fallback_repository: Optional[
             FileRepository | RepositoryId
         ] = FALLBACK_REPOSITORY,
+        request: Optional[Request] = None,
     ) -> Image:
         obj = super().from_bytes(
             data,
@@ -89,6 +91,7 @@ class Image(File):
             file_name=file_name,
             repository=repository,
             fallback_repository=fallback_repository,
+            request=request,
         )
         obj.width = size.width if size else None
         obj.height = size.height if size else None
@@ -104,6 +107,7 @@ class Image(File):
         fallback_repository: Optional[
             FileRepository | RepositoryId
         ] = FALLBACK_REPOSITORY,
+        request: Optional[Request] = None,
     ) -> Image:
         size = ImageSize(width=pil_image.width, height=pil_image.height)
         if format is None:
@@ -133,6 +137,7 @@ class Image(File):
             file_name,
             repository,
             fallback_repository=fallback_repository,
+            request=request,
         )
 
     def to_pil(self, mode: str = "RGB") -> PILImage.Image:

{fal-1.5.1 → fal-1.5.2}/tests/integration_test.py

@@ -23,6 +23,8 @@ from fal.toolkit.utils.download_utils import _get_git_revision_hash, _hash_url
 from pydantic import BaseModel, Field
 from pydantic import __version__ as pydantic_version
 
+EXAMPLE_FILE_URL = "https://raw.githubusercontent.com/fal-ai/fal/main/projects/fal/tests/assets/cat.png"
+
 
 @pytest.mark.flaky(max_runs=3)
 def test_isolated(isolated_client: Callable[..., Callable[..., IsolatedFunction]]):
@@ -197,11 +199,9 @@ def mock_fal_persistent_dirs(monkeypatch):
 def test_download_file(isolated_client, mock_fal_persistent_dirs):
     from fal.toolkit.utils.download_utils import FAL_PERSISTENT_DIR
 
-    EXAMPLE_FILE_URL = "https://raw.githubusercontent.com/fal-ai/isolate/d553f927348206530208442556f481f39b161732/README.md"
-
     relative_directory = "test"
     output_directory = FAL_PERSISTENT_DIR / relative_directory
-    expected_path = output_directory / "README.md"
+    expected_path = output_directory / "cat.png"
 
     @isolated_client()
     def absolute_path_persistent_dir():
@@ -301,8 +301,7 @@ def test_download_model_weights(isolated_client, mock_fal_persistent_dirs):
 
     print(FAL_MODEL_WEIGHTS_DIR)
 
-    EXAMPLE_FILE_URL = "https://raw.githubusercontent.com/fal-ai/isolate/d553f927348206530208442556f481f39b161732/README.md"
-    expected_path = FAL_MODEL_WEIGHTS_DIR / _hash_url(EXAMPLE_FILE_URL) / "README.md"
+    expected_path = FAL_MODEL_WEIGHTS_DIR / _hash_url(EXAMPLE_FILE_URL) / "cat.png"
 
     @isolated_client()
     def download_weights():
@@ -510,8 +509,8 @@ def test_fal_file_save(isolated_client):
     "file_url, expected_content",
     [
         (
-            "https://raw.githubusercontent.com/fal-ai/fal/fe0e2a1aa4b46a42a93bad0fbd9aca4aefcb4296/README.md",
-            "projects/fal/README.md",
+            EXAMPLE_FILE_URL,
+            "projects/fal/cat.png",
         ),
         ("data:text/plain;charset=UTF-8,fal", "fal"),
     ],

{fal-1.5.1 → fal-1.5.2}/tests/test_stability.py

@@ -580,6 +580,7 @@ def test_worker_env_vars(isolated_client):
             "https://storage.googleapis.com/isolate-dev-smiling-shark_toolkit_bucket/",
         ),
         ("fal_v2", "https://v2.fal.media/files"),
+        ("fal_v3", "https://v3.fal.media/files"),
     ],
 )
 def test_fal_storage(isolated_client, repo_type, url_prefix):