daarion-admin/microdao-daarion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: move CheckConstraint to __table_args__ in AgentMemoryEvent model

Browse Source 
- Fix syntax error on line 115-116
- Move CheckConstraint from Column parameters to __table_args__
- Add proper constraint names
This commit is contained in:
Apple
2025-11-15 11:49:34 -08:00
parent d50765f2ff
commit f7c0a0fc08
1 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
services/memory-service/app/models.py
View File

@@ -109,18 +109,10 @@ class AgentMemoryEvent(Base):
user_id = Column(String, ForeignKey("users.id", ondelete="CASCADE"), nullable=True, index=True) user_id = Column(String, ForeignKey("users.id", ondelete="CASCADE"), nullable=True, index=True)
# Scope: short_term, mid_term, long_term # Scope: short_term, mid_term, long_term
scope = Column( scope = Column(String, nullable=False)
String,
nullable=False,
CheckConstraint("scope IN ('short_term', 'mid_term', 'long_term')")
)
# Kind: message, fact, summary, note # Kind: message, fact, summary, note
kind = Column( kind = Column(String, nullable=False)
String,
nullable=False,
CheckConstraint("kind IN ('message', 'fact', 'summary', 'note')")
)
# Тіло події # Тіло події
body_text = Column(Text, nullable=True) body_text = Column(Text, nullable=True)
@@ -129,6 +121,8 @@ class AgentMemoryEvent(Base):
created_at = Column(TIMESTAMP(timezone=True), nullable=False, server_default=func.now()) created_at = Column(TIMESTAMP(timezone=True), nullable=False, server_default=func.now())
__table_args__ = ( __table_args__ = (
CheckConstraint("scope IN ('short_term', 'mid_term', 'long_term')", name="ck_agent_memory_scope"),
CheckConstraint("kind IN ('message', 'fact', 'summary', 'note')", name="ck_agent_memory_kind"),
Index("idx_agent_memory_events_agent_team_scope", "agent_id", "team_id", "scope"), Index("idx_agent_memory_events_agent_team_scope", "agent_id", "team_id", "scope"),
Index("idx_agent_memory_events_channel", "agent_id", "channel_id"), Index("idx_agent_memory_events_channel", "agent_id", "channel_id"),
Index("idx_agent_memory_events_created_at", "created_at"), Index("idx_agent_memory_events_created_at", "created_at"),