@tiagojct/subsub 0.23.0 → 0.24.1

package/src/screens.ts

@@ -237,6 +237,15 @@ export interface LibraryItem {
 	title?: string;
 	authors?: string;
 	year?: string | number;
+	/** A group library's item: "group:<id>" and the group's name. Groups are read-only. */
+	library?: string;
+	group?: string;
+}
+
+/** How a command names an item: the citekey (or key), and the group library when it is in one. */
+export function itemRef(it: LibraryItem): string {
+	const ref = it.citekey || it.key;
+	return it.library ? `${ref} in ${it.library}` : ref;
 }
 
 export interface LibraryLayout {
@@ -270,7 +279,7 @@ export function libraryLines(o: LibraryLayout): string[] {
 		const start = Math.max(0, Math.min(o.selected - Math.floor(fit / 2), o.items.length - fit));
 		for (const [k, it] of o.items.slice(start, start + fit).entries()) {
 			const chosen = start + k === o.selected;
-			const meta = [it.authors, it.year, it.citekey].filter(Boolean).join(" · ");
+			const meta = [it.authors, it.year, it.citekey, it.group].filter(Boolean).join(" · ");
 			out.push(chosen ? `${p.mark(g.pointer)} ${p.text(cut(it.title || it.key, w - 2))}` : `  ${p.muted(cut(it.title || it.key, w - 2))}`, `  ${p.dim(cut(meta, w - 2))}`);
 		}
 	}
@@ -480,6 +489,8 @@ export interface InputOptions {
 	title: string;
 	placeholder?: string;
 	prefill?: string;
+	/** An API key: each character is shown as a star. */
+	secret?: boolean;
 	submit: (text: string) => void;
 	cancel: () => void;
 	lang: Lang;
@@ -506,7 +517,8 @@ export class InputView {
 	}
 
 	render(width: number): string[] {
-		return fill(inputLines({ title: this.o.title, placeholder: this.o.placeholder, text: this.text, width, lang: this.o.lang, paint: this.o.paint, glyphs: this.o.glyphs }), width);
+		const text = this.o.secret ? "*".repeat([...this.text].length) : this.text;
+		return fill(inputLines({ title: this.o.title, placeholder: this.o.placeholder, text, width, lang: this.o.lang, paint: this.o.paint, glyphs: this.o.glyphs }), width);
 	}
 
 	invalidate(): void {}

package/src/subsub.ts

@@ -45,14 +45,15 @@ import { describeArgs, formatPreview, isEmptyPreview } from "./preview.ts";
 import { perMinuteWait, providerTrouble, TROUBLE_TEXT, TROUBLE_TEXT_PT } from "./keys.ts";
 import { headerLines, LIBRARIAN_THEME, type LibraryState, paintPreview, QUOTES, type Scheme, schemeFromAnsi, themeName } from "./look.ts";
 import { footerLines, toolRenderers, usageTotals } from "./chrome.ts";
+import { type CostLevel, costStatus, money, usesModel } from "./cost.ts";
 import { ApprovalView } from "./dialogs.ts";
 import { allowedFile, listNotes, OPENABLE, openFile } from "./files.ts";
-import { LibraryView, NotesView, type LibraryItem } from "./screens.ts";
+import { itemRef, LibraryView, NotesView, type LibraryItem } from "./screens.ts";
 import { glyphs, paintFromTheme, workingFrames } from "./skin.ts";
 import { loadWords, toolWord, type Words } from "./words.ts";
 import { systemAddition } from "./prompt.ts";
 import { logUsage } from "./usage.ts";
-import { gateKind, SERVER_PREFIXES, toolsFor, unavailableReason, outboundProblem } from "./roles.ts";
+import { activeTools, gateKind, SERVER_PREFIXES, toolsFor, unavailableReason, outboundProblem } from "./roles.ts";
 
 const MODE_ENTRY = "subsub-mode";
 
@@ -181,6 +182,8 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 	let resumeTimer: ReturnType<typeof setTimeout> | undefined;
 	let turnStart = 0;
 	let turnTools = 0;
+	/** The cost level already noted in this session, so each note comes once. */
+	let costNoted: CostLevel | undefined;
 	let webCtx: ExtensionContext | undefined;
 	function emitWeb(): void {
 		const ctx = webCtx;
@@ -195,6 +198,8 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 			userName: cfg.userName ?? null,
 			vault: cfg.vault ?? null,
 			starbuck: starbuckEnabled(cfg),
+			// Shown next to the model only when there is a limit.
+			cost: cfg.costLimit ? { spent: usageTotals(ctx.sessionManager.getEntries()).cost, limit: cfg.costLimit } : null,
 		};
 		try {
 			ctx.ui.setStatus("subsub.web", JSON.stringify(state));
@@ -277,15 +282,11 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 	/** /library: search as you type; Enter puts /read <item> in the box (the Librarian gets the key), ctrl+n puts /lit-note <item>. */
 	async function openLibrary(ctx: ExtensionContext): Promise<void> {
 		if (!screenOk(ctx)) return;
-		const ref = (it: LibraryItem) => it.citekey || it.key;
+		const ref = itemRef;
 		await ctx.ui.custom<void>(
 			(tui, theme, _keys, done) =>
 				new LibraryView({
-					search: async (query) => {
-						const r = await current().call("zotero_find_items", { query, limit: 25 });
-						if (r.isError) throw new Error(r.text);
-						return ((r.data as { items?: LibraryItem[] })?.items ?? []).filter((it) => it && it.key);
-					},
+					search: searchLibraries,
 					choose: (it) => {
 						ctx.ui.setEditorText(mode === "researcher" ? `/read ${ref(it)}` : ref(it));
 						done();
@@ -326,6 +327,33 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 				: text;
 	}
 
+	/** The group libraries in Zotero, from the last library check (read-only; searched with the user's library). */
+	let groups: Array<{ library: string; name: string }> = [];
+
+	/** Search the user's library, then each group library; a group's items carry the group. */
+	async function searchLibraries(query: string): Promise<LibraryItem[]> {
+		const first = await current().call("zotero_find_items", { query, limit: 25 });
+		if (first.isError) throw new Error(first.text);
+		const items = ((first.data as { items?: LibraryItem[] })?.items ?? []).filter((it) => it && it.key);
+		for (const g of groups) {
+			const r = await current().call("zotero_find_items", { query, limit: 25, library: g.library });
+			if (r.isError) continue;
+			for (const it of (r.data as { items?: LibraryItem[] })?.items ?? []) if (it?.key) items.push({ ...it, library: g.library, group: g.name });
+		}
+		return items;
+	}
+
+	async function refreshGroups(): Promise<void> {
+		if (!current().has("zotero_list_libraries")) return;
+		try {
+			const r = await current().call("zotero_list_libraries", {});
+			const all = ((r.data as { libraries?: Array<{ library: string; name?: string }> })?.libraries ?? []);
+			groups = r.isError ? [] : all.filter((l) => l.library?.startsWith("group:")).map((l) => ({ library: l.library, name: l.name || l.library }));
+		} catch {
+			groups = [];
+		}
+	}
+
 	async function refreshLibrary(): Promise<void> {
 		try {
 			const st = await current().call("zotero_status", {});
@@ -336,6 +364,7 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 				const ov = await current().call("zotero_library_overview", {});
 				const o = (ov.data ?? {}) as Record<string, unknown>;
 				Object.assign(library, { zotero: "reachable", items: o.items as number, toReview: o.awaiting_review as number });
+				await refreshGroups();
 			}
 		} catch {
 			library.zotero = "down";
@@ -410,7 +439,7 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 	/** Set the mode's tools and, unless `switchModel` is false, the mode's model. */
 	async function applyMode(ctx: ExtensionContext, next: Mode, remember: boolean, switchModel = true): Promise<void> {
 		mode = next;
-		pi.setActiveTools(toolsFor(mode, registered(), toolOpts()));
+		pi.setActiveTools(activeTools(pi.getActiveTools(), toolsFor(mode, registered(), toolOpts()), standalone));
 		const spec = cfg.models[mode];
 		if (spec && switchModel) {
 			const i = spec.indexOf("/");
@@ -442,6 +471,7 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 	}
 
 	pi.on("session_start", async (_event, ctx) => {
+		costNoted = undefined; // a new or another conversation: its own cost
 		if (!bridge) {
 			const b = new Bridge();
 			await b.start(deps.specs ?? serverSpecs(cfg), cfg.startTimeout * 1000);
@@ -531,6 +561,17 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 	// A per-minute limit (free tiers): wait as long as the provider says, then continue the task,
 	// up to three times per request of the user. Anything the user types cancels the wait.
 	pi.on("agent_end", async (_event, ctx) => {
+		const spent = costStatus(usageTotals(ctx.sessionManager.getEntries()).cost, cfg.costLimit);
+		if (spent.level !== "ok" && costNoted !== spent.level) {
+			costNoted = spent.level;
+			ctx.ui.notify(
+				spent.level === "near"
+					? L(`Sub-Sub: this conversation has used 80% of your cost limit (${money(spent.spent)} of ${money(spent.limit!)}).`, `Sub-Sub: esta conversa já usou 80% do seu limite de custo (${money(spent.spent)} de ${money(spent.limit!)}).`)
+					: L(`Sub-Sub: this conversation has reached your cost limit (${money(spent.spent)} of ${money(spent.limit!)}). Sub-Sub will ask before each new request.`, `Sub-Sub: esta conversa atingiu o seu limite de custo (${money(spent.spent)} de ${money(spent.limit!)}). O Sub-Sub vai perguntar antes de cada pedido novo.`),
+				"warning",
+			);
+		}
+		if (cfg.costLimit) emitWeb(); // the cost next to the model in the browser view
 		if (turnStart) logUsage(cfg.usageLog, { ev: "turn", seconds: Math.round((Date.now() - turnStart) / 1000), tools: turnTools, error: providerTrouble(lastError) ?? (lastError ? "other" : undefined) });
 		turnStart = 0;
 		const wait = perMinuteWait(lastError);
@@ -558,6 +599,19 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 		resumeTimer = undefined;
 		resumes = 0;
 		const cmd = /^\/([\w-]+)(?:\s|$)/.exec(event.text.trim())?.[1];
+		// Over the cost limit: ask before each new request to the model (a running answer is not cut).
+		const spent = costStatus(usageTotals(ctx.sessionManager.getEntries()).cost, cfg.costLimit);
+		if (spent.level === "over" && usesModel(event.text)) {
+			const text = L(
+				`This conversation has cost ${money(spent.spent)}, over your limit of ${money(spent.limit!)} (costLimit in the settings). Send this request to the model?`,
+				`Esta conversa já custou ${money(spent.spent)}, acima do seu limite de ${money(spent.limit!)} (costLimit nas definições). Enviar este pedido ao modelo?`,
+			);
+			const go = ctx.hasUI ? await ctx.ui.confirm(L("Sub-Sub: cost limit", "Sub-Sub: limite de custo"), text) : false;
+			if (!go) {
+				if (!ctx.hasUI) ctx.ui.notify(text, "warning");
+				return { action: "handled" };
+			}
+		}
 		// The usage log keeps the kind of request and its length, never its text.
 		if (cmd !== "subsub-refresh") logUsage(cfg.usageLog, { ev: "ask", kind: cmd ? `/${cmd}` : "message", words: event.text.trim().split(/\s+/).filter(Boolean).length });
 		if (mode === "librarian" && cmd && RESEARCH_PROMPTS.includes(cmd)) {
@@ -578,7 +632,7 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 	});
 
 	pi.on("before_agent_start", async (event, ctx) => ({
-		systemPrompt: `${event.systemPrompt}\n\n${systemAddition(mode, cfg, ctx.cwd)}`,
+		systemPrompt: `${event.systemPrompt}\n\n${systemAddition(mode, cfg, ctx.cwd, { zoteroDown: library.zotero === "down" })}`,
 	}));
 
 	/** Model test (subsub-bench): record every change instead of asking, and apply none. */
@@ -846,6 +900,8 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 							? ov.text
 							: `Items: ${o.items}. Without tags: ${o.without_manual_tags}. Missing facets: ${JSON.stringify(o.missing_facet)}. Awaiting review (_agent): ${o.awaiting_review}.`,
 					);
+					await refreshGroups();
+					if (groups.length) lines.push(`Group libraries (read-only, searched with yours): ${groups.map((g) => g.name).join(", ")}.`);
 				}
 			}
 			redraw();
@@ -870,8 +926,7 @@ export async function createSubsub(pi: ExtensionAPI, deps: SubsubDeps = {}): Pro
 			const q = String(args ?? "").trim();
 			let out: Record<string, unknown>;
 			try {
-				const r = await current().call("zotero_find_items", { query: q, limit: 25 });
-				out = r.isError ? { error: r.text } : { items: (r.data as { items?: unknown[] })?.items ?? [] };
+				out = { items: await searchLibraries(q) };
 			} catch (err) {
 				out = { error: (err as Error).message };
 			}

package/src/tui-view.ts

@@ -252,7 +252,7 @@ export function sideLines(s: SideState, width: number, rows: number, p: Paint, g
 	const w = Math.max(16, width);
 	const words = SIDE_WORDS[s.lang];
 	const mode = s.mode === "librarian" ? "Librarian" : "Researcher";
-	const lib = s.library.zotero === "reachable" ? `${s.library.items ?? "?"} ${s.lang === "pt" ? "itens" : "items"}` : s.library.zotero === "down" ? `${g.hollow} Zotero` : "Zotero...";
+	const lib = s.library.zotero === "reachable" ? `${s.library.items ?? "?"} ${s.lang === "pt" ? "itens" : "items"}` : s.library.zotero === "down" ? `${g.hollow} Zotero: ${s.lang === "pt" ? "só pesquisa" : "search only"}` : "Zotero...";
 	const lines = [p.mark(cut(` Sub-Sub ${s.version}`, w)), p.text(cut(` ${mode}`, w)), p.muted(cut(` ${lib}`, w)), p.dim(cut(` ${s.model}`, w)), rule(w, "solid", p.muted, g)];
 	for (const [key, word] of [["F2", words.notes], ["F3", words.library], ["F4", words.actions], ["F5", words.mode]] as const) lines.push(p.muted(cut(` ${key.padEnd(4)}${word}`, w)));
 	lines.push(rule(w, "solid", p.muted, g), p.text(cut(` ${words.conversations}`, w)));

package/src/tui.ts

@@ -12,12 +12,15 @@ import { CombinedAutocompleteProvider, Editor, HStack, Markdown, matchesKey, Pro
 import { agentDirFor, chooseCwd, PACKAGE_DIR, packageVersion } from "./cli.ts";
 import { fill, footerLines } from "./chrome.ts";
 import { loadConfig, uiLanguage } from "./config.ts";
+import { modelsAfterLogin } from "./init.ts";
 import { type LibraryState, LIBRARIAN_THEME, DEFAULT_THEME, paintPreview, QUOTES } from "./look.ts";
 import { PURPOSES } from "./read.ts";
 import { allowedFile, listNotes, OPENABLE, openFile } from "./files.ts";
 import { Agent, findLibrary, isDialog, Side, withHints } from "./rpc.ts";
+import { KEY_PROVIDERS, saveApiKey } from "./keys.ts";
+import { CUSTOM_DEFAULT_NAME, saveCustomProvider } from "./providers.ts";
 import { ApprovalView } from "./dialogs.ts";
-import { type Choice, ChoiceView, InputView, LibraryView, type LibraryItem, NotesView } from "./screens.ts";
+import { itemRef, type Choice, ChoiceView, InputView, LibraryView, type LibraryItem, NotesView } from "./screens.ts";
 import { glyphs as chooseGlyphs, type Glyphs, workingFrames } from "./skin.ts";
 import { bannerLines, type Entry, entriesFromMessages, loadPalette, messageText, type Palette, schemeFromEnv, type Scheme, SIDE_WIDTH, sideLines, transcriptLines } from "./tui-view.ts";
 import { loadWords, type Words } from "./words.ts";
@@ -58,6 +61,10 @@ export interface TuiDeps {
 	searchDelay?: number;
 	/** Called when the user quits or pi stops. */
 	onQuit?: (code: number) => void;
+	/** pi's agent folder, where /login saves keys (auth.json) and other providers (models.json). */
+	agentDir?: string;
+	/** After /login saved a provider: set the models when none are set (spec: the other provider's model). */
+	afterLogin?: (provider: string, spec?: string) => void;
 }
 
 const SHORT = (s: string, n: number) => (s.length > n ? `${s.slice(0, n - 3)}...` : s);
@@ -213,6 +220,8 @@ export class TuiApp {
 			{ name: "library", description: pt ? "Pesquisar na biblioteca à medida que escreve" : "Search your library as you type" },
 			{ name: "actions", description: pt ? "As ações da vista no navegador" : "The actions of the browser view" },
 			{ name: "sessions", description: pt ? "Conversas anteriores" : "Earlier conversations" },
+			{ name: "resume", description: pt ? "Retomar uma conversa anterior" : "Resume an earlier conversation" },
+			{ name: "login", description: pt ? "Ligar um fornecedor do modelo" : "Connect a model provider" },
 			{ name: "new", description: pt ? "Uma conversa nova" : "A new conversation" },
 			{ name: "model", description: pt ? "Escolher o modelo" : "Choose the model" },
 			{ name: "restart", description: pt ? "Recomeçar o pi na mesma conversa" : "Start pi again in the same conversation" },
@@ -492,7 +501,7 @@ export class TuiApp {
 	/** Alt+L, /library: search as you type; Enter puts /read <item> in the editor (the Librarian gets the key), ctrl+n puts /lit-note <item>. */
 	async openLibrary(): Promise<void> {
 		if (this.overlay) return;
-		const ref = (it: LibraryItem) => it.citekey || it.key;
+		const ref = itemRef;
 		this.open(
 			new LibraryView({
 				...this.common(),
@@ -558,6 +567,70 @@ export class TuiApp {
 		);
 	}
 
+	/** Ask one line on its own screen; undefined when the user presses Esc. */
+	private ask(title: string, opts: { placeholder?: string; secret?: boolean } = {}): Promise<string | undefined> {
+		return new Promise((resolve) => {
+			this.open(
+				new InputView({
+					...this.common(),
+					title,
+					...opts,
+					submit: (text) => (this.close(), resolve(text.trim())),
+					cancel: () => (this.close(), resolve(undefined)),
+				}),
+				true,
+			);
+		});
+	}
+
+	/** /login: a provider that pi has (its API key) or another provider that speaks the OpenAI API; then pi starts again. */
+	async openLogin(): Promise<void> {
+		if (this.overlay) return;
+		const pt = this.lang === "pt";
+		const agentDir = this.d.agentDir;
+		if (!agentDir) return void this.notice("error", pt ? "Sem pasta do agente: use subsub web." : "No agent folder: use subsub web.");
+		const choices: Choice[] = [
+			...KEY_PROVIDERS.map((p) => ({ label: p.label, hint: p.url, value: p.id })),
+			{ label: pt ? "Outro fornecedor (API da OpenAI)" : "Another provider (OpenAI API)", hint: pt ? "servidor da instituição, Ollama, LM Studio" : "your institution's server, Ollama, LM Studio", value: CUSTOM_DEFAULT_NAME },
+		];
+		const provider = await new Promise<string | undefined>((resolve) => {
+			this.open(
+				new ChoiceView({
+					...this.common(),
+					title: pt ? "Ligar um fornecedor" : "Connect a provider",
+					choices,
+					choose: (v) => (this.close(), resolve(v)),
+					cancel: () => (this.close(), resolve(undefined)),
+				}),
+				true,
+			);
+		});
+		if (!provider) return;
+		let spec: string | undefined;
+		try {
+			if (provider === CUSTOM_DEFAULT_NAME) {
+				const baseUrl = await this.ask(pt ? "Endereço do servidor" : "Server address", { placeholder: "https://llm.example.org/v1" });
+				if (baseUrl === undefined) return;
+				const model = await this.ask(pt ? "Identificador do modelo" : "Model id");
+				if (model === undefined) return;
+				const key = await this.ask(pt ? "Chave de API (Enter sem chave, num servidor local)" : "API key (Enter for none, for a local server)", { secret: true });
+				if (key === undefined) return;
+				[spec] = saveCustomProvider(agentDir, { name: CUSTOM_DEFAULT_NAME, baseUrl, apiKey: key, models: [model] });
+			} else {
+				const label = KEY_PROVIDERS.find((p) => p.id === provider)?.label ?? provider;
+				const key = await this.ask(pt ? `Chave de API de ${label}` : `API key for ${label}`, { secret: true });
+				if (!key) return;
+				saveApiKey(agentDir, provider, key);
+			}
+			this.d.afterLogin?.(provider, spec);
+		} catch (err) {
+			this.notice("error", (err as Error).message);
+			return void this.tui.requestRender();
+		}
+		this.notice("info", pt ? "Guardado. O pi recomeça com o fornecedor." : "Saved. pi starts again with the provider.");
+		await this.restart();
+	}
+
 	/** /new: an empty conversation. */
 	async newConversation(): Promise<void> {
 		try {
@@ -633,12 +706,12 @@ export class TuiApp {
 		this.editor.addToHistory?.(line);
 		this.editor.setText("");
 		if (/^\/(quit|exit)\s*$/.test(line)) return void (await this.quit());
-		const unavailable = /^\/(login|logout|settings|tree|resume|fork|share|export|import|hotkeys|reload)\b/.exec(line);
+		const unavailable = /^\/(logout|settings|tree|fork|share|export|import|hotkeys|reload)\b/.exec(line);
 		if (unavailable) {
 			this.notice("info", this.lang === "pt" ? `/${unavailable[1]} não existe em subsub tui. Use subsub no terminal ou subsub web.` : `/${unavailable[1]} is not in subsub tui. Use subsub in the terminal, or subsub web.`);
 			return void this.tui.requestRender();
 		}
-		const own: Record<string, () => Promise<void> | void> = { "/restart": () => this.restart(), "/model": () => this.openModels(), "/notes": () => this.openNotes(), "/library": () => this.openLibrary(), "/actions": () => this.openActions(), "/sessions": () => this.openSessions(), "/new": () => this.newConversation() };
+		const own: Record<string, () => Promise<void> | void> = { "/restart": () => this.restart(), "/model": () => this.openModels(), "/notes": () => this.openNotes(), "/library": () => this.openLibrary(), "/actions": () => this.openActions(), "/sessions": () => this.openSessions(), "/resume": () => this.openSessions(), "/login": () => this.openLogin(), "/new": () => this.newConversation() };
 		if (own[line]) return void (await own[line]());
 		try {
 			await this.d.agent.request({ type: "prompt", message: line, ...(this.busy ? { streamingBehavior: "followUp" } : {}) });
@@ -703,6 +776,8 @@ export async function runTui(args: string[], env: NodeJS.ProcessEnv = process.en
 			themes: { researcher: typeof cfg.themes === "string" ? cfg.themes : undefined, librarian: cfg.themesLibrarian ?? (typeof cfg.themes === "string" ? cfg.themes : undefined) },
 			sessions: async () => (await SessionManager.list(cwd)).filter((s) => s.messageCount > 0).sort((a, b) => +new Date(b.modified) - +new Date(a.modified)).map((s) => ({ title: s.name ?? s.firstMessage.slice(0, 60), path: s.path })),
 			onQuit: resolve,
+			agentDir: env.PI_CODING_AGENT_DIR,
+			afterLogin: (provider, spec) => modelsAfterLogin(env, provider, spec),
 		});
 		void app.start();
 	});

package/src/web.ts

@@ -21,8 +21,9 @@ import { homedir } from "node:os";
 import { basename, dirname, extname, join, normalize, relative, resolve, sep } from "node:path";
 import { agentDirFor, chooseCwd, PACKAGE_DIR, packageVersion } from "./cli.ts";
 import { configPath, loadConfig, RECOMMENDED_MODELS, readConfigFile, saveConfig, type SubsubConfig, uiLanguage, withStarbuck } from "./config.ts";
-import { TESTED_MODELS } from "./init.ts";
+import { modelsAfterLogin } from "./init.ts";
 import { KEY_PROVIDERS, saveApiKey } from "./keys.ts";
+import { CUSTOM_DEFAULT_NAME, saveCustomProvider } from "./providers.ts";
 
 // ---------------------------------------------------------------- options
 
@@ -479,9 +480,13 @@ export async function startWeb(opts: WebOptions, env: NodeJS.ProcessEnv = proces
 					if (busy || switching) return send(res, 409, { error: "busy" });
 					const body = await readJson(req);
 					return withRestart(res, () => {
+						if (body.provider === CUSTOM_DEFAULT_NAME) {
+							// Another provider (OpenAI API): models.json; its model becomes the default when none is set.
+							const [spec] = saveCustomProvider(agentDir, { name: CUSTOM_DEFAULT_NAME, baseUrl: String(body.baseUrl ?? ""), apiKey: String(body.key ?? ""), models: [String(body.model ?? "")] });
+							return modelsAfterLogin(env, CUSTOM_DEFAULT_NAME, spec);
+						}
 						saveApiKey(agentDir, String(body.provider ?? ""), String(body.key ?? ""));
-						const current = loadConfig(env);
-						if (body.provider === "opencode-go" && Object.keys(current.models).length === 0) saveConfig({ models: TESTED_MODELS, model: undefined }, env);
+						modelsAfterLogin(env, String(body.provider ?? ""));
 					});
 				}
 				case "GET /api/notes":

package/web/app.js

@@ -12,6 +12,8 @@ let session = null; // pi's session state
 let busy = false;
 let commands = [];
 let providers = [];
+/** The login form's choice for a provider that pi does not include (src/providers.ts CUSTOM_DEFAULT_NAME). */
+const OTHER_PROVIDER = "custom";
 const dialogs = []; // waiting dialog requests, oldest first
 let shownDialog = null;
 let currentMsg = null; // assistant message being streamed
@@ -255,7 +257,10 @@ function renderHeader() {
 	$("starbuck-label").title = t("starbuckHint");
 	const m = state?.model ?? session?.model;
 	const none = !m || !m.id || m.id === "unknown";
-	$("model-btn").textContent = $("model-btn-side").textContent = none ? t("noModel") : m.id;
+	// With a cost limit in the settings, the conversation's cost and the limit follow the model.
+	const c = state?.cost;
+	const cost = c && typeof c.spent === "number" ? ` · $${c.spent.toFixed(2)}/$${c.limit.toFixed(2)}` : "";
+	$("model-btn").textContent = $("model-btn-side").textContent = none ? t("noModel") : m.id + cost;
 	for (const b of [$("model-btn"), $("model-btn-side")]) b.classList.toggle("needs", none);
 	const lib = $("library");
 	const txt = lib.querySelector(".library-text") ?? lib;
@@ -363,7 +368,9 @@ function renderItems(items, message) {
 	if (message) return ul.append(el("li", { class: "muted", text: message }));
 	const researcher = mode() === "researcher";
 	for (const it of items) {
-		const ref = it.citekey || it.key;
+		// A group library's item says its library (keep in step with itemRef in src/screens.ts).
+		const ref = it.library ? `${it.citekey || it.key} in ${it.library}` : it.citekey || it.key;
+		const scope = it.library ? `groups/${it.library.split(":")[1]}` : "library";
 		const run = (cmd) => () => {
 			closeSide();
 			sendMessage(`/${cmd} ${ref}`);
@@ -373,13 +380,13 @@ function renderItems(items, message) {
 				"li",
 				{},
 				el("div", { class: "item-title", text: it.title || it.key }),
-				el("div", { class: "item-meta", text: [it.authors, it.year, it.citekey].filter(Boolean).join(" · ") }),
+				el("div", { class: "item-meta", text: [it.authors, it.year, it.citekey, it.group].filter(Boolean).join(" · ") }),
 				el(
 					"div",
 					{ class: "item-actions" },
 					researcher ? el("button", { type: "button", text: t("readItem"), onclick: run("read") }) : null,
 					researcher ? el("button", { type: "button", text: t("noteItem"), onclick: run("lit-note") }) : null,
-					el("a", { href: `zotero://select/library/items/${encodeURIComponent(it.key)}`, text: "Zotero" }),
+					el("a", { href: `zotero://select/${scope}/items/${encodeURIComponent(it.key)}`, text: "Zotero" }),
 				),
 			),
 		);
@@ -1260,15 +1267,22 @@ async function openModels() {
 	const ul = $("models");
 	ul.replaceChildren(el("li", { class: "empty-models", text: "…" }));
 	const sel = $("login-provider");
-	sel.replaceChildren(...providers.map((p) => el("option", { value: p.id, text: p.label })));
+	sel.replaceChildren(...providers.map((p) => el("option", { value: p.id, text: p.label })), el("option", { value: OTHER_PROVIDER, text: t("otherProvider") }));
 	const link = $("login-link");
 	const setLink = () => {
 		const p = providers.find((x) => x.id === sel.value);
+		const other = sel.value === OTHER_PROVIDER;
+		link.hidden = other;
 		link.href = p?.url ?? "#";
+		$("login-url-row").hidden = !other;
+		$("login-model-row").hidden = !other;
+		$("login-url").required = other;
+		$("login-model").required = other;
+		$("login-key").required = !other;
 		const free = { google: "freeGoogle", openrouter: "freeOpenRouter" }[sel.value];
 		$("login-free").hidden = !free;
 		$("login-free").textContent = free ? t(free) : "";
-		$("login-where").textContent = p?.where ? t(`where_${p.where}`) : "";
+		$("login-where").textContent = other ? t("otherNote") : p?.where ? t(`where_${p.where}`) : "";
 	};
 	sel.onchange = setLink;
 	setLink();
@@ -1361,7 +1375,7 @@ $("login-form")?.addEventListener("submit", async (e) => {
 	btn.disabled = true;
 	btn.textContent = t("saving");
 	try {
-		await api("/api/login", { provider: $("login-provider").value, key: $("login-key").value });
+		await api("/api/login", { provider: $("login-provider").value, key: $("login-key").value, baseUrl: $("login-url").value, model: $("login-model").value });
 		$("login-key").value = "";
 		note(t("saved"));
 		await openModels();

package/web/i18n.js

@@ -24,7 +24,7 @@ export const STRINGS = {
 		starbuckReport: "Write the Starbuck report to {x}?",
 		approveHint: "Nothing changes until you select Yes.",
 		zoteroChecking: "Zotero: checking",
-		zoteroDown: "Zotero is not running",
+		zoteroDown: "Zotero is closed: search only",
 		items: "{n} items",
 		toReview: "{n} to check",
 		toReviewHint: "Items that Sub-Sub tagged and that wait for your check (they carry the _agent tag). Select Library overview, or ask Sub-Sub to list them.",
@@ -59,6 +59,10 @@ export const STRINGS = {
 		save: "Save",
 		getKey: "Get a key",
 		keyNote: "The key is saved on this computer, in Sub-Sub's settings folder, and is sent only to that provider.",
+		otherProvider: "Another provider (OpenAI API)",
+		serverUrl: "Server address",
+		modelId: "Model id",
+		otherNote: "A server that speaks the OpenAI API: your institution's, or a local one (Ollama, LM Studio). The key can stay empty for a local server. Your messages go to that server.",
 		freeGoogle: "Free to try: a Google AI Studio key needs no card. The free tier limits the text sent each minute and the requests each day, enough to try Sub-Sub, and gemini-3.1-flash-lite is the tested free model. You must be 18 or older. In the EU, the UK and Switzerland, Google does not use your prompts to improve its products; elsewhere, on the free tier, it may.",
 		freeOpenRouter: "Free to try: models whose names end in :free cost nothing, with 50 requests a day (1000 after you buy 10 dollars of credit). The providers of some free models may keep your prompts.",
 		trouble_quota: "The model provider refused the request: a limit of your account was reached. Free tiers limit the text sent each minute and the requests each day. Wait a minute and ask Sub-Sub to continue; if it happens again, the daily limit is used up (it resets the next day). Or choose another model with the model button.",
@@ -167,7 +171,7 @@ export const STRINGS = {
 		starbuckReport: "Escrever o relatório do Starbuck em {x}?",
 		approveHint: "Nada muda até selecionar Sim.",
 		zoteroChecking: "Zotero: a verificar",
-		zoteroDown: "O Zotero não está aberto",
+		zoteroDown: "O Zotero está fechado: só pesquisa",
 		items: "{n} itens",
 		toReview: "{n} por verificar",
 		toReviewHint: "Itens que o Sub-Sub etiquetou e que esperam a sua verificação (têm a etiqueta _agent). Selecione Resumo da biblioteca, ou peça ao Sub-Sub para os listar.",
@@ -202,6 +206,10 @@ export const STRINGS = {
 		save: "Guardar",
 		getKey: "Obter uma chave",
 		keyNote: "A chave fica guardada neste computador, na pasta de definições do Sub-Sub, e só é enviada a esse fornecedor.",
+		otherProvider: "Outro fornecedor (API da OpenAI)",
+		serverUrl: "Endereço do servidor",
+		modelId: "Identificador do modelo",
+		otherNote: "Um servidor que fala a API da OpenAI: o da sua instituição, ou um servidor local (Ollama, LM Studio). Num servidor local, a chave pode ficar vazia. As suas mensagens vão para esse servidor.",
 		freeGoogle: "Para experimentar sem pagar: uma chave do Google AI Studio não pede cartão. O nível gratuito limita o texto enviado por minuto e os pedidos por dia, o que chega para experimentar o Sub-Sub; o modelo gratuito testado é o gemini-3.1-flash-lite. Tem de ter 18 anos ou mais. Na UE, no Reino Unido e na Suíça, a Google não usa os seus pedidos para melhorar os produtos; noutros países, no nível gratuito, pode usar.",
 		freeOpenRouter: "Para experimentar sem pagar: os modelos cujo nome termina em :free não custam nada, com 50 pedidos por dia (1000 depois de comprar 10 dólares de crédito). Os fornecedores de alguns modelos gratuitos podem guardar os seus pedidos.",
 		trouble_quota: "O fornecedor do modelo recusou o pedido: atingiu um limite da sua conta. Os níveis gratuitos limitam o texto enviado por minuto e os pedidos por dia. Espere um minuto e peça ao Sub-Sub para continuar; se voltar a acontecer, esgotou o limite diário (renova-se no dia seguinte). Ou escolha outro modelo no botão do modelo.",
@@ -316,6 +324,7 @@ export const ACTIONS = {
 		{ cmd: "subsub", send: true, en: "Library overview", pt: "Resumo da biblioteca" },
 		{ cmd: "tag-batch 10", send: true, en: "Tag 10 items", pt: "Etiquetar 10 itens" },
 		{ cmd: "clean-tags", send: true, en: "Clean up tags", pt: "Limpar etiquetas" },
+		{ cmd: "check-library", send: true, en: "Check the library", pt: "Verificar a biblioteca" },
 		{ cmd: "import-queue", send: true, en: "Import the queue", pt: "Importar a fila" },
 		{ cmd: "history", send: true, en: "Recent changes", pt: "Alterações recentes" },
 		{ cmd: "undo", send: true, en: "Undo the last change", pt: "Desfazer a última alteração" },
@@ -341,6 +350,7 @@ export const SUGGESTIONS = {
 export const TOOLS = {
 	en: {
 		find_items: "search the library",
+		list_libraries: "list the libraries",
 		get_figure: "cut a figure out of the PDF",
 		standalone_items: "list files without a parent item",
 		set_parent_items: "give files a parent item",
@@ -402,6 +412,7 @@ export const TOOLS = {
 	},
 	pt: {
 		find_items: "pesquisar na biblioteca",
+		list_libraries: "listar as bibliotecas",
 		get_figure: "recortar uma figura do PDF",
 		standalone_items: "listar ficheiros sem item principal",
 		set_parent_items: "dar um item principal aos ficheiros",

package/web/index.html

@@ -120,6 +120,12 @@
 			<label><span data-i18n="provider">Provider</span>
 				<select id="login-provider"></select>
 			</label>
+			<label id="login-url-row" hidden><span data-i18n="serverUrl">Server address</span>
+				<input type="url" id="login-url" autocomplete="off" spellcheck="false" placeholder="https://llm.example.org/v1">
+			</label>
+			<label id="login-model-row" hidden><span data-i18n="modelId">Model id</span>
+				<input type="text" id="login-model" autocomplete="off" spellcheck="false">
+			</label>
 			<label><span data-i18n="apiKey">API key</span>
 				<input type="password" id="login-key" autocomplete="off" spellcheck="false" required>
 			</label>