AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 05:55:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: wrong token usage in iteration node for streaming result (#5336)

Browse Source
This commit is contained in:
rerorero 2024-06-18 14:08:40 +09:00 committed by GitHub
parent 132f5fb3de
commit 9b7fdadce4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/app/entities/task_entities.py
View File

@ -367,7 +367,7 @@ class IterationNodeNextStreamResponse(StreamResponse):
class IterationNodeCompletedStreamResponse(StreamResponse): class IterationNodeCompletedStreamResponse(StreamResponse):
""" """
NodeStartStreamResponse entity NodeCompletedStreamResponse entity
""" """
class Data(BaseModel): class Data(BaseModel):
""" """
@ -385,6 +385,7 @@ class IterationNodeCompletedStreamResponse(StreamResponse):
error: Optional[str] = None error: Optional[str] = None
elapsed_time: float elapsed_time: float
total_tokens: int total_tokens: int
execution_metadata: Optional[dict] = None
finished_at: int finished_at: int
steps: int steps: int

6
api/core/app/task_pipeline/workflow_iteration_cycle_manage.py
View File

@ -95,6 +95,9 @@ class WorkflowIterationCycleManage(WorkflowCycleStateManager):
error=None, error=None,
elapsed_time=time.perf_counter() - current_iteration.started_at, elapsed_time=time.perf_counter() - current_iteration.started_at,
total_tokens=current_iteration.total_tokens, total_tokens=current_iteration.total_tokens,
execution_metadata={
'total_tokens': current_iteration.total_tokens,
},
finished_at=int(time.time()), finished_at=int(time.time()),
steps=current_iteration.current_index steps=current_iteration.current_index
) )
@ -276,6 +279,9 @@ class WorkflowIterationCycleManage(WorkflowCycleStateManager):
error=error, error=error,
elapsed_time=time.perf_counter() - current_iteration.started_at, elapsed_time=time.perf_counter() - current_iteration.started_at,
total_tokens=current_iteration.total_tokens, total_tokens=current_iteration.total_tokens,
execution_metadata={
'total_tokens': current_iteration.total_tokens,
},
finished_at=int(time.time()), finished_at=int(time.time()),
steps=current_iteration.current_index steps=current_iteration.current_index
) )