fal 1.22.0__tar.gz → 1.23.1__tar.gz

{fal-1.22.0/fal.egg-info → fal-1.23.1}/PKG-INFO

@@ -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 → fal-1.23.1/fal.egg-info}/PKG-INFO

@@ -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 → fal-1.23.1}/fal.egg-info/SOURCES.txt

@@ -131,6 +131,7 @@ src/fal/logging/trace.py
 src/fal/logging/user.py
 src/fal/toolkit/__init__.py
 src/fal/toolkit/exceptions.py
+src/fal/toolkit/kv.py
 src/fal/toolkit/optimize.py
 src/fal/toolkit/types.py
 src/fal/toolkit/audio/__init__.py
@@ -161,6 +162,7 @@ tests/conftest.py
 tests/integration_test.py
 tests/mainify_target.py
 tests/test_apps.py
+tests/test_kv.py
 tests/test_stability.py
 tests/assets/cat.png
 tests/cli/__init__.py

{fal-1.22.0 → fal-1.23.1}/src/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-1.22.0 → fal-1.23.1}/src/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-1.22.0 → fal-1.23.1}/src/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-1.23.1/src/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-1.22.0 → fal-1.23.1}/src/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 → fal-1.23.1}/tests/integration_test.py

@@ -23,7 +23,7 @@ from fal.toolkit import (
     Image as FalImage,
 )
 from fal.toolkit.file.file import CompressedFile
-from fal.toolkit.utils.download_utils import _get_git_revision_hash, _hash_url
+from fal.toolkit.utils.download_utils import _git_rev_parse, _hash_url
 
 EXAMPLE_FILE_URL = "https://raw.githubusercontent.com/fal-ai/fal/main/projects/fal/tests/assets/cat.png"
 
@@ -376,13 +376,13 @@ def test_clone_repository(isolated_client, mock_fal_persistent_dirs):
         first_path = clone_repository(
             EXAMPLE_REPO_URL, commit_hash=EXAMPLE_REPO_FIRST_COMMIT
         )
-        first_repo_hash = _get_git_revision_hash(first_path)
+        first_repo_hash = _git_rev_parse(first_path, "HEAD")
 
         second_path = clone_repository(
             EXAMPLE_REPO_URL, commit_hash=EXAMPLE_REPO_SECOND_COMMIT
         )
 
-        second_repo_hash = _get_git_revision_hash(second_path)
+        second_repo_hash = _git_rev_parse(second_path, "HEAD")
 
         return first_path, first_repo_hash, second_path, second_repo_hash
 

fal-1.23.1/tests/test_kv.py

@@ -0,0 +1,8 @@
+from fal.toolkit.kv import KVStore
+
+
+def test_kv_store():
+    kv = KVStore("test")
+    kv.set("test", "test")
+    assert kv.get("test") == "test"
+    assert kv.get("test2") is None