Address feedback

This commit is contained in:
Krzysztof Czerwinski 2025-01-07 14:11:24 +01:00
parent 5ec40c27b0
commit f28b298dd7
2 changed files with 7 additions and 7 deletions

View File

@ -306,7 +306,7 @@ class UserCredit(UserCreditBase):
raise ValueError("Either session_id or user_id must be provided")
# Retrieve CreditTransaction
credit_transaction = await CreditTransaction.prisma().find_first_or_raise(
credit_transaction = await CreditTransaction.prisma().find_first(
where={
"OR": [
(
@ -315,13 +315,14 @@ class UserCredit(UserCreditBase):
else {"transactionKey": ""}
),
{"userId": user_id} if user_id is not None else {"userId": ""},
]
],
"isActive": False,
},
order={"createdAt": "desc"},
)
# This can be called multiple times for one id, so ignore if already fulfilled
if credit_transaction.isActive:
if not credit_transaction:
return
# Retrieve the Checkout Session from the API
@ -366,8 +367,7 @@ class DisabledUserCredit(UserCreditBase):
def get_user_credit_model() -> UserCreditBase:
# return UserCredit()
if settings.config.enable_credit.lower() == "true":
if settings.config.enable_credit:
return UserCredit()
else:
return DisabledUserCredit()

View File

@ -81,8 +81,8 @@ class Config(UpdateTrackingModel["Config"], BaseSettings):
default=True,
description="If authentication is enabled or not",
)
enable_credit: str = Field(
default="false",
enable_credit: bool = Field(
default=False,
description="If user credit system is enabled or not",
)
num_user_credits_refill: int = Field(