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>
52 lines
2.4 KiB
TypeScript
52 lines
2.4 KiB
TypeScript
// Other models: the list the model is given (with notes), a subagent on another model, and a
|
|
// fallback when a model's server cannot be reached at all.
|
|
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 { Event } from "../src/bus/index.ts"
|
|
import { paths } from "../src/config/paths.ts"
|
|
import { delta, fakeProvider, toolCall, type Fake } from "./fake-provider.ts"
|
|
|
|
let fake: Fake | undefined
|
|
afterEach(() => fake?.stop())
|
|
|
|
function setup(script: Parameters<typeof fakeProvider>[0], connections: (url: string) => string) {
|
|
fake = fakeProvider(script)
|
|
mkdirSync(paths.config, { recursive: true })
|
|
writeFileSync(join(paths.config, "connections.yaml"), connections(fake.url), { mode: 0o600 })
|
|
writeFileSync(join(paths.config, "config.yaml"), "model: f/main\ntitles: prompt\n")
|
|
const app = createApp({ cwd: mkdtempSync(join(tmpdir(), "ph-models-")), store: false, snapshots: false, asker: { ask: async () => ({ kind: "once" }) } })
|
|
const events: Event[] = []
|
|
app.bus.on((e) => events.push(e))
|
|
return { app, events }
|
|
}
|
|
|
|
const two = (url: string) =>
|
|
`connections:\n f:\n type: webui\n url: ${url}\n models:\n f/main: {}\n f/big: { name: Big One, notes: slow but careful; best at reviews }\n`
|
|
|
|
test("the model is told which other models there are, with their notes", async () => {
|
|
const { app } = setup([{ chunks: [delta({ content: "ok" }, "stop")] }], two)
|
|
await app.engine.prompt("hi")
|
|
const system = fake!.requests[0]!.messages[0].content as string
|
|
expect(system).toContain("- f/big (Big One) — slow but careful; best at reviews")
|
|
expect(system).not.toContain("- f/main")
|
|
})
|
|
|
|
test("task runs a subagent on the model asked for", async () => {
|
|
const { app } = setup(
|
|
[
|
|
{ chunks: [toolCall(0, "t1", "task", JSON.stringify({ description: "second opinion", prompt: "Is 2+2 4?", model: "f/big" }))] },
|
|
{ chunks: [delta({ content: "Yes." }, "stop")] },
|
|
{ chunks: [delta({ content: "Big says yes." }, "stop")] },
|
|
],
|
|
two,
|
|
)
|
|
expect(await app.engine.prompt("ask the big one")).toBe("stop")
|
|
expect(fake!.requests.map((r) => r.model)).toEqual(["f/main", "f/big", "f/main"])
|
|
const result = app.engine.messages.find((m) => m.role === "tool")
|
|
expect(result && result.role === "tool" && result.content).toContain("Yes.")
|
|
})
|
|
|