michal/mcpctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(db): add personalities + agent-direct prompts schema (Stage 1)

Browse Source 
A Personality is a named overlay on top of an Agent — same agent,
same LLM, but a different bundle of prompts injected into the system
block at chat time. VLAN-on-ethernet semantics: ethernet still works
without VLAN; with a VLAN tag, frames are segmented but still ethernet.

Schema additions:
- Prompt.agentId (nullable FK + index, cascade on delete) so prompts
  can attach directly to an agent without going through a project.
- Personality { id, name, description, agentId, priority } with
  unique (name, agentId).
- PersonalityPrompt join table with per-binding priority override.
- Agent.defaultPersonalityId (SetNull on delete) so an agent can pick
  one personality as the default when no --personality flag is passed.

Backwards-compatible by construction: every new column is nullable;
existing rows are valid as-is; the chat.service systemBlock changes
land in Stage 3.

8 new prisma-level assertions in agent-schema.test.ts cover unique
constraints, cascade behavior, the SetNull on defaultPersonalityId,
and shared-prompt-across-personalities. All 16 db tests pass; mcpd
typecheck + 777 mcpd unit tests still green.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Michal
2026-04-26 19:12:22 +01:00
parent 9389ffff3c
commit f60f00f1fd
4 changed files with 347 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
src/db/prisma/migrations/20260426180927_add_personalities_and_agent_prompts/migration.sql Normal file
View File

@@ -0,0 +1,59 @@
-- Add agent-direct prompts and per-agent personalities (with personality<->prompt join).
-- ── Prompt: optional agentId, mirrors projectId pattern ──
ALTER TABLE "Prompt" ADD COLUMN "agentId" TEXT;
CREATE UNIQUE INDEX "Prompt_name_agentId_key" ON "Prompt"("name", "agentId");
CREATE INDEX "Prompt_agentId_idx" ON "Prompt"("agentId");
ALTER TABLE "Prompt" ADD CONSTRAINT "Prompt_agentId_fkey"
FOREIGN KEY ("agentId") REFERENCES "Agent"("id") ON DELETE CASCADE ON UPDATE CASCADE;
-- ── Agent: defaultPersonalityId (optional FK; SET NULL on delete to keep agent alive) ──
ALTER TABLE "Agent" ADD COLUMN "defaultPersonalityId" TEXT;
CREATE INDEX "Agent_defaultPersonalityId_idx" ON "Agent"("defaultPersonalityId");
-- ── Personality table ──
CREATE TABLE "Personality" (
"id" TEXT NOT NULL,
"name" TEXT NOT NULL,
"description" TEXT NOT NULL DEFAULT '',
"agentId" TEXT NOT NULL,
"priority" INTEGER NOT NULL DEFAULT 5,
"createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
"updatedAt" TIMESTAMP(3) NOT NULL,
CONSTRAINT "Personality_pkey" PRIMARY KEY ("id")
);
CREATE UNIQUE INDEX "Personality_name_agentId_key" ON "Personality"("name", "agentId");
CREATE INDEX "Personality_agentId_idx" ON "Personality"("agentId");
ALTER TABLE "Personality" ADD CONSTRAINT "Personality_agentId_fkey"
FOREIGN KEY ("agentId") REFERENCES "Agent"("id") ON DELETE CASCADE ON UPDATE CASCADE;
-- Now that Personality exists, the Agent.defaultPersonalityId FK can point at it.
ALTER TABLE "Agent" ADD CONSTRAINT "Agent_defaultPersonalityId_fkey"
FOREIGN KEY ("defaultPersonalityId") REFERENCES "Personality"("id") ON DELETE SET NULL ON UPDATE CASCADE;
-- ── PersonalityPrompt join table ──
CREATE TABLE "PersonalityPrompt" (
"id" TEXT NOT NULL,
"personalityId" TEXT NOT NULL,
"promptId" TEXT NOT NULL,
"priority" INTEGER NOT NULL DEFAULT 5,
"createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
CONSTRAINT "PersonalityPrompt_pkey" PRIMARY KEY ("id")
);
CREATE UNIQUE INDEX "PersonalityPrompt_personalityId_promptId_key" ON "PersonalityPrompt"("personalityId", "promptId");
CREATE INDEX "PersonalityPrompt_personalityId_idx" ON "PersonalityPrompt"("personalityId");
CREATE INDEX "PersonalityPrompt_promptId_idx" ON "PersonalityPrompt"("promptId");
ALTER TABLE "PersonalityPrompt" ADD CONSTRAINT "PersonalityPrompt_personalityId_fkey"
FOREIGN KEY ("personalityId") REFERENCES "Personality"("id") ON DELETE CASCADE ON UPDATE CASCADE;
ALTER TABLE "PersonalityPrompt" ADD CONSTRAINT "PersonalityPrompt_promptId_fkey"
FOREIGN KEY ("promptId") REFERENCES "Prompt"("id") ON DELETE CASCADE ON UPDATE CASCADE;

228
src/db/prisma/schema.prisma
View File

@@ -21,13 +21,13 @@ model User {
createdAt DateTime @default(now()) createdAt DateTime @default(now())
updatedAt DateTime @updatedAt updatedAt DateTime @updatedAt
sessions Session[] sessions Session[]
auditLogs AuditLog[] auditLogs AuditLog[]
ownedProjects Project[] ownedProjects Project[]
groupMemberships GroupMember[] groupMemberships GroupMember[]
mcpTokens McpToken[] mcpTokens McpToken[]
ownedAgents Agent[] ownedAgents Agent[]
chatThreads ChatThread[] chatThreads ChatThread[]
@@index([email]) @@index([email])
} }
@@ -56,23 +56,23 @@ model Session {
// ── MCP Servers ── // ── MCP Servers ──
model McpServer { model McpServer {
id String @id @default(cuid()) id String @id @default(cuid())
name String @unique name String @unique
description String @default("") description String @default("")
packageName String? packageName String?
runtime String? runtime String?
dockerImage String? dockerImage String?
transport Transport @default(STDIO) transport Transport @default(STDIO)
repositoryUrl String? repositoryUrl String?
externalUrl String? externalUrl String?
command Json? command Json?
containerPort Int? containerPort Int?
replicas Int @default(1) replicas Int @default(1)
env Json @default("[]") env Json @default("[]")
healthCheck Json? healthCheck Json?
version Int @default(1) version Int @default(1)
createdAt DateTime @default(now()) createdAt DateTime @default(now())
updatedAt DateTime @updatedAt updatedAt DateTime @updatedAt
templateName String? templateName String?
templateVersion String? templateVersion String?
@@ -125,15 +125,15 @@ model McpTemplate {
model SecretBackend { model SecretBackend {
id String @id @default(cuid()) id String @id @default(cuid())
name String @unique name String @unique
type String // plaintext | openbao | (future: vault, aws-sm, ...) type String // plaintext | openbao | (future: vault, aws-sm, ...)
config Json @default("{}") // type-specific: url, mount, namespace, tokenSecretRef config Json @default("{}") // type-specific: url, mount, namespace, tokenSecretRef
// Runtime metadata for auto-rotating backend credentials (openbao token // Runtime metadata for auto-rotating backend credentials (openbao token
// auth). Fields: generatedAt, nextRenewalAt, validUntil, lastRotationAt, // auth). Fields: generatedAt, nextRenewalAt, validUntil, lastRotationAt,
// lastRotationError, rotatable (true only for wizard-provisioned tokens). // lastRotationError, rotatable (true only for wizard-provisioned tokens).
// Empty object for backends that don't use rotation (plaintext, kubernetes // Empty object for backends that don't use rotation (plaintext, kubernetes
// auth, or static tokens). Managed entirely by the rotator service. // auth, or static tokens). Managed entirely by the rotator service.
tokenMeta Json @default("{}") tokenMeta Json @default("{}")
isDefault Boolean @default(false) // exactly one row has isDefault=true isDefault Boolean @default(false) // exactly one row has isDefault=true
description String @default("") description String @default("")
version Int @default(1) version Int @default(1)
createdAt DateTime @default(now()) createdAt DateTime @default(now())
@@ -156,8 +156,8 @@ model Secret {
// on next mcpd startup. New rows written by SecretService always carry a // on next mcpd startup. New rows written by SecretService always carry a
// valid FK immediately. // valid FK immediately.
backendId String @default("") backendId String @default("")
data Json @default("{}") // populated by plaintext backend only data Json @default("{}") // populated by plaintext backend only
externalRef String @default("") // populated by non-plaintext backends (e.g. "mount/path#v3") externalRef String @default("") // populated by non-plaintext backends (e.g. "mount/path#v3")
// Sorted list of the secret's data keys WITHOUT their values. Populated on // Sorted list of the secret's data keys WITHOUT their values. Populated on
// every create/update/migrate so list views and describe-without-reveal can // every create/update/migrate so list views and describe-without-reveal can
// show "this secret has GRAFANA_URL + GRAFANA_TOKEN" without fetching the // show "this secret has GRAFANA_URL + GRAFANA_TOKEN" without fetching the
@@ -186,14 +186,14 @@ model Secret {
model Llm { model Llm {
id String @id @default(cuid()) id String @id @default(cuid())
name String @unique name String @unique
type String // anthropic | openai | deepseek | vllm | ollama | gemini-cli type String // anthropic | openai | deepseek | vllm | ollama | gemini-cli
model String // e.g. claude-3-5-sonnet-20241022 model String // e.g. claude-3-5-sonnet-20241022
url String @default("") // endpoint (empty for provider default) url String @default("") // endpoint (empty for provider default)
tier String @default("fast") // fast | heavy tier String @default("fast") // fast | heavy
description String @default("") description String @default("")
apiKeySecretId String? // FK to Secret apiKeySecretId String? // FK to Secret
apiKeySecretKey String? // key inside the Secret's data apiKeySecretKey String? // key inside the Secret's data
extraConfig Json @default("{}") // per-type extras extraConfig Json @default("{}") // per-type extras
version Int @default(1) version Int @default(1)
createdAt DateTime @default(now()) createdAt DateTime @default(now())
updatedAt DateTime @updatedAt updatedAt DateTime @updatedAt
@@ -252,19 +252,19 @@ model RbacDefinition {
// ── Projects ── // ── Projects ──
model Project { model Project {
id String @id @default(cuid()) id String @id @default(cuid())
name String @unique name String @unique
description String @default("") description String @default("")
prompt String @default("") prompt String @default("")
proxyModel String @default("") proxyModel String @default("")
gated Boolean @default(true) gated Boolean @default(true)
llmProvider String? llmProvider String?
llmModel String? llmModel String?
serverOverrides Json? serverOverrides Json?
ownerId String ownerId String
version Int @default(1) version Int @default(1)
createdAt DateTime @default(now()) createdAt DateTime @default(now())
updatedAt DateTime @updatedAt updatedAt DateTime @updatedAt
owner User @relation(fields: [ownerId], references: [id], onDelete: Cascade) owner User @relation(fields: [ownerId], references: [id], onDelete: Cascade)
servers ProjectServer[] servers ProjectServer[]
@@ -322,11 +322,11 @@ model McpToken {
// ── MCP Instances (running containers) ── // ── MCP Instances (running containers) ──
model McpInstance { model McpInstance {
id String @id @default(cuid()) id String @id @default(cuid())
serverId String serverId String
containerId String? containerId String?
status InstanceStatus @default(STOPPED) status InstanceStatus @default(STOPPED)
port Int? port Int?
metadata Json @default("{}") metadata Json @default("{}")
healthStatus String? healthStatus String?
lastHealthCheck DateTime? lastHealthCheck DateTime?
@@ -356,6 +356,7 @@ model Prompt {
name String name String
content String @db.Text content String @db.Text
projectId String? projectId String?
agentId String?
priority Int @default(5) priority Int @default(5)
summary String? @db.Text summary String? @db.Text
chapters Json? chapters Json?
@@ -364,10 +365,14 @@ model Prompt {
createdAt DateTime @default(now()) createdAt DateTime @default(now())
updatedAt DateTime @updatedAt updatedAt DateTime @updatedAt
project Project? @relation(fields: [projectId], references: [id], onDelete: Cascade) project Project? @relation(fields: [projectId], references: [id], onDelete: Cascade)
agent Agent? @relation(fields: [agentId], references: [id], onDelete: Cascade)
personalities PersonalityPrompt[]
@@unique([name, projectId]) @@unique([name, projectId])
@@unique([name, agentId])
@@index([projectId]) @@index([projectId])
@@index([agentId])
} }
// ── Prompt Requests (pending proposals from LLM sessions) ── // ── Prompt Requests (pending proposals from LLM sessions) ──
@@ -392,21 +397,21 @@ model PromptRequest {
// ── Audit Events (pipeline/gate/tool trace from mcplocal) ── // ── Audit Events (pipeline/gate/tool trace from mcplocal) ──
model AuditEvent { model AuditEvent {
id String @id @default(cuid()) id String @id @default(cuid())
timestamp DateTime timestamp DateTime
sessionId String sessionId String
projectName String projectName String
eventKind String eventKind String
source String source String
verified Boolean @default(false) verified Boolean @default(false)
serverName String? serverName String?
correlationId String? correlationId String?
parentEventId String? parentEventId String?
userName String? userName String?
tokenName String? tokenName String?
tokenSha String? tokenSha String?
payload Json payload Json
createdAt DateTime @default(now()) createdAt DateTime @default(now())
@@index([sessionId]) @@index([sessionId])
@@index([projectName]) @@index([projectName])
@@ -423,7 +428,7 @@ model BackupPending {
id String @id @default(cuid()) id String @id @default(cuid())
resourceKind String resourceKind String
resourceName String resourceName String
action String // 'create' | 'update' | 'delete' action String // 'create' | 'update' | 'delete'
userName String userName String
yamlContent String? @db.Text yamlContent String? @db.Text
createdAt DateTime @default(now()) createdAt DateTime @default(now())
@@ -439,29 +444,80 @@ model BackupPending {
// Per-call LiteLLM-style overrides stack on top of `defaultParams`. // Per-call LiteLLM-style overrides stack on top of `defaultParams`.
model Agent { model Agent {
id String @id @default(cuid()) id String @id @default(cuid())
name String @unique name String @unique
description String @default("") // shown in MCP tools/list description String @default("") // shown in MCP tools/list
systemPrompt String @default("") @db.Text // agent persona systemPrompt String @default("") @db.Text // agent persona
llmId String llmId String
projectId String? projectId String?
proxyModelName String? // optional informational override defaultPersonalityId String? // applied at chat time when no --personality flag
defaultParams Json @default("{}") // LiteLLM-style: temperature, top_p, top_k, max_tokens, stop, ... proxyModelName String? // optional informational override
extras Json @default("{}") // future LoRA / tool-allowlist defaultParams Json @default("{}") // LiteLLM-style: temperature, top_p, top_k, max_tokens, stop, ...
ownerId String extras Json @default("{}") // future LoRA / tool-allowlist
version Int @default(1) ownerId String
createdAt DateTime @default(now()) version Int @default(1)
updatedAt DateTime @updatedAt createdAt DateTime @default(now())
updatedAt DateTime @updatedAt
llm Llm @relation(fields: [llmId], references: [id], onDelete: Restrict) llm Llm @relation(fields: [llmId], references: [id], onDelete: Restrict)
project Project? @relation(fields: [projectId], references: [id], onDelete: SetNull) project Project? @relation(fields: [projectId], references: [id], onDelete: SetNull)
owner User @relation(fields: [ownerId], references: [id], onDelete: Cascade) owner User @relation(fields: [ownerId], references: [id], onDelete: Cascade)
threads ChatThread[] threads ChatThread[]
prompts Prompt[]
personalities Personality[] @relation("AgentPersonalities")
defaultPersonality Personality? @relation("AgentDefaultPersonality", fields: [defaultPersonalityId], references: [id], onDelete: SetNull)
@@index([name]) @@index([name])
@@index([llmId]) @@index([llmId])
@@index([projectId]) @@index([projectId])
@@index([ownerId]) @@index([ownerId])
@@index([defaultPersonalityId])
}
// ── Personalities (named overlay bundles of prompts on top of an Agent) ──
//
// VLAN-on-ethernet semantics: an Agent works without a Personality (today's
// flow). Selecting a Personality adds its bound Prompts to the system block
// after the Agent's own systemPrompt, agent-direct prompts, and project
// prompts — additive, never replacing.
model Personality {
id String @id @default(cuid())
name String
description String @default("")
agentId String
priority Int @default(5)
createdAt DateTime @default(now())
updatedAt DateTime @updatedAt
agent Agent @relation("AgentPersonalities", fields: [agentId], references: [id], onDelete: Cascade)
prompts PersonalityPrompt[]
defaultForAgent Agent[] @relation("AgentDefaultPersonality")
@@unique([name, agentId])
@@index([agentId])
}
// ── Personality ↔ Prompt join table ──
//
// A Prompt can be bound to many Personalities; a Personality can bind many
// Prompts. The `priority` here overrides the Prompt's own priority *within
// this Personality's overlay slice* (so the same prompt can be high-priority
// in one personality and low in another).
model PersonalityPrompt {
id String @id @default(cuid())
personalityId String
promptId String
priority Int @default(5)
createdAt DateTime @default(now())
personality Personality @relation(fields: [personalityId], references: [id], onDelete: Cascade)
prompt Prompt @relation(fields: [promptId], references: [id], onDelete: Cascade)
@@unique([personalityId, promptId])
@@index([personalityId])
@@index([promptId])
} }
// ── Chat Threads (persisted conversation per Agent) ── // ── Chat Threads (persisted conversation per Agent) ──
@@ -493,11 +549,11 @@ model ChatMessage {
id String @id @default(cuid()) id String @id @default(cuid())
threadId String threadId String
turnIndex Int turnIndex Int
role String // 'system' | 'user' | 'assistant' | 'tool' role String // 'system' | 'user' | 'assistant' | 'tool'
content String @db.Text content String @db.Text
toolCalls Json? // assistant turn: [{id,name,arguments}] toolCalls Json? // assistant turn: [{id,name,arguments}]
toolCallId String? // tool turn: which call this answers toolCallId String? // tool turn: which call this answers
status String @default("complete") // 'pending' | 'complete' | 'error' status String @default("complete") // 'pending' | 'complete' | 'error'
createdAt DateTime @default(now()) createdAt DateTime @default(now())
thread ChatThread @relation(fields: [threadId], references: [id], onDelete: Cascade) thread ChatThread @relation(fields: [threadId], references: [id], onDelete: Cascade)
@@ -509,13 +565,13 @@ model ChatMessage {
// ── Audit Logs ── // ── Audit Logs ──
model AuditLog { model AuditLog {
id String @id @default(cuid()) id String @id @default(cuid())
userId String userId String
action String action String
resource String resource String
resourceId String? resourceId String?
details Json @default("{}") details Json @default("{}")
createdAt DateTime @default(now()) createdAt DateTime @default(now())
user User @relation(fields: [userId], references: [id], onDelete: Cascade) user User @relation(fields: [userId], references: [id], onDelete: Cascade)

142
src/db/tests/agent-schema.test.ts
View File

@@ -201,4 +201,146 @@ describe('agent / chat-thread / chat-message schema', () => {
}); });
expect(ordered.map((t) => t.id)).toEqual([t2.id, t1.id]); expect(ordered.map((t) => t.id)).toEqual([t2.id, t1.id]);
}); });
// ── Agent-direct prompts (Prompt.agentId) ──
it('attaches a prompt directly to an agent (no project)', async () => {
const user = await makeUser();
const llm = await makeLlm('llm-agent-prompt');
const agent = await makeAgent({ name: 'directprompt', llmId: llm.id, ownerId: user.id });
const prompt = await prisma.prompt.create({
data: { name: 'agent-only', content: 'hi from agent', agentId: agent.id },
});
expect(prompt.agentId).toBe(agent.id);
expect(prompt.projectId).toBeNull();
});
it('cascade-deletes agent-direct prompts when the agent is removed', async () => {
const user = await makeUser();
const llm = await makeLlm('llm-agent-prompt-cascade');
const agent = await makeAgent({ name: 'goingaway', llmId: llm.id, ownerId: user.id });
await prisma.prompt.create({
data: { name: 'p1', content: 'c', agentId: agent.id },
});
await prisma.agent.delete({ where: { id: agent.id } });
expect(await prisma.prompt.count({ where: { agentId: agent.id } })).toBe(0);
});
it('enforces (name, agentId) uniqueness independently of (name, projectId)', async () => {
const user = await makeUser();
const llm = await makeLlm('llm-uniq-prompt-agent');
const agent = await makeAgent({ name: 'has-prompts', llmId: llm.id, ownerId: user.id });
await prisma.prompt.create({
data: { name: 'shared', content: 'a', agentId: agent.id },
});
await expect(
prisma.prompt.create({
data: { name: 'shared', content: 'b', agentId: agent.id },
}),
).rejects.toThrow();
// Same name on a different agent is fine.
const agent2 = await makeAgent({ name: 'other-agent', llmId: llm.id, ownerId: user.id });
const ok = await prisma.prompt.create({
data: { name: 'shared', content: 'c', agentId: agent2.id },
});
expect(ok.id).toBeDefined();
});
// ── Personalities ──
it('creates a personality bound to an agent', async () => {
const user = await makeUser();
const llm = await makeLlm('llm-pers-1');
const agent = await makeAgent({ name: 'with-pers', llmId: llm.id, ownerId: user.id });
const p = await prisma.personality.create({
data: { name: 'grumpy', description: 'curmudgeonly', agentId: agent.id },
});
expect(p.id).toBeDefined();
expect(p.priority).toBe(5);
});
it('enforces (name, agentId) uniqueness on personalities', async () => {
const user = await makeUser();
const llm = await makeLlm('llm-pers-uniq');
const agent = await makeAgent({ name: 'pers-uniq', llmId: llm.id, ownerId: user.id });
await prisma.personality.create({ data: { name: 'mode', agentId: agent.id } });
await expect(
prisma.personality.create({ data: { name: 'mode', agentId: agent.id } }),
).rejects.toThrow();
});
it('cascade-deletes personalities + bindings when the agent is removed', async () => {
const user = await makeUser();
const llm = await makeLlm('llm-pers-cascade');
const agent = await makeAgent({ name: 'kaboom', llmId: llm.id, ownerId: user.id });
const personality = await prisma.personality.create({
data: { name: 'snarky', agentId: agent.id },
});
const prompt = await prisma.prompt.create({
data: { name: 'tone', content: 'be snarky', agentId: agent.id },
});
await prisma.personalityPrompt.create({
data: { personalityId: personality.id, promptId: prompt.id },
});
await prisma.agent.delete({ where: { id: agent.id } });
expect(await prisma.personality.count({ where: { id: personality.id } })).toBe(0);
expect(await prisma.personalityPrompt.count({ where: { personalityId: personality.id } })).toBe(0);
// Prompts directly attached to the agent also go via Prompt.agentId cascade.
expect(await prisma.prompt.count({ where: { id: prompt.id } })).toBe(0);
});
it('SetNull on Agent.defaultPersonalityId when the personality is deleted', async () => {
const user = await makeUser();
const llm = await makeLlm('llm-default-pers');
const agent = await makeAgent({ name: 'has-default', llmId: llm.id, ownerId: user.id });
const personality = await prisma.personality.create({
data: { name: 'def', agentId: agent.id },
});
await prisma.agent.update({
where: { id: agent.id },
data: { defaultPersonalityId: personality.id },
});
await prisma.personality.delete({ where: { id: personality.id } });
const reloaded = await prisma.agent.findUnique({ where: { id: agent.id } });
expect(reloaded?.defaultPersonalityId).toBeNull();
});
it('binds the same prompt to multiple personalities of an agent', async () => {
const user = await makeUser();
const llm = await makeLlm('llm-shared-prompt');
const agent = await makeAgent({ name: 'shared', llmId: llm.id, ownerId: user.id });
const p1 = await prisma.personality.create({ data: { name: 'a', agentId: agent.id } });
const p2 = await prisma.personality.create({ data: { name: 'b', agentId: agent.id } });
const prompt = await prisma.prompt.create({
data: { name: 'shared-text', content: 'reusable', agentId: agent.id },
});
await prisma.personalityPrompt.create({
data: { personalityId: p1.id, promptId: prompt.id, priority: 9 },
});
await prisma.personalityPrompt.create({
data: { personalityId: p2.id, promptId: prompt.id, priority: 1 },
});
expect(await prisma.personalityPrompt.count({ where: { promptId: prompt.id } })).toBe(2);
// Same (personalityId, promptId) cannot collide.
await expect(
prisma.personalityPrompt.create({
data: { personalityId: p1.id, promptId: prompt.id },
}),
).rejects.toThrow();
});
}); });

4
src/db/tests/helpers.ts
View File

@@ -32,6 +32,10 @@ export async function clearAllTables(client: PrismaClient): Promise<void> {
await client.auditLog.deleteMany(); await client.auditLog.deleteMany();
await client.chatMessage.deleteMany(); await client.chatMessage.deleteMany();
await client.chatThread.deleteMany(); await client.chatThread.deleteMany();
await client.personalityPrompt.deleteMany();
// Break Agent.defaultPersonalityId before personalities can be removed.
await client.agent.updateMany({ data: { defaultPersonalityId: null } });
await client.personality.deleteMany();
await client.agent.deleteMany(); await client.agent.deleteMany();
await client.llm.deleteMany(); await client.llm.deleteMany();
await client.mcpInstance.deleteMany(); await client.mcpInstance.deleteMany();