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

add AdvancedChatAppGenerateTaskPipeline

Browse Source
This commit is contained in:
takatost
2024-03-04 02:04:40 +08:00
parent 602bc67495
commit be709d4b84
14 changed files with 570 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,4 @@
from core.app.entities.app_invoke_entities import ChatAppGenerateEntity
from core.app.entities.app_invoke_entities import AgentChatAppGenerateEntity, 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,10 @@ from models.provider import Provider, ProviderType
@message_was_created.connect
def handle(sender, **kwargs):
message = sender
application_generate_entity: ChatAppGenerateEntity = kwargs.get('application_generate_entity')
application_generate_entity = kwargs.get('application_generate_entity')
if not isinstance(application_generate_entity, ChatAppGenerateEntity | AgentChatAppGenerateEntity):
return
model_config = application_generate_entity.model_config
provider_model_bundle = model_config.provider_model_bundle

3
api/events/event_handlers/generate_conversation_name_when_first_message_created.py
View File

@ -1,6 +1,7 @@
from core.llm_generator.llm_generator import LLMGenerator
from events.message_event import message_was_created
from extensions.ext_database import db
from models.model import AppMode
@message_was_created.connect
@ -15,7 +16,7 @@ def handle(sender, **kwargs):
auto_generate_conversation_name = extras.get('auto_generate_conversation_name', True)
if auto_generate_conversation_name and is_first_message:
if conversation.mode == 'chat':
if conversation.mode != AppMode.COMPLETION.value:
app_model = conversation.app
if not app_model:
return

7
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 ChatAppGenerateEntity
from core.app.entities.app_invoke_entities import AgentChatAppGenerateEntity, ChatAppGenerateEntity
from events.message_event import message_was_created
from extensions.ext_database import db
from models.provider import Provider
@ -9,7 +9,10 @@ from models.provider import Provider
@message_was_created.connect
def handle(sender, **kwargs):
message = sender
application_generate_entity: ChatAppGenerateEntity = kwargs.get('application_generate_entity')
application_generate_entity = kwargs.get('application_generate_entity')
if not isinstance(application_generate_entity, ChatAppGenerateEntity | AgentChatAppGenerateEntity):
return
db.session.query(Provider).filter(
Provider.tenant_id == application_generate_entity.app_config.tenant_id,