mirror of
https://github.com/langgenius/dify.git
synced 2026-10-04 13:47:55 +08:00
Charles YaoandClaude Opus 4.8
97b8354173
The build/edit test handlers wrapped run_draft in a bare, non-logging `except Exception` that turned any launch-time failure into a blank failed run (dify_run_id="", per_node=[], no error). Diagnose then had nothing to act on and the repair loop ran blind — the prod symptom where a stale test input (ValueError "<var> is required in input form") produced an undiagnosable await_repair loop. Now the launch error is logged, captured, and persisted so diagnose/routing can act on it: - Run.error field + dify_builder_runs.error column (migration a9cc8d552db4) - handlers_build/edit: logger.exception + launch_error_text(exc) -> Run.error - is_input_failure() also inspects Run.error (+ "in input form" signal), so a stale-input launch failure routes to the testdata gate, not blind repair - diagnose/_degraded_diagnosis surface the launch error when no node executed Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
191 lines
8.0 KiB
Python
191 lines
8.0 KiB
Python
from datetime import datetime
|
|
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import DateTime, String, func
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from libs.uuid_utils import uuidv7
|
|
|
|
from .base import Base
|
|
from .types import AdjustedJSON, StringUUID
|
|
|
|
|
|
class DifyBuilderSession(Base):
|
|
"""
|
|
DifyBuilder Session Model
|
|
|
|
Durable state for a single dify-builder conversation: which app/tenant
|
|
it belongs to, the entry mode it was started in, its current FSM state,
|
|
and the `version` column used as the CAS counter for
|
|
`Repository.compare_and_advance`.
|
|
"""
|
|
|
|
__tablename__ = "dify_builder_sessions"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="dify_builder_session_pkey"),
|
|
sa.Index("dify_builder_session_tenant_idx", "tenant_id"),
|
|
sa.Index("dify_builder_session_app_idx", "app_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
app_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
tenant_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
owner_account_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
entry_mode: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
current_state: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
version: Mapped[int] = mapped_column(sa.Integer, nullable=False, server_default=sa.text("1"))
|
|
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, server_default=func.current_timestamp())
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime, nullable=False, server_default=func.current_timestamp(), onupdate=func.current_timestamp()
|
|
)
|
|
|
|
|
|
class DifyBuilderSessionCommit(Base):
|
|
"""
|
|
DifyBuilder Session Commit Model
|
|
|
|
One immutable row per session version, carrying the serialized
|
|
`DifyBuilderContext` (`context`) as of that version. The latest commit by
|
|
`version` for a given `session_id` is the session's current context.
|
|
"""
|
|
|
|
__tablename__ = "dify_builder_session_commits"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="dify_builder_session_commit_pkey"),
|
|
sa.Index("dify_builder_session_commit_session_idx", "session_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
session_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
version: Mapped[int] = mapped_column(sa.Integer, nullable=False)
|
|
state: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
context: Mapped[dict] = mapped_column(AdjustedJSON, nullable=False)
|
|
actor: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, server_default=func.current_timestamp())
|
|
|
|
|
|
class DifyBuilderSnapshot(Base):
|
|
"""
|
|
DifyBuilder Snapshot Model
|
|
|
|
A content-addressed snapshot of a workflow graph (`hash` + `graph`)
|
|
captured during a dify_builder session, referenced by checkpoints.
|
|
"""
|
|
|
|
__tablename__ = "dify_builder_snapshots"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="dify_builder_snapshot_pkey"),
|
|
sa.Index("dify_builder_snapshot_session_idx", "session_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
session_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
hash: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
graph: Mapped[dict] = mapped_column(AdjustedJSON, nullable=False)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, server_default=func.current_timestamp())
|
|
|
|
|
|
class DifyBuilderCheckpoint(Base):
|
|
"""
|
|
DifyBuilder Checkpoint Model
|
|
|
|
A named point in a dify_builder session's history (`state`) pinned to a
|
|
`DifyBuilderSnapshot` (`snapshot_id`, no FK per house style).
|
|
"""
|
|
|
|
__tablename__ = "dify_builder_checkpoints"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="dify_builder_checkpoint_pkey"),
|
|
sa.Index("dify_builder_checkpoint_session_idx", "session_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
session_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
state: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
snapshot_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, server_default=func.current_timestamp())
|
|
|
|
|
|
class DifyBuilderRun(Base):
|
|
"""
|
|
DifyBuilder Run Model
|
|
|
|
A single execution (verify/publish/etc.) driven by the dify_builder, including
|
|
per-node results and whether the row is `immutable` once finalized.
|
|
"""
|
|
|
|
__tablename__ = "dify_builder_runs"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="dify_builder_run_pkey"),
|
|
sa.Index("dify_builder_run_session_idx", "session_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
session_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
kind: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
dify_run_id: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
status: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
per_node: Mapped[dict | None] = mapped_column(AdjustedJSON, nullable=True)
|
|
culprit_node_id: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
# Launch-failure text captured when a verify run throws before any node runs.
|
|
error: Mapped[str | None] = mapped_column(sa.Text, nullable=True)
|
|
inputs_ref: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
tokens: Mapped[int | None] = mapped_column(sa.Integer, nullable=True)
|
|
elapsed_ms: Mapped[int | None] = mapped_column(sa.Integer, nullable=True)
|
|
immutable: Mapped[bool] = mapped_column(sa.Boolean, nullable=False, server_default=sa.text("false"))
|
|
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, server_default=func.current_timestamp())
|
|
|
|
|
|
class DifyBuilderConversationItem(Base):
|
|
"""
|
|
DifyBuilder Conversation Item Model
|
|
|
|
One ordered message/event (`seq`) in a dify_builder session's conversation
|
|
log. `(session_id, seq)` is unique — the DB is the seq-authority.
|
|
"""
|
|
|
|
__tablename__ = "dify_builder_conversation_items"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="dify_builder_conversation_item_pkey"),
|
|
sa.UniqueConstraint("session_id", "seq", name="dify_builder_conversation_item_session_seq_key"),
|
|
sa.Index("dify_builder_conversation_item_session_idx", "session_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
session_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
seq: Mapped[int] = mapped_column(sa.Integer, nullable=False)
|
|
kind: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
payload: Mapped[dict] = mapped_column(AdjustedJSON, nullable=False)
|
|
at_version: Mapped[int] = mapped_column(sa.Integer, nullable=False)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, server_default=func.current_timestamp())
|
|
|
|
|
|
class DifyBuilderTestInput(Base):
|
|
"""
|
|
DifyBuilder Test Input Model
|
|
|
|
A saved set of workflow test inputs (`inputs`) captured during a dify_builder
|
|
session, keyed to the workflow's start-schema shape (`start_schema_hash`)
|
|
so stale inputs can be detected.
|
|
"""
|
|
|
|
__tablename__ = "dify_builder_test_inputs"
|
|
__table_args__ = (
|
|
sa.PrimaryKeyConstraint("id", name="dify_builder_test_input_pkey"),
|
|
sa.Index("dify_builder_test_input_session_idx", "session_id"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(StringUUID, default=lambda: str(uuidv7()))
|
|
session_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
|
|
source: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
inputs: Mapped[dict] = mapped_column(AdjustedJSON, nullable=False)
|
|
start_schema_hash: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, server_default=func.current_timestamp())
|