Browse Source

fix: missing usage of metadata in the chatflow app (#9167)

tags/0.9.2
takatost 1 year ago
parent
commit
477beae3bb
No account linked to committer's email address
1 changed files with 5 additions and 0 deletions
  1. 5
    0
      api/core/app/apps/advanced_chat/generate_task_pipeline.py

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

@@ -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(

Loading…
Cancel
Save