Function createHarness

src/node/services/aiService.test.ts:1206–1286

(
    muxHomePath: string,
    metadata: WorkspaceMetadata,
    options?: {
      routeProvider?: ProviderName;
      allTools?: Record<string, Tool>;
      postPolicyTools?: Record<string, Tool>;
      sessionUsageService?: SessionUsageService;
      effectiveModelString?: string;
      canonicalProviderName?: ProviderName;
      canonicalModelId?: string;
      useRequestedModelString?: boolean;
      experimentsService?: ExperimentsService;
    }
  )

  }

  function createHarness(
    muxHomePath: string,
    metadata: WorkspaceMetadata,
    options?: {
      routeProvider?: ProviderName;
      allTools?: Record<string, Tool>;
      postPolicyTools?: Record<string, Tool>;
      sessionUsageService?: SessionUsageService;
      effectiveModelString?: string;
      canonicalProviderName?: ProviderName;
      canonicalModelId?: string;
      useRequestedModelString?: boolean;
      experimentsService?: ExperimentsService;
    }
  ): StreamMessageHarness {
    const { config, historyService, initStateManager, service } = createBasicAIService(
      muxHomePath,
      {
        sessionUsageService: options?.sessionUsageService,
        experimentsService: options?.experimentsService,
      }
    );
    const planPayloadMessageIds: string[][] = [];
    const preparedPayloadMessageIds: string[][] = [];
    const preparedToolNamesForSentinel: string[][] = [];
    const streamSystemContextMuxScopes: MuxToolScope[] = [];
    const streamSystemContextAdvisorFlags: Array<boolean | undefined> = [];
    const streamSystemContextMemoryToolFlags: Array<boolean | undefined> = [];
    const streamSystemContextHotMemoriesBlocks: Array<string | undefined> = [];
    const startStreamCalls: unknown[][] = [];

    const getToolsForModelSpy = stubCommonStreamMessageDependencies({
      service,
      config,
      historyService,
      initStateManager,
      metadata,
      startStreamCalls,
      routeProvider: options?.routeProvider,
      allTools: options?.allTools,
      effectiveModelString: options?.effectiveModelString,
      canonicalProviderName: options?.canonicalProviderName,
      canonicalModelId: options?.canonicalModelId,
      useRequestedModelString: options?.useRequestedModelString,
      onPlanPayloadMessageIds: (messageIds) => planPayloadMessageIds.push(messageIds),
      onBuildStreamSystemContext: (contextArgs) => {
        if (!contextArgs.muxScope) {
          throw new Error("Expected muxScope in stream system context build args");
        }
        streamSystemContextMuxScopes.push(contextArgs.muxScope);
        streamSystemContextAdvisorFlags.push(contextArgs.advisorToolAvailable);
        streamSystemContextMemoryToolFlags.push(contextArgs.memoryToolAvailable);
        streamSystemContextHotMemoriesBlocks.push(contextArgs.hotMemoriesBlock);
      },
      onPrepareMessagesForProvider: (pipelineArgs) => {
        preparedPayloadMessageIds.push(
          pipelineArgs.messagesWithSentinel.map((message) => message.id)
        );