@mastra/libsql 1.29.0-alpha.0 → 1.29.0-alpha.1

package/dist/index.js

@@ -2643,7 +2643,8 @@ var AgentsLibSQL = class extends AgentsStorage {
 				"skillsFormat",
 				"durable",
 				"browser",
-				"toolProviders"
+				"toolProviders",
+				"modelList"
 			]
 		});
 		await this.#migrateToolsToJsonbFormat();
@@ -3060,8 +3061,9 @@ var AgentsLibSQL = class extends AgentsStorage {
 					versionNumber: input.versionNumber,
 					name: input.name ?? null,
 					description: input.description ?? null,
-					instructions: this.serializeInstructions(input.instructions),
-					model: input.model,
+					instructions: this.serializeInstructions(input.instructions ?? ""),
+					model: input.model ?? {},
+					modelList: input.modelList ?? null,
 					tools: input.tools ?? null,
 					defaultOptions: input.defaultOptions ?? null,
 					workflows: input.workflows ?? null,
@@ -3318,6 +3320,7 @@ var AgentsLibSQL = class extends AgentsStorage {
 			description: row.description,
 			instructions: this.deserializeInstructions(row.instructions),
 			model: this.parseJson(row.model, "model"),
+			modelList: this.parseJson(row.modelList, "modelList"),
 			tools: this.parseJson(row.tools, "tools"),
 			defaultOptions: this.parseJson(row.defaultOptions, "defaultOptions"),
 			workflows: this.parseJson(row.workflows, "workflows"),
@@ -6667,13 +6670,23 @@ function createKnowledgePagesCoreLoader(features, loadStorageModule) {
 	};
 }
 /**
-* Keyword-search words: the query split on whitespace, lowercased, deduplicated. Every word must
-* appear (as a case-insensitive substring) in a hit; an empty list means the search returns nothing.
+* Keyword-search words: the query split on whitespace, lowercased, deduplicated. A hit needs at
+* least one of them (as a case-insensitive substring); an empty list means the search returns nothing.
 */
 function knowledgeSearchWords(query) {
 	return [...new Set(query.toLowerCase().split(/\s+/).filter(Boolean))];
 }
 /**
+* Points one matched search word earns on a page, by the best field it appears in: name, then
+* description, then body. Pages rank by how many query words they match first; these points only
+* break ties between pages that match the same number of words.
+*/
+const KNOWLEDGE_SEARCH_WEIGHTS = {
+	name: 3,
+	description: 2,
+	body: 1
+};
+/**
 * Why `key` can't be a Knowledge page key, or undefined when it can. Keys are one path segment:
 * non-empty, with no `/` (path separator), `#` (scope/path separator) or `%` (escape character).
 */
@@ -7543,6 +7556,15 @@ var LibSQLKnowledgePages = class {
 		const rows = await this.#query(this.#read, `SELECT n.* FROM "${T.notes}" n WHERE n.pageId = ?${input.includeDeleted ? "" : " AND n.deletedAt IS NULL"} AND ${readable.sql}${after} ORDER BY n.createdAt, n.id LIMIT ?`, args);
 		return this.#noteOutput(rows, limit);
 	}
+	async countChildPages(input) {
+		const ids = [...new Set(input.parentIds)];
+		const counts = Object.fromEntries(ids.map((id) => [id, 0]));
+		if (ids.length === 0) return counts;
+		const readable = pageReadable("p", input.read);
+		const rows = await this.#query(this.#read, `SELECT p.parentId AS parentId, COUNT(*) AS total FROM "${T.pages}" p WHERE p.parentId IN (SELECT value FROM json_each(?)) AND p.isScope = 0 AND p.deletedAt IS NULL AND ${readable.sql} GROUP BY p.parentId`, [jsonList(ids), ...readable.args]);
+		for (const row of rows) counts[String(row.parentId)] = Number(row.total);
+		return counts;
+	}
 	async countNotes(input) {
 		const ids = [...new Set(input.pageIds)];
 		const counts = Object.fromEntries(ids.map((id) => [id, 0]));
@@ -7714,18 +7736,22 @@ var LibSQLKnowledgePages = class {
 		const readable = pageReadable("p", input.read);
 		const scopeFilter = input.scopeIds ? ` AND p.scopeId IN (SELECT value FROM json_each(?))` : "";
 		const scopeArgs = input.scopeIds ? [jsonList(input.scopeIds)] : [];
-		const pageMatch = words.map(() => `(lower(p.name) LIKE ? ESCAPE '=' OR lower(COALESCE(p.description, '')) LIKE ? ESCAPE '=' OR lower(COALESCE(p.body, '')) LIKE ? ESCAPE '=')`).join(" AND ");
-		const nameMatch = words.map(() => `lower(p.name) LIKE ? ESCAPE '='`).join(" AND ");
-		const hits = (await this.#query(this.#read, `SELECT p.* FROM "${T.pages}" p WHERE p.deletedAt IS NULL AND p.isScope = 0 AND ${readable.sql}${scopeFilter} AND ${pageMatch} ORDER BY CASE WHEN instr(p.canonicalName, ?) > 0 THEN 0 WHEN ${nameMatch} THEN 1 ELSE 2 END, p.updatedAt DESC, p.id ASC LIMIT ?`, [
+		const field = (column) => `lower(COALESCE(${column}, '')) LIKE ? ESCAPE '='`;
+		const pageMatch = words.map(() => `(${field("p.name")} OR ${field("p.description")} OR ${field("p.body")})`).join(" OR ");
+		const pageScore = words.map(() => `(CASE WHEN ${field("p.name")} THEN ${KNOWLEDGE_SEARCH_WEIGHTS.name} WHEN ${field("p.description")} THEN ${KNOWLEDGE_SEARCH_WEIGHTS.description} WHEN ${field("p.body")} THEN ${KNOWLEDGE_SEARCH_WEIGHTS.body} ELSE 0 END)`).join(" + ");
+		const pageCount = words.map(() => `(CASE WHEN ${field("p.name")} OR ${field("p.description")} OR ${field("p.body")} THEN 1 ELSE 0 END)`).join(" + ");
+		const triple = likes.flatMap((like) => [
+			like,
+			like,
+			like
+		]);
+		const hits = (await this.#query(this.#read, `SELECT p.* FROM "${T.pages}" p WHERE p.deletedAt IS NULL AND p.isScope = 0 AND ${readable.sql}${scopeFilter} AND (${pageMatch}) ORDER BY CASE WHEN instr(p.canonicalName, ?) > 0 THEN 0 ELSE 1 END, (${pageCount}) DESC, (${pageScore}) DESC, p.updatedAt DESC, p.id ASC LIMIT ?`, [
 			...readable.args,
 			...scopeArgs,
-			...likes.flatMap((like) => [
-				like,
-				like,
-				like
-			]),
+			...triple,
 			phrase,
-			...likes,
+			...triple,
+			...triple,
 			limit
 		])).map((row) => ({
 			type: "page",
@@ -7733,11 +7759,12 @@ var LibSQLKnowledgePages = class {
 		}));
 		if (hits.length >= limit) return hits;
 		const noteVisible = noteReadable("n", input.read);
-		const noteMatch = words.map(() => `lower(n.text) LIKE ? ESCAPE '='`).join(" AND ");
-		const noteRows = await this.#query(this.#read, `SELECT n.* FROM "${T.notes}" n JOIN "${T.pages}" p ON p.id = n.pageId WHERE n.deletedAt IS NULL AND p.deletedAt IS NULL${scopeFilter} AND ${noteMatch} AND ${noteVisible.sql} ORDER BY n.updatedAt DESC, n.id DESC LIMIT ?`, [
+		const noteHas = words.map(() => `lower(n.text) LIKE ? ESCAPE '='`);
+		const noteRows = await this.#query(this.#read, `SELECT n.* FROM "${T.notes}" n JOIN "${T.pages}" p ON p.id = n.pageId WHERE n.deletedAt IS NULL AND p.deletedAt IS NULL${scopeFilter} AND (${noteHas.join(" OR ")}) AND ${noteVisible.sql} ORDER BY (${noteHas.map((has) => `(CASE WHEN ${has} THEN 1 ELSE 0 END)`).join(" + ")}) DESC, n.updatedAt DESC, n.id DESC LIMIT ?`, [
 			...scopeArgs,
 			...likes,
 			...noteVisible.args,
+			...likes,
 			limit - hits.length
 		]);
 		const notes = await this.#hydrateNotes(this.#read, noteRows);
@@ -8067,6 +8094,9 @@ var KnowledgeLibSQL = class extends KnowledgeStorage {
 	listNotes(input) {
 		return this.#pages.listNotes(input);
 	}
+	countChildPages(input) {
+		return this.#pages.countChildPages(input);
+	}
 	countNotes(input) {
 		return this.#pages.countNotes(input);
 	}