Function requestFilesFromRepoMap

core/context/retrieval/repoMapRequest.ts:34–108

(
  defaultLlm: ILLM,
  config: ContinueConfig,
  ide: IDE,
  input: string,
  filterDirUri?: string,
)

}

export async function requestFilesFromRepoMap(
  defaultLlm: ILLM,
  config: ContinueConfig,
  ide: IDE,
  input: string,
  filterDirUri?: string,
): Promise<Chunk[]> {
  const llm = getModelByRole(config, "repoMapFileSelection") ?? defaultLlm;

  // Only supported for Claude models right now
  if (!isSupportedModel(config, llm.title)) {
    return [];
  }

  try {
    const repoMap = await generateRepoMap(llm, ide, {
      dirUris: filterDirUri ? [filterDirUri] : undefined,
      includeSignatures: false,
      outputRelativeUriPaths: true,
    });

    const prompt = `${repoMap}

Given the above repo map, your task is to decide which files are most likely to be relevant in answering a question. Before giving your answer, you should write your reasoning about which files/folders are most important. This thinking should start with a <reasoning> tag, followed by a paragraph explaining your reasoning, and then a closing </reasoning> tag on the last line.

After this, your response should begin with a <results> tag, followed by a list of each file, one per line, and then a closing </results> tag on the last line. You should select between 5 and 10 files. The names that you list should be the exact relative path that you saw in the repo map, not just the basename of the file.

This is the question that you should select relevant files for: "${input}"`;

    const response = await llm.chat(
      [
        { role: "user", content: prompt },
        { role: "assistant", content: "<reasoning>" },
      ],
      new AbortController().signal,
    );
    const content = renderChatMessage(response);

    if (!content.includes("\n")) {
      return [];
    }

    const filepaths = content
      .split("<results>")[1]
      ?.split("</results>")[0]
      ?.split("\n")
      .filter(Boolean)
      .map((filepath) => filepath.trim());

    const chunks = await Promise.all(
      filepaths.map(async (filepath) => {
        const uri = await resolveRelativePathInDir(filepath, ide);
        if (!uri) {
          return undefined;
        }
        const content = await ide.readFile(uri);
        const lineCount = content.split("\n").length;
        const chunk: Chunk = {