forked from EduCraft/curriculum-project-hub
fix: bind agent sessions to role
This commit is contained in:
@@ -0,0 +1,9 @@
|
||||
-- ADR-0017 refinement: Hub sessions are provider/role/model-bound.
|
||||
-- Existing sessions predate role-aware binding; treat them as draft sessions.
|
||||
ALTER TABLE "AgentSession" ADD COLUMN "roleId" TEXT NOT NULL DEFAULT 'draft';
|
||||
ALTER TABLE "AgentSession" ALTER COLUMN "roleId" DROP DEFAULT;
|
||||
|
||||
CREATE INDEX "AgentSession_provider_roleId_model_idx" ON "AgentSession"("provider", "roleId", "model");
|
||||
CREATE INDEX "AgentSession_projectId_provider_roleId_model_archivedAt_idx" ON "AgentSession"("projectId", "provider", "roleId", "model", "archivedAt");
|
||||
|
||||
DROP INDEX "AgentSession_provider_model_idx";
|
||||
@@ -190,14 +190,15 @@ model ProjectGroupBinding {
|
||||
|
||||
// --- AgentRun, session, lock (ADR-0002, 0017) -----------------------------
|
||||
|
||||
/// ADR-0017: session is provider/model-bound. `provider` + `model` capture
|
||||
/// the binding; provider-specific runtime cursors live in `metadata`
|
||||
/// (e.g. metadata.claudeSessionId). Same provider+model ⇒ reuse across runs
|
||||
/// (ADR-0002); a switch ⇒ new Hub session.
|
||||
/// ADR-0017: session is provider/role/model-bound. `provider` + `roleId` +
|
||||
/// `model` capture the binding; provider-specific runtime cursors live in
|
||||
/// `metadata` (e.g. metadata.claudeSessionId). Same provider+role+model ⇒
|
||||
/// reuse across runs (ADR-0002); a switch ⇒ new Hub session.
|
||||
model AgentSession {
|
||||
id String @id @default(cuid())
|
||||
projectId String
|
||||
provider String
|
||||
roleId String
|
||||
model String
|
||||
title String?
|
||||
metadata Json
|
||||
@@ -210,7 +211,8 @@ model AgentSession {
|
||||
messages AgentMessage[] @relation("sessionMessages")
|
||||
|
||||
@@index([projectId, archivedAt])
|
||||
@@index([provider, model])
|
||||
@@index([provider, roleId, model])
|
||||
@@index([projectId, provider, roleId, model, archivedAt])
|
||||
@@index([updatedAt])
|
||||
}
|
||||
|
||||
|
||||
@@ -180,23 +180,26 @@ export function makeTriggerHandler(deps: TriggerDeps): TriggerHandler {
|
||||
});
|
||||
const promptForAgent = withFeishuTriggerContext(agentPrompt, feishuTriggerContext);
|
||||
|
||||
// ADR-0017: provider+model-bound session. Reuse if one exists for this
|
||||
// (project, provider, model) triple; otherwise create.
|
||||
// ADR-0017: provider+role+model-bound session. Reuse if one exists for
|
||||
// this (project, provider, role, model) tuple; otherwise create. Role is
|
||||
// part of the key because role prompts/tool surfaces can differ even when
|
||||
// the underlying model is the same.
|
||||
const existingSession = await deps.prisma.agentSession.findFirst({
|
||||
where: { projectId, provider: providerId, model, archivedAt: null },
|
||||
where: { projectId, provider: providerId, roleId, model, archivedAt: null },
|
||||
select: { id: true, metadata: true },
|
||||
});
|
||||
const session =
|
||||
existingSession !== null
|
||||
? await deps.prisma.agentSession.update({
|
||||
where: { id: existingSession.id },
|
||||
data: { provider: providerId, model, updatedAt: new Date() },
|
||||
data: { provider: providerId, roleId, model, updatedAt: new Date() },
|
||||
select: { id: true, metadata: true },
|
||||
})
|
||||
: await deps.prisma.agentSession.create({
|
||||
data: {
|
||||
projectId,
|
||||
provider: providerId,
|
||||
roleId,
|
||||
model,
|
||||
title: agentPrompt.slice(0, 40) || null,
|
||||
metadata: {},
|
||||
|
||||
@@ -421,6 +421,34 @@ describe("trigger full lifecycle (integration)", () => {
|
||||
});
|
||||
});
|
||||
|
||||
it("keeps role sessions separate even when roles share a model", async () => {
|
||||
await seedProject("proj-12b", "chat-12b");
|
||||
await prisma.roleTriggerGrant.create({
|
||||
data: { projectId: "proj-12b", roleId: "review", principalType: "USER", principalId: "ou_test_user" },
|
||||
});
|
||||
const trigger = makeTriggerHandler({ prisma, settings, logger: silentLogger, runAgent, messageBatcherOptions: { maxMessages: 1 } });
|
||||
|
||||
await trigger(makeEvent("chat-12b", "@_user_1 /draft 写第三单元"), rt);
|
||||
await vi.waitFor(async () => {
|
||||
const runs = await prisma.agentRun.findMany();
|
||||
expect(runs).toHaveLength(1);
|
||||
expect(runs[0]?.status).toBe("COMPLETED");
|
||||
});
|
||||
|
||||
await trigger(makeEvent("chat-12b", "@_user_1 /review 看看这节"), rt);
|
||||
await vi.waitFor(async () => {
|
||||
const runs = await prisma.agentRun.findMany();
|
||||
expect(runs).toHaveLength(2);
|
||||
expect(runs.every((run) => run.status === "COMPLETED")).toBe(true);
|
||||
});
|
||||
|
||||
const sessions = await prisma.agentSession.findMany({ orderBy: { roleId: "asc" } });
|
||||
expect(sessions.map((session) => session.roleId)).toEqual(["draft", "review"]);
|
||||
expect(new Set(sessions.map((session) => session.model))).toEqual(new Set(["mock-model"]));
|
||||
expect(new Set(sessions.map((session) => session.id)).size).toBe(2);
|
||||
expect(runAgentCalls[1]?.resumeSessionId).toBeUndefined();
|
||||
});
|
||||
|
||||
it("dedups a redelivered event by event_id (no second run)", async () => {
|
||||
await seedProject("proj-13", "chat-13");
|
||||
const trigger = makeTriggerHandler({ prisma, settings, logger: silentLogger, runAgent, messageBatcherOptions: { maxMessages: 1 } });
|
||||
|
||||
Reference in New Issue
Block a user