Remove hardcoded capability check

newfold-labs · Aug 20, 2024 · cc18c5c · cc18c5c
1 parent 4b11291
commit cc18c5c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/includes/SiteGen/SiteGen.php b/includes/SiteGen/SiteGen.php
@@ -57,7 +57,9 @@ class SiteGen {
 	 * Function to check capabilities
 	 */
 	private static function check_capabilities() {
-		return true;
+		$capability      = new SiteCapabilities();
+		$sitegen_enabled = $capability->get( 'hasAISiteGen' );
+		return $sitegen_enabled;
 	}
 
 	/**