Method retrieveChunksFromQuery

core/indexing/docs/DocsService.ts:742–763

(
    query: string,
    startUrl: string,
    nRetrieve: number,
  )

  // Retrieve docs embeds based on user input
  async retrieveChunksFromQuery(
    query: string,
    startUrl: string,
    nRetrieve: number,
  ) {
    const { provider } = await this.getEmbeddingsProvider();

    if (!provider) {
      void this.ide.showToast(
        "error",
        "Set up an embeddings model to use the @docs context provider. See: " +
          "https://docs.continue.dev/customize/model-roles/embeddings",
      );
      return [];
    }

    // Try to get embeddings for the query
    const [vector] = await provider.embed([query]);

    // Retrieve chunks using the query vector
    return await this.retrieveChunks(startUrl, vector, nRetrieve);
  }

  private lanceDBRowToChunk(row: LanceDbDocsRow): Chunk {
    return {