@lincoln504/pi-research 0.1.12 → 1.0.0

package/src/orchestration/deep-research-reducer.ts

@@ -1,129 +0,0 @@
-/**
- * Deep Research Reducer
- *
- * PURE FUNCTION logic for state transitions.
- * Contains 0 side effects. 100% testable without mocks.
- */
-
-import type { SystemResearchState, DeepResearchEvent } from './deep-research-types.ts';
-
-export function deepResearchReducer(state: SystemResearchState, event: DeepResearchEvent): SystemResearchState {
-  // Deep clone state to ensure true purity
-  const newState: SystemResearchState = {
-    ...state,
-    lastUpdated: Date.now(),
-    initialAgenda: [...state.initialAgenda],
-    allScrapedLinks: [...state.allScrapedLinks],
-    aspects: { ...state.aspects }
-  };
-
-  switch (event.type) {
-    case 'PLANNING_COMPLETE': {
-      newState.initialAgenda = event.agenda;
-      newState.status = 'researching';
-      // Assign first batch to aspects
-      event.agenda.slice(0, event.initialCount).forEach((q: string, i: number) => {
-        const id = `1.${i + 1}`;
-        newState.aspects[id] = { id, query: q, status: 'pending' };
-      });
-      break;
-    }
-
-    case 'SIBLING_STARTED': {
-      const aspect = newState.aspects[event.id];
-      if (aspect) {
-        newState.aspects[event.id] = {
-          ...aspect,
-          status: 'running'
-        };
-      }
-      break;
-    }
-
-    case 'SIBLING_COMPLETED': {
-      const aspect = newState.aspects[event.id];
-      if (aspect) {
-        newState.aspects[event.id] = {
-          ...aspect,
-          status: 'completed',
-          report: event.report
-        };
-      }
-      break;
-    }
-
-    case 'SIBLING_FAILED': {
-      const aspect = newState.aspects[event.id];
-      if (aspect) {
-        newState.aspects[event.id] = {
-          ...aspect,
-          status: 'failed',
-          error: event.error
-        };
-      }
-      break;
-    }
-
-    case 'SIBLING_TOKENS': {
-      const aspect = newState.aspects[event.id];
-      if (aspect) {
-        // Accumulate tokens and cost for this sibling
-        const currentTokens = aspect.tokens || 0;
-        const currentCost = aspect.cost || 0;
-        newState.aspects[event.id] = {
-          ...aspect,
-          tokens: currentTokens + event.tokens,
-          cost: currentCost + event.cost
-        };
-      }
-      break;
-    }
-
-    case 'LINKS_SCRAPED': {
-      newState.allScrapedLinks = [...new Set([...newState.allScrapedLinks, ...event.links])];
-      break;
-    }
-
-    case 'PROMOTION_STARTED': {
-      newState.promotedId = event.id;
-      break;
-    }
-
-    case 'PROMOTION_DECISION': {
-      if (event.finalSynthesis !== undefined) {
-        // Case 1: Lead Evaluator chose to synthesize - complete with synthesis
-        newState.finalSynthesis = event.finalSynthesis;
-        newState.status = 'completed';
-        newState.promotedId = undefined;
-      } else if (event.nextQueries.length > 0 && newState.currentRound < event.maxRounds) {
-        // Case 2: Lead Evaluator chose to continue AND we're under max rounds
-        newState.currentRound++;
-        newState.status = 'researching';
-        newState.promotedId = undefined; // Reset for new round
-        event.nextQueries.forEach((q: string, i: number) => {
-          const id = `${newState.currentRound}.${i + 1}`;
-          newState.aspects[id] = { id, query: q, status: 'pending' };
-        });
-      } else {
-        // Case 3: Max rounds reached OR no next queries
-        // Note: Per lead-evaluator prompt, when max rounds are reached, the evaluator
-        // should synthesize. If nextQueries are provided at max rounds, this is a
-        // protocol violation, and we enforce the constraint by completing anyway.
-        newState.status = 'completed';
-        newState.promotedId = undefined;
-      }
-      break;
-    }
-  }
-
-  return newState;
-}
-
-/**
- * Utility to check if a round is finished
- */
-export function isRoundComplete(state: SystemResearchState, round: number): boolean {
-  const roundAspects = Object.values(state.aspects).filter(a => a.id.startsWith(`${round}.`));
-  if (roundAspects.length === 0) return false;
-  return roundAspects.every(a => a.status === 'completed' || a.status === 'failed');
-}

package/src/orchestration/id-utils.ts

@@ -1,63 +0,0 @@
-/**
- * Researcher ID Utilities
- *
- * Abstracts hierarchical IDs (1.1, 1.2, 2.1) to sequential display numbers (1, 2, 3)
- * for cleaner UI while maintaining internal structure.
- */
-
-import type { SystemResearchState } from './deep-research-types.ts';
-
-/**
- * Get sequential display number for a researcher ID
- *
- * Maps internal hierarchical IDs to sequential display numbers:
- * - Round 1: 1.1 → 1, 1.2 → 2, 1.3 → 3
- * - Round 2: 2.1 → 4, 2.2 → 5, 2.3 → 6
- * - Continue sequentially across all rounds
- */
-export function getDisplayNumber(state: SystemResearchState, internalId: string): string {
-  const aspects = Object.values(state.aspects).sort((a, b) => {
-    const aParts = a.id.split('.');
-    const bParts = b.id.split('.');
-    const aRound = parseInt(aParts[0] ?? '0');
-    const aNum = parseInt(aParts[1] ?? '0');
-    const bRound = parseInt(bParts[0] ?? '0');
-    const bNum = parseInt(bParts[1] ?? '0');
-
-    if (aRound !== bRound) return aRound - bRound;
-    return aNum - bNum;
-  });
-
-  const index = aspects.findIndex(a => a.id === internalId);
-  return index >= 0 ? (index + 1).toString() : internalId;
-}
-
-/**
- * Get a researcher's role context
- *
- * Returns role information for a researcher:
- * - Number in this round (1, 2, 3)
- * - Total siblings in this round
- * - Current round
- */
-export function getResearcherRoleContext(
-  state: SystemResearchState,
-  internalId: string
-): {
-  displayNumber: string;
-  roundNumber: number;
-  totalInRound: number;
-  isLastInRound: boolean;
-} {
-  const parts = internalId.split('.');
-  const round = parseInt(parts[0] ?? '0');
-  const sibling = parseInt(parts[1] ?? '0');
-  const roundAspects = Object.values(state.aspects).filter(a => a.id.startsWith(`${round}.`));
-
-  return {
-    displayNumber: getDisplayNumber(state, internalId),
-    roundNumber: round,
-    totalInRound: roundAspects.length,
-    isLastInRound: sibling === roundAspects.length,
-  };
-}