feat(coderd/database): add chat_context_resources table (#26430)

Adds chat_context_resources: a per-chat pinned copy of the agent context
resources a chat is hydrated against. The agent-side table
(workspace_agent_context_resources) is last-writer-wins with no history,
so a chat copies its resources at hydration/refresh to keep a stable view
while the agent drifts.

Schema foundation only (no queries/dbauthz/prepareGeneration/SDK yet).
chat_id FK ON DELETE CASCADE for cleanup parity; no agent FK so the pin
survives agent replacement; PK (chat_id, source); reuses the 000522 enum
types.
This commit is contained in:
Kyle Carberry
2026-06-16 14:28:57 -07:00
committed by GitHub
parent 3e68dd304a
commit 53a6459ecd
10 changed files with 182 additions and 6 deletions
@@ -0,0 +1,4 @@
-- The workspace_agent_context_* enum types are owned by migration
-- 000522 and are still in use by workspace_agent_context_resources, so
-- they are intentionally left in place here.
DROP TABLE IF EXISTS chat_context_resources;
@@ -0,0 +1,30 @@
-- Creates chat_context_resources: a per-chat pinned copy of
-- workspace_agent_context_resources (semantics in COMMENT ON TABLE
-- below). Migration-specific notes: there is deliberately no FK to
-- workspace_agents so the pin survives agent replacement and workspace
-- rebuilds, and the body_kind/status enum types are reused from 000522
-- and must not be recreated here.
CREATE TABLE chat_context_resources (
chat_id UUID NOT NULL REFERENCES chats(id) ON DELETE CASCADE,
source TEXT NOT NULL,
body_kind workspace_agent_context_body_kind NOT NULL,
body JSONB NOT NULL,
content_hash BYTEA NOT NULL,
size_bytes BIGINT NOT NULL,
status workspace_agent_context_resource_status NOT NULL,
error TEXT NOT NULL DEFAULT '',
source_path TEXT NOT NULL DEFAULT '',
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
updated_at TIMESTAMPTZ NOT NULL DEFAULT now(),
PRIMARY KEY (chat_id, source)
);
COMMENT ON TABLE chat_context_resources IS 'Per-chat pinned copy of the agent context resources a chat is hydrated against. Copied from workspace_agent_context_resources at chat hydration and context refresh; survives agent replacement and workspace rebuilds.';
COMMENT ON COLUMN chat_context_resources.source IS 'Resource locator: canonical file path for file-backed kinds, or the MCP server name for mcp_server resources.';
COMMENT ON COLUMN chat_context_resources.body_kind IS 'Discriminator for the body JSON shape. Matches the proto oneof variant: instruction_file, skill, mcp_config, mcp_server. PLUGIN/HOOK/SUBAGENT/COMMAND are reserved for the Claude Code plugin RFC.';
COMMENT ON COLUMN chat_context_resources.body IS 'protojson-encoded variant body matching body_kind. Always populated; non-OK statuses use the variant zero value so the wire kind is still attributable.';
COMMENT ON COLUMN chat_context_resources.content_hash IS 'sha256 over the resource''s original bytes (or transport-encoded server tool list).';
COMMENT ON COLUMN chat_context_resources.size_bytes IS 'Original payload size in bytes; populated regardless of status.';
COMMENT ON COLUMN chat_context_resources.status IS 'Per-resource status. ok carries a populated body; oversize, unreadable, invalid, and excluded carry an empty body plus an error string.';
COMMENT ON COLUMN chat_context_resources.error IS 'Per-resource error or warning string. Populated whenever status is non-ok; may also carry a non-fatal warning when status is ok.';
COMMENT ON COLUMN chat_context_resources.source_path IS 'User-declared scan root that produced this resource. Empty for built-in scan roots.';
@@ -0,0 +1,82 @@
-- Pinned context resources covering each non-reserved body kind plus a
-- non-OK status. The earlier chat fixtures already insert at least one row
-- into chats; we attach the resources to the first such chat (ordered
-- deterministically) so migration tests see a non-empty
-- chat_context_resources table without hard-coding a specific chat ID.
INSERT INTO chat_context_resources (
chat_id,
source,
body_kind,
body,
content_hash,
size_bytes,
status,
error,
source_path
)
SELECT
c.id,
v.source,
v.body_kind::workspace_agent_context_body_kind,
v.body::jsonb,
decode(v.content_hash, 'hex'),
v.size_bytes,
v.status::workspace_agent_context_resource_status,
v.error,
v.source_path
FROM (
SELECT id FROM chats ORDER BY created_at, id LIMIT 1
) AS c
CROSS JOIN (
VALUES
(
'/home/coder/workspace/AGENTS.md',
'instruction_file',
'{"content":"aGVsbG8="}',
'1111111111111111111111111111111111111111111111111111111111111111',
5::bigint,
'ok',
'',
''
),
(
'/home/coder/workspace/.agents/skills/example/SKILL.md',
'skill',
'{"meta":"LS0tCm5hbWU6IGV4YW1wbGUKLS0tCmJvZHk=","name":"example","description":"Example skill"}',
'2222222222222222222222222222222222222222222222222222222222222222',
32::bigint,
'ok',
'',
'/home/coder/workspace'
),
(
'/home/coder/workspace/.mcp.json',
'mcp_config',
'{}',
'3333333333333333333333333333333333333333333333333333333333333333',
128::bigint,
'ok',
'',
''
),
(
'mcp:echo',
'mcp_server',
'{"server_name":"echo","description":"echoes input"}',
'4444444444444444444444444444444444444444444444444444444444444444',
256::bigint,
'ok',
'',
'/home/coder/workspace/.mcp.json'
),
(
'/home/coder/workspace/big.md',
'instruction_file',
'{}',
'5555555555555555555555555555555555555555555555555555555555555555',
99999::bigint,
'oversize',
'file exceeds 64KiB per-resource cap',
''
)
) AS v(source, body_kind, body, content_hash, size_bytes, status, error, source_path);