fix: ai balance for unlimited plan

This commit is contained in:
Vasily Zubarev
2025-05-03 09:48:38 +02:00
parent 6c3641c469
commit 69bc4b3889
3 changed files with 18 additions and 16 deletions

View File

@@ -1,5 +1,5 @@
import config from "@/lib/config"
import { getSelfHostedUser, getUserByEmail, getUserById } from "@/models/users"
import { getSelfHostedUser, getUserByEmail, getUserById, SELF_HOSTED_USER } from "@/models/users"
import { User } from "@prisma/client"
import { betterAuth } from "better-auth"
import { prismaAdapter } from "better-auth/adapters/prisma"
@@ -101,3 +101,10 @@ export function isSubscriptionExpired(user: User) {
}
return user.membershipExpiresAt && user.membershipExpiresAt < new Date()
}
export function isAiBalanceExhausted(user: User) {
if (config.selfHosted.isEnabled || user.membershipPlan === SELF_HOSTED_USER.membershipPlan) {
return false
}
return user.aiBalance <= 0
}