| @@ -45,6 +45,7 @@ from core.app.entities.task_entities import ( | |||
| from core.app.task_pipeline.based_generate_task_pipeline import BasedGenerateTaskPipeline | |||
| from core.app.task_pipeline.message_cycle_manage import MessageCycleManage | |||
| from core.app.task_pipeline.workflow_cycle_manage import WorkflowCycleManage | |||
| from core.model_runtime.entities.llm_entities import LLMUsage | |||
| from core.model_runtime.utils.encoders import jsonable_encoder | |||
| from core.ops.ops_trace_manager import TraceQueueManager | |||
| from core.workflow.enums import SystemVariableKey | |||
| @@ -509,6 +510,10 @@ class AdvancedChatAppGenerateTaskPipeline(BasedGenerateTaskPipeline, WorkflowCyc | |||
| self._message.total_price = usage.total_price | |||
| self._message.currency = usage.currency | |||
| self._task_state.metadata["usage"] = jsonable_encoder(usage) | |||
| else: | |||
| self._task_state.metadata["usage"] = jsonable_encoder(LLMUsage.empty_usage()) | |||
| db.session.commit() | |||
| message_was_created.send( | |||