use period usage count for remaining calculations on any plan except …

…free
Automattic · Aug 28, 2024 · a06ce78 · a06ce78
1 parent 5fd7f76
commit a06ce78
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/projects/plugins/jetpack/extensions/blocks/ai-assistant/hooks/use-ai-feature/index.ts b/projects/plugins/jetpack/extensions/blocks/ai-assistant/hooks/use-ai-feature/index.ts
@@ -2,11 +2,7 @@
  * External dependencies
  */
 import { useDispatch, useSelect } from '@wordpress/data';
-import {
-	PLAN_TYPE_FREE,
-	PLAN_TYPE_TIERED,
-	usePlanType as getPlanType,
-} from '../../../../shared/use-plan-type';
+import { PLAN_TYPE_FREE, usePlanType as getPlanType } from '../../../../shared/use-plan-type';
 import type { WordPressPlansSelectors } from 'extensions/store/wordpress-com';
 
 export default function useAiFeature() {
@@ -29,7 +25,7 @@ export default function useAiFeature() {
 		const currentTierLimit = currentTier?.limit || freeRequestsLimit;
 
 		const actualRequestsCount =
-			planType === PLAN_TYPE_TIERED ? usagePeriod?.requestsCount : allTimeRequestsCount;
+			planType === PLAN_TYPE_FREE ? allTimeRequestsCount : usagePeriod?.requestsCount;
 		const actualRequestsLimit = planType === PLAN_TYPE_FREE ? freeRequestsLimit : currentTierLimit;
 
 		return {