fal 1.16.0__py3-none-any.whl → 1.16.1__py3-none-any.whl

fal/_fal_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.16.0'
-__version_tuple__ = version_tuple = (1, 16, 0)
+__version__ = version = '1.16.1'
+__version_tuple__ = version_tuple = (1, 16, 1)

fal/_version.py

@@ -1,6 +1,89 @@
+import json
+import os
+import tempfile
+from typing import Any, Dict, Optional
+
 try:
     from ._fal_version import version as __version__  # type: ignore[import]
     from ._fal_version import version_tuple  # type: ignore[import]
 except ImportError:
     __version__ = "UNKNOWN"
     version_tuple = (0, 0, __version__)  # type: ignore[assignment]
+
+
+_PYPI_URL = "https://pypi.org/pypi/fal/json"
+_PYPI_CACHE_TTL = 60 * 60  # 1 hour
+_PYPI_CACHE_PATH = os.path.expanduser("~/.fal/cache/pypi.json")
+_URLOPEN_TIMEOUT = 1
+
+
+def _write_pypi_cache(data: Dict[str, Any]) -> None:
+    cache_dir = os.path.dirname(_PYPI_CACHE_PATH)
+    os.makedirs(cache_dir, exist_ok=True)
+    prefix = os.path.basename(_PYPI_CACHE_PATH) + ".tmp."
+    with tempfile.NamedTemporaryFile(
+        mode="w",
+        dir=cache_dir,
+        prefix=prefix,
+        delete=False,
+    ) as fobj:
+        fobj.write(json.dumps(data))
+        os.rename(fobj.name, _PYPI_CACHE_PATH)
+
+
+def _get_pypi_cache() -> Optional[Dict[str, Any]]:
+    import time
+
+    try:
+        mtime = os.path.getmtime(_PYPI_CACHE_PATH)
+    except FileNotFoundError:
+        return None
+
+    if mtime + _PYPI_CACHE_TTL < time.time():
+        return None
+
+    with open(_PYPI_CACHE_PATH) as fobj:
+        try:
+            return json.load(fobj)
+        except ValueError:
+            return None
+
+
+def _fetch_pypi_data() -> Dict[str, Any]:
+    from urllib.request import urlopen
+
+    response = urlopen(_PYPI_URL, timeout=_URLOPEN_TIMEOUT)
+    if response.status != 200:
+        raise Exception(f"Failed to fetch {_PYPI_URL}")
+
+    data = response.read()
+    return json.loads(data)
+
+
+def get_latest_version() -> str:
+    from fal.logging import get_logger
+
+    logger = get_logger(__name__)
+
+    try:
+        data = _get_pypi_cache()
+    except Exception:
+        logger.warning("Failed to get pypi cache", exc_info=True)
+        data = None
+
+    if data is None:
+        try:
+            data = _fetch_pypi_data()
+        except Exception:
+            logger.warning("Failed to get latest fal version", exc_info=True)
+            data = {}
+
+        try:
+            _write_pypi_cache(data)
+        except Exception:
+            logger.warning("Failed to write pypi cache", exc_info=True)
+
+    try:
+        return data["info"]["version"]
+    except KeyError:
+        return "0.0.0"

fal/api.py

@@ -501,12 +501,7 @@ class FalServerlessHost(Host):
         if isinstance(func, ServeWrapper):
             # Assigning in a separate property leaving a place for the user
             # to add more metadata in the future
-            try:
-                metadata["openapi"] = func.openapi()
-            except Exception as e:
-                print(
-                    f"[warning] Failed to generate OpenAPI metadata for function: {e}"
-                )
+            metadata["openapi"] = func.openapi()
 
         for partial_result in self._connection.register(
             partial_func,
@@ -1169,7 +1164,12 @@ class BaseServable:
         Build the OpenAPI specification for the served function.
         Attach needed metadata for a better integration to fal.
         """
-        return self._build_app().openapi()
+        try:
+            return self._build_app().openapi()
+        except Exception as e:
+            raise FalServerlessException(
+                "Failed to generate OpenAPI metadata for function"
+            ) from e
 
     def serve(self) -> None:
         import asyncio

fal/app.py

@@ -105,15 +105,12 @@ def wrap_app(cls: type[App], **kwargs) -> IsolatedFunction:
         app.serve()
 
     metadata = {}
-    try:
-        app = cls(_allow_init=True)
-        metadata["openapi"] = app.openapi()
-    except Exception:
-        logger.warning("Failed to build OpenAPI specification for %s", cls.__name__)
-        realtime_app = False
-    else:
-        routes = app.collect_routes()
-        realtime_app = any(route.is_websocket for route in routes)
+    app = cls(_allow_init=True)
+
+    metadata["openapi"] = app.openapi()
+
+    routes = app.collect_routes()
+    realtime_app = any(route.is_websocket for route in routes)
 
     kind = cls.host_kwargs.pop("kind", "virtualenv")
     if kind == "container":

fal/auth/__init__.py

@@ -63,8 +63,13 @@ def key_credentials() -> tuple[str, str] | None:
 
     key = os.environ.get("FAL_KEY") or config.get("key") or get_colab_token()
     if key:
-        key_id, key_secret = key.split(":", 1)
-        return (key_id, key_secret)
+        try:
+            key_id, key_secret = key.split(":", 1)
+            return (key_id, key_secret)
+        except ValueError:
+            print(f"Invalid key format: {key}")
+            return None
+
     elif "FAL_KEY_ID" in os.environ and "FAL_KEY_SECRET" in os.environ:
         return (os.environ["FAL_KEY_ID"], os.environ["FAL_KEY_SECRET"])
     else:

fal/cli/main.py

@@ -77,11 +77,48 @@ def _print_error(msg):
     console.print(f"{CROSS_ICON} {msg}")
 
 
+def _check_latest_version():
+    from packaging.version import parse
+    from rich.emoji import Emoji
+    from rich.panel import Panel
+    from rich.text import Text
+
+    from fal._version import get_latest_version, version_tuple
+
+    latest_version = get_latest_version()
+    parsed = parse(latest_version)
+    latest_version_tuple = (parsed.major, parsed.minor, parsed.micro)
+    if latest_version_tuple <= version_tuple:
+        return
+
+    if not console.is_terminal:
+        return
+
+    line1 = Text.assemble(
+        (Emoji.replace(":warning-emoji:  "), "bold white"),
+        ("A new version of fal is available: ", "bold white"),
+        (latest_version, "bold green"),
+    )
+    line2 = Text.assemble(("pip install --upgrade fal", "bold cyan"))
+    line2.align("center", width=len(line1))
+
+    panel = Panel(
+        line1 + "\n\n" + line2,
+        border_style="yellow",
+        padding=(1, 2),
+        highlight=True,
+        expand=False,
+    )
+    console.print(panel)
+
+
 def main(argv=None) -> int:
     import grpc
 
     from fal.api import UserFunctionException
 
+    _check_latest_version()
+
     ret = 1
     try:
         args = parse_args(argv)

fal/cli/profile.py

@@ -66,7 +66,7 @@ def _delete(args):
         if config.profile == args.PROFILE:
             config.set_internal("profile", None)
 
-        config.delete(args.PROFILE)
+        config.delete_profile(args.PROFILE)
         args.console.print(f"Profile [cyan]{args.PROFILE}[/] deleted.")
 
 

fal/cli/runners.py

@@ -21,17 +21,21 @@ def runners_table(runners: List[RunnerInfo]):
     table.add_column("Revision")
 
     for runner in runners:
+        external_metadata = runner.external_metadata
+        present = external_metadata.get("present_in_group", True)
+
         num_leases_with_request = len(
             [
                 lease
-                for lease in runner.external_metadata.get("leases", [])
+                for lease in external_metadata.get("leases", [])
                 if lease.get("request_id") is not None
             ]
         )
 
         table.add_row(
             runner.alias,
-            runner.runner_id,
+            # Mark lost runners in red
+            runner.runner_id if present else f"[red]{runner.runner_id}[/]",
             str(runner.in_flight_requests),
             str(runner.in_flight_requests - num_leases_with_request),
             (

fal/config.py

@@ -99,7 +99,7 @@ class Config:
     def unset_internal(self, key: str) -> None:
         self._config.get(SETTINGS_SECTION, {}).pop(key, None)
 
-    def delete(self, profile: str) -> None:
+    def delete_profile(self, profile: str) -> None:
         del self._config[profile]
 
     @contextmanager

fal/toolkit/utils/endpoint.py

@@ -0,0 +1,29 @@
+from contextlib import asynccontextmanager
+
+from anyio import create_task_group
+from fastapi import Request
+
+
+@asynccontextmanager
+async def cancel_on_disconnect(request: Request):
+    """
+    Async context manager for async code that needs to be cancelled if client
+    disconnects prematurely.
+    The client disconnect is monitored through the Request object.
+    """
+    async with create_task_group() as tg:
+
+        async def watch_disconnect():
+            while True:
+                message = await request.receive()
+
+                if message["type"] == "http.disconnect":
+                    tg.cancel_scope.cancel()
+                    break
+
+        tg.start_soon(watch_disconnect)
+
+        try:
+            yield
+        finally:
+            tg.cancel_scope.cancel()

{fal-1.16.0.dist-info → fal-1.16.1.dist-info}/METADATA

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

{fal-1.16.0.dist-info → fal-1.16.1.dist-info}/RECORD

@@ -1,12 +1,12 @@
 fal/__init__.py,sha256=wXs1G0gSc7ZK60-bHe-B2m0l_sA6TrFk4BxY0tMoLe8,784
 fal/__main__.py,sha256=4JMK66Wj4uLZTKbF-sT3LAxOsr6buig77PmOkJCRRxw,83
-fal/_fal_version.py,sha256=82WJEz-zkeAg7wgq_Qa170slIBg2QEGzstYPwtLKAR8,513
+fal/_fal_version.py,sha256=fHsj4qb8quZFYixIg7OlfrAd4kLbNz0T5Cu5K-h6vsE,513
 fal/_serialization.py,sha256=npXNsFJ5G7jzBeBIyVMH01Ww34mGY4XWhHpRbSrTtnQ,7598
-fal/_version.py,sha256=EBGqrknaf1WygENX-H4fBefLvHryvJBBGtVJetaB0NY,266
-fal/api.py,sha256=cQuE9tr8lBHiBZ08DJD8jKfqjM0-deP4k1M24HS6mrQ,46335
-fal/app.py,sha256=R-DYIpB2GE0t_9GMvcARXjSBwAKDKG_QUhkMzCRwrsE,24322
+fal/_version.py,sha256=1BbTFnucNC_6ldKJ_ZoC722_UkW4S9aDBSW9L0fkKAw,2315
+fal/api.py,sha256=moDNT8wt20uzsI-NTEsbVTpjFXFkSuuRXJx7Apux3SI,46329
+fal/app.py,sha256=S5VHxDaj5J9YVC8ECenHCZJlTHalapHyOyHbCBNsDfs,24153
 fal/apps.py,sha256=pzCd2mrKl5J_4oVc40_pggvPtFahXBCdrZXWpnaEJVs,12130
-fal/config.py,sha256=19Q7fymEkfxCd9AIy8SxhaQaRvb_vKvYAG3AeZAI6uk,3116
+fal/config.py,sha256=BEMH10B2bfWJ9yNawnLG6v3kBLnLmkhMe201EAODzs4,3124
 fal/container.py,sha256=OvR-Zq-NPbYFHTnw0SBUUFxr890Fgbe68J2kSJEpLOk,1905
 fal/files.py,sha256=LHJxT4fs2jDs1hH26YoXdq77hUQp4IiaNJ0TE2-RFjo,2773
 fal/flags.py,sha256=48pgtc9xb4LMpR9RE5KG2A2sH7zQRk_VjrgpND-H4Tc,942
@@ -17,7 +17,7 @@ fal/sdk.py,sha256=OvNgoV6ERnFup7ulylBDSohiXQpBa1ycqNuycPZb1-Q,25816
 fal/sync.py,sha256=ZuIJA2-hTPNANG9B_NNJZUsO68EIdTH0dc9MzeVE2VU,4340
 fal/utils.py,sha256=iQTBG3-i6JZgHkkwbY_I4210g0xoW-as51yrke608u0,2208
 fal/workflows.py,sha256=Zl4f6Bs085hY40zmqScxDUyCu7zXkukDbW02iYOLTTI,14805
-fal/auth/__init__.py,sha256=2tki_o_IaQbaZeCTDAS1wBtrvcAOPRTQMtPSXAqk_Ig,6157
+fal/auth/__init__.py,sha256=2mEKdk6_1GclF3cPC3uWSRKFf0KHNIUNAi0xYRbdJ1A,6278
 fal/auth/auth0.py,sha256=g5OgEKe4rsbkLQp6l7EauOAVL6WsmKjuA1wmzmyvvhc,5354
 fal/auth/local.py,sha256=sndkM6vKpeVny6NHTacVlTbiIFqaksOmw0Viqs_RN1U,1790
 fal/cli/__init__.py,sha256=padK4o0BFqq61kxAA1qQ0jYr2SuhA2mf90B3AaRkmJA,37
@@ -32,11 +32,11 @@ fal/cli/deploy.py,sha256=CWf0Y56w-hNCrht-qrfgiOi9nuvve1Kl5NFZJpt_oRA,7770
 fal/cli/doctor.py,sha256=U4ne9LX5gQwNblsYQ27XdO8AYDgbYjTO39EtxhwexRM,983
 fal/cli/files.py,sha256=zOJeRy1W1CsNw0QMxt2vT8Q352phh3l4lZSOLiTQa2w,1968
 fal/cli/keys.py,sha256=7Sf4DT4le89G42eAOt0ltRjbZAtE70AVQ62hmjZhUy0,3059
-fal/cli/main.py,sha256=CNh-i1xL0G2pbYMsk0VUC6qsxBT9rrQuLCIeDSiRuQs,2260
+fal/cli/main.py,sha256=ao8EEV_Fkd7AdN5En6k_dZWp158Et5DrqNRutl98MHY,3273
 fal/cli/parser.py,sha256=jYsGQ0BLQuKI7KtN1jnLVYKMbLtez7hPjwTNfG3UPSk,2964
-fal/cli/profile.py,sha256=SzAswTckfc5qhpVsFZIQedzGYHkH8wWaXwQxwe5mc8E,3937
+fal/cli/profile.py,sha256=lYOz0S1kr5DW4_r5pB5cEaCHsUEbiPQFmm8HT6_bx9k,3945
 fal/cli/run.py,sha256=nAC12Qss4Fg1XmV0qOS9RdGNLYcdoHeRgQMvbTN4P9I,1202
-fal/cli/runners.py,sha256=z7WkZZC9rCW2mU5enowVQsxd1W18iBtLNOnPjrzhEf0,3491
+fal/cli/runners.py,sha256=7efNX9vm6D1aBlg0M5-u5plw3HHC41Sj-N7eRNIHnqw,3689
 fal/cli/secrets.py,sha256=QKSmazu-wiNF6fOpGL9v2TDYxAjX9KTi7ot7vnv6f5E,2474
 fal/cli/teams.py,sha256=6fR2rKJtiUJPThP7QsO4NLo9UdhUxraGvQZk3_Di6Ow,1218
 fal/console/__init__.py,sha256=ernZ4bzvvliQh5SmrEqQ7lA5eVcbw6Ra2jalKtA7dxg,132
@@ -72,6 +72,7 @@ fal/toolkit/image/nsfw_filter/model.py,sha256=63mu8D15z_IosoRUagRLGHy6VbLqFmrG-y
 fal/toolkit/image/nsfw_filter/requirements.txt,sha256=3Pmrd0Ny6QAeBqUNHCgffRyfaCARAPJcfSCX5cRYpbM,37
 fal/toolkit/utils/__init__.py,sha256=CrmM9DyCz5-SmcTzRSm5RaLgxy3kf0ZsSEN9uhnX2Xo,97
 fal/toolkit/utils/download_utils.py,sha256=NgOMNs-bQGSg3gWnu123BgZitJgJrvtRexIefTMuylY,19739
+fal/toolkit/utils/endpoint.py,sha256=5EXoshA2PD_brjEfhNWAWasjqLOCRrjBnfhj6QGuMt8,782
 fal/toolkit/utils/retry.py,sha256=mHcQvvNIpu-Hi29P1HXSZuyvolRd48dMaJToqzlG0NY,1353
 openapi_fal_rest/__init__.py,sha256=ziculmF_i6trw63LzZGFX-6W3Lwq9mCR8_UpkpvpaHI,152
 openapi_fal_rest/client.py,sha256=G6BpJg9j7-JsrAUGddYwkzeWRYickBjPdcVgXoPzxuE,2817
@@ -136,8 +137,8 @@ openapi_fal_rest/models/workflow_node_type.py,sha256=-FzyeY2bxcNmizKbJI8joG7byRi
 openapi_fal_rest/models/workflow_schema.py,sha256=4K5gsv9u9pxx2ItkffoyHeNjBBYf6ur5bN4m_zePZNY,2019
 openapi_fal_rest/models/workflow_schema_input.py,sha256=2OkOXWHTNsCXHWS6EGDFzcJKkW5FIap-2gfO233EvZQ,1191
 openapi_fal_rest/models/workflow_schema_output.py,sha256=EblwSPAGfWfYVWw_WSSaBzQVju296is9o28rMBAd0mc,1196
-fal-1.16.0.dist-info/METADATA,sha256=ofJflcV0SxzRjZCXOAOQTC3mYGknv4sjdhlbIBXcNe8,4084
-fal-1.16.0.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-fal-1.16.0.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
-fal-1.16.0.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
-fal-1.16.0.dist-info/RECORD,,
+fal-1.16.1.dist-info/METADATA,sha256=KIyDdj36vshtcq463l_DSZU49gm6optvf9wvVqX8-BQ,4084
+fal-1.16.1.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+fal-1.16.1.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
+fal-1.16.1.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
+fal-1.16.1.dist-info/RECORD,,