youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 02:18:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor app generate

Browse Source
This commit is contained in:
takatost
2024-03-03 04:18:38 +08:00
parent 171b2bdc20
commit 406a625c98
35 changed files with 1236 additions and 751 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/events/event_handlers/deduct_quota_when_messaeg_created.py
View File

@ -1,4 +1,4 @@
from core.app.entities.app_invoke_entities import EasyUIBasedAppGenerateEntity
from core.app.entities.app_invoke_entities import ChatAppGenerateEntity
from core.entities.provider_entities import QuotaUnit
from events.message_event import message_was_created
from extensions.ext_database import db
@ -8,7 +8,7 @@ from models.provider import Provider, ProviderType
@message_was_created.connect
def handle(sender, **kwargs):
message = sender
application_generate_entity: EasyUIBasedAppGenerateEntity = kwargs.get('application_generate_entity')
application_generate_entity: ChatAppGenerateEntity = kwargs.get('application_generate_entity')
model_config = application_generate_entity.model_config
provider_model_bundle = model_config.provider_model_bundle

4
api/events/event_handlers/update_provider_last_used_at_when_messaeg_created.py
View File

@ -1,6 +1,6 @@
from datetime import datetime
from core.app.entities.app_invoke_entities import EasyUIBasedAppGenerateEntity
from core.app.entities.app_invoke_entities import ChatAppGenerateEntity
from events.message_event import message_was_created
from extensions.ext_database import db
from models.provider import Provider
@ -9,7 +9,7 @@ from models.provider import Provider
@message_was_created.connect
def handle(sender, **kwargs):
message = sender
application_generate_entity: EasyUIBasedAppGenerateEntity = kwargs.get('application_generate_entity')
application_generate_entity: ChatAppGenerateEntity = kwargs.get('application_generate_entity')
db.session.query(Provider).filter(
Provider.tenant_id == application_generate_entity.app_config.tenant_id,