Method _embed

core/llm/llms/NCompass.ts:43–63

(chunks: string[])

  }

  protected async _embed(chunks: string[]): Promise<number[][]> {
    const resp = await this.fetch(NCompass.embeddingsApiEndpoint, {
      method: "POST",
      body: JSON.stringify({
        input: chunks,
        model: this.model,
        ...this.extraBodyProperties(),
      }),
      headers: {
        Authorization: `Bearer ${this.apiKey}`,
        "Content-Type": "application/json",
      },
    });

    if (!resp.ok) {
      throw new Error(await resp.text());
    }

    const data = (await resp.json()) as any;
    return data.data.map((result: { embedding: number[] }) => result.embedding);
  }
}

export default NCompass;