Files
AgentCompany/tests/unit/prompt-manager-classification.test.ts
T
ZacharyZhang-NY e94939a562 test(connector-id): migrate fixtures and assertions to pi-mono ids
Replaces legacy adapter ids (codex_local / claude_local / gemini_local /
claude_compatible) across ~42 test files with pi-mono equivalents:
- codex_local → openai-codex
- claude_local → claude-pro
- gemini_local → gemini-cli
- claude_compatible → anthropic

Hand-rewrites where fixture semantics changed:
- connector-matching: TASK_TYPE_PREFERENCE ordering rebased (claude-pro
  first for code/content/management; google first for research)
- connector-model-policy: getModels(provider) from pi-ai instead of the
  deleted adapter.module.models array
- session-management + edge-case-session: adapter exemption removed;
  rotation now uniform under pi-mono
- ipc-core-admin-handlers: CSV compat model replaced with pi-ai anthropic
- connectors.spec: lookup by openai-codex id (catalog now starts with claude-pro)
- onboarding-model-pool: listProviderModels(provider) helper; label text
  changed to "ChatGPT Plus/Pro (Codex)"
- database integration: added vault insecure-fallback env + electron mock
- command-paths: uses new @main/connector-helpers normalizeCommandValue
- pi-mono-migrations: added `as string` casts for 3 negative-existence
  assertions verifying legacy ids were removed

Drops:
- pnpm-workspace.yaml packages: [] (was "packages/*"; directory empty)
- obsolete CONNECTOR_STRENGTHS import; .module.models probing

Verification:
- pnpm typecheck → 0 errors
- pnpm lint → 0 errors / 0 warnings
- pnpm build:verify → 0 errors / 0 warnings
- pnpm test:unit → 508/508 pass
- pnpm test:integration → 96/96 pass
- All 6 adapter hard gates pass (NOT FOUND / 0 matches / empty)
- pi-ai / pi-agent-core / pi-coding-agent all resolved @0.67.6
2026-04-17 02:36:53 -04:00

83 lines
2.4 KiB
TypeScript

import { describe, expect, it } from "vitest";
import { buildSystemPrompt } from "../../src/main/prompt-builder";
import type { PromptContext } from "../../src/main/prompt-builder/index";
function buildCtx(overrides: Partial<PromptContext> = {}): PromptContext {
return {
agent: {
id: "agent-1",
companyId: "company-1",
name: "CTO",
role: "CTO",
title: "Chief Technology Officer",
roleLevel: "c_suite",
department: "engineering",
status: "active",
reportsTo: "ceo-1",
connectorId: "openai-codex",
workspaceId: null,
model: null,
capabilities: "Lead engineering",
budgetMonthlyUsd: 200,
spentMonthlyUsd: 0,
metadataJson: "{}",
heartbeatEnabled: true,
heartbeatIntervalSec: 120,
lastHeartbeatAt: null,
createdAt: new Date().toISOString(),
updatedAt: new Date().toISOString(),
},
apiUrl: "http://127.0.0.1:3210",
apiKey: "token",
runId: "run-1",
task: null,
companyName: "Prompt Co",
companyDescription: "Tests leadership prompt classification",
goals: [],
projects: [],
directReports: [],
chainOfCommand: [{
id: "ceo-1",
companyId: "company-1",
name: "CEO",
role: "CEO",
title: "Chief Executive Officer",
roleLevel: "c_suite",
department: "executive",
status: "active",
reportsTo: null,
connectorId: "openai-codex",
workspaceId: null,
model: null,
capabilities: "Lead the company",
budgetMonthlyUsd: 250,
spentMonthlyUsd: 0,
metadataJson: "{}",
heartbeatEnabled: true,
heartbeatIntervalSec: 120,
lastHeartbeatAt: null,
createdAt: new Date().toISOString(),
updatedAt: new Date().toISOString(),
}],
assignedTasks: [],
allAgents: [],
pendingApprovals: [],
tasksAwaitingReview: [],
wakeReason: null,
allTasks: [],
recentComments: [],
socialAccounts: [],
...overrides,
};
}
describe("buildSystemPrompt leadership classification", () => {
it("treats first-layer executive leaders as managers even before they have direct reports", () => {
const prompt = buildSystemPrompt(buildCtx());
expect(prompt).toContain("Operating Instructions (Manager)");
expect(prompt).toContain("New hires for your team should report to you by default");
expect(prompt).not.toContain("Operating Instructions (Individual Contributor)");
});
});