mirror of https://github.com/langgenius/dify.git
Apply suggestion from @asukaminato0721
Co-authored-by: Asuka Minato <i@asukaminato.eu.org>
This commit is contained in:
parent
483288b1bb
commit
bb857df799
|
|
@ -22,7 +22,7 @@ from core.workflow.enums import WorkflowExecutionStatus
|
|||
from libs.helper import generate_string # type: ignore[import-not-found]
|
||||
|
||||
from .account import Account, Tenant
|
||||
from .base import Base
|
||||
from .base import Base, TypeBase
|
||||
from .engine import db
|
||||
from .enums import CreatorUserRole
|
||||
from .provider_ids import GenericProviderID
|
||||
|
|
@ -1771,7 +1771,7 @@ class MessageChain(Base):
|
|||
created_at = mapped_column(sa.DateTime, nullable=False, server_default=sa.func.current_timestamp())
|
||||
|
||||
|
||||
class MessageAgentThought(Base):
|
||||
class MessageAgentThought(TypeBase):
|
||||
__tablename__ = "message_agent_thoughts"
|
||||
__table_args__ = (
|
||||
sa.PrimaryKeyConstraint("id", name="message_agent_thought_pkey"),
|
||||
|
|
|
|||
Loading…
Reference in New Issue