yihong
10 月之前
+ 1
- 4
api/core/app/apps/advanced_chat/generate_task_pipeline.py
查看文件
| self._conversation_name_generate_thread = None | self._conversation_name_generate_thread = None | ||||
| self._recorded_files: list[Mapping[str, Any]] = [] | self._recorded_files: list[Mapping[str, Any]] = [] | ||||
| self.total_tokens: int = 0 | |||||
| def process(self): | def process(self): | ||||
| """ | """ | ||||
| if not workflow_run: | if not workflow_run: | ||||
| raise Exception("Workflow run not initialized.") | raise Exception("Workflow run not initialized.") | ||||
| # FIXME for issue #11221 quick fix maybe have a better solution | |||||
| self.total_tokens += event.metadata.get("total_tokens", 0) if event.metadata else 0 | |||||
| yield self._workflow_iteration_completed_to_stream_response( | yield self._workflow_iteration_completed_to_stream_response( | ||||
| task_id=self._application_generate_entity.task_id, workflow_run=workflow_run, event=event | task_id=self._application_generate_entity.task_id, workflow_run=workflow_run, event=event | ||||
| ) | ) | ||||
| workflow_run = self._handle_workflow_run_success( | workflow_run = self._handle_workflow_run_success( | ||||
| workflow_run=workflow_run, | workflow_run=workflow_run, | ||||
| start_at=graph_runtime_state.start_at, | start_at=graph_runtime_state.start_at, | ||||
| total_tokens=graph_runtime_state.total_tokens or self.total_tokens, | |||||
| total_tokens=graph_runtime_state.total_tokens, | |||||
| total_steps=graph_runtime_state.node_run_steps, | total_steps=graph_runtime_state.node_run_steps, | ||||
| outputs=event.outputs, | outputs=event.outputs, | ||||
| conversation_id=self._conversation.id, | conversation_id=self._conversation.id, |
+ 1
- 4
api/core/app/apps/workflow/generate_task_pipeline.py
查看文件
| self._task_state = WorkflowTaskState() | self._task_state = WorkflowTaskState() | ||||
| self._wip_workflow_node_executions = {} | self._wip_workflow_node_executions = {} | ||||
| self.total_tokens: int = 0 | |||||
| def process(self) -> Union[WorkflowAppBlockingResponse, Generator[WorkflowAppStreamResponse, None, None]]: | def process(self) -> Union[WorkflowAppBlockingResponse, Generator[WorkflowAppStreamResponse, None, None]]: | ||||
| """ | """ | ||||
| if not workflow_run: | if not workflow_run: | ||||
| raise Exception("Workflow run not initialized.") | raise Exception("Workflow run not initialized.") | ||||
| # FIXME for issue #11221 quick fix maybe have a better solution | |||||
| self.total_tokens += event.metadata.get("total_tokens", 0) if event.metadata else 0 | |||||
| yield self._workflow_iteration_completed_to_stream_response( | yield self._workflow_iteration_completed_to_stream_response( | ||||
| task_id=self._application_generate_entity.task_id, workflow_run=workflow_run, event=event | task_id=self._application_generate_entity.task_id, workflow_run=workflow_run, event=event | ||||
| ) | ) | ||||
| workflow_run = self._handle_workflow_run_success( | workflow_run = self._handle_workflow_run_success( | ||||
| workflow_run=workflow_run, | workflow_run=workflow_run, | ||||
| start_at=graph_runtime_state.start_at, | start_at=graph_runtime_state.start_at, | ||||
| total_tokens=graph_runtime_state.total_tokens or self.total_tokens, | |||||
| total_tokens=graph_runtime_state.total_tokens, | |||||
| total_steps=graph_runtime_state.node_run_steps, | total_steps=graph_runtime_state.node_run_steps, | ||||
| outputs=event.outputs, | outputs=event.outputs, | ||||
| conversation_id=None, | conversation_id=None, |
Loading…