mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-27 21:50:36 +00:00
e8c7df9042
* feat(channels): arbitrate IM transports with a per-provider leader lease Every Gateway instance connected every enabled IM platform with the same bot credentials, so channels could not run on more than one instance (Telegram 409 storms, Discord answering twice, scattered per-process state). channels.leader_lease was reserved by #6746; this implements it. - channel_leases table (migration 0041): one row per provider. Acquire is one dialect-native upsert decided against the database clock and bumps a fencing epoch; renew extends only a live lease at the holder's epoch; release sets the expiry to 0 so a peer takes over on its next tick. - app/channels/leadership.py: the holder's local monotonic deadline (send time of the last confirmed renewal + lease_seconds); holds() turns false one renewal interval early, so the transport stops at least grace_seconds before a peer may connect. DB calls are bounded. - ChannelService: start fence in _start_channel (no DB call) behind every start path; fixed-rate supervisor renews/acquires and runs transport starts/stops as per-provider tasks; takeover reloads the provider's saved config; failed starts back off; shutdown releases only leases whose transport stopped. get_status reports leader/leader_here and the holder's transport state on every instance. - Notification worker leaves rows of providers another instance holds. - Gate: the enabled=false refusal now recommends auto/true; an explicit true with database.backend memory is refused when a channel is enabled. - Docs, Helm, deployment guides, AGENTS.md and CHANGELOG updated. * docs(changelog): link the channel leader lease entry to #6843 * fix(channels): see peer-held leases for providers absent from local config Review on #6843: a provider connected from the web UI on a peer never enters this instance's channel config, so the notification worker here still claimed its rows (and parked them for 900s) and GET /api/channels/ reported no leader for it. Every supervision round, and startup, now also reads the whole lease table once (bounded), and notification exclusions and status use those observations: a provider whose live lease a peer holds is left to it even without local config. Running that read concurrently with renewals exposed that, through SQLAlchemy's aiosqlite adapter, an UPDATE ... RETURNING racing a read on another connection can hand back one of its own bind values instead of the column (stored rows stay correct; plain sqlite3 and raw aiosqlite do not show it). A lease must never take such a row for a confirmed renewal, so acquire/renew/release now decide by affected-row count and acquire reads its epoch back in the same transaction. --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
78 lines
3.1 KiB
Python
78 lines
3.1 KiB
Python
"""ORM model registration entry point.
|
|
|
|
Importing this module ensures all ORM models are registered with
|
|
``Base.metadata`` so Alembic autogenerate detects every table.
|
|
|
|
The actual ORM classes have moved to entity-specific subpackages:
|
|
- ``deerflow.persistence.thread_meta``
|
|
- ``deerflow.persistence.run``
|
|
- ``deerflow.persistence.feedback``
|
|
- ``deerflow.persistence.user``
|
|
|
|
``RunEventRow`` remains in ``deerflow.persistence.models.run_event`` because
|
|
its storage implementation lives in ``deerflow.runtime.events.store.db`` and
|
|
there is no matching entity directory.
|
|
"""
|
|
|
|
from deerflow.persistence.agents.model import AgentRow
|
|
from deerflow.persistence.channel_connections.model import (
|
|
ChannelConnectionRow,
|
|
ChannelConversationRow,
|
|
ChannelCredentialRow,
|
|
ChannelOAuthStateRow,
|
|
)
|
|
from deerflow.persistence.channel_leases.model import ChannelLeaseRow
|
|
from deerflow.persistence.channel_thread_bindings.model import ChannelThreadBindingRow
|
|
from deerflow.persistence.feedback.model import FeedbackRow
|
|
from deerflow.persistence.login_throttle.model import LoginThrottleRow
|
|
from deerflow.persistence.managed_subagents.model import ManagedSubagentRow
|
|
from deerflow.persistence.mcp_tasks.model import McpTaskRow
|
|
from deerflow.persistence.models.run_event import RunEventRow, RunEventThreadSeqRow
|
|
from deerflow.persistence.notification_deliveries.model import NotificationDeliveryRow
|
|
from deerflow.persistence.personal_access_tokens.model import PersonalAccessTokenRow
|
|
from deerflow.persistence.projects.model import ProjectDocumentRow, ProjectRow
|
|
from deerflow.persistence.run.model import RunChangeClockRow, RunRow
|
|
from deerflow.persistence.scheduled_task_events.model import ScheduledTaskEventRow
|
|
from deerflow.persistence.scheduled_task_runs.model import ScheduledTaskRunRow
|
|
from deerflow.persistence.scheduled_tasks.model import ScheduledTaskRow
|
|
from deerflow.persistence.subagent_batches.model import SubagentBatchItemRow, SubagentBatchRow
|
|
from deerflow.persistence.thread_meta.model import ThreadMetaRow
|
|
from deerflow.persistence.thread_reads.model import ThreadReadMarkerRow, ThreadReadVersionRow
|
|
from deerflow.persistence.user.im_notifications import ImNotificationPreferenceRow
|
|
from deerflow.persistence.user.model import UserPreferenceRow, UserRow
|
|
from deerflow.persistence.webhook_delivery.model import WebhookDeliveryRow
|
|
|
|
__all__ = [
|
|
"AgentRow",
|
|
"ChannelConnectionRow",
|
|
"ChannelConversationRow",
|
|
"ChannelCredentialRow",
|
|
"ChannelLeaseRow",
|
|
"ChannelOAuthStateRow",
|
|
"ChannelThreadBindingRow",
|
|
"FeedbackRow",
|
|
"LoginThrottleRow",
|
|
"McpTaskRow",
|
|
"ManagedSubagentRow",
|
|
"NotificationDeliveryRow",
|
|
"PersonalAccessTokenRow",
|
|
"ProjectDocumentRow",
|
|
"ProjectRow",
|
|
"RunEventRow",
|
|
"RunEventThreadSeqRow",
|
|
"RunChangeClockRow",
|
|
"RunRow",
|
|
"ScheduledTaskEventRow",
|
|
"ScheduledTaskRow",
|
|
"ScheduledTaskRunRow",
|
|
"SubagentBatchRow",
|
|
"SubagentBatchItemRow",
|
|
"ThreadMetaRow",
|
|
"ThreadReadMarkerRow",
|
|
"ThreadReadVersionRow",
|
|
"UserPreferenceRow",
|
|
"ImNotificationPreferenceRow",
|
|
"UserRow",
|
|
"WebhookDeliveryRow",
|
|
]
|