opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-29 04:26:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

use enum instead

Browse Source
This commit is contained in:
takatost 2024-03-04 02:05:47 +08:00
parent e9004a06a5
commit d9b8a938c6
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/app/apps/advanced_chat/generate_task_pipeline.py
View File

@ -30,6 +30,7 @@ from core.model_runtime.entities.llm_entities import LLMUsage
from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
from core.moderation.output_moderation import ModerationRule, OutputModeration from core.moderation.output_moderation import ModerationRule, OutputModeration
from core.tools.tool_file_manager import ToolFileManager from core.tools.tool_file_manager import ToolFileManager
from core.workflow.entities.NodeEntities import NodeType
from events.message_event import message_was_created from events.message_event import message_was_created
from extensions.ext_database import db from extensions.ext_database import db
from models.model import Conversation, Message, MessageFile from models.model import Conversation, Message, MessageFile
@ -111,7 +112,7 @@ class AdvancedChatAppGenerateTaskPipeline:
elif isinstance(event, QueueNodeFinishedEvent): elif isinstance(event, QueueNodeFinishedEvent):
workflow_node_execution = self._get_workflow_node_execution(event.workflow_node_execution_id) workflow_node_execution = self._get_workflow_node_execution(event.workflow_node_execution_id)
if workflow_node_execution.status == WorkflowNodeExecutionStatus.SUCCEEDED.value: if workflow_node_execution.status == WorkflowNodeExecutionStatus.SUCCEEDED.value:
if workflow_node_execution.node_type == 'llm': # todo use enum if workflow_node_execution.node_type == NodeType.LLM.value:
outputs = workflow_node_execution.outputs_dict outputs = workflow_node_execution.outputs_dict
usage_dict = outputs.get('usage', {}) usage_dict = outputs.get('usage', {})
self._task_state.metadata['usage'] = usage_dict self._task_state.metadata['usage'] = usage_dict
@ -201,7 +202,7 @@ class AdvancedChatAppGenerateTaskPipeline:
elif isinstance(event, QueueNodeFinishedEvent): elif isinstance(event, QueueNodeFinishedEvent):
workflow_node_execution = self._get_workflow_node_execution(event.workflow_node_execution_id) workflow_node_execution = self._get_workflow_node_execution(event.workflow_node_execution_id)
if workflow_node_execution.status == WorkflowNodeExecutionStatus.SUCCEEDED.value: if workflow_node_execution.status == WorkflowNodeExecutionStatus.SUCCEEDED.value:
if workflow_node_execution.node_type == 'llm': # todo use enum if workflow_node_execution.node_type == NodeType.LLM.value:
outputs = workflow_node_execution.outputs_dict outputs = workflow_node_execution.outputs_dict
usage_dict = outputs.get('usage', {}) usage_dict = outputs.get('usage', {})
self._task_state.metadata['usage'] = usage_dict self._task_state.metadata['usage'] = usage_dict