LibSQLVector ストア
LibSQLストレージ実装は、SQLite互換のベクトル検索LibSQL (ベクトル拡張機能を持つSQLiteのフォーク)と、ベクトル拡張機能を持つTurso を提供し、軽量で効率的なベクトルデータベースソリューションを提供します。
これは@mastra/core
パッケージの一部であり、メタデータフィルタリングによる効率的なベクトル類似性検索を提供します。
インストール
デフォルトのベクトルストアはコアパッケージに含まれています:
npm install @mastra/core@latest
使用方法
import { LibSQLVector } from "@mastra/core/vector/libsql";
// Create a new vector store instance
const store = new LibSQLVector({
connectionUrl: process.env.DATABASE_URL,
// Optional: for Turso cloud databases
authToken: process.env.DATABASE_AUTH_TOKEN,
});
// Create an index
await store.createIndex({
indexName: "myCollection",
dimension: 1536,
});
// Add vectors with metadata
const vectors = [[0.1, 0.2, ...], [0.3, 0.4, ...]];
const metadata = [
{ text: "first document", category: "A" },
{ text: "second document", category: "B" }
];
await store.upsert({
indexName: "myCollection",
vectors,
metadata,
});
// Query similar vectors
const queryVector = [0.1, 0.2, ...];
const results = await store.query({
indexName: "myCollection",
queryVector,
topK: 10, // top K results
filter: { category: "A" } // optional metadata filter
});
コンストラクタオプション
connectionUrl:
string
LibSQLデータベースURL。インメモリデータベースには':memory:'、ローカルファイルには'file:dbname.db'、または'libsql://your-database.turso.io'のようなLibSQL互換の接続文字列を使用します。
authToken?:
string
Tursoクラウドデータベース用の認証トークン
syncUrl?:
string
データベースレプリケーション用のURL(Turso固有)
syncInterval?:
number
データベース同期の間隔(ミリ秒)(Turso固有)
メソッド
createIndex()
新しいベクトルコレクションを作成します。インデックス名は文字またはアンダースコアで始まり、文字、数字、アンダースコアのみを含むことができます。次元は正の整数である必要があります。
indexName:
string
作成するインデックスの名前
dimension:
number
ベクトルの次元サイズ(埋め込みモデルと一致する必要があります)
metric?:
'cosine' | 'euclidean' | 'dotproduct'
= cosine
類似性検索の距離メトリック。注意:現在、LibSQLではコサイン類似度のみがサポートされています。
upsert()
ベクトルとそのメタデータをインデックスに追加または更新します。トランザクションを使用して、すべてのベクトルが原子的に挿入されることを保証します - 挿入が失敗した場合、操作全体がロールバックされます。
indexName:
string
挿入先のインデックスの名前
vectors:
number[][]
埋め込みベクトルの配列
metadata?:
Record<string, any>[]
各ベクトルのメタデータ
ids?:
string[]
オプションのベクトルID(提供されない場合は自動生成)
query()
オプションのメタデータフィルタリングを使用して類似ベクトルを検索します。
indexName:
string
検索するインデックスの名前
queryVector:
number[]
類似ベクトルを検索するためのクエリベクトル
topK?:
number
= 10
返す結果の数
filter?:
Filter
メタデータフィルター
includeVector?:
boolean
= false
結果にベクトルデータを含めるかどうか
minScore?:
number
= 0
最小類似度スコアのしきい値
describeIndex()
インデックスに関する情報を取得します。
indexName:
string
説明するインデックスの名前
戻り値:
interface IndexStats {
dimension: number;
count: number;
metric: "cosine" | "euclidean" | "dotproduct";
}
deleteIndex()
インデックスとそのすべてのデータを削除します。
indexName:
string
削除するインデックスの名前
listIndexes()
データベース内のすべてのベクトルインデックスを一覧表示します。
戻り値:Promise<string[]>
truncateIndex()
インデックス構造を維持しながら、インデックスからすべてのベクトルを削除します。
indexName:
string
切り詰めるインデックスの名前
updateVector()
IDによって特定のベクトルエントリを新しいベクトルデータやメタデータで更新します。
indexName:
string
ベクトルを含むインデックスの名前
id:
string
更新するベクトルエントリのID
update:
object
ベクトルやメタデータを含む更新データ
update.vector?:
number[]
更新する新しいベクトルデータ
update.metadata?:
Record<string, any>
更新する新しいメタデータ
deleteVector()
IDによってインデックスから特定のベクトルエントリを削除します。
indexName:
string
ベクトルを含むインデックスの名前
id:
string
削除するベクトルエントリのID
レスポンスタイプ
クエリ結果は以下の形式で返されます:
interface QueryResult {
id: string;
score: number;
metadata: Record<string, any>;
vector?: number[]; // Only included if includeVector is true
}
エラー処理
ストアは異なる失敗ケースに対して特定のエラーをスローします:
try {
await store.query({
indexName: "my-collection",
queryVector: queryVector,
});
} catch (error) {
// Handle specific error cases
if (error.message.includes("Invalid index name format")) {
console.error(
"Index name must start with a letter/underscore and contain only alphanumeric characters",
);
} else if (error.message.includes("Table not found")) {
console.error("The specified index does not exist");
} else {
console.error("Vector store error:", error.message);
}
}
一般的なエラーケースには以下が含まれます:
- 無効なインデックス名の形式
- 無効なベクトル次元
- テーブル/インデックスが見つからない
- データベース接続の問題
- アップサート中のトランザクション失敗