fal 1.15.0__tar.gz → 1.16.0__tar.gz

{fal-1.15.0/fal.egg-info → fal-1.16.0}/PKG-INFO

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

{fal-1.15.0 → fal-1.16.0/fal.egg-info}/PKG-INFO

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

{fal-1.15.0 → fal-1.16.0}/src/fal/_fal_version.py

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

{fal-1.15.0 → fal-1.16.0}/src/fal/_serialization.py

@@ -6,7 +6,7 @@ from typing import Any, Callable
 import cloudpickle
 
 
-def _register_pickle_by_value(name) -> None:
+def include_module(name) -> None:
     # cloudpickle.register_pickle_by_value wants an imported module object,
     # but there is really no reason to go through that complication, as
     # it might be prone to errors.
@@ -22,7 +22,7 @@ def include_package_from_path(raw_path: str) -> None:
         parent = parent.parent
 
     if parent != path:
-        _register_pickle_by_value(parent.name)
+        include_module(parent.name)
 
 
 def include_modules_from(obj: Any) -> None:
@@ -33,7 +33,7 @@ def include_modules_from(obj: Any) -> None:
     if "." in module_name:
         # Just include the whole package
         package_name, *_ = module_name.partition(".")
-        _register_pickle_by_value(package_name)
+        include_module(package_name)
         return
 
     if module_name == "__main__":
@@ -44,7 +44,7 @@ def include_modules_from(obj: Any) -> None:
         include_package_from_path(__main__.__file__)
         return
 
-    _register_pickle_by_value(module_name)
+    include_module(module_name)
 
 
 def _register(cls: Any, func: Callable) -> None:
@@ -230,4 +230,4 @@ def patch_pickle() -> None:
     _patch_console_thread_locals()
     _patch_exceptions()
 
-    _register_pickle_by_value("fal")
+    include_module("fal")

{fal-1.15.0 → fal-1.16.0}/src/fal/api.py

@@ -39,7 +39,7 @@ from pydantic import __version__ as pydantic_version
 from typing_extensions import Concatenate, ParamSpec
 
 import fal.flags as flags
-from fal._serialization import include_modules_from, patch_pickle
+from fal._serialization import include_module, include_modules_from, patch_pickle
 from fal.container import ContainerImage
 from fal.exceptions import (
     AppException,
@@ -694,6 +694,7 @@ def function(
     serve: Literal[False] = False,
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
 ) -> Callable[
     [Callable[Concatenate[ArgsT], ReturnT]], IsolatedFunction[ArgsT, ReturnT]
 ]: ...
@@ -710,6 +711,7 @@ def function(
     serve: Literal[True],
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
 ) -> Callable[
     [Callable[Concatenate[ArgsT], ReturnT]], ServedIsolatedFunction[ArgsT, ReturnT]
 ]: ...
@@ -727,6 +729,7 @@ def function(
     serve: Literal[False] = False,
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
     # FalServerlessHost options
     metadata: dict[str, Any] | None = None,
     machine_type: str | list[str] = FAL_SERVERLESS_DEFAULT_MACHINE_TYPE,
@@ -756,6 +759,7 @@ def function(
     serve: Literal[True],
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
     # FalServerlessHost options
     metadata: dict[str, Any] | None = None,
     machine_type: str | list[str] = FAL_SERVERLESS_DEFAULT_MACHINE_TYPE,
@@ -792,6 +796,7 @@ def function(
     serve: Literal[False] = False,
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
 ) -> Callable[
     [Callable[Concatenate[ArgsT], ReturnT]], IsolatedFunction[ArgsT, ReturnT]
 ]: ...
@@ -813,6 +818,7 @@ def function(
     serve: Literal[True],
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
 ) -> Callable[
     [Callable[Concatenate[ArgsT], ReturnT]], ServedIsolatedFunction[ArgsT, ReturnT]
 ]: ...
@@ -835,6 +841,7 @@ def function(
     serve: Literal[False] = False,
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
     # FalServerlessHost options
     metadata: dict[str, Any] | None = None,
     machine_type: str | list[str] = FAL_SERVERLESS_DEFAULT_MACHINE_TYPE,
@@ -869,6 +876,7 @@ def function(
     serve: Literal[True],
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
     # FalServerlessHost options
     metadata: dict[str, Any] | None = None,
     machine_type: str | list[str] = FAL_SERVERLESS_DEFAULT_MACHINE_TYPE,
@@ -897,6 +905,7 @@ def function(
     serve: Literal[False] = False,
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
     # FalServerlessHost options
     metadata: dict[str, Any] | None = None,
     machine_type: str | list[str] = FAL_SERVERLESS_DEFAULT_MACHINE_TYPE,
@@ -925,6 +934,7 @@ def function(
     serve: Literal[True],
     exposed_port: int | None = None,
     max_concurrency: int | None = None,
+    local_python_modules: list[str] | None = None,
     # FalServerlessHost options
     metadata: dict[str, Any] | None = None,
     machine_type: str | list[str] = FAL_SERVERLESS_DEFAULT_MACHINE_TYPE,
@@ -948,6 +958,7 @@ def function(  # type: ignore
     kind: str = "virtualenv",
     *,
     host: Host | None = None,
+    local_python_modules: list[str] | None = None,
     **config: Any,
 ):
     if host is None:
@@ -956,6 +967,10 @@ def function(  # type: ignore
 
     def wrapper(func: Callable[ArgsT, ReturnT]):
         include_modules_from(func)
+
+        for module_name in local_python_modules or []:
+            include_module(module_name)
+
         proxy = IsolatedFunction(
             host=host,  # type: ignore
             raw_func=func,  # type: ignore

{fal-1.15.0 → fal-1.16.0}/src/fal/app.py

@@ -122,6 +122,7 @@ def wrap_app(cls: type[App], **kwargs) -> IsolatedFunction:
     wrapper = fal_function(
         kind,
         requirements=cls.requirements,
+        local_python_modules=cls.local_python_modules,
         machine_type=cls.machine_type,
         num_gpus=cls.num_gpus,
         **cls.host_kwargs,
@@ -265,6 +266,7 @@ def _print_python_packages() -> None:
 
 class App(BaseServable):
     requirements: ClassVar[list[str]] = []
+    local_python_modules: ClassVar[list[str]] = []
     machine_type: ClassVar[str] = "S"
     num_gpus: ClassVar[int | None] = None
     host_kwargs: ClassVar[dict[str, Any]] = {

{fal-1.15.0 → fal-1.16.0}/src/fal/cli/files.py

@@ -18,14 +18,14 @@ def _download(args):
     from fal.files import FalFileSystem
 
     fs = FalFileSystem()
-    fs.get(args.remote_path, args.local_path)
+    fs.get(args.remote_path, args.local_path, recursive=True)
 
 
 def _upload(args):
     from fal.files import FalFileSystem
 
     fs = FalFileSystem()
-    fs.put(args.local_path, args.remote_path)
+    fs.put(args.local_path, args.remote_path, recursive=True)
 
 
 def add_parser(main_subparsers, parents):

{fal-1.15.0 → fal-1.16.0}/src/fal/cli/profile.py

@@ -55,6 +55,12 @@ def _key_set(args):
         args.console.print(f"Key set for profile [cyan]{config.profile}[/].")
 
 
+def _host_set(args):
+    with Config().edit() as config:
+        config.set("host", args.HOST)
+        args.console.print(f"Fal host set to [cyan]{args.HOST}[/].")
+
+
 def _delete(args):
     with Config().edit() as config:
         if config.profile == args.PROFILE:
@@ -121,6 +127,19 @@ def add_parser(main_subparsers, parents):
     )
     key_set_parser.set_defaults(func=_key_set)
 
+    host_set_help = "Set fal host."
+    host_set_parser = subparsers.add_parser(
+        "host",
+        description=host_set_help,
+        help=host_set_help,
+        parents=parents,
+    )
+    host_set_parser.add_argument(
+        "HOST",
+        help="Fal host.",
+    )
+    host_set_parser.set_defaults(func=_host_set)
+
     delete_help = "Delete profile."
     delete_parser = subparsers.add_parser(
         "delete",

{fal-1.15.0 → fal-1.16.0}/src/fal/files.py

@@ -1,3 +1,4 @@
+import os
 import posixpath
 from functools import cached_property
 from typing import TYPE_CHECKING
@@ -28,14 +29,16 @@ class FalFileSystem(AbstractFileSystem):
         )
 
     def ls(self, path, detail=True, **kwargs):
-        response = self._client.get(f"/files/list/{path.lstrip('/')}")
-        response.raise_for_status()
-        files = response.json()
-        if detail:
-            return sorted(
+        if path in self.dircache:
+            entries = self.dircache[path]
+        else:
+            response = self._client.get(f"/files/list/{path.lstrip('/')}")
+            response.raise_for_status()
+            files = response.json()
+            entries = sorted(
                 (
                     {
-                        "name": entry["path"],
+                        "name": entry["path"].lstrip("/data/"),
                         "size": entry["size"],
                         "type": "file" if entry["is_file"] else "directory",
                         "mtime": entry["updated_time"],
@@ -44,8 +47,12 @@ class FalFileSystem(AbstractFileSystem):
                 ),
                 key=lambda x: x["name"],
             )
-        else:
-            return sorted(entry["path"] for entry in files)
+        self.dircache[path] = entries
+
+        if detail:
+            return entries
+
+        return [entry["name"] for entry in entries]
 
     def info(self, path, **kwargs):
         parent = posixpath.dirname(path)
@@ -56,19 +63,28 @@ class FalFileSystem(AbstractFileSystem):
         raise FileNotFoundError(f"File not found: {path}")
 
     def get_file(self, rpath, lpath, **kwargs):
+        if self.isdir(rpath):
+            os.makedirs(lpath, exist_ok=True)
+            return
+
         with open(lpath, "wb") as fobj:
             response = self._client.get(f"/files/file/{rpath.lstrip('/')}")
             response.raise_for_status()
             fobj.write(response.content)
 
     def put_file(self, lpath, rpath, mode="overwrite", **kwargs):
+        if os.path.isdir(lpath):
+            return
+
         with open(lpath, "rb") as fobj:
             response = self._client.post(
                 f"/files/file/local/{rpath.lstrip('/')}",
                 files={"file_upload": (posixpath.basename(lpath), fobj, "text/plain")},
             )
             response.raise_for_status()
+        self.dircache.clear()
 
     def rm(self, path, **kwargs):
         response = self._client.delete(f"/files/file/{path.lstrip('/')}")
         response.raise_for_status()
+        self.dircache.clear()

{fal-1.15.0 → fal-1.16.0}/src/fal/flags.py

@@ -2,6 +2,8 @@ from __future__ import annotations
 
 import os
 
+from fal.config import Config
+
 
 def bool_envvar(name: str):
     if name in os.environ:
@@ -14,7 +16,10 @@ DEBUG = bool_envvar("DEBUG")
 TEST_MODE = bool_envvar("ISOLATE_TEST_MODE")
 AUTH_DISABLED = bool_envvar("ISOLATE_AUTH_DISABLED")
 
-GRPC_HOST = os.getenv("FAL_HOST", "api.alpha.fal.ai")
+config = Config()
+config_host = config.get("host")
+
+GRPC_HOST = config_host or os.getenv("FAL_HOST") or "api.alpha.fal.ai"
 if not TEST_MODE:
     assert GRPC_HOST.startswith("api"), "FAL_HOST must start with 'api'"
 

{fal-1.15.0 → fal-1.16.0}/src/fal/utils.py

@@ -21,8 +21,11 @@ def load_function_from(
     file_path: str,
     function_name: str | None = None,
 ) -> LoadedFunction:
+    import os
     import runpy
+    import sys
 
+    sys.path.append(os.getcwd())
     module = runpy.run_path(file_path)
     if function_name is None:
         fal_objects = {