Jaroslav Beneš
6cffcb357d
The other half of background execution: a job that finishes while nobody is looking prompts the model back with its result, rather than sitting unread until the model happens to run again. The vehicle is the queue, because it is the only wiring that already delivers a turn into or after a reply. A per-job poller notices completion and calls jobs.wake. If a reply is being written the completion is left queued for that reply's _inject/_drain; if the chat is idle a fresh reply is started to answer it -- the send_queued_now move. All of it under a per-chat lock with no await between the running-check and ensure, so two jobs finishing at once cannot each spin up a generation: the second sees the first's reply already live and leaves its completion for it. That is the invariant the queue exists to hold, reached from outside a request for the first time. The completion is a user-role turn whose content names itself a machine event -- "A background job you started has finished" -- not a bare person turn. _inject sends a queued turn verbatim, so the framing cannot live there; it lives in the words, the way execute_plan quotes the plan, and a tool.background fragment tells the model these arrive and are a machine event rather than the person speaking. The poller reconnects a fresh connection each tick rather than holding one open -- holding one is the exact live-connection state the whole ssh.py/base.py design forbids, and poll is self-healing besides. Bounded by background_max_jobs and a six-hour ceiling, after which the remote job may keep running but we stop watching it. A Job table, and here the terminal/generation "lost on restart" precedent does NOT transfer: those are seconds long with a human watching, a background job is hours long with nobody watching -- the one case a restart forgetting it would silently break the feature's whole promise. So the row lets a lifespan startup hook rehydrate the watcher and wake as if nothing happened. Cancelling a watcher never stops the detached remote job; it runs on and is picked back up. Tested end to end against a real local shell: launch a detached command, poll it to completion through a watcher, and assert the model was woken with the exit code and output -- plus the lock proving two simultaneous completions start one reply, not two. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
141 lines
2.6 KiB
Python
141 lines
2.6 KiB
Python
"""All ORM models.
|
|
|
|
Importing this package registers every table on ``Base.metadata``, which is
|
|
what ``init_db()`` relies on to create the schema at startup. Any new model
|
|
module must be imported here or its table will silently never be created.
|
|
"""
|
|
|
|
from lembas.db.models.agent import (
|
|
AUTH_KEY,
|
|
AUTH_METHODS,
|
|
AUTH_PASSWORD,
|
|
Job,
|
|
SshProfile,
|
|
)
|
|
from lembas.db.models.attachment import (
|
|
KIND_DOCUMENT,
|
|
KIND_IMAGE,
|
|
KIND_TEXT,
|
|
Attachment,
|
|
)
|
|
from lembas.db.models.chat import (
|
|
KIND_AGENT,
|
|
KIND_CHAT,
|
|
KINDS,
|
|
ROLE_ASSISTANT,
|
|
ROLE_SYSTEM,
|
|
ROLE_TOOL,
|
|
ROLE_USER,
|
|
Chat,
|
|
Folder,
|
|
Message,
|
|
)
|
|
from lembas.db.models.connection import Connection, Model, model_groups
|
|
from lembas.db.models.library import (
|
|
AUTHOR_MODEL,
|
|
AUTHOR_USER,
|
|
PRINCIPAL_GROUP,
|
|
PRINCIPAL_USER,
|
|
RESOURCE_BASE,
|
|
RESOURCE_NOTE,
|
|
RESOURCE_SKILL,
|
|
SOURCE_LINK,
|
|
SOURCE_UPLOAD,
|
|
Document,
|
|
KnowledgeBase,
|
|
Memory,
|
|
Note,
|
|
Share,
|
|
Skill,
|
|
SkillRevision,
|
|
chat_knowledge_bases,
|
|
)
|
|
from lembas.db.models.setting import Setting
|
|
from lembas.db.models.suggestion import Suggestion
|
|
from lembas.db.models.tool import (
|
|
RESPONSE_JSON,
|
|
RESPONSE_MODES,
|
|
RESPONSE_RAW,
|
|
RESPONSE_TEXT,
|
|
SECRET_BEARER,
|
|
SECRET_HEADER,
|
|
SECRET_NONE,
|
|
SECRET_PLACEMENTS,
|
|
SECRET_QUERY,
|
|
CustomTool,
|
|
McpServer,
|
|
custom_tool_groups,
|
|
mcp_server_groups,
|
|
)
|
|
from lembas.db.models.user import (
|
|
ROLE_ADMIN,
|
|
ROLE_PENDING,
|
|
Group,
|
|
Session,
|
|
User,
|
|
user_groups,
|
|
)
|
|
|
|
__all__ = [
|
|
"AUTHOR_MODEL",
|
|
"AUTH_KEY",
|
|
"AUTH_METHODS",
|
|
"AUTH_PASSWORD",
|
|
"AUTHOR_USER",
|
|
"Attachment",
|
|
"KINDS",
|
|
"KIND_AGENT",
|
|
"KIND_CHAT",
|
|
"KIND_DOCUMENT",
|
|
"KIND_IMAGE",
|
|
"KIND_TEXT",
|
|
"PRINCIPAL_GROUP",
|
|
"PRINCIPAL_USER",
|
|
"RESOURCE_BASE",
|
|
"RESOURCE_NOTE",
|
|
"RESOURCE_SKILL",
|
|
"RESPONSE_JSON",
|
|
"RESPONSE_MODES",
|
|
"RESPONSE_RAW",
|
|
"RESPONSE_TEXT",
|
|
"ROLE_ADMIN",
|
|
"ROLE_ASSISTANT",
|
|
"ROLE_PENDING",
|
|
"ROLE_SYSTEM",
|
|
"ROLE_TOOL",
|
|
"ROLE_USER",
|
|
"SECRET_BEARER",
|
|
"SECRET_HEADER",
|
|
"SECRET_NONE",
|
|
"SECRET_PLACEMENTS",
|
|
"SECRET_QUERY",
|
|
"SOURCE_LINK",
|
|
"SOURCE_UPLOAD",
|
|
"Chat",
|
|
"Job",
|
|
"Connection",
|
|
"CustomTool",
|
|
"Document",
|
|
"Folder",
|
|
"Group",
|
|
"KnowledgeBase",
|
|
"McpServer",
|
|
"Memory",
|
|
"Message",
|
|
"Model",
|
|
"Note",
|
|
"Session",
|
|
"Setting",
|
|
"Share",
|
|
"Skill",
|
|
"SshProfile",
|
|
"SkillRevision",
|
|
"Suggestion",
|
|
"User",
|
|
"chat_knowledge_bases",
|
|
"custom_tool_groups",
|
|
"mcp_server_groups",
|
|
"model_groups",
|
|
"user_groups",
|
|
]
|