@octelium/console-agent 0.1.0-b-ca02.1.1

package/dist/agent/pi/auth.js

@@ -0,0 +1,279 @@
+import * as crypto from "node:crypto";
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { badRequest, conflict, notFound } from "../../errors.js";
+const preferredSelections = {
+    anthropic: "copy_code",
+};
+const loginTTLMs = 15 * 60 * 1000;
+const maxEvents = 50;
+const maxAnswerLength = 8192;
+const uuidPattern = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
+export const toLoginEvent = (event) => {
+    switch (event.type) {
+        case "info":
+            return {
+                type: "info",
+                message: event.message,
+                links: event.links?.map((l) => ({ url: l.url, label: l.label })),
+            };
+        case "auth_url":
+            return {
+                type: "auth_url",
+                url: event.url,
+                instructions: event.instructions,
+            };
+        case "device_code":
+            return {
+                type: "device_code",
+                userCode: event.userCode,
+                verificationUri: event.verificationUri,
+                intervalSeconds: event.intervalSeconds,
+                expiresInSeconds: event.expiresInSeconds,
+            };
+        case "progress":
+            return { type: "progress", message: event.message };
+    }
+};
+export class AuthManager {
+    opts;
+    providers;
+    logins = new Map();
+    constructor(opts) {
+        this.opts = opts;
+        this.providers = opts.providers;
+    }
+    getDeviceId() {
+        const filePath = path.join(this.opts.dataDir, "device-id");
+        try {
+            const value = fs.readFileSync(filePath, "utf8").trim();
+            if (uuidPattern.test(value)) {
+                return value;
+            }
+        }
+        catch { }
+        const id = crypto.randomUUID();
+        fs.mkdirSync(this.opts.dataDir, { recursive: true, mode: 0o700 });
+        fs.writeFileSync(filePath, id, { mode: 0o600 });
+        return id;
+    }
+    async listProviders() {
+        const rt = this.opts.modelRuntime;
+        const ret = [];
+        for (const id of this.providers) {
+            const provider = rt.getProvider(id);
+            const oauth = provider?.auth.oauth;
+            if (!provider || !oauth) {
+                continue;
+            }
+            ret.push({
+                id,
+                name: oauth.name,
+                loginLabel: oauth.loginLabel,
+                subscription: !!oauth.isSubscription,
+                configured: rt.hasConfiguredAuth(id),
+                oauth: rt.isUsingOAuth(id),
+            });
+        }
+        return ret;
+    }
+    startLogin(req) {
+        const provider = this.opts.modelRuntime.getProvider(req.provider);
+        if (!this.providers.includes(req.provider) || !provider?.auth.oauth) {
+            throw badRequest(`Signing in is not supported for the provider "${req.provider}"`);
+        }
+        for (const entry of this.logins.values()) {
+            if (entry.session.provider === req.provider &&
+                entry.session.status === "pending") {
+                this.cancelLogin(entry.session.id);
+            }
+        }
+        const now = new Date().toISOString();
+        const id = crypto.randomUUID();
+        const entry = {
+            session: {
+                id,
+                provider: req.provider,
+                status: "pending",
+                events: [],
+                createdAt: now,
+                updatedAt: now,
+            },
+            controller: new AbortController(),
+            timer: setTimeout(() => this.expire(id), loginTTLMs),
+        };
+        entry.timer.unref();
+        this.logins.set(id, entry);
+        void this.runLogin(entry, req.selectModel ?? true);
+        return structuredClone(entry.session);
+    }
+    getLogin(id) {
+        const entry = this.logins.get(id);
+        return entry ? structuredClone(entry.session) : undefined;
+    }
+    answerLogin(id, promptId, value) {
+        const entry = this.logins.get(id);
+        if (!entry) {
+            throw notFound(`Login not found: ${id}`);
+        }
+        if (typeof value !== "string" || value.length > maxAnswerLength) {
+            throw badRequest('The "value" field must be a string');
+        }
+        if (entry.session.status !== "pending" ||
+            !entry.prompt ||
+            entry.prompt.id !== promptId) {
+            throw conflict("The prompt is not pending anymore");
+        }
+        entry.prompt.resolve(value);
+        return structuredClone(entry.session);
+    }
+    cancelLogin(id) {
+        const entry = this.logins.get(id);
+        if (!entry) {
+            throw notFound(`Login not found: ${id}`);
+        }
+        if (entry.session.status === "pending") {
+            entry.session.status = "cancelled";
+            this.touch(entry);
+            entry.controller.abort();
+            entry.prompt?.reject(new Error("The login was cancelled"));
+        }
+        return structuredClone(entry.session);
+    }
+    async logout(provider) {
+        if (!this.providers.includes(provider)) {
+            throw badRequest(`Signing out is not supported for the provider "${provider}"`);
+        }
+        await this.opts.modelRuntime.logout(provider);
+        await this.opts.onLogout?.(provider);
+    }
+    dispose() {
+        for (const entry of this.logins.values()) {
+            clearTimeout(entry.timer);
+            if (entry.session.status === "pending") {
+                this.cancelLogin(entry.session.id);
+            }
+        }
+        this.logins.clear();
+    }
+    touch(entry) {
+        entry.session.updatedAt = new Date().toISOString();
+    }
+    expire(id) {
+        const entry = this.logins.get(id);
+        if (!entry) {
+            return;
+        }
+        if (entry.session.status === "pending") {
+            this.cancelLogin(id);
+        }
+        this.logins.delete(id);
+    }
+    async runLogin(entry, selectModel) {
+        const { provider } = entry.session;
+        const interaction = {
+            signal: entry.controller.signal,
+            notify: (event) => {
+                entry.session.events.push(toLoginEvent(event));
+                if (entry.session.events.length > maxEvents) {
+                    entry.session.events.splice(0, entry.session.events.length - maxEvents);
+                }
+                this.touch(entry);
+            },
+            prompt: (prompt) => this.prompt(entry, prompt),
+        };
+        try {
+            await this.opts.modelRuntime.login(provider, "oauth", interaction, {
+                getDeviceId: () => this.getDeviceId(),
+            });
+            if (entry.session.status !== "pending") {
+                return;
+            }
+            if (this.opts.onLogin) {
+                try {
+                    entry.session.model = await this.opts.onLogin(provider, selectModel);
+                }
+                catch (err) {
+                    this.opts.logger.warn("Could not select a model after signing in", {
+                        provider,
+                        error: err,
+                    });
+                }
+            }
+            entry.session.status = "completed";
+            this.opts.logger.info("Signed in to the LLM provider", { provider });
+        }
+        catch (err) {
+            if (entry.session.status === "pending") {
+                entry.session.status = "failed";
+                entry.session.error = {
+                    code: "login_failed",
+                    message: err.message ?? String(err),
+                };
+                this.opts.logger.warn("Could not sign in to the LLM provider", {
+                    provider,
+                    error: err,
+                });
+            }
+        }
+        finally {
+            entry.prompt = undefined;
+            entry.session.prompt = undefined;
+            this.touch(entry);
+        }
+    }
+    prompt(entry, prompt) {
+        if (prompt.type === "select") {
+            const preferred = preferredSelections[entry.session.provider];
+            if (preferred && prompt.options.some((o) => o.id === preferred)) {
+                return Promise.resolve(preferred);
+            }
+        }
+        return new Promise((resolve, reject) => {
+            if (prompt.signal?.aborted || entry.controller.signal.aborted) {
+                reject(new Error("The prompt was cancelled"));
+                return;
+            }
+            const id = crypto.randomUUID();
+            const cleanup = () => {
+                prompt.signal?.removeEventListener("abort", onAbort);
+                if (entry.prompt?.id === id) {
+                    entry.prompt = undefined;
+                    entry.session.prompt = undefined;
+                    this.touch(entry);
+                }
+            };
+            const onAbort = () => {
+                cleanup();
+                reject(new Error("The prompt was cancelled"));
+            };
+            prompt.signal?.addEventListener("abort", onAbort, { once: true });
+            entry.prompt = {
+                id,
+                resolve: (value) => {
+                    cleanup();
+                    resolve(value);
+                },
+                reject: (err) => {
+                    cleanup();
+                    reject(err);
+                },
+            };
+            entry.session.prompt = {
+                id,
+                type: prompt.type,
+                message: prompt.message,
+                placeholder: prompt.type === "select" ? undefined : prompt.placeholder,
+                options: prompt.type === "select"
+                    ? prompt.options.map((o) => ({
+                        id: o.id,
+                        label: o.label,
+                        description: o.description,
+                    }))
+                    : undefined,
+            };
+            this.touch(entry);
+        });
+    }
+}
+//# sourceMappingURL=auth.js.map

package/dist/agent/pi/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","sourceRoot":"","sources":["../../../src/agent/pi/auth.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AACtC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAOlC,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAWjE,MAAM,mBAAmB,GAA2B;IAClD,SAAS,EAAE,WAAW;CACvB,CAAC;AAEF,MAAM,UAAU,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAClC,MAAM,SAAS,GAAG,EAAE,CAAC;AACrB,MAAM,eAAe,GAAG,IAAI,CAAC;AAE7B,MAAM,WAAW,GACf,iEAAiE,CAAC;AA2BpE,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,KAAgB,EAAc,EAAE;IAC3D,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,MAAM;YACT,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;aACjE,CAAC;QACJ,KAAK,UAAU;YACb,OAAO;gBACL,IAAI,EAAE,UAAU;gBAChB,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,YAAY,EAAE,KAAK,CAAC,YAAY;aACjC,CAAC;QACJ,KAAK,aAAa;YAChB,OAAO;gBACL,IAAI,EAAE,aAAa;gBACnB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,eAAe,EAAE,KAAK,CAAC,eAAe;gBACtC,eAAe,EAAE,KAAK,CAAC,eAAe;gBACtC,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;aACzC,CAAC;QACJ,KAAK,UAAU;YACb,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;IACxD,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,OAAO,WAAW;IACd,IAAI,CAAqB;IACzB,SAAS,CAAW;IACpB,MAAM,GAAG,IAAI,GAAG,EAAsB,CAAC;IAE/C,YAAY,IAAwB;QAClC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;IAClC,CAAC;IAED,WAAW;QACT,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC3D,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;YACvD,IAAI,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5B,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;QAEV,MAAM,EAAE,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAC/B,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAChD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC;QAClC,MAAM,GAAG,GAAmB,EAAE,CAAC;QAC/B,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAG,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YACpC,MAAM,KAAK,GAAG,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC;YACnC,IAAI,CAAC,QAAQ,IAAI,CAAC,KAAK,EAAE,CAAC;gBACxB,SAAS;YACX,CAAC;YACD,GAAG,CAAC,IAAI,CAAC;gBACP,EAAE;gBACF,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,cAAc;gBACpC,UAAU,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACpC,KAAK,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;aAC3B,CAAC,CAAC;QACL,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,UAAU,CAAC,GAAsB;QAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;YACpE,MAAM,UAAU,CACd,iDAAiD,GAAG,CAAC,QAAQ,GAAG,CACjE,CAAC;QACJ,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IACE,KAAK,CAAC,OAAO,CAAC,QAAQ,KAAK,GAAG,CAAC,QAAQ;gBACvC,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,EAClC,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,EAAE,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAe;YACxB,OAAO,EAAE;gBACP,EAAE;gBACF,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,EAAE;gBACV,SAAS,EAAE,GAAG;gBACd,SAAS,EAAE,GAAG;aACf;YACD,UAAU,EAAE,IAAI,eAAe,EAAE;YACjC,KAAK,EAAE,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC;SACrD,CAAC;QACF,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAE3B,KAAK,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,GAAG,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC;QACnD,OAAO,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,OAAO,KAAK,CAAC,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5D,CAAC;IAED,WAAW,CAAC,EAAU,EAAE,QAAgB,EAAE,KAAa;QACrD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,QAAQ,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,eAAe,EAAE,CAAC;YAChE,MAAM,UAAU,CAAC,oCAAoC,CAAC,CAAC;QACzD,CAAC;QACD,IACE,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS;YAClC,CAAC,KAAK,CAAC,MAAM;YACb,KAAK,CAAC,MAAM,CAAC,EAAE,KAAK,QAAQ,EAC5B,CAAC;YACD,MAAM,QAAQ,CAAC,mCAAmC,CAAC,CAAC;QACtD,CAAC;QACD,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC5B,OAAO,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,WAAW,CAAC,EAAU;QACpB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,QAAQ,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACvC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,WAAW,CAAC;YACnC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAClB,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACzB,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAgB;QAC3B,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvC,MAAM,UAAU,CACd,kDAAkD,QAAQ,GAAG,CAC9D,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IAED,OAAO;QACL,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC1B,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACvC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAEO,KAAK,CAAC,KAAiB;QAC7B,KAAK,CAAC,OAAO,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACrD,CAAC;IAEO,MAAM,CAAC,EAAU;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACvC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACvB,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,KAAiB,EAAE,WAAoB;QAC5D,MAAM,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC;QACnC,MAAM,WAAW,GAAoB;YACnC,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC,MAAM;YAC/B,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;gBAChB,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC/C,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,SAAS,EAAE,CAAC;oBAC5C,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CACzB,CAAC,EACD,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,SAAS,CACxC,CAAC;gBACJ,CAAC;gBACD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACpB,CAAC;YACD,MAAM,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;SAC/C,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE;gBACjE,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE;aACtC,CAAC,CAAC;YACH,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACvC,OAAO;YACT,CAAC;YACD,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACtB,IAAI,CAAC;oBACH,KAAK,CAAC,OAAO,CAAC,KAAK,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;gBACvE,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2CAA2C,EAAE;wBACjE,QAAQ;wBACR,KAAK,EAAE,GAAY;qBACpB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,WAAW,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QACvE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACvC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,QAAQ,CAAC;gBAChC,KAAK,CAAC,OAAO,CAAC,KAAK,GAAG;oBACpB,IAAI,EAAE,cAAc;oBACpB,OAAO,EAAG,GAAa,CAAC,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC;iBAC/C,CAAC;gBACF,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE;oBAC7D,QAAQ;oBACR,KAAK,EAAE,GAAY;iBACpB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;YACzB,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;YACjC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,KAAiB,EAAE,MAAkB;QAClD,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,mBAAmB,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC9D,IAAI,SAAS,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,SAAS,CAAC,EAAE,CAAC;gBAChE,OAAO,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,OAAO,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC7C,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC9D,MAAM,CAAC,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC,CAAC;gBAC9C,OAAO;YACT,CAAC;YAED,MAAM,EAAE,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBACrD,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;oBAC5B,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;oBACzB,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;oBACjC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACpB,CAAC;YACH,CAAC,CAAC;YACF,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,OAAO,EAAE,CAAC;gBACV,MAAM,CAAC,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC,CAAC;YAChD,CAAC,CAAC;YACF,MAAM,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAElE,KAAK,CAAC,MAAM,GAAG;gBACb,EAAE;gBACF,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,OAAO,EAAE,CAAC;oBACV,OAAO,CAAC,KAAK,CAAC,CAAC;gBACjB,CAAC;gBACD,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE;oBACd,OAAO,EAAE,CAAC;oBACV,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;aACF,CAAC;YACF,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG;gBACrB,EAAE;gBACF,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,WAAW,EAAE,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW;gBACtE,OAAO,EACL,MAAM,CAAC,IAAI,KAAK,QAAQ;oBACtB,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBACzB,EAAE,EAAE,CAAC,CAAC,EAAE;wBACR,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,WAAW,EAAE,CAAC,CAAC,WAAW;qBAC3B,CAAC,CAAC;oBACL,CAAC,CAAC,SAAS;aAChB,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/agent/pi/backend.d.ts

@@ -0,0 +1,77 @@
+import type { Api, Model } from "@earendil-works/pi-ai";
+import { ModelRuntime } from "@earendil-works/pi-coding-agent";
+import { type Config } from "../../config.ts";
+import type { Logger } from "../../log.ts";
+import type { APICatalog } from "../../octelium/catalog.ts";
+import type { OcteliumClient } from "../../octelium/client.ts";
+import type { ModelInfo, ModelsResponse, SetModelRequest } from "../../protocol/index.ts";
+import type { ConversationStore } from "../../store/conversations.ts";
+import type { FileStore } from "../../store/files.ts";
+import type { AgentBackend, BackendRunContext, BackendRunInput, BackendRunResult, ToolContextProvider } from "../types.ts";
+import { AuthManager } from "./auth.ts";
+export declare const PREFERRED_MODELS: Record<string, string>;
+export interface Identity {
+    user?: {
+        name?: string;
+        uid?: string;
+        displayName?: string;
+    };
+    workspace?: {
+        name?: string;
+        hostname?: string;
+    };
+}
+export interface PiBackendOptions {
+    config: Config;
+    catalog: APICatalog;
+    octelium: OcteliumClient;
+    octeliumUnavailableReason?: string;
+    runtime: ToolContextProvider;
+    store: ConversationStore;
+    files: FileStore;
+    logger: Logger;
+    skillPaths?: string[];
+    identity?: Identity;
+    modelRuntime?: ModelRuntime;
+    model?: Model<Api>;
+}
+export declare const formatBytes: (n: number) => string;
+export declare class PiBackend implements AgentBackend {
+    model?: ModelInfo;
+    thinkingLevel?: string;
+    auth?: AuthManager;
+    private opts;
+    private settings;
+    private agentDir;
+    private modelRuntime?;
+    private resolvedModel?;
+    private modelError?;
+    private resolving?;
+    private sessions;
+    private pendingSessions;
+    private systemPrompt;
+    private toolNames;
+    private builtinTools;
+    constructor(opts: PiBackendOptions);
+    get issues(): string[];
+    init(): Promise<void>;
+    private promptContext;
+    get prompt(): string;
+    private ensureModel;
+    private resolveInitialModel;
+    listModels(): Promise<ModelsResponse>;
+    setModel(req: SetModelRequest): Promise<ModelsResponse>;
+    private applyModel;
+    private syncSession;
+    private onLogin;
+    private onLogout;
+    private createTools;
+    private getSession;
+    private createSession;
+    private scheduleIdle;
+    private buildPrompt;
+    run(input: BackendRunInput, ctx: BackendRunContext): Promise<BackendRunResult>;
+    generateTitle(text: string, signal?: AbortSignal): Promise<string | undefined>;
+    forgetConversation(conversationId: string): void;
+    dispose(): Promise<void>;
+}