From 58d5fa1e4cec2783962328d887b12810d6de0c8c Mon Sep 17 00:00:00 2001
From: Teknium <127238744+teknium1@users.noreply.github.com>
Date: Thu, 21 Aug 2025 08:10:54 -0700
Subject: [PATCH] update fal requirements

---
 image_generation_tool.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/image_generation_tool.py b/image_generation_tool.py
index 307fb90f..09d4e51d 100644
--- a/image_generation_tool.py
+++ b/image_generation_tool.py
@@ -323,8 +323,8 @@ async def image_generate_tool(
             raise ValueError("Prompt must be 1000 characters or less")
         
         # Check API key availability
-        if not os.getenv("FAL_API_KEY"):
-            raise ValueError("FAL_API_KEY environment variable not set")
+        if not os.getenv("FAL_KEY"):
+            raise ValueError("FAL_KEY environment variable not set")
         
         # Validate parameters
         validated_params = _validate_parameters(
@@ -445,7 +445,7 @@ def check_fal_api_key() -> bool:
     Returns:
         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:
@@ -502,8 +502,8 @@ if __name__ == "__main__":
     api_available = check_fal_api_key()
     
     if not api_available:
-        print("❌ FAL_API_KEY environment variable not set")
-        print("Please set your API key: export FAL_API_KEY='your-key-here'")
+        print("❌ FAL_KEY environment variable not set")
+        print("Please set your API key: export FAL_KEY='your-key-here'")
         print("Get API key at: https://fal.ai/")
         exit(1)
     else: