youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 09:58:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(api): restore node state snapshots via /events api

Browse Source
This commit is contained in:
QuantumGhost
2026-01-19 09:49:58 +08:00
parent 6bf6bf6a2a
commit b085df9425
6 changed files with 849 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/controllers/web/workflow_events.py
View File

@ -5,7 +5,7 @@ Web App Workflow Resume APIs.
import json
from collections.abc import Generator
from flask import Response
from flask import Response, request
from sqlalchemy.orm import sessionmaker
from controllers.web import api
@ -19,6 +19,7 @@ from extensions.ext_database import db
from models.enums import CreatorUserRole
from models.model import App, AppMode, EndUser
from repositories.factory import DifyAPIRepositoryFactory
from services.workflow_event_snapshot_service import build_workflow_event_stream
class WorkflowEventsApi(WebApiResource):
@ -76,7 +77,19 @@ class WorkflowEventsApi(WebApiResource):
else:
raise InvalidArgumentError(f"cannot subscribe to workflow run, workflow_run_id={workflow_run.id}")
include_state_snapshot = request.args.get("include_state_snapshot", "false").lower() == "true"
def _generate_stream_events():
if include_state_snapshot:
return generator.convert_to_event_stream(
build_workflow_event_stream(
app_mode=app_mode,
workflow_run=workflow_run,
tenant_id=app_model.tenant_id,
app_id=app_model.id,
session_maker=session_maker,
)
)
return generator.convert_to_event_stream(
msg_generator.retrieve_events(app_mode, workflow_run.id),
)