update fal requirements

This commit is contained in:
Teknium 2025-08-21 08:10:54 -07:00
parent f4ff1f496b
commit 58d5fa1e4c

View file

@ -323,8 +323,8 @@ async def image_generate_tool(
raise ValueError("Prompt must be 1000 characters or less") raise ValueError("Prompt must be 1000 characters or less")
# Check API key availability # Check API key availability
if not os.getenv("FAL_API_KEY"): if not os.getenv("FAL_KEY"):
raise ValueError("FAL_API_KEY environment variable not set") raise ValueError("FAL_KEY environment variable not set")
# Validate parameters # Validate parameters
validated_params = _validate_parameters( validated_params = _validate_parameters(
@ -445,7 +445,7 @@ def check_fal_api_key() -> bool:
Returns: Returns:
bool: True if API key is set, False otherwise bool: True if API key is set, False otherwise
""" """
return bool(os.getenv("FAL_API_KEY")) return bool(os.getenv("FAL_KEY"))
def check_image_generation_requirements() -> bool: def check_image_generation_requirements() -> bool:
@ -502,8 +502,8 @@ if __name__ == "__main__":
api_available = check_fal_api_key() api_available = check_fal_api_key()
if not api_available: if not api_available:
print("❌ FAL_API_KEY environment variable not set") print("❌ FAL_KEY environment variable not set")
print("Please set your API key: export FAL_API_KEY='your-key-here'") print("Please set your API key: export FAL_KEY='your-key-here'")
print("Get API key at: https://fal.ai/") print("Get API key at: https://fal.ai/")
exit(1) exit(1)
else: else: