tbenandClaude Opus 5.5
e7b23b7454
Knowledgebases give bots RAG over documents, scoped like extensions: to particular bots, to a workspace, or to every bot, with their own button beside Tools in the composer. Documents are chunked and embedded by the file-ingest worker with the install's embedding model, which is now an install-wide setting rather than a per-bot one. - Word search runs on an FTS5 index ranked by BM25, with a scored LIKE fallback for scripts written without spaces. - Changing the embedding model re-indexes documents automatically. - Ingest jobs wait without spending attempts while every embedding model rests, fail at once on errors retrying can't fix, and keep passages searchable by words when the model refuses. - Settings > Knowledgebases shows each knowledgebase's and document's index state: spinner, warning, grey check (words only) or green check. Co-Authored-By: Claude Opus 5.5 <[email protected]>
93 lines
2.7 KiB
TypeScript
93 lines
2.7 KiB
TypeScript
import BunSQLite from "@moduletrace/bun-sqlite";
|
|
import { Harnesses, isInstallWideHarness, type HarnessKind } from "@/src/dict/harness-dict";
|
|
import AppError from "@/src/utils/app-error";
|
|
import assertHarnessModel from "./assert-harness-model";
|
|
import assertHarnessProvider from "./assert-harness-provider";
|
|
|
|
type Params = {
|
|
/**
|
|
* A worker bot the caller has checked belongs to the session user.
|
|
*/
|
|
bot_id: number;
|
|
harness: HarnessKind;
|
|
/**
|
|
* Drop the bot's own setting and use the default again.
|
|
*/
|
|
inherit: boolean;
|
|
model_id: number | null;
|
|
/**
|
|
* Limits the harness to one provider; null: any.
|
|
*/
|
|
provider_id?: number | null;
|
|
fallback: boolean;
|
|
enabled: boolean;
|
|
options?: Record<string, unknown>;
|
|
};
|
|
|
|
/**
|
|
* Saves (or, with `inherit`, removes) a worker bot's own setting for one
|
|
* harness. Same rules as the default: the model must be available and fit
|
|
* the job, and text can't be turned off.
|
|
*/
|
|
export default async function saveBotHarnessSetting({
|
|
bot_id,
|
|
harness,
|
|
inherit,
|
|
model_id,
|
|
provider_id = null,
|
|
fallback,
|
|
enabled,
|
|
options,
|
|
}: Params) {
|
|
if (inherit) {
|
|
await BunSQLite.sql({
|
|
sql: `DELETE FROM bot_harness_settings WHERE bot_id = ? AND harness = ?`,
|
|
values: [bot_id, harness],
|
|
});
|
|
return;
|
|
}
|
|
|
|
if (isInstallWideHarness({ harness })) {
|
|
const title = Harnesses.find((h) => h.value == harness)?.title || harness;
|
|
|
|
throw new AppError(`${title} is set for the whole install, in Settings, under Models.`);
|
|
}
|
|
|
|
if (harness == "text" && !enabled) {
|
|
throw new AppError(`The text model can't be turned off.`);
|
|
}
|
|
|
|
if (provider_id) {
|
|
await assertHarnessProvider({ provider_id });
|
|
}
|
|
|
|
if (model_id) {
|
|
await assertHarnessModel({ harness, model_id, provider_id });
|
|
}
|
|
|
|
const now = Date.now();
|
|
|
|
const res = await BunSQLite.sql({
|
|
sql: `INSERT INTO bot_harness_settings (created_at, updated_at, bot_id, harness, model_id, provider_id, fallback, enabled, options)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
ON CONFLICT(bot_id, harness) DO UPDATE SET updated_at = excluded.updated_at,
|
|
model_id = excluded.model_id, provider_id = excluded.provider_id, fallback = excluded.fallback,
|
|
enabled = excluded.enabled, options = excluded.options`,
|
|
values: [
|
|
now,
|
|
now,
|
|
bot_id,
|
|
harness,
|
|
model_id,
|
|
provider_id,
|
|
fallback ? 1 : 0,
|
|
enabled ? 1 : 0,
|
|
JSON.stringify(options || {}),
|
|
],
|
|
});
|
|
|
|
if (!res.success) {
|
|
throw new Error(`Couldn't save the bot's model setting: ${res.error}`);
|
|
}
|
|
}
|