JiuContinent
d7db84571d
feat: 添加飞书用户模块和订阅功能支持 - 新增feishu_users模块用于处理飞书用户身份验证和权限管理 - 新增subscriptions模块用于处理订阅相关功能 - 新增personalization模块用于个性化服务 - 在alembic迁移配置中注册新的模型模块 - 在API路由器中添加feishu_users和subscriptions路由 - 实现事件调度服务的改进，包括错误处理和状态更新优化 - 添加飞书命令处理的权限检查机制 - 实现飞书应用票据事件处理 - 改进审计日志记录功能 ```
61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import JSON, DateTime, ForeignKey, Integer, String, Text, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.core.constants import ActorValue
|
|
from app.core.database import Base
|
|
from app.core.utils.time import utc_now
|
|
from app.modules.feishu_users.models import FeishuUser
|
|
from app.modules.ai_memory.constants import (
|
|
AIMemoryKind,
|
|
AIMemoryScope,
|
|
AIMemorySource,
|
|
AIMemoryStatus,
|
|
)
|
|
|
|
|
|
class AIMemoryEntry(Base):
|
|
__tablename__ = "ai_memory_entries"
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"owner_id",
|
|
"fingerprint",
|
|
name="uq_ai_memory_owner_fingerprint",
|
|
),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
fingerprint: Mapped[str | None] = mapped_column(
|
|
String(64), nullable=True, index=True
|
|
)
|
|
owner_id: Mapped[int | None] = mapped_column(
|
|
ForeignKey("feishu_users.id", ondelete="CASCADE"),
|
|
nullable=True,
|
|
index=True,
|
|
)
|
|
owner: Mapped[FeishuUser | None] = relationship()
|
|
kind: Mapped[str] = mapped_column(
|
|
String(32),
|
|
default=AIMemoryKind.MEMORY,
|
|
index=True,
|
|
)
|
|
scope: Mapped[str] = mapped_column(String(64), default=AIMemoryScope.GLOBAL, index=True)
|
|
subject: Mapped[str] = mapped_column(String(128), index=True)
|
|
content: Mapped[str] = mapped_column(Text)
|
|
summary: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
tags: Mapped[list | None] = mapped_column(JSON, nullable=True)
|
|
source: Mapped[str] = mapped_column(String(64), default=AIMemorySource.AUTO, index=True)
|
|
importance: Mapped[int] = mapped_column(Integer, default=1, index=True)
|
|
status: Mapped[str] = mapped_column(String(32), default=AIMemoryStatus.ACTIVE, index=True)
|
|
actor: Mapped[str] = mapped_column(String(128), default=ActorValue.SYSTEM, index=True)
|
|
last_used_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True, index=True)
|
|
expires_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True, index=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime,
|
|
default=utc_now,
|
|
onupdate=utc_now,
|
|
)
|