Files
PaperClipAI/tests/runner-e2e/chat-catalog-domains/internal-document-context.ts
T

58 lines
4.3 KiB
TypeScript

import type { ChatCatalogContext } from "../chat-catalog-contract.js";
import type { IssueDocument, IssueAttachment } from "../../../packages/shared/src/types/issue.js";
import type { Entity } from "./internal-support.js";
import { uploadFactualQueryAttachment } from "../chat-query-attachments.js";
/** Visible source metadata plus an actual human query attachment: conversation
* descriptions alone are intentionally omitted from native wake context. */
export async function selectDocumentSource(ctx: ChatCatalogContext, input: {
task: Entity;
document?: { id: string; key: string; latestRevisionId: string | null };
annotationId?: string;
attachment?: IssueAttachment;
workspaceFilename?: string;
}) {
const { api, fixtures } = ctx.input;
const target = await api.get<Entity>(`/api/issues/${input.task.id}`);
if (target.id !== input.task.id || target.companyId !== fixtures.company.id || target.projectId !== input.task.projectId)
throw new Error("Selected document source does not belong to the expected task/company/project");
const taskHref = `/${fixtures.company.issuePrefix}/issues/${target.identifier ?? target.id}`;
const lines = [`Selected source: [${target.title}](${taskHref}).`];
if (input.document) {
const doc = await api.get<IssueDocument>(`/api/issues/${target.id}/documents/${encodeURIComponent(input.document.key)}`);
if (doc.id !== input.document.id || doc.issueId !== target.id || doc.latestRevisionId !== input.document.latestRevisionId)
throw new Error("Selected source document or revision changed before the canonical turn");
lines.push(`Selected document: [${doc.title ?? doc.key}](${taskHref}#document-${encodeURIComponent(doc.key)}).`);
if (input.annotationId) {
const annotations = await api.get<Entity[]>(`/api/issues/${target.id}/documents/${encodeURIComponent(doc.key)}/annotations?status=all&includeComments=true`);
const annotation = annotations.find(row => row.id === input.annotationId);
if (!annotation || annotation.documentId !== doc.id || annotation.originalRevisionId !== doc.latestRevisionId || annotation.status !== "open")
throw new Error("Selected review annotation is not open on the actual source revision");
lines.push(`Selected review comment: ${annotation.id} on this document.`);
}
} else if (input.annotationId) throw new Error("A selected annotation requires its actual document");
if (input.attachment) {
const file = (await api.get<IssueAttachment[]>(`/api/issues/${target.id}/attachments`)).find(row => row.id === input.attachment!.id);
if (!file || file.issueId !== target.id || file.sha256 !== input.attachment.sha256 || file.contentPath !== input.attachment.contentPath)
throw new Error("Selected prior delivery is not the original source attachment");
lines.push(`Selected existing delivery: [${file.originalFilename}](${file.contentPath}).`);
}
if (input.workspaceFilename) lines.push(`Selected workspace file: ${input.workspaceFilename}, in the selected task's authorized workspace.`);
const description = lines.join("\n\n");
await api.post(`/api/companies/${fixtures.company.id}/chats/${fixtures.agent.id}`);
await ctx.refreshIssue();
const source = ctx.issue();
if (!source || source.id === target.id) throw new Error("Source selection requires the actual separate standing conversation");
await api.patch(`/api/issues/${source.id}`, { projectId: target.projectId, description });
const saved = await api.get<Entity>(`/api/issues/${source.id}`);
if (saved.id !== source.id || saved.companyId !== fixtures.company.id || saved.projectId !== target.projectId || saved.description !== description)
throw new Error("Visible source selection did not persist on the canonical conversation");
await ctx.refreshIssue();
const factualContextAttachment = await uploadFactualQueryAttachment(ctx, { filename: "selected-source.md", body: description });
const receipt = { sourceIssueId: saved.id, targetIssueId: target.id, projectId: target.projectId, description,
documentId: input.document?.id, revisionId: input.document?.latestRevisionId, annotationId: input.annotationId,
attachmentId: input.attachment?.id, attachmentSha256: input.attachment?.sha256, workspaceFilename: input.workspaceFilename, factualContextAttachment };
await ctx.input.evidence("chat-catalog-selected-document-source.json", receipt);
return receipt;
}