fal 1.23.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.23.0'
-__version_tuple__ = version_tuple = (1, 23, 0)
+__version__ = version = '1.23.1'
+__version_tuple__ = version_tuple = (1, 23, 1)

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-1.23.0.dist-info → fal-1.23.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fal
-Version: 1.23.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.23.0.dist-info → fal-1.23.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 fal/__init__.py,sha256=wXs1G0gSc7ZK60-bHe-B2m0l_sA6TrFk4BxY0tMoLe8,784
 fal/__main__.py,sha256=4JMK66Wj4uLZTKbF-sT3LAxOsr6buig77PmOkJCRRxw,83
-fal/_fal_version.py,sha256=5phfzDrMzuxkew7dkMe3AQHf5yuGHlEWJi5-3xyi6aM,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
@@ -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
@@ -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.23.0.dist-info/METADATA,sha256=PK_7YpWltcIwUVpO_iPYg4zEhfyX6ZJuagR1pMbeNQE,4085
-fal-1.23.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fal-1.23.0.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
-fal-1.23.0.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
-fal-1.23.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,,