fal 1.11.2__tar.gz → 1.11.4__tar.gz

{fal-1.11.2/fal.egg-info → fal-1.11.4}/PKG-INFO

@@ -1,17 +1,16 @@
 Metadata-Version: 2.4
 Name: fal
-Version: 1.11.2
+Version: 1.11.4
 Summary: fal is an easy-to-use Serverless Python Framework
 Author: Features & Labels <support@fal.ai>
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: isolate[build]<0.17.0,>=0.16.1
-Requires-Dist: isolate-proto<0.8.0,>=0.7.0
+Requires-Dist: isolate-proto<0.8.0,>=0.7.2
 Requires-Dist: grpcio==1.64.0
 Requires-Dist: dill==0.3.7
 Requires-Dist: cloudpickle==3.0.0
 Requires-Dist: typing-extensions<5,>=4.7.1
-Requires-Dist: click<9,>=8.1.3
 Requires-Dist: structlog<23,>=22.3.0
 Requires-Dist: opentelemetry-api<2,>=1.15.0
 Requires-Dist: opentelemetry-sdk<2,>=1.15.0
@@ -22,7 +21,7 @@ Requires-Dist: rich<14,>=13.3.2
 Requires-Dist: rich_argparse
 Requires-Dist: packaging>=21.3
 Requires-Dist: pathspec<1,>=0.11.1
-Requires-Dist: pydantic!=2.0.*,!=2.1.*,!=2.2.*,!=2.3.*,!=2.4.*,<3
+Requires-Dist: pydantic!=2.0.*,!=2.1.*,!=2.2.*,!=2.3.*,!=2.4.*,<2.11
 Requires-Dist: structlog>=22.0
 Requires-Dist: fastapi<1,>=0.99.1
 Requires-Dist: starlette-exporter>=0.21.0
@@ -47,6 +46,7 @@ Provides-Extra: test
 Requires-Dist: pytest<8; extra == "test"
 Requires-Dist: pytest-asyncio; extra == "test"
 Requires-Dist: pytest-xdist; extra == "test"
+Requires-Dist: pytest-timeout; extra == "test"
 Requires-Dist: flaky; extra == "test"
 Requires-Dist: boto3; extra == "test"
 Provides-Extra: dev

{fal-1.11.2 → fal-1.11.4/fal.egg-info}/PKG-INFO

@@ -1,17 +1,16 @@
 Metadata-Version: 2.4
 Name: fal
-Version: 1.11.2
+Version: 1.11.4
 Summary: fal is an easy-to-use Serverless Python Framework
 Author: Features & Labels <support@fal.ai>
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: isolate[build]<0.17.0,>=0.16.1
-Requires-Dist: isolate-proto<0.8.0,>=0.7.0
+Requires-Dist: isolate-proto<0.8.0,>=0.7.2
 Requires-Dist: grpcio==1.64.0
 Requires-Dist: dill==0.3.7
 Requires-Dist: cloudpickle==3.0.0
 Requires-Dist: typing-extensions<5,>=4.7.1
-Requires-Dist: click<9,>=8.1.3
 Requires-Dist: structlog<23,>=22.3.0
 Requires-Dist: opentelemetry-api<2,>=1.15.0
 Requires-Dist: opentelemetry-sdk<2,>=1.15.0
@@ -22,7 +21,7 @@ Requires-Dist: rich<14,>=13.3.2
 Requires-Dist: rich_argparse
 Requires-Dist: packaging>=21.3
 Requires-Dist: pathspec<1,>=0.11.1
-Requires-Dist: pydantic!=2.0.*,!=2.1.*,!=2.2.*,!=2.3.*,!=2.4.*,<3
+Requires-Dist: pydantic!=2.0.*,!=2.1.*,!=2.2.*,!=2.3.*,!=2.4.*,<2.11
 Requires-Dist: structlog>=22.0
 Requires-Dist: fastapi<1,>=0.99.1
 Requires-Dist: starlette-exporter>=0.21.0
@@ -47,6 +46,7 @@ Provides-Extra: test
 Requires-Dist: pytest<8; extra == "test"
 Requires-Dist: pytest-asyncio; extra == "test"
 Requires-Dist: pytest-xdist; extra == "test"
+Requires-Dist: pytest-timeout; extra == "test"
 Requires-Dist: flaky; extra == "test"
 Requires-Dist: boto3; extra == "test"
 Provides-Extra: dev

{fal-1.11.2 → fal-1.11.4}/fal.egg-info/requires.txt

@@ -1,10 +1,9 @@
 isolate[build]<0.17.0,>=0.16.1
-isolate-proto<0.8.0,>=0.7.0
+isolate-proto<0.8.0,>=0.7.2
 grpcio==1.64.0
 dill==0.3.7
 cloudpickle==3.0.0
 typing-extensions<5,>=4.7.1
-click<9,>=8.1.3
 structlog<23,>=22.3.0
 opentelemetry-api<2,>=1.15.0
 opentelemetry-sdk<2,>=1.15.0
@@ -15,7 +14,7 @@ rich<14,>=13.3.2
 rich_argparse
 packaging>=21.3
 pathspec<1,>=0.11.1
-pydantic!=2.0.*,!=2.1.*,!=2.2.*,!=2.3.*,!=2.4.*,<3
+pydantic!=2.0.*,!=2.1.*,!=2.2.*,!=2.3.*,!=2.4.*,<2.11
 structlog>=22.0
 fastapi<1,>=0.99.1
 starlette-exporter>=0.21.0
@@ -48,5 +47,6 @@ sphinx-autodoc-typehints
 pytest<8
 pytest-asyncio
 pytest-xdist
+pytest-timeout
 flaky
 boto3

{fal-1.11.2 → fal-1.11.4}/pyproject.toml

@@ -23,12 +23,11 @@ readme = "README.md"
 requires-python = ">=3.8"
 dependencies = [
     "isolate[build]>=0.16.1,<0.17.0",
-    "isolate-proto>=0.7.0,<0.8.0",
+    "isolate-proto>=0.7.2,<0.8.0",
     "grpcio==1.64.0",
     "dill==0.3.7",
     "cloudpickle==3.0.0",
     "typing-extensions>=4.7.1,<5",
-    "click>=8.1.3,<9",
     "structlog>=22.3.0,<23",
     "opentelemetry-api>=1.15.0,<2",
     "opentelemetry-sdk>=1.15.0,<2",
@@ -39,7 +38,7 @@ dependencies = [
     "rich_argparse",
     "packaging>=21.3",
     "pathspec>=0.11.1,<1",
-    "pydantic!=2.0.*,!=2.1.*,!=2.2.*,!=2.3.*,!=2.4.*,<3",
+    "pydantic!=2.0.*,!=2.1.*,!=2.2.*,!=2.3.*,!=2.4.*,<2.11",
     # serve=True dependencies
     "structlog>=22.0",
     "fastapi>=0.99.1,<1",
@@ -71,6 +70,7 @@ test = [
     "pytest<8",
     "pytest-asyncio",
     "pytest-xdist",
+    "pytest-timeout",
     "flaky",
     "boto3",
 ]
@@ -93,3 +93,6 @@ known-first-party = ["fal"]
 
 [tool.ruff.lint.pyupgrade]
 keep-runtime-typing = true
+
+[tool.pytest.ini_options]
+timeout = 300

{fal-1.11.2 → fal-1.11.4}/src/fal/_fal_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.11.2'
-__version_tuple__ = version_tuple = (1, 11, 2)
+__version__ = version = '1.11.4'
+__version_tuple__ = version_tuple = (1, 11, 4)

{fal-1.11.2 → fal-1.11.4}/src/fal/api.py

@@ -1088,7 +1088,10 @@ class BaseServable:
                     type(exc), exc, exc.__traceback__
                 )
 
-            print(json.dumps({"traceback": "".join(formatted_exception[::-1])}))
+            print(
+                json.dumps({"traceback": "".join(formatted_exception[::-1])}),
+                flush=True,
+            )
 
             if _is_cuda_oom_exception(exc):
                 return await cuda_out_of_memory_exception_handler(

{fal-1.11.2 → fal-1.11.4}/src/fal/app.py

@@ -268,7 +268,7 @@ class App(fal.api.BaseServable):
         "keep_alive": 60,
     }
     app_name: ClassVar[str]
-    app_auth: ClassVar[Literal["private", "public", "shared"]] = "private"
+    app_auth: ClassVar[Literal["private", "public", "shared", None]] = None
     request_timeout: ClassVar[int | None] = None
     startup_timeout: ClassVar[int | None] = None
 

{fal-1.11.2 → fal-1.11.4}/src/fal/auth/__init__.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from threading import Lock
 from typing import Optional
 
-import click
-
 from fal.auth import auth0, local
 from fal.config import Config
 from fal.console import console
 from fal.console.icons import CHECK_ICON
+from fal.exceptions import FalServerlessException
 from fal.exceptions.auth import UnauthenticatedException
 
 
@@ -85,7 +84,7 @@ def login():
 def logout():
     refresh_token, _ = local.load_token()
     if refresh_token is None:
-        raise click.ClickException(message="You're not logged in")
+        raise FalServerlessException("You're not logged in")
     auth0.revoke(refresh_token)
     with local.lock_token():
         local.delete_token()

{fal-1.11.2 → fal-1.11.4}/src/fal/auth/auth0.py

@@ -4,12 +4,12 @@ import functools
 import time
 import warnings
 
-import click
 import httpx
 
 from fal.console import console
 from fal.console.icons import CHECK_ICON
 from fal.console.ux import maybe_open_browser_tab
+from fal.exceptions import FalServerlessException
 
 WEBSITE_URL = "https://fal.ai"
 
@@ -55,7 +55,7 @@ def login() -> dict:
     )
 
     if device_code_response.status_code != 200:
-        raise click.ClickException("Error generating the device code")
+        raise FalServerlessException("Error generating the device code")
 
     device_code_data = device_code_response.json()
     device_user_code = device_code_data["user_code"]
@@ -92,7 +92,7 @@ def login() -> dict:
 
             elif token_data["error"] not in ("authorization_pending", "slow_down"):
                 status.update(spinner=None)
-                raise click.ClickException(token_data["error_description"])
+                raise FalServerlessException(token_data["error_description"])
 
             else:
                 time.sleep(device_code_data["interval"])
@@ -115,7 +115,7 @@ def refresh(token: str) -> dict:
 
         return token_data
     else:
-        raise click.ClickException(token_data["error_description"])
+        raise FalServerlessException(token_data["error_description"])
 
 
 def revoke(token: str):
@@ -130,7 +130,7 @@ def revoke(token: str):
 
     if token_response.status_code != 200:
         token_data = token_response.json()
-        raise click.ClickException(token_data["error_description"])
+        raise FalServerlessException(token_data["error_description"])
 
     _open_browser(logout_url(WEBSITE_URL), None)
 
@@ -142,7 +142,7 @@ def get_user_info(bearer_token: str) -> dict:
     )
 
     if userinfo_response.status_code != 200:
-        raise click.ClickException(userinfo_response.content.decode("utf-8"))
+        raise FalServerlessException(userinfo_response.content.decode("utf-8"))
 
     return userinfo_response.json()
 

{fal-1.11.2 → fal-1.11.4}/src/fal/cli/apps.py

@@ -2,6 +2,8 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING
 
+import fal.cli.runners as runners
+
 from ._utils import get_client
 from .parser import FalClientParser
 
@@ -268,59 +270,17 @@ def _add_set_rev_parser(subparsers, parents):
 
 
 def _runners(args):
-    from rich.table import Table
-
     client = get_client(args.host, args.team)
     with client.connect() as connection:
-        runners = connection.list_alias_runners(alias=args.app_name)
-
-    table = Table()
-    table.add_column("Runner ID")
-    table.add_column("In Flight Requests")
-    table.add_column("Missing Leases")
-    table.add_column("Expires In")
-    table.add_column("Uptime")
-
-    for runner in runners:
-        num_leases_with_request = len(
-            [
-                lease
-                for lease in runner.external_metadata.get("leases", [])
-                if lease.get("request_id") is not None
-            ]
-        )
-
-        table.add_row(
-            runner.runner_id,
-            str(runner.in_flight_requests),
-            str(runner.in_flight_requests - num_leases_with_request),
-            (
-                "N/A (active)"
-                if runner.expiration_countdown is None
-                else f"{runner.expiration_countdown}s"
-            ),
-            f"{runner.uptime} ({runner.uptime.total_seconds()}s)",
-        )
-
-    args.console.print(f"Runners: {len(runners)}")
-    args.console.print(table)
-
-    requests_table = Table()
-    requests_table.add_column("Runner ID")
-    requests_table.add_column("Request ID")
-    requests_table.add_column("Caller ID")
-
-    for runner in runners:
-        for lease in runner.external_metadata.get("leases", []):
-            if not (req_id := lease.get("request_id")):
-                continue
+        alias_runners = connection.list_alias_runners(alias=args.app_name)
 
-            requests_table.add_row(
-                runner.runner_id,
-                req_id,
-                lease.get("caller_user_id") or "",
-            )
+    runners_table = runners.runners_table(alias_runners)
+    args.console.print(f"Runners: {len(alias_runners)}")
+    # Drop the alias column, which is the first column
+    runners_table.columns.pop(0)
+    args.console.print(runners_table)
 
+    requests_table = runners.runners_requests_table(alias_runners)
     args.console.print(f"Requests: {len(requests_table.rows)}")
     args.console.print(requests_table)
 

{fal-1.11.2 → fal-1.11.4}/src/fal/cli/deploy.py

@@ -97,7 +97,7 @@ def _deploy_from_reference(
     )
     isolated_function = loaded.function
     app_name = app_name or loaded.app_name  # type: ignore
-    app_auth = auth or loaded.app_auth or "private"
+    app_auth = auth or loaded.app_auth
     deployment_strategy = deployment_strategy or "recreate"
 
     app_id = host.register(

{fal-1.11.2 → fal-1.11.4}/src/fal/cli/main.py

@@ -2,7 +2,7 @@ import argparse
 
 import rich
 
-from fal import __version__
+from fal._version import __version__
 from fal.console import console
 from fal.console.icons import CROSS_ICON
 

fal-1.11.4/src/fal/cli/runners.py

@@ -0,0 +1,133 @@
+from __future__ import annotations
+
+from typing import List
+
+from fal.sdk import RunnerInfo
+
+from ._utils import get_client
+from .parser import FalClientParser
+
+
+def runners_table(runners: List[RunnerInfo]):
+    from rich.table import Table
+
+    table = Table()
+    table.add_column("Alias")
+    table.add_column("Runner ID")
+    table.add_column("In Flight Requests")
+    table.add_column("Missing Leases")
+    table.add_column("Expires In")
+    table.add_column("Uptime")
+    table.add_column("Revision")
+
+    for runner in runners:
+        num_leases_with_request = len(
+            [
+                lease
+                for lease in runner.external_metadata.get("leases", [])
+                if lease.get("request_id") is not None
+            ]
+        )
+
+        table.add_row(
+            runner.alias,
+            runner.runner_id,
+            str(runner.in_flight_requests),
+            str(runner.in_flight_requests - num_leases_with_request),
+            (
+                "N/A (active)"
+                if runner.expiration_countdown is None
+                else f"{runner.expiration_countdown}s"
+            ),
+            f"{runner.uptime} ({runner.uptime.total_seconds()}s)",
+            runner.revision,
+        )
+
+    return table
+
+
+def runners_requests_table(runners: list[RunnerInfo]):
+    from rich.table import Table
+
+    table = Table()
+    table.add_column("Runner ID")
+    table.add_column("Request ID")
+    table.add_column("Caller ID")
+
+    for runner in runners:
+        for lease in runner.external_metadata.get("leases", []):
+            if not (req_id := lease.get("request_id")):
+                continue
+
+            table.add_row(
+                runner.runner_id,
+                req_id,
+                lease.get("caller_user_id") or "",
+            )
+
+    return table
+
+
+def _kill(args):
+    client = get_client(args.host, args.team)
+    with client.connect() as connection:
+        connection.kill_runner(args.id)
+
+
+def _list(args):
+    client = get_client(args.host, args.team)
+    with client.connect() as connection:
+        runners = connection.list_runners()
+        args.console.print(f"Runners: {len(runners)}")
+        args.console.print(runners_table(runners))
+
+        requests_table = runners_requests_table(runners)
+        args.console.print(f"Requests: {len(requests_table.rows)}")
+        args.console.print(requests_table)
+
+
+def _add_kill_parser(subparsers, parents):
+    kill_help = "Kill a runner."
+    parser = subparsers.add_parser(
+        "kill",
+        description=kill_help,
+        help=kill_help,
+        parents=parents,
+    )
+    parser.add_argument(
+        "id",
+        help="Runner ID.",
+    )
+    parser.set_defaults(func=_kill)
+
+
+def _add_list_parser(subparsers, parents):
+    list_help = "List runners."
+    parser = subparsers.add_parser(
+        "list",
+        description=list_help,
+        help=list_help,
+        parents=parents,
+    )
+    parser.set_defaults(func=_list)
+
+
+def add_parser(main_subparsers, parents):
+    runners_help = "Manage fal runners."
+    parser = main_subparsers.add_parser(
+        "runners",
+        description=runners_help,
+        help=runners_help,
+        parents=parents,
+        aliases=["machine"],  # backwards compatibility
+    )
+
+    subparsers = parser.add_subparsers(
+        title="Commands",
+        metavar="command",
+        required=True,
+        parser_class=FalClientParser,
+    )
+
+    _add_kill_parser(subparsers, parents)
+    _add_list_parser(subparsers, parents)

{fal-1.11.2 → fal-1.11.4}/src/fal/config.py

@@ -72,7 +72,7 @@ class Config:
         if not self.profile:
             raise ValueError("No profile set.")
 
-        del self._config[self.profile][key]
+        self._config.get(self.profile, {}).pop(key, None)
 
     def get_internal(self, key: str) -> Optional[str]:
         if SETTINGS_SECTION not in self._config:

{fal-1.11.2 → fal-1.11.4}/src/fal/sdk.py

@@ -246,6 +246,8 @@ class RunnerInfo:
     expiration_countdown: Optional[int]
     uptime: timedelta
     external_metadata: dict[str, Any]
+    revision: str
+    alias: str
 
 
 @dataclass
@@ -392,6 +394,8 @@ def _from_grpc_runner_info(message: isolate_proto.RunnerInfo) -> RunnerInfo:
         ),
         uptime=timedelta(seconds=message.uptime),
         external_metadata=external_metadata,
+        revision=message.revision,
+        alias=message.alias,
     )
 
 
@@ -571,11 +575,12 @@ class FalServerlessConnection:
         else:
             wrapped_requirements = None
 
+        auth_mode = None
         if application_auth_mode == "public":
             auth_mode = isolate_proto.ApplicationAuthMode.PUBLIC
         elif application_auth_mode == "shared":
             auth_mode = isolate_proto.ApplicationAuthMode.SHARED
-        else:
+        elif application_auth_mode == "private":
             auth_mode = isolate_proto.ApplicationAuthMode.PRIVATE
 
         struct_metadata = None
@@ -750,3 +755,8 @@ class FalServerlessConnection:
     def kill_runner(self, runner_id: str) -> None:
         request = isolate_proto.KillRunnerRequest(runner_id=runner_id)
         self.stub.KillRunner(request)
+
+    def list_runners(self) -> list[RunnerInfo]:
+        request = isolate_proto.ListRunnersRequest()
+        response = self.stub.ListRunners(request)
+        return [from_grpc(runner) for runner in response.runners]

{fal-1.11.2 → fal-1.11.4}/tests/test_apps.py

@@ -576,6 +576,41 @@ def test_app_client_async():
     assert result == {"slept": True}
 
 
+# If the logging subsystem is not working for some nodes, this test will flake
+@pytest.mark.flaky(max_runs=10)
+def test_traceback_logs(test_exception_app: AppClient):
+    date = (datetime.utcnow() - timedelta(seconds=1)).isoformat()
+
+    with pytest.raises(AppClientError):
+        test_exception_app.fail({})
+
+    with httpx.Client(
+        base_url=REST_CLIENT.base_url,
+        headers=REST_CLIENT.get_headers(),
+        timeout=300,
+    ) as client:
+        # Give some time for logs to propagate through the logging subsystem.
+        for _ in range(10):
+            time.sleep(2)
+            response = client.get(
+                REST_CLIENT.base_url + f"/logs/?traceback=true&since={date}"
+            )
+
+            logs = response.json()
+            if len(logs) > 0:
+                break
+
+        assert len(logs) > 0
+        for log in logs:
+            assert log["message"].count("\n") > 1, "Logs should be multi-line"
+            assert (
+                '{"traceback":' not in log["message"]
+            ), "Logs should not be JSON-wrapped"
+            assert (
+                "this app is designed to fail" in log["message"]
+            ), "Logs should contain the traceback message"
+
+
 def test_app_openapi_spec_metadata(test_app: str, request: pytest.FixtureRequest):
     user_id, _, app_id = test_app.partition("/")
     res = app_metadata.sync_detailed(
@@ -617,6 +652,18 @@ def test_404_response(test_app: str, request: pytest.FixtureRequest):
         apps.run(test_app, path="/other", arguments={"lhs": 1, "rhs": 2})
 
 
+def test_app_no_auth():
+    # This will just pass for users with shared apps access
+    app_alias = str(uuid.uuid4()) + "-alias"
+    with pytest.raises(api.FalServerlessError, match="Must specify auth_mode"):
+        addition_app.host.register(
+            func=addition_app.func,
+            options=addition_app.options,
+            # random enough
+            application_name=app_alias,
+        )
+
+
 def test_app_deploy_scale(aliased_app: Tuple[str, str]):
     from dataclasses import replace
 
@@ -658,6 +705,8 @@ def test_app_deploy_scale(aliased_app: Tuple[str, str]):
         assert found.max_multiplexing == 30
 
 
+# List aliases is taking long
+@pytest.mark.timeout(600)
 def test_app_update_app(aliased_app: Tuple[str, str]):
     app_revision, app_alias = aliased_app
 
@@ -708,6 +757,8 @@ def test_app_update_app(aliased_app: Tuple[str, str]):
         assert res.max_multiplexing == new_max_multiplexing
 
 
+# List aliases is taking long
+@pytest.mark.timeout(600)
 def test_app_set_delete_alias(aliased_app: Tuple[str, str]):
     app_revision, app_alias = aliased_app
 
@@ -831,41 +882,6 @@ def test_workflows(test_app: str):
             assert data["result"] == 10
 
 
-# If the logging subsystem is not working for some nodes, this test will flake
-@pytest.mark.flaky(max_runs=5)
-def test_traceback_logs(test_exception_app: AppClient):
-    date = (datetime.utcnow() - timedelta(seconds=1)).isoformat()
-
-    with pytest.raises(AppClientError):
-        test_exception_app.fail({})
-
-    with httpx.Client(
-        base_url=REST_CLIENT.base_url,
-        headers=REST_CLIENT.get_headers(),
-        timeout=300,
-    ) as client:
-        # Give some time for logs to propagate through the logging subsystem.
-        for _ in range(10):
-            time.sleep(2)
-            response = client.get(
-                REST_CLIENT.base_url + f"/logs/?traceback=true&since={date}"
-            )
-
-            logs = response.json()
-            if len(logs) > 0:
-                break
-
-        assert len(logs) > 0
-        for log in logs:
-            assert log["message"].count("\n") > 1, "Logs should be multi-line"
-            assert (
-                '{"traceback":' not in log["message"]
-            ), "Logs should not be JSON-wrapped"
-            assert (
-                "this app is designed to fail" in log["message"]
-            ), "Logs should contain the traceback message"
-
-
 def test_app_exceptions(test_exception_app: AppClient):
     with pytest.raises(AppClientError) as app_exc:
         test_exception_app.app_exception({})

{fal-1.11.2 → fal-1.11.4}/tests/test_stability.py

@@ -559,6 +559,8 @@ def test_serve_on_off(isolated_client):
 
 
 def test_worker_env_vars(isolated_client):
+    from fal.flags import GRPC_HOST
+
     @isolated_client("virtualenv", keep_alive=5)
     def get_env_var(name: str) -> str | None:
         import os
@@ -567,8 +569,8 @@ def test_worker_env_vars(isolated_client):
 
     fal_host = get_env_var("FAL_HOST")
     assert fal_host, "FAL_HOST is not set"
-    assert fal_host.startswith("api.")
-    assert fal_host.endswith(".shark.fal.ai")
+    assert fal_host == GRPC_HOST, "FAL_HOST is not set to the correct value"
+    print(f"FAL_HOST: {fal_host}, GRPC_HOST: {GRPC_HOST}")
 
     fal_key_id = get_env_var("FAL_KEY_ID")
     assert fal_key_id, "FAL_KEY_ID is not set"