JiuContinent/company-ai-platform
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

```

Browse Source 
feat: 添加飞书用户模块和订阅功能支持

- 新增feishu_users模块用于处理飞书用户身份验证和权限管理
- 新增subscriptions模块用于处理订阅相关功能
- 新增personalization模块用于个性化服务
- 在alembic迁移配置中注册新的模型模块
- 在API路由器中添加feishu_users和subscriptions路由
- 实现事件调度服务的改进，包括错误处理和状态更新优化
- 添加飞书命令处理的权限检查机制
- 实现飞书应用票据事件处理
- 改进审计日志记录功能
```
This commit is contained in:
JiuContinent
2026-07-27 08:02:17 +08:00
parent db751f03b4
commit d7db84571d
148 changed files with 17110 additions and 765 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
alembic/versions/202607260001_v3_runtime_reliability.py Normal file
View File

@@ -0,0 +1,51 @@
"""Harden V3 heartbeat identity and concurrent updates.
Revision ID: 202607260001
Revises: 202607150001
Create Date: 2026-07-26
"""
from alembic import op
import sqlalchemy as sa
revision = "202607260001"
down_revision = "202607150001"
branch_labels = None
depends_on = None
HEARTBEAT_TABLE = "system_heartbeats"
HEARTBEAT_IDENTITY_CONSTRAINT = "uq_system_heartbeat_component_instance"
def upgrade() -> None:
op.execute(
sa.text(
"""
DELETE FROM system_heartbeats
WHERE id IN (
SELECT id
FROM (
SELECT
id,
ROW_NUMBER() OVER (
PARTITION BY component, instance_id
ORDER BY last_seen_at DESC, id DESC
) AS duplicate_rank
FROM system_heartbeats
) ranked
WHERE duplicate_rank > 1
)
"""
)
)
with op.batch_alter_table(HEARTBEAT_TABLE) as batch_op:
batch_op.create_unique_constraint(
HEARTBEAT_IDENTITY_CONSTRAINT,
["component", "instance_id"],
)
def downgrade() -> None:
with op.batch_alter_table(HEARTBEAT_TABLE) as batch_op:
batch_op.drop_constraint(HEARTBEAT_IDENTITY_CONSTRAINT, type_="unique")

64
alembic/versions/202607260002_v3_workflow_memory_idempotency.py Normal file
View File

@@ -0,0 +1,64 @@
"""Add V3 workflow-event and AI-memory idempotency keys.
Revision ID: 202607260002
Revises: 202607260001
Create Date: 2026-07-26
"""
from alembic import op
import sqlalchemy as sa
revision = "202607260002"
down_revision = "202607260001"
branch_labels = None
depends_on = None
AI_MEMORY_TABLE = "ai_memory_entries"
AI_MEMORY_FINGERPRINT_INDEX = "ix_ai_memory_entries_fingerprint"
WORKFLOW_ACTION_TABLE = "workflow_actions"
WORKFLOW_SOURCE_EVENT_INDEX = "ix_workflow_actions_source_event_id"
WORKFLOW_SOURCE_EVENT_FOREIGN_KEY = "fk_workflow_actions_source_event_id"
def upgrade() -> None:
with op.batch_alter_table(AI_MEMORY_TABLE) as batch_op:
batch_op.add_column(
sa.Column("fingerprint", sa.String(length=64), nullable=True)
)
batch_op.create_index(
AI_MEMORY_FINGERPRINT_INDEX,
["fingerprint"],
unique=True,
)
with op.batch_alter_table(WORKFLOW_ACTION_TABLE) as batch_op:
batch_op.add_column(
sa.Column("source_event_id", sa.String(length=64), nullable=True)
)
batch_op.create_index(
WORKFLOW_SOURCE_EVENT_INDEX,
["source_event_id"],
unique=True,
)
batch_op.create_foreign_key(
WORKFLOW_SOURCE_EVENT_FOREIGN_KEY,
"domain_events",
["source_event_id"],
["event_id"],
ondelete="RESTRICT",
)
def downgrade() -> None:
with op.batch_alter_table(WORKFLOW_ACTION_TABLE) as batch_op:
batch_op.drop_constraint(
WORKFLOW_SOURCE_EVENT_FOREIGN_KEY,
type_="foreignkey",
)
batch_op.drop_index(WORKFLOW_SOURCE_EVENT_INDEX)
batch_op.drop_column("source_event_id")
with op.batch_alter_table(AI_MEMORY_TABLE) as batch_op:
batch_op.drop_index(AI_MEMORY_FINGERPRINT_INDEX)
batch_op.drop_column("fingerprint")

481
alembic/versions/202607260003_feishu_personalization_subscriptions.py Normal file
View File

@@ -0,0 +1,481 @@
"""Add Feishu identities, personalization, and durable subscriptions.
Revision ID: 202607260003
Revises: 202607260002
Create Date: 2026-07-26
"""
from alembic import op
import sqlalchemy as sa
revision = "202607260003"
down_revision = "202607260002"
branch_labels = None
depends_on = None
AI_MEMORY_TABLE = "ai_memory_entries"
MARKET_WATCHLIST_TABLE = "market_watchlists"
def _create_indexes(
table_name: str,
definitions: tuple[tuple[str, tuple[str, ...], bool], ...],
) -> None:
for name, columns, unique in definitions:
op.create_index(op.f(name), table_name, list(columns), unique=unique)
def upgrade() -> None:
_create_feishu_users()
_upgrade_ai_memory()
_upgrade_market_watchlists()
_create_personalization_tables()
_create_subscription_tables()
def _create_feishu_users() -> None:
op.create_table(
"feishu_users",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("code", sa.String(length=64), nullable=False),
sa.Column("tenant_key", sa.String(length=128), nullable=False),
sa.Column("open_id", sa.String(length=128), nullable=False),
sa.Column("union_id", sa.String(length=128), nullable=True),
sa.Column("user_id", sa.String(length=128), nullable=True),
sa.Column("role", sa.String(length=32), nullable=False),
sa.Column("status", sa.String(length=32), nullable=False),
sa.Column("timezone", sa.String(length=64), nullable=False),
sa.Column("quiet_hours_start", sa.Time(), nullable=True),
sa.Column("quiet_hours_end", sa.Time(), nullable=True),
sa.Column("last_active_at", sa.DateTime(), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint(
"tenant_key",
"open_id",
name="uq_feishu_user_tenant_open_id",
),
)
_create_indexes(
"feishu_users",
(
("ix_feishu_users_code", ("code",), True),
("ix_feishu_users_created_at", ("created_at",), False),
("ix_feishu_users_last_active_at", ("last_active_at",), False),
("ix_feishu_users_open_id", ("open_id",), False),
("ix_feishu_users_role", ("role",), False),
("ix_feishu_users_status", ("status",), False),
("ix_feishu_users_tenant_key", ("tenant_key",), False),
("ix_feishu_users_union_id", ("union_id",), False),
("ix_feishu_users_user_id", ("user_id",), False),
),
)
def _upgrade_ai_memory() -> None:
op.drop_index(
"ix_ai_memory_entries_fingerprint",
table_name=AI_MEMORY_TABLE,
)
with op.batch_alter_table(AI_MEMORY_TABLE) as batch_op:
batch_op.add_column(
sa.Column("owner_id", sa.Integer(), nullable=True)
)
batch_op.add_column(
sa.Column(
"kind",
sa.String(length=32),
nullable=False,
server_default="memory",
)
)
batch_op.create_foreign_key(
"fk_ai_memory_entries_owner_id",
"feishu_users",
["owner_id"],
["id"],
ondelete="CASCADE",
)
batch_op.create_unique_constraint(
"uq_ai_memory_owner_fingerprint",
["owner_id", "fingerprint"],
)
batch_op.create_index(
"ix_ai_memory_entries_fingerprint",
["fingerprint"],
unique=False,
)
batch_op.create_index(
"ix_ai_memory_entries_owner_id",
["owner_id"],
unique=False,
)
batch_op.create_index(
"ix_ai_memory_entries_kind",
["kind"],
unique=False,
)
connection = op.get_bind()
connection.execute(
sa.text(
"""
UPDATE ai_memory_entries
SET
owner_id = NULL,
kind = 'company_rule',
source = 'legacy_company',
status = 'active'
WHERE source = 'user_rule'
"""
)
)
connection.execute(
sa.text(
"""
UPDATE ai_memory_entries
SET kind = 'memory', status = 'archived'
WHERE owner_id IS NULL AND source IN ('auto', 'hermes')
"""
)
)
with op.batch_alter_table(AI_MEMORY_TABLE) as batch_op:
batch_op.alter_column(
"kind",
existing_type=sa.String(length=32),
nullable=False,
server_default=None,
)
def _upgrade_market_watchlists() -> None:
with op.batch_alter_table(MARKET_WATCHLIST_TABLE) as batch_op:
batch_op.drop_constraint(
"uq_market_watchlist_actor_symbol",
type_="unique",
)
batch_op.add_column(sa.Column("owner_id", sa.Integer(), nullable=True))
batch_op.create_foreign_key(
"fk_market_watchlists_owner_id",
"feishu_users",
["owner_id"],
["id"],
ondelete="CASCADE",
)
batch_op.create_unique_constraint(
"uq_market_watchlist_owner_symbol",
["owner_id", "symbol"],
)
batch_op.create_index(
"ix_market_watchlists_owner_id",
["owner_id"],
unique=False,
)
def _create_personalization_tables() -> None:
op.create_table(
"user_preferences",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("code", sa.String(length=64), nullable=False),
sa.Column("owner_id", sa.Integer(), nullable=False),
sa.Column("category", sa.String(length=32), nullable=False),
sa.Column("value", sa.Text(), nullable=False),
sa.Column("normalized_value", sa.String(length=1000), nullable=False),
sa.Column("source", sa.String(length=32), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(
["owner_id"],
["feishu_users.id"],
name="fk_user_preferences_owner_id",
ondelete="CASCADE",
),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint(
"owner_id",
"category",
"normalized_value",
name="uq_user_preference_owner_category_value",
),
)
_create_indexes(
"user_preferences",
(
("ix_user_preferences_category", ("category",), False),
("ix_user_preferences_code", ("code",), True),
("ix_user_preferences_created_at", ("created_at",), False),
("ix_user_preferences_owner_id", ("owner_id",), False),
("ix_user_preferences_source", ("source",), False),
),
)
op.create_table(
"ai_conversations",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("code", sa.String(length=64), nullable=False),
sa.Column("owner_id", sa.Integer(), nullable=False),
sa.Column("chat_type", sa.String(length=32), nullable=False),
sa.Column("chat_key", sa.String(length=256), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(
["owner_id"],
["feishu_users.id"],
name="fk_ai_conversations_owner_id",
ondelete="CASCADE",
),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint(
"owner_id",
"chat_type",
"chat_key",
name="uq_ai_conversation_owner_chat",
),
)
_create_indexes(
"ai_conversations",
(
("ix_ai_conversations_chat_key", ("chat_key",), False),
("ix_ai_conversations_chat_type", ("chat_type",), False),
("ix_ai_conversations_code", ("code",), True),
("ix_ai_conversations_created_at", ("created_at",), False),
("ix_ai_conversations_owner_id", ("owner_id",), False),
("ix_ai_conversations_updated_at", ("updated_at",), False),
),
)
op.create_table(
"ai_conversation_messages",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("conversation_id", sa.Integer(), nullable=False),
sa.Column("role", sa.String(length=32), nullable=False),
sa.Column("content", sa.Text(), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(
["conversation_id"],
["ai_conversations.id"],
name="fk_ai_conversation_messages_conversation_id",
ondelete="CASCADE",
),
sa.PrimaryKeyConstraint("id"),
)
_create_indexes(
"ai_conversation_messages",
(
(
"ix_ai_conversation_messages_conversation_id",
("conversation_id",),
False,
),
("ix_ai_conversation_messages_created_at", ("created_at",), False),
("ix_ai_conversation_messages_role", ("role",), False),
),
)
op.create_table(
"personal_data_erasure_requests",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("owner_id", sa.Integer(), nullable=False),
sa.Column("token_hash", sa.String(length=64), nullable=False),
sa.Column("expires_at", sa.DateTime(), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(
["owner_id"],
["feishu_users.id"],
name="fk_personal_data_erasure_requests_owner_id",
ondelete="CASCADE",
),
sa.PrimaryKeyConstraint("id"),
)
_create_indexes(
"personal_data_erasure_requests",
(
(
"ix_personal_data_erasure_requests_created_at",
("created_at",),
False,
),
(
"ix_personal_data_erasure_requests_expires_at",
("expires_at",),
False,
),
(
"ix_personal_data_erasure_requests_owner_id",
("owner_id",),
True,
),
),
)
def _create_subscription_tables() -> None:
op.create_table(
"push_subscriptions",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("code", sa.String(length=64), nullable=False),
sa.Column("owner_id", sa.Integer(), nullable=False),
sa.Column("target_type", sa.String(length=16), nullable=False),
sa.Column("target_id", sa.String(length=256), nullable=False),
sa.Column("prompt", sa.Text(), nullable=False),
sa.Column("schedule_type", sa.String(length=32), nullable=False),
sa.Column("schedule_config", sa.JSON(), nullable=False),
sa.Column("timezone", sa.String(length=64), nullable=False),
sa.Column("next_run_at", sa.DateTime(), nullable=True),
sa.Column("status", sa.String(length=32), nullable=False),
sa.Column("consented_at", sa.DateTime(), nullable=False),
sa.Column("last_run_at", sa.DateTime(), nullable=True),
sa.Column("locked_by", sa.String(length=128), nullable=True),
sa.Column("locked_until", sa.DateTime(), nullable=True),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(
["owner_id"],
["feishu_users.id"],
name="fk_push_subscriptions_owner_id",
ondelete="CASCADE",
),
sa.PrimaryKeyConstraint("id"),
)
_create_indexes(
"push_subscriptions",
(
("ix_push_subscriptions_code", ("code",), True),
("ix_push_subscriptions_created_at", ("created_at",), False),
("ix_push_subscriptions_locked_by", ("locked_by",), False),
("ix_push_subscriptions_locked_until", ("locked_until",), False),
("ix_push_subscriptions_next_run_at", ("next_run_at",), False),
("ix_push_subscriptions_owner_id", ("owner_id",), False),
("ix_push_subscriptions_schedule_type", ("schedule_type",), False),
("ix_push_subscriptions_status", ("status",), False),
("ix_push_subscriptions_target_type", ("target_type",), False),
),
)
op.create_table(
"push_deliveries",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("code", sa.String(length=64), nullable=False),
sa.Column("subscription_id", sa.Integer(), nullable=False),
sa.Column("scheduled_for", sa.DateTime(), nullable=False),
sa.Column("idempotency_key", sa.String(length=64), nullable=False),
sa.Column("message_uuid", sa.String(length=36), nullable=False),
sa.Column("status", sa.String(length=32), nullable=False),
sa.Column("attempt_count", sa.Integer(), nullable=False),
sa.Column("next_attempt_at", sa.DateTime(), nullable=True),
sa.Column("rendered_content", sa.Text(), nullable=True),
sa.Column("provider_message_id", sa.String(length=256), nullable=True),
sa.Column("last_error", sa.Text(), nullable=True),
sa.Column("locked_by", sa.String(length=128), nullable=True),
sa.Column("locked_until", sa.DateTime(), nullable=True),
sa.Column("sent_at", sa.DateTime(), nullable=True),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(
["subscription_id"],
["push_subscriptions.id"],
name="fk_push_deliveries_subscription_id",
ondelete="CASCADE",
),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint(
"subscription_id",
"scheduled_for",
name="uq_push_delivery_subscription_schedule",
),
)
_create_indexes(
"push_deliveries",
(
("ix_push_deliveries_code", ("code",), True),
("ix_push_deliveries_created_at", ("created_at",), False),
(
"ix_push_deliveries_idempotency_key",
("idempotency_key",),
True,
),
("ix_push_deliveries_locked_by", ("locked_by",), False),
("ix_push_deliveries_locked_until", ("locked_until",), False),
("ix_push_deliveries_message_uuid", ("message_uuid",), True),
(
"ix_push_deliveries_next_attempt_at",
("next_attempt_at",),
False,
),
(
"ix_push_deliveries_provider_message_id",
("provider_message_id",),
False,
),
("ix_push_deliveries_scheduled_for", ("scheduled_for",), False),
("ix_push_deliveries_sent_at", ("sent_at",), False),
("ix_push_deliveries_status", ("status",), False),
(
"ix_push_deliveries_subscription_id",
("subscription_id",),
False,
),
),
)
def downgrade() -> None:
op.drop_table("push_deliveries")
op.drop_table("push_subscriptions")
op.drop_table("personal_data_erasure_requests")
op.drop_table("ai_conversation_messages")
op.drop_table("ai_conversations")
op.drop_table("user_preferences")
with op.batch_alter_table(MARKET_WATCHLIST_TABLE) as batch_op:
batch_op.drop_constraint(
"uq_market_watchlist_owner_symbol",
type_="unique",
)
batch_op.drop_constraint(
"fk_market_watchlists_owner_id",
type_="foreignkey",
)
batch_op.drop_index("ix_market_watchlists_owner_id")
batch_op.drop_column("owner_id")
batch_op.create_unique_constraint(
"uq_market_watchlist_actor_symbol",
["actor", "symbol"],
)
op.execute(
sa.text(
"""
UPDATE ai_memory_entries
SET source = 'user_rule'
WHERE owner_id IS NULL
AND kind = 'company_rule'
AND source = 'legacy_company'
"""
)
)
with op.batch_alter_table(AI_MEMORY_TABLE) as batch_op:
batch_op.drop_constraint(
"uq_ai_memory_owner_fingerprint",
type_="unique",
)
batch_op.drop_constraint(
"fk_ai_memory_entries_owner_id",
type_="foreignkey",
)
batch_op.drop_index("ix_ai_memory_entries_fingerprint")
batch_op.drop_index("ix_ai_memory_entries_kind")
batch_op.drop_index("ix_ai_memory_entries_owner_id")
batch_op.drop_column("kind")
batch_op.drop_column("owner_id")
batch_op.create_index(
"ix_ai_memory_entries_fingerprint",
["fingerprint"],
unique=True,
)
op.drop_table("feishu_users")

43
alembic/versions/202607260004_feishu_app_tickets.py Normal file
View File

@@ -0,0 +1,43 @@
"""Persist verified Feishu app tickets.
Revision ID: 202607260004
Revises: 202607260003
Create Date: 2026-07-26
"""
from alembic import op
import sqlalchemy as sa
revision = "202607260004"
down_revision = "202607260003"
branch_labels = None
depends_on = None
def upgrade() -> None:
op.create_table(
"feishu_app_tickets",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("app_id", sa.String(length=128), nullable=False),
sa.Column("app_ticket", sa.Text(), nullable=False),
sa.Column("received_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.PrimaryKeyConstraint("id"),
)
op.create_index(
op.f("ix_feishu_app_tickets_app_id"),
"feishu_app_tickets",
["app_id"],
unique=True,
)
op.create_index(
op.f("ix_feishu_app_tickets_received_at"),
"feishu_app_tickets",
["received_at"],
unique=False,
)
def downgrade() -> None:
op.drop_table("feishu_app_tickets")

41
alembic/versions/202607260005_feishu_admin_bootstrap_tombstones.py Normal file
View File

@@ -0,0 +1,41 @@
"""Prevent erased initial administrators from being bootstrapped again.
Revision ID: 202607260005
Revises: 202607260004
Create Date: 2026-07-27
"""
from alembic import op
import sqlalchemy as sa
revision = "202607260005"
down_revision = "202607260004"
branch_labels = None
depends_on = None
def upgrade() -> None:
op.create_table(
"feishu_admin_bootstrap_tombstones",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("identity_hash", sa.String(length=64), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.PrimaryKeyConstraint("id"),
)
op.create_index(
op.f("ix_feishu_admin_bootstrap_tombstones_created_at"),
"feishu_admin_bootstrap_tombstones",
["created_at"],
unique=False,
)
op.create_index(
op.f("ix_feishu_admin_bootstrap_tombstones_identity_hash"),
"feishu_admin_bootstrap_tombstones",
["identity_hash"],
unique=True,
)
def downgrade() -> None:
op.drop_table("feishu_admin_bootstrap_tombstones")