Embed Chunk Array

ドキュメントをチャンク化した後、テキストチャンクを類似性検索に使用できる数値ベクトルに変換する必要があります。embed メソッドは、選択したプロバイダーとモデルを使用してテキストチャンクを埋め込みに変換します。この例では、テキストチャンクの配列に対して埋め込みを生成する方法を示します。


import { openai } from '@ai-sdk/openai';
import { MDocument } from '@mastra/rag';
import { embed } from 'ai';
 
const doc = MDocument.fromText("Your text content...");
 
const chunks = await doc.chunk();
 
const { embeddings } = await embedMany({
  model: openai.embedding('text-embedding-3-small'),
  values: chunks.map(chunk => chunk.text),
});

GitHubで例を見る