from sqlalchemy import Column, String, Text, DateTime, ForeignKey, Integer
|
from sqlalchemy.orm import relationship
|
from sqlalchemy.ext.declarative import declarative_base
|
import datetime
|
|
Base = declarative_base()
|
|
|
class ChatGroup(Base):
|
__tablename__ = 'chat_groups'
|
id = Column(String(64), primary_key=True) # 企业微信群ID
|
name = Column(String(255))
|
|
|
class User(Base):
|
__tablename__ = 'users'
|
id = Column(String(64), primary_key=True) # 企业微信用户ID
|
name = Column(String(255))
|
|
|
class GroupUser(Base):
|
__tablename__ = 'group_users'
|
id = Column(String(64), primary_key=True)
|
group_id = Column(String(64), ForeignKey('chat_groups.id'))
|
user_id = Column(String(64), ForeignKey('users.id'))
|
|
group = relationship("ChatGroup", backref="members")
|
user = relationship("User", backref="groups")
|
|
|
class GroupSMSAccount(Base):
|
__tablename__ = 'group_sms_accounts'
|
id = Column(String(64), primary_key=True)
|
group_id = Column(String(64), ForeignKey('chat_groups.id'))
|
sms_account = Column(String(64))
|
|
group = relationship("ChatGroup", backref="sms_accounts")
|
|
|
class MessageHistory(Base):
|
__tablename__ = 'message_history'
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
user_id = Column(String(64))
|
conversation_id = Column(String(64))
|
role = Column(String(10)) # 'user' 或 'assistant'
|
content = Column(Text)
|
timestamp = Column(DateTime, default=datetime.datetime.utcnow)
|
