HomerandClaude Opus 5.5
28ff205ed0
LLeMbas CLI 2.0 is LLeMbas-server-specific (CLI rule 2). Removed: the anthropic, gemini, ollama and responses clients, context discovery and llama-swap unloading (discover.ts), the Connection schema with its dialect, auth, key_cmd, discover, unload_url and one_model_at_a_time, the model `fallback` (it only ever moved to another server), effort_map and cache, and the multi-connection refs. Kept: openai-chat, http, sse, think, effort, tokens and all of src/lembas. - connections.yaml holds the login (`type: webui`, as lembas login writes it). A custom entry from 1.x is read past with a warning that says how to log in, so an old file still starts; with several logins left, the latest is used and the others named. - A model's ref is the id the server serves it under; the older forms (`<login>/<id>`, `<login>/<model>`) still resolve and config.yaml is rewritten once. Without a login there is no model, and the error (and the first-run screen, now the sign-in alone) says `lembas login <address>`. - A login to another server signs this machine out of the previous one. - The window is the server's word (no discovery); a capacity group comes from the server's `capacity.group` alone; embeddings use the server's /v1/embeddings. - `lembas models` lists the server's models; install.sh's starter files, the README and the JSON schemas say the same. Tests: every fixture is a login now (the fake server lists models, answers other GETs with 404, and can hand one model to another script, as a server with several backends); the dialect suites and the cross-connection tests are gone with their code. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
113 lines
7.7 KiB
TypeScript
113 lines
7.7 KiB
TypeScript
// library: lembas: the account's library through the instance's /mcp, in place of this
|
|
// machine's — the tools under their own names, as a LLeMbas chat has them, with the device token.
|
|
import { afterEach, expect, test } from "bun:test"
|
|
import { mkdirSync, mkdtempSync, writeFileSync } from "node:fs"
|
|
import { tmpdir } from "node:os"
|
|
import { join } from "node:path"
|
|
import { createApp } from "../src/app.ts"
|
|
import type { AskReply } from "../src/bus/index.ts"
|
|
import { paths } from "../src/config/paths.ts"
|
|
|
|
const stops: (() => void)[] = []
|
|
afterEach(() => {
|
|
for (const s of stops.splice(0)) s()
|
|
})
|
|
|
|
function fakeLembasMcp() {
|
|
const calls: { auth: string; name?: string; args?: unknown; meta?: unknown }[] = []
|
|
const server = Bun.serve({
|
|
port: 0,
|
|
async fetch(req) {
|
|
const u = new URL(req.url)
|
|
if (u.pathname !== "/mcp") return new Response("no", { status: 404 })
|
|
if (req.method !== "POST") return new Response(null, { status: 405 })
|
|
const msg = (await req.json()) as any
|
|
const auth = req.headers.get("authorization") ?? ""
|
|
if (auth !== "Bearer lmb_lib") return new Response("no", { status: 401 })
|
|
if (msg.id === undefined) return new Response(null, { status: 202 })
|
|
const reply = (result: unknown) => Response.json({ jsonrpc: "2.0", id: msg.id, result })
|
|
if (msg.method === "initialize") return reply({ protocolVersion: msg.params.protocolVersion, capabilities: { tools: {} }, serverInfo: { name: "lembas", version: "1.18.0" } })
|
|
if (msg.method === "tools/list")
|
|
return reply({
|
|
tools: [
|
|
{ name: "memory_add", description: "Remember one fact.", inputSchema: { type: "object", properties: { content: { type: "string" } }, required: ["content"] }, annotations: { readOnlyHint: false } },
|
|
{ name: "notes_search", description: "Search notes.", inputSchema: { type: "object", properties: { query: { type: "string" } } }, annotations: { readOnlyHint: true } },
|
|
],
|
|
})
|
|
if (msg.method === "tools/call") {
|
|
// `_meta.lembas` alone: the SDK adds a progress token of its own beside it.
|
|
calls.push({ auth, name: msg.params.name, args: msg.params.arguments, ...(msg.params._meta?.lembas ? { meta: msg.params._meta.lembas } : {}) })
|
|
return reply({ content: [{ type: "text", text: `ok: ${msg.params.name}` }], isError: false })
|
|
}
|
|
return Response.json({ jsonrpc: "2.0", id: msg.id, error: { code: -32601, message: "no" } })
|
|
},
|
|
})
|
|
stops.push(() => server.stop(true))
|
|
return { base: `http://127.0.0.1:${server.port}`, calls }
|
|
}
|
|
|
|
test("the instance's library replaces this machine's, under the tools' own names", async () => {
|
|
const lembas = fakeLembasMcp()
|
|
mkdirSync(join(paths.config, "lembas"), { recursive: true })
|
|
writeFileSync(join(paths.config, "lembas.json"), JSON.stringify({ instances: { ai: { base_url: lembas.base, connection: "ai", logged_in_at: "" } } }))
|
|
writeFileSync(join(paths.config, "lembas", "ai.key"), "lmb_lib\n", { mode: 0o600 })
|
|
writeFileSync(join(paths.config, "connections.yaml"), "connections:\n f:\n type: webui\n url: http://127.0.0.1:1/v1\n models: { f/m: {} }\n", { mode: 0o600 })
|
|
writeFileSync(join(paths.config, "config.yaml"), "model: f/m\nlibrary: lembas\n")
|
|
const app = createApp({ cwd: mkdtempSync(join(tmpdir(), "ph-lib-")), store: false, asker: { ask: async (): Promise<AskReply> => ({ kind: "once" }) } })
|
|
await app.mcpReady
|
|
await Bun.sleep(50)
|
|
const names = app.engine.o.tools.map((t) => t.name)
|
|
expect(names).toContain("memory_add")
|
|
expect(names).toContain("notes_search")
|
|
// This machine's library tools are gone; notes_search is the instance's, once.
|
|
for (const local of ["memory", "note_view", "note_manage", "skills_list", "skill_manage", "knowledge_search"]) expect(names).not.toContain(local)
|
|
expect(names.filter((n) => n === "notes_search").length).toBe(1)
|
|
const add = app.engine.o.tools.find((t) => t.name === "memory_add")!
|
|
expect(add.permission({ content: "x" } as never, { ...app.engine.o.toolCtx, signal: new AbortController().signal }).class).toBe("interact")
|
|
const out = await add.run({ content: "Prefers tabs." } as never, { ...app.engine.o.toolCtx, signal: new AbortController().signal })
|
|
expect(out.output).toContain("ok: memory_add")
|
|
expect(lembas.calls).toEqual([{ auth: "Bearer lmb_lib", name: "memory_add", args: { content: "Prefers tabs." } }])
|
|
})
|
|
|
|
test("a session the client gave a library scope sends its id on every call to the instance's library; base passes through", async () => {
|
|
const lembas = fakeLembasMcp()
|
|
mkdirSync(join(paths.config, "lembas"), { recursive: true })
|
|
writeFileSync(join(paths.config, "lembas.json"), JSON.stringify({ instances: { ai: { base_url: lembas.base, connection: "ai", logged_in_at: "" } } }))
|
|
writeFileSync(join(paths.config, "lembas", "ai.key"), "lmb_lib\n", { mode: 0o600 })
|
|
writeFileSync(join(paths.config, "connections.yaml"), "connections:\n f:\n type: webui\n url: http://127.0.0.1:1/v1\n models: { f/m: {} }\n", { mode: 0o600 })
|
|
writeFileSync(join(paths.config, "config.yaml"), "model: f/m\nlibrary: lembas\n")
|
|
const app = createApp({ cwd: mkdtempSync(join(tmpdir(), "ph-lib-")), asker: { ask: async (): Promise<AskReply> => ({ kind: "once" }) } })
|
|
stops.push(() => void app.close())
|
|
await app.mcpReady
|
|
await Bun.sleep(50)
|
|
const search = app.engine.o.tools.find((t) => t.name === "notes_search")!
|
|
const ctx = () => ({ ...app.engine.o.toolCtx, signal: new AbortController().signal })
|
|
// Nothing said about the session's library: no scope, as before.
|
|
await search.run({ query: "a" } as never, ctx())
|
|
expect(lembas.calls[0]).toEqual({ auth: "Bearer lmb_lib", name: "notes_search", args: { query: "a" } })
|
|
// The client named the bases: the session's id rides on the call as _meta.lembas.scope, and
|
|
// the instance answers within that chat's scope (beside the SDK's own progress token); `base`
|
|
// is the model's word, passed untouched.
|
|
app.setContext({ library: { bases: [{ id: "kb1", name: "Manuals" }] } }, "ai")
|
|
await search.run({ query: "b", base: "Manuals" } as never, ctx())
|
|
expect(lembas.calls[1]).toEqual({ auth: "Bearer lmb_lib", name: "notes_search", args: { query: "b", base: "Manuals" }, meta: { scope: app.engine.sessionId } })
|
|
// The prompt names what the session may search.
|
|
expect(app.engine.o.system("manual", app.engine.model)).toContain("Knowledge searches here cover only: Manuals. Finding nothing there means nothing is there, not that the library is empty.")
|
|
// Said again with no bases: the account's whole library — nothing named to the model, while the
|
|
// scope still rides (the instance answers in the chat's data group).
|
|
app.setContext({ library: { bases: [] } }, "ai")
|
|
await search.run({ query: "c" } as never, ctx())
|
|
expect(lembas.calls[2]!.meta).toEqual({ scope: app.engine.sessionId })
|
|
expect(app.engine.o.system("manual", app.engine.model)).not.toContain("Knowledge searches here cover only")
|
|
expect(app.engine.o.system("manual", app.engine.model)).toContain("search it with knowledge_search")
|
|
})
|
|
|
|
test("without a login it says so and keeps this machine's library", () => {
|
|
writeFileSync(join(paths.config, "lembas.json"), JSON.stringify({ instances: {} }))
|
|
writeFileSync(join(paths.config, "connections.yaml"), "connections:\n f:\n type: webui\n url: http://127.0.0.1:1/v1\n models: { f/m: {} }\n", { mode: 0o600 })
|
|
writeFileSync(join(paths.config, "config.yaml"), "model: f/m\nlibrary: lembas\n")
|
|
const app = createApp({ cwd: mkdtempSync(join(tmpdir(), "ph-lib-")), store: false, asker: { ask: async (): Promise<AskReply> => ({ kind: "once" }) } })
|
|
expect(app.loaded.warnings.join(" ")).toContain("not logged in")
|
|
expect(app.engine.o.tools.map((t) => t.name)).toContain("memory")
|
|
})
|