mirror of
https://github.com/langgenius/dify.git
synced 2025-08-12 11:17:28 +00:00
6eb155ae69
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com>
123 lines
4.0 KiB
Python
123 lines
4.0 KiB
Python
import threading
|
|
from collections.abc import Sequence
|
|
from typing import Optional
|
|
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
import contexts
|
|
from extensions.ext_database import db
|
|
from libs.infinite_scroll_pagination import InfiniteScrollPagination
|
|
from models import (
|
|
Account,
|
|
App,
|
|
EndUser,
|
|
WorkflowNodeExecutionModel,
|
|
WorkflowRun,
|
|
WorkflowRunTriggeredFrom,
|
|
)
|
|
from repositories.factory import DifyAPIRepositoryFactory
|
|
|
|
|
|
class WorkflowRunService:
|
|
def __init__(self):
|
|
"""Initialize WorkflowRunService with repository dependencies."""
|
|
session_maker = sessionmaker(bind=db.engine, expire_on_commit=False)
|
|
self._node_execution_service_repo = DifyAPIRepositoryFactory.create_api_workflow_node_execution_repository(
|
|
session_maker
|
|
)
|
|
self._workflow_run_repo = DifyAPIRepositoryFactory.create_api_workflow_run_repository(session_maker)
|
|
|
|
def get_paginate_advanced_chat_workflow_runs(self, app_model: App, args: dict) -> InfiniteScrollPagination:
|
|
"""
|
|
Get advanced chat app workflow run list
|
|
Only return triggered_from == advanced_chat
|
|
|
|
:param app_model: app model
|
|
:param args: request args
|
|
"""
|
|
|
|
class WorkflowWithMessage:
|
|
message_id: str
|
|
conversation_id: str
|
|
|
|
def __init__(self, workflow_run: WorkflowRun):
|
|
self._workflow_run = workflow_run
|
|
|
|
def __getattr__(self, item):
|
|
return getattr(self._workflow_run, item)
|
|
|
|
pagination = self.get_paginate_workflow_runs(app_model, args)
|
|
|
|
with_message_workflow_runs = []
|
|
for workflow_run in pagination.data:
|
|
message = workflow_run.message
|
|
with_message_workflow_run = WorkflowWithMessage(workflow_run=workflow_run)
|
|
if message:
|
|
with_message_workflow_run.message_id = message.id
|
|
with_message_workflow_run.conversation_id = message.conversation_id
|
|
|
|
with_message_workflow_runs.append(with_message_workflow_run)
|
|
|
|
pagination.data = with_message_workflow_runs
|
|
return pagination
|
|
|
|
def get_paginate_workflow_runs(self, app_model: App, args: dict) -> InfiniteScrollPagination:
|
|
"""
|
|
Get debug workflow run list
|
|
Only return triggered_from == debugging
|
|
|
|
:param app_model: app model
|
|
:param args: request args
|
|
"""
|
|
limit = int(args.get("limit", 20))
|
|
last_id = args.get("last_id")
|
|
|
|
return self._workflow_run_repo.get_paginated_workflow_runs(
|
|
tenant_id=app_model.tenant_id,
|
|
app_id=app_model.id,
|
|
triggered_from=WorkflowRunTriggeredFrom.DEBUGGING.value,
|
|
limit=limit,
|
|
last_id=last_id,
|
|
)
|
|
|
|
def get_workflow_run(self, app_model: App, run_id: str) -> Optional[WorkflowRun]:
|
|
"""
|
|
Get workflow run detail
|
|
|
|
:param app_model: app model
|
|
:param run_id: workflow run id
|
|
"""
|
|
return self._workflow_run_repo.get_workflow_run_by_id(
|
|
tenant_id=app_model.tenant_id,
|
|
app_id=app_model.id,
|
|
run_id=run_id,
|
|
)
|
|
|
|
def get_workflow_run_node_executions(
|
|
self,
|
|
app_model: App,
|
|
run_id: str,
|
|
user: Account | EndUser,
|
|
) -> Sequence[WorkflowNodeExecutionModel]:
|
|
"""
|
|
Get workflow run node execution list
|
|
"""
|
|
workflow_run = self.get_workflow_run(app_model, run_id)
|
|
|
|
contexts.plugin_tool_providers.set({})
|
|
contexts.plugin_tool_providers_lock.set(threading.Lock())
|
|
|
|
if not workflow_run:
|
|
return []
|
|
|
|
# Get tenant_id from user
|
|
tenant_id = user.tenant_id if isinstance(user, EndUser) else user.current_tenant_id
|
|
if tenant_id is None:
|
|
raise ValueError("User tenant_id cannot be None")
|
|
|
|
return self._node_execution_service_repo.get_executions_by_workflow_run(
|
|
tenant_id=tenant_id,
|
|
app_id=app_model.id,
|
|
workflow_run_id=run_id,
|
|
)
|