diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index 6de89f9f..f2fb5c0e 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -301,6 +301,11 @@ def subscription_to_state(subscription: Subscription) -> str: return SubscriptionState.INVALID.value elif subscription.type == Subscription.Type.TRIAL: # Check if the trial has expired + if not subscription.renewal_date: + # If the renewal date is not set, set it to the current date + trial length and evaluate + subscription.renewal_date = subscription.created_at + timedelta(days=LENGTH_OF_FREE_TRIAL) + subscription.save() + if subscription.renewal_date and datetime.now(tz=timezone.utc) > subscription.renewal_date: return SubscriptionState.EXPIRED.value return SubscriptionState.TRIAL.value