Function stubCommonStreamMessageDependencies

src/node/services/aiService.test.ts:286–402

(args: {
  service: AIService;
  config: Config;
  historyService: HistoryService;
  initStateManager: InitStateManager;
  metadata: WorkspaceMetadata;
  startStreamCalls?: unknown[][];
  routeProvider?: ProviderName;
  allTools?: Record<string, Tool>;
  workspacePathOverride?: string;
  historySequence?: number;
  effectiveModelString?: string;
  canonicalProviderName?: ProviderName;
  canonicalModelId?: string;
  useRequestedModelString?: boolean;
  onPlanPayloadMessageIds?: (messageIds: string[]) => void;
  onBuildStreamSystemContext?: (
    args: Parameters<typeof streamContextBuilder.buildStreamSystemContext>[0]
  ) => void;
  onPrepareMessagesForProvider?: (
    args: Parameters<typeof messagePipeline.prepareMessagesForProvider>[0]
  ) => void;
})

}

function stubCommonStreamMessageDependencies(args: {
  service: AIService;
  config: Config;
  historyService: HistoryService;
  initStateManager: InitStateManager;
  metadata: WorkspaceMetadata;
  startStreamCalls?: unknown[][];
  routeProvider?: ProviderName;
  allTools?: Record<string, Tool>;
  workspacePathOverride?: string;
  historySequence?: number;
  effectiveModelString?: string;
  canonicalProviderName?: ProviderName;
  canonicalModelId?: string;
  useRequestedModelString?: boolean;
  onPlanPayloadMessageIds?: (messageIds: string[]) => void;
  onBuildStreamSystemContext?: (
    args: Parameters<typeof streamContextBuilder.buildStreamSystemContext>[0]
  ) => void;
  onPrepareMessagesForProvider?: (
    args: Parameters<typeof messagePipeline.prepareMessagesForProvider>[0]
  ) => void;
}): ReturnType<typeof spyOn<typeof toolsModule, "getToolsForModel">> {
  spyOn(agentResolution, "resolveAgentForStream").mockResolvedValue(
    resolvedAgentResultFor(args.metadata)
  );
  spyOn(streamContextBuilder, "buildPlanInstructions").mockImplementation((planArgs) => {
    args.onPlanPayloadMessageIds?.(planArgs.requestPayloadMessages.map((message) => message.id));
    return Promise.resolve({
      effectiveAdditionalInstructions: undefined,
      planFilePath: path.join(args.metadata.projectPath, "plan.md"),
      planContentForTransition: undefined,
    });
  });
  spyOn(streamContextBuilder, "buildStreamSystemContext").mockImplementation((contextArgs) => {
    args.onBuildStreamSystemContext?.(contextArgs);
    return Promise.resolve({
      agentSystemPromptSections: ["test-agent-prompt"],
      systemMessage: "test-system-message",
      systemMessageTokens: 1,
      agentDefinitions: undefined,
      availableSkills: undefined,
      ancestorPlanFilePaths: [],
    });
  });
  spyOn(messagePipeline, "prepareMessagesForProvider").mockImplementation((pipelineArgs) => {
    args.onPrepareMessagesForProvider?.(pipelineArgs);
    return Promise.resolve(
      pipelineArgs.messagesWithSentinel as unknown as Awaited<
        ReturnType<typeof messagePipeline.prepareMessagesForProvider>
      >
    );
  });
  const getToolsForModelSpy = spyOn(toolsModule, "getToolsForModel").mockResolvedValue(
    args.allTools ?? {}
  );
  spyOn(systemMessageModule, "readToolInstructions").mockResolvedValue({});