fal 1.22.0__py3-none-any.whl → 1.23.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.22.0'
-__version_tuple__ = version_tuple = (1, 22, 0)
+__version__ = version = '1.23.1'
+__version_tuple__ = version_tuple = (1, 23, 1)

fal/container.py

@@ -19,6 +19,7 @@ class ContainerImage:
     builder: Optional[Builder] = field(default=None)
     compression: str = DEFAULT_COMPRESSION
     force_compression: bool = DEFAULT_FORCE_COMPRESSION
+    secrets: Dict[str, str] = field(default_factory=dict)
 
     def __post_init__(self) -> None:
         if self.registries:
@@ -51,4 +52,5 @@ class ContainerImage:
             "builder": self.builder,
             "compression": self.compression,
             "force_compression": self.force_compression,
+            "secrets": self.secrets,
         }

fal/toolkit/exceptions.py

@@ -8,3 +8,9 @@ class FileUploadException(FalTookitException):
     """Raised when file upload fails"""
 
     pass
+
+
+class KVStoreException(FalTookitException):
+    """Raised when KV store operation fails"""
+
+    pass

fal/toolkit/kv.py

@@ -0,0 +1,85 @@
+import json
+from typing import Any, Dict, Optional
+from urllib.error import HTTPError
+from urllib.request import Request
+
+from fal.toolkit.exceptions import KVStoreException
+from fal.toolkit.file.providers.fal import _maybe_retry_request, fal_v3_token_manager
+
+FAL_KV_HOST = "https://kv.fal.media"
+
+
+class KVStore:
+    """A key-value store client for interacting with the FAL KV service.
+
+    Args:
+        db_name: The name of the database/namespace to use for this KV store.
+    """
+
+    def __init__(self, db_name: str):
+        self.db_name = db_name
+
+    @property
+    def auth_headers(self) -> Dict[str, str]:
+        token = fal_v3_token_manager.get_token()
+        return {
+            "Authorization": f"{token.token_type} {token.token}",
+            "User-Agent": "fal/0.1.0",
+        }
+
+    def get(self, key: str) -> Optional[str]:
+        """Retrieve a value from the key-value store.
+
+        Args:
+            key: The key to retrieve the value for.
+
+        Returns:
+            The value associated with the key, or None if the key doesn't exist.
+        """
+        response = self._send_request(
+            method="GET",
+            path=f"/get/{self.db_name}/{key}",
+        )
+
+        if response is None:
+            return None
+
+        return response["value"]
+
+    def set(self, key: str, value: str) -> None:
+        """Store a value in the key-value store.
+
+        Args:
+            key: The key to store the value under.
+            value: The value to store.
+        """
+        self._send_request(
+            method="PUT",
+            path=f"/set/{self.db_name}/{key}",
+            data=value.encode(),
+        )
+
+    def _send_request(
+        self,
+        method: str,
+        path: str,
+        data: Optional[bytes] = None,
+    ) -> Optional[Dict[str, Any]]:
+        headers = {
+            **self.auth_headers,
+            "Accept": "application/json",
+        }
+
+        url = FAL_KV_HOST + path
+        request = Request(url, headers=headers, method=method, data=data)
+        try:
+            with _maybe_retry_request(request) as response:
+                result = json.load(response)
+        except HTTPError as e:
+            if e.status == 404:
+                return None
+            raise KVStoreException(
+                f"Error sending request. Status {e.status}: {e.reason}"
+            )
+
+        return result

fal/toolkit/utils/download_utils.py

@@ -427,13 +427,22 @@ def clone_repository(
     if repo_name is None:
         repo_name = Path(https_url).stem
         if commit_hash:
+            if len(commit_hash) < 8:
+                raise ValueError(f"Commit hash '{commit_hash}' is too short.")
             repo_name += f"-{commit_hash[:8]}"
 
     local_repo_path = Path(target_dir) / repo_name  # type: ignore[arg-type]
 
     if local_repo_path.exists():
-        local_repo_commit_hash = _get_git_revision_hash(local_repo_path)
-        if local_repo_commit_hash == commit_hash and not force:
+        local_repo_commit_hash = _git_rev_parse(local_repo_path, "HEAD")
+        full_commit_hash = (
+            _git_rev_parse(local_repo_path, commit_hash) if commit_hash else None
+        )
+        if (
+            full_commit_hash
+            and local_repo_commit_hash == full_commit_hash
+            and not force
+        ):
             if include_to_path:
                 __add_local_path_to_sys_path(local_repo_path)
             return local_repo_path
@@ -516,12 +525,12 @@ def __add_local_path_to_sys_path(local_path: Path | str):
         sys.path.insert(0, local_path_str)
 
 
-def _get_git_revision_hash(repo_path: Path) -> str:
+def _git_rev_parse(repo_path: Path, ref: str) -> str:
     import subprocess
 
     try:
         return subprocess.check_output(
-            ["git", "rev-parse", "HEAD"],
+            ["git", "rev-parse", ref],
             cwd=repo_path,
             text=True,
             stderr=subprocess.STDOUT,

{fal-1.22.0.dist-info → fal-1.23.1.dist-info}/METADATA

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

{fal-1.22.0.dist-info → fal-1.23.1.dist-info}/RECORD

@@ -1,13 +1,13 @@
 fal/__init__.py,sha256=wXs1G0gSc7ZK60-bHe-B2m0l_sA6TrFk4BxY0tMoLe8,784
 fal/__main__.py,sha256=4JMK66Wj4uLZTKbF-sT3LAxOsr6buig77PmOkJCRRxw,83
-fal/_fal_version.py,sha256=kabBVDEUDLijsKri4qrNeWMmy6rv5W4lO7kJYfMb9Mg,513
+fal/_fal_version.py,sha256=-TCybTpn5d1ogABxKp1Mlx4Oiz9LQCNVDc5ZmAiLfyE,513
 fal/_serialization.py,sha256=npXNsFJ5G7jzBeBIyVMH01Ww34mGY4XWhHpRbSrTtnQ,7598
 fal/_version.py,sha256=1BbTFnucNC_6ldKJ_ZoC722_UkW4S9aDBSW9L0fkKAw,2315
 fal/api.py,sha256=vTYWeKnNg3japot1vOV_JSk812aeB6a2Y7lC1ed-VmM,47277
 fal/app.py,sha256=pMf7P9iVEcw5HiFCYgSdHkjX6f-1SPe06EOwYH1ImGA,24079
 fal/apps.py,sha256=pzCd2mrKl5J_4oVc40_pggvPtFahXBCdrZXWpnaEJVs,12130
 fal/config.py,sha256=BEMH10B2bfWJ9yNawnLG6v3kBLnLmkhMe201EAODzs4,3124
-fal/container.py,sha256=OvR-Zq-NPbYFHTnw0SBUUFxr890Fgbe68J2kSJEpLOk,1905
+fal/container.py,sha256=FTsa5hOW4ars-yV1lUoc0BNeIIvAZcpw7Ftyt3A4m_w,2000
 fal/files.py,sha256=1eOyrj1M0hTixZdbtQ1ogJqWpLd7UfiOt1Rxihnqh8g,3565
 fal/flags.py,sha256=QonyDM7R2GqfAB1bJr46oriu-fHJCkpUwXuSdanePWg,987
 fal/project.py,sha256=QgfYfMKmNobMPufrAP_ga1FKcIAlSbw18Iar1-0qepo,2650
@@ -52,7 +52,8 @@ fal/logging/style.py,sha256=ckIgHzvF4DShM5kQh8F133X53z_vF46snuDHVmo_h9g,386
 fal/logging/trace.py,sha256=OhzB6d4rQZimBc18WFLqH_9BGfqFFumKKTAGSsmWRMg,1904
 fal/logging/user.py,sha256=H7Pg-nqhpzsUb5f6uXyZUeLWAsr3oImQEaYSCIIAlqo,818
 fal/toolkit/__init__.py,sha256=GR5KxAsNODlhs-DTarJcb5raujActubn0afR7FPueWs,886
-fal/toolkit/exceptions.py,sha256=elHZ7dHCJG5zlHGSBbz-ilkZe9QUvQMomJFi8Pt91LA,198
+fal/toolkit/exceptions.py,sha256=8-EMuqDXEofPu-eVoWowc7WEM-ifusithyv6tnsm2MM,301
+fal/toolkit/kv.py,sha256=5kMk-I5PMRORK4TYc0jqqowjqKkbk7zUIgz9rAIztxE,2364
 fal/toolkit/optimize.py,sha256=p75sovF0SmRP6zxzpIaaOmqlxvXB_xEz3XPNf59EF7w,1339
 fal/toolkit/types.py,sha256=kkbOsDKj1qPGb1UARTBp7yuJ5JUuyy7XQurYUBCdti8,4064
 fal/toolkit/audio/__init__.py,sha256=sqNVfrKbppWlIGLoFTaaNTxLpVXsFHxOSHLA5VG547A,35
@@ -73,7 +74,7 @@ fal/toolkit/image/nsfw_filter/inference.py,sha256=BhIPF_zxRLetThQYxDDF0sdx9VRwvu
 fal/toolkit/image/nsfw_filter/model.py,sha256=63mu8D15z_IosoRUagRLGHy6VbLqFmrG-yZqnu2vVm4,457
 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=sy1MNmYAELyBScSsG7QNxzOCCxFIPusol_RPhNyqZ_w,20191
+fal/toolkit/utils/download_utils.py,sha256=0OtZoiLGQZUXlYG7MxiTSZgfT7L4qOxQIflvmbPcqSs,20501
 fal/toolkit/utils/endpoint.py,sha256=5EXoshA2PD_brjEfhNWAWasjqLOCRrjBnfhj6QGuMt8,782
 fal/toolkit/utils/retry.py,sha256=0pnKqs1Y2dADMAk2944FZr68ZL3wQC_5hqApfgyMf_8,1531
 fal/toolkit/video/__init__.py,sha256=YV0jWpuvoA_CDFQXhd3zOvilFLKH7DYARrbzR7hWhpE,35
@@ -141,8 +142,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.22.0.dist-info/METADATA,sha256=1h6ZFhJJi6nYLhYDC8vY_rVsGkjd0SY6cwAjtoHU8v4,4085
-fal-1.22.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fal-1.22.0.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
-fal-1.22.0.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
-fal-1.22.0.dist-info/RECORD,,
+fal-1.23.1.dist-info/METADATA,sha256=MvSJ1DBSdozdRvsUUw-9ycHm1yWWuZ--JvifArmgOMo,4085
+fal-1.23.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fal-1.23.1.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
+fal-1.23.1.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
+fal-1.23.1.dist-info/RECORD,,