The Worker Definition

export interface WorkerDefinition {
  /** Unique identifier — used in URLs: /portal/staffing/[id] */
  id: string;

  /** Display name shown in the UI */
  name: string;

  /** Job title / role description */
  title: string;

  /** Short description of what this employee does */
  description: string;

  /** Longer description for the detail page */
  longDescription?: string;

  /** Emoji or icon identifier */
  icon: string;

  /** Whether this employee is available or coming soon */
  status: 'active' | 'coming_soon';

  /**
   * Tool names this employee is allowed to use.
   * These are looked up from the global tool registry.
   * If empty, the employee has access to all tools.
   */
  allowedTools: string[];

  /** Loop configuration overrides (merged with global defaults) */
  loopConfig: Partial<LoopConfig>;

  /**
   * Initial living document sections.
   * Keys become the section names tracked across passes.
   */
  sections: Record<string, string>;

  /** Pre-enrichment configuration */
  preEnrichment: WorkerPreEnrichmentConfig;

  /** Build the system prompt for this employee */
  buildSystemPrompt: (context: WorkerContext) => string;

  /** Build the synthesis prompt for final response generation */
  buildSynthesisPrompt?: (context: WorkerContext) => string;

  /** Starter prompts shown on the empty state */
  starterPrompts: string[];

  /** Approval button label (default: "Approve") */
  approvalLabel?: string;

  /** Empty state description shown before first message */
  emptyStateDescription?: string;
}

function buildSystemPrompt(context: WorkerContext): string {
  return `
You are [Employee Name], an AI assistant specialized in [domain].

## Your Role
[2-3 sentences describing the job]

## Data Model
[Explain the data structures the AI will work with]

## Your Tools
${buildToolMenuText(allowedTools)}
// This dynamically formats all allowed tools with params

## Workflow
Pass 1: Confirm the user's criteria and load context
Pass 2-3: Search and gather data using parallel tool calls
Pass 4: Cross-reference and filter results
Pass 5+: Score, rank, and prepare final output

## Output Format
[Exact Markdown format the AI must produce]

## JSON Response Contract
Every response MUST be valid JSON:
{
  "thinking": "Your reasoning here",
  "tool_calls": [{ "tool": "...", "params": {} }],
  "should_respond": false,
  "confidence": "low|medium|high",
  "document_updates": { "section_name": "content" }
}

## Rules
1. Never call the same tool with the same params twice
2. Always confirm criteria before executing write tools
3. [domain-specific rules...]
`;
}

import type { WorkerDefinition } from '../types';
import { buildToolMenuText } from '../prompts';

const ALLOWED_TOOLS = [
  'data.searchRecords',
  'data.getRecordDetail',
  'data.getStats',
  'analysis.crossReference',
  'analysis.batchScore',
  'analysis.buildList',
  'data.updateRecord',  // write — requires approval
  'web.searchCompany',
];

export const recordAnalystWorker: WorkerDefinition = {
  id: 'record-analyst',
  name: 'Record Analyst',
  title: 'AI Data Research Specialist',
  description: 'Searches, cross-references, and scores records to build qualified lists.',
  icon: '🔍',
  status: 'active',

  allowedTools: ALLOWED_TOOLS,

  loopConfig: {
    maxPasses: 8,
    costBudget: 1.00,
    autoApprove: false,
    thinkModel: 'google/gemini-3-flash-preview',
    synthesizeModel: 'anthropic/claude-haiku-4.5',
    escalationModel: 'anthropic/claude-sonnet-4.6',
  },

  sections: {
    criteria: '',
    plan: '',
    work_queue: '',
    findings: '',
    results: '',
    confidence: 'low',
  },

  preEnrichment: {
    enabled: true,
    maxTools: 2,
    defaultTools: [
      { tool: 'data.getStats', params: {} },
    ],
    buildPrompt: (userMessage: string) => `
      The user said: "${userMessage}"
      Available pre-load tools:
      - data.getStats: Get overall database statistics
      - data.searchRecords: Search with filters
      Which tools (up to 2) should run before analysis?
      Return JSON: [{"tool": "...", "params": {}, "reason": "..."}]
    `,
  },

  buildSystemPrompt: ({ organizationId }) => `
You are Record Analyst, an AI data research specialist.

## Your Role
You search, cross-reference, and score records to build qualified
lists for the user. You work methodically across multiple passes,
building up a living document of your findings.

## Tools Available
${buildToolMenuText(ALLOWED_TOOLS)}

## Workflow
Pass 1: Confirm criteria with user, review pre-loaded stats
Pass 2-3: Search records using parallel tool calls
Pass 4: Cross-reference results against existing data
Pass 5-6: Score and rank surviving records
Pass 7+: Prepare final output table

## JSON Response Contract
{
  "thinking": "your reasoning",
  "tool_calls": [{ "tool": "...", "params": {} }],
  "should_respond": false,
  "confidence": "low|medium|high",
  "document_updates": { "section": "content" }
}

## Rules
- Never call the same tool with the same params twice
- Run multiple search tools in parallel when possible
- Confirm write actions before executing
- Organization context: ${organizationId}
  `,

  starterPrompts: [
    'Find 50 qualified records matching [criteria]',
    'Build a list of [category] records in [location]',
    'Cross-reference my existing list against the database',
  ],

  approvalLabel: 'Approve Updates',
  emptyStateDescription: 'Describe what records you need and I will search, score, and build a qualified list.',
};