fal 1.23.0__tar.gz → 1.23.1__tar.gz

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

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

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

{fal-1.23.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.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