jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Feat/billing enhancement (#2239)

Browse source 
Co-authored-by: takatost <takatost@gmail.com>
This commit is contained in:
Garfield Dai 2024-01-26 18:26:15 +08:00 • committed by GitHub
commit bb5d5fc683
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 43 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/events/event_handlers/deduct_quota_when_messaeg_created.py
View file

@ -33,6 +33,11 @@ def handle(sender, **kwargs):
if quota_unit:
if quota_unit == QuotaUnit.TOKENS:
used_quota = message.message_tokens + message.answer_tokens
elif quota_unit == QuotaUnit.CREDITS:
used_quota = 1
if 'gpt-4' in model_config.model:
used_quota = 20
else:
used_quota = 1