@guava-ai/guava-sdk 0.22.0 → 0.24.0

package/dist/src/health.d.ts

@@ -0,0 +1,35 @@
+import { type Logger } from "./logging.ts";
+export type HealthState = "starting" | "live" | "stopped";
+export declare class HealthContext {
+    private _state;
+    setState(newState: HealthState): void;
+    ready(): void;
+    stopped(): void;
+    isLive(): boolean;
+}
+export declare class MultiHealthContext {
+    private _ctxs;
+    createCtx(): HealthContext;
+    isLive(): boolean;
+}
+export interface HealthServerOptions {
+    host?: string;
+    port?: number;
+    logger?: Logger;
+}
+export declare class HealthServer implements AsyncDisposable {
+    private readonly _server;
+    private readonly _logger;
+    private _stopping;
+    private constructor();
+    static start(healthCtx: HealthContext | MultiHealthContext, options?: HealthServerOptions): Promise<HealthServer>;
+    stop(): Promise<void>;
+    [Symbol.asyncDispose](): Promise<void>;
+}
+/**
+ * Returns an AsyncDisposable that, when the GUAVA_HEALTH_SERVER env var is
+ * truthy ("yes", "true", "on"), starts an HTTP health check server exposing
+ * `/live` on port 4828. Otherwise returns a no-op disposable. Intended for use
+ * with `await using`.
+ */
+export declare function getHealthServer(healthCtx: HealthContext | MultiHealthContext): Promise<AsyncDisposable>;

package/dist/src/health.js

@@ -0,0 +1,99 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HealthServer = exports.MultiHealthContext = exports.HealthContext = void 0;
+exports.getHealthServer = getHealthServer;
+const node_http_1 = require("node:http");
+const logging_ts_1 = require("./logging.js");
+class HealthContext {
+    _state = "starting";
+    setState(newState) {
+        this._state = newState;
+    }
+    ready() {
+        this.setState("live");
+    }
+    stopped() {
+        this.setState("stopped");
+    }
+    isLive() {
+        return this._state === "live";
+    }
+}
+exports.HealthContext = HealthContext;
+class MultiHealthContext {
+    _ctxs = [];
+    createCtx() {
+        const ctx = new HealthContext();
+        this._ctxs.push(ctx);
+        return ctx;
+    }
+    isLive() {
+        return this._ctxs.length > 0 && this._ctxs.every((ctx) => ctx.isLive());
+    }
+}
+exports.MultiHealthContext = MultiHealthContext;
+class HealthServer {
+    _server;
+    _logger;
+    _stopping = null;
+    constructor(server, logger) {
+        this._server = server;
+        this._logger = logger;
+    }
+    static async start(healthCtx, options = {}) {
+        const host = options.host ?? "0.0.0.0";
+        const port = options.port ?? 4828;
+        const logger = options.logger ?? (0, logging_ts_1.getDefaultLogger)();
+        const server = (0, node_http_1.createServer)((req, res) => {
+            if (req.method === "GET" && req.url === "/live") {
+                res.statusCode = healthCtx.isLive() ? 200 : 503;
+            }
+            else {
+                res.statusCode = 404;
+            }
+            res.end();
+        });
+        await new Promise((resolve, reject) => {
+            server.once("error", reject);
+            server.listen(port, host, () => {
+                server.off("error", reject);
+                resolve();
+            });
+        });
+        logger.info("Health check server listening on http://%s:%d/live", host, port);
+        return new HealthServer(server, logger);
+    }
+    stop() {
+        if (this._stopping)
+            return this._stopping;
+        this._logger.info("Health check server shutting down.");
+        this._stopping = new Promise((resolve, reject) => {
+            this._server.closeAllConnections();
+            this._server.close((err) => (err ? reject(err) : resolve()));
+        });
+        return this._stopping;
+    }
+    async [Symbol.asyncDispose]() {
+        await this.stop();
+    }
+}
+exports.HealthServer = HealthServer;
+const NOOP_DISPOSABLE = {
+    async [Symbol.asyncDispose]() { },
+};
+function isEnabled() {
+    const value = (process.env.GUAVA_HEALTH_SERVER ?? "false").toLowerCase().trim();
+    return value === "yes" || value === "true" || value === "on";
+}
+/**
+ * Returns an AsyncDisposable that, when the GUAVA_HEALTH_SERVER env var is
+ * truthy ("yes", "true", "on"), starts an HTTP health check server exposing
+ * `/live` on port 4828. Otherwise returns a no-op disposable. Intended for use
+ * with `await using`.
+ */
+async function getHealthServer(healthCtx) {
+    if (!isEnabled())
+        return NOOP_DISPOSABLE;
+    return HealthServer.start(healthCtx);
+}
+//# sourceMappingURL=health.js.map

package/dist/src/health.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"health.js","sourceRoot":"","sources":["../../src/health.ts"],"names":[],"mappings":";;;AAkHA,0CAKC;AAvHD,yCAAsD;AACtD,6CAA6D;AAI7D,MAAa,aAAa;IAChB,MAAM,GAAgB,UAAU,CAAC;IAEzC,QAAQ,CAAC,QAAqB;QAC5B,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;IACzB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACxB,CAAC;IAED,OAAO;QACL,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC3B,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,MAAM,KAAK,MAAM,CAAC;IAChC,CAAC;CACF;AAlBD,sCAkBC;AAED,MAAa,kBAAkB;IACrB,KAAK,GAAoB,EAAE,CAAC;IAEpC,SAAS;QACP,MAAM,GAAG,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1E,CAAC;CACF;AAZD,gDAYC;AAQD,MAAa,YAAY;IACN,OAAO,CAAS;IAChB,OAAO,CAAS;IACzB,SAAS,GAAyB,IAAI,CAAC;IAE/C,YAAoB,MAAc,EAAE,MAAc;QAChD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,KAAK,CAChB,SAA6C,EAC7C,UAA+B,EAAE;QAEjC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,SAAS,CAAC;QACvC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC;QAClC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,IAAA,6BAAgB,GAAE,CAAC;QAEpD,MAAM,MAAM,GAAG,IAAA,wBAAY,EAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACvC,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,GAAG,CAAC,GAAG,KAAK,OAAO,EAAE,CAAC;gBAChD,GAAG,CAAC,UAAU,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;YAClD,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,UAAU,GAAG,GAAG,CAAC;YACvB,CAAC;YACD,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1C,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC7B,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;gBAC7B,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBAC5B,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,oDAAoD,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAC9E,OAAO,IAAI,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC,SAAS,CAAC;QAC1C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;QACxD,IAAI,CAAC,SAAS,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrD,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;YACnC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC;QACzB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;IACpB,CAAC;CACF;AApDD,oCAoDC;AAED,MAAM,eAAe,GAAoB;IACvC,KAAK,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,KAAI,CAAC;CACjC,CAAC;AAEF,SAAS,SAAS;IAChB,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IAChF,OAAO,KAAK,KAAK,KAAK,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,IAAI,CAAC;AAC/D,CAAC;AAED;;;;;GAKG;AACI,KAAK,UAAU,eAAe,CACnC,SAA6C;IAE7C,IAAI,CAAC,SAAS,EAAE;QAAE,OAAO,eAAe,CAAC;IACzC,OAAO,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;AACvC,CAAC"}

package/dist/src/helpers/index.d.ts

@@ -0,0 +1 @@
+export { DatetimeFilter, DocumentQA, IntentRecognizer } from "./llm.ts";

package/dist/src/helpers/index.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IntentRecognizer = exports.DocumentQA = exports.DatetimeFilter = void 0;
+/* This file must not include the OpenAI helpers. */
+var llm_ts_1 = require("./llm.js");
+Object.defineProperty(exports, "DatetimeFilter", { enumerable: true, get: function () { return llm_ts_1.DatetimeFilter; } });
+Object.defineProperty(exports, "DocumentQA", { enumerable: true, get: function () { return llm_ts_1.DocumentQA; } });
+Object.defineProperty(exports, "IntentRecognizer", { enumerable: true, get: function () { return llm_ts_1.IntentRecognizer; } });
+//# sourceMappingURL=index.js.map

package/dist/src/helpers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/helpers/index.ts"],"names":[],"mappings":";;;AAAA,oDAAoD;AACpD,mCAAwE;AAA/D,wGAAA,cAAc,OAAA;AAAE,oGAAA,UAAU,OAAA;AAAE,0GAAA,gBAAgB,OAAA"}

package/dist/src/helpers/llm.d.ts

@@ -1,2 +1,34 @@
-import type { Client } from "../client.ts";
+import type { SuggestedAction } from "../agent.ts";
+import { Client } from "../client.ts";
 export declare function _generate(client: Client, prompt: string, jsonSchema?: object): Promise<string>;
+export declare class DocumentQA {
+    private readonly _rag;
+    private readonly _instructions;
+    private readonly _initialized;
+    constructor({ documents, ids, instructions, namespace, }?: {
+        documents?: string | string[];
+        ids?: string[];
+        instructions?: string;
+        namespace?: string;
+    });
+    upsertDocument(key: string, text: string): Promise<void>;
+    addDocument(text: string): Promise<void>;
+    deleteDocument(key: string): Promise<void>;
+    clear(): Promise<void>;
+    ask(question: string): Promise<string>;
+}
+export declare class DatetimeFilter {
+    private readonly _client;
+    private readonly _slotsStr;
+    constructor(sourceList: string[]);
+    filter(query: string, { maxResults }?: {
+        maxResults?: number;
+    }): Promise<[string[], string[]]>;
+}
+export declare class IntentRecognizer {
+    private readonly _client;
+    private readonly _intentChoices;
+    private readonly _schema;
+    constructor(intentChoices: string[] | Record<string, string>);
+    classify(intent: string): Promise<SuggestedAction[] | null>;
+}

package/dist/src/helpers/llm.js

@@ -1,7 +1,45 @@
 "use strict";
+var __esDecorate = (this && this.__esDecorate) || function (ctor, descriptorIn, decorators, contextIn, initializers, extraInitializers) {
+    function accept(f) { if (f !== void 0 && typeof f !== "function") throw new TypeError("Function expected"); return f; }
+    var kind = contextIn.kind, key = kind === "getter" ? "get" : kind === "setter" ? "set" : "value";
+    var target = !descriptorIn && ctor ? contextIn["static"] ? ctor : ctor.prototype : null;
+    var descriptor = descriptorIn || (target ? Object.getOwnPropertyDescriptor(target, contextIn.name) : {});
+    var _, done = false;
+    for (var i = decorators.length - 1; i >= 0; i--) {
+        var context = {};
+        for (var p in contextIn) context[p] = p === "access" ? {} : contextIn[p];
+        for (var p in contextIn.access) context.access[p] = contextIn.access[p];
+        context.addInitializer = function (f) { if (done) throw new TypeError("Cannot add initializers after decoration has completed"); extraInitializers.push(accept(f || null)); };
+        var result = (0, decorators[i])(kind === "accessor" ? { get: descriptor.get, set: descriptor.set } : descriptor[key], context);
+        if (kind === "accessor") {
+            if (result === void 0) continue;
+            if (result === null || typeof result !== "object") throw new TypeError("Object expected");
+            if (_ = accept(result.get)) descriptor.get = _;
+            if (_ = accept(result.set)) descriptor.set = _;
+            if (_ = accept(result.init)) initializers.unshift(_);
+        }
+        else if (_ = accept(result)) {
+            if (kind === "field") initializers.unshift(_);
+            else descriptor[key] = _;
+        }
+    }
+    if (target) Object.defineProperty(target, contextIn.name, descriptor);
+    done = true;
+};
+var __runInitializers = (this && this.__runInitializers) || function (thisArg, initializers, value) {
+    var useValue = arguments.length > 2;
+    for (var i = 0; i < initializers.length; i++) {
+        value = useValue ? initializers[i].call(thisArg, value) : initializers[i].call(thisArg);
+    }
+    return useValue ? value : void 0;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.IntentRecognizer = exports.DatetimeFilter = exports.DocumentQA = void 0;
 exports._generate = _generate;
+const client_ts_1 = require("../client.js");
+const telemetry_ts_1 = require("../telemetry.js");
 const utils_ts_1 = require("../utils.js");
+const server_rag_ts_1 = require("./server-rag.js");
 async function _generate(client, prompt, jsonSchema) {
     const url = new URL("v1/llm/generate", client.getHttpBase());
     const body = { prompt };
@@ -11,7 +49,189 @@ async function _generate(client, prompt, jsonSchema) {
         method: "POST",
         headers: { ...(await client.headers()), "Content-Type": "application/json" },
         body: JSON.stringify(body),
+        signal: AbortSignal.timeout(60_000),
     });
     return (await response.json()).text;
 }
+let DocumentQA = (() => {
+    let _classDecorators = [telemetry_ts_1.telemetryClient.trackClass()];
+    let _classDescriptor;
+    let _classExtraInitializers = [];
+    let _classThis;
+    var DocumentQA = class {
+        static { _classThis = this; }
+        static {
+            const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(null) : void 0;
+            __esDecorate(null, _classDescriptor = { value: _classThis }, _classDecorators, { kind: "class", name: _classThis.name, metadata: _metadata }, null, _classExtraInitializers);
+            DocumentQA = _classThis = _classDescriptor.value;
+            if (_metadata) Object.defineProperty(_classThis, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
+            __runInitializers(_classThis, _classExtraInitializers);
+        }
+        _rag;
+        _instructions;
+        _initialized;
+        constructor({ documents, ids, instructions, namespace, } = {}) {
+            this._rag = new server_rag_ts_1.ServerRAG(new client_ts_1.Client(), { namespace });
+            this._instructions = instructions;
+            if (documents !== undefined) {
+                const docs = typeof documents === "string" ? [documents] : documents;
+                this._initialized = this._rag.reconcile(docs, ids);
+            }
+            else {
+                this._initialized = Promise.resolve();
+            }
+        }
+        async upsertDocument(key, text) {
+            await this._initialized;
+            await this._rag.upsertDocument(key, text);
+        }
+        async addDocument(text) {
+            await this._initialized;
+            await this._rag.addDocument(text);
+        }
+        async deleteDocument(key) {
+            await this._initialized;
+            await this._rag.removeDocument(key);
+        }
+        async clear() {
+            await this._initialized;
+            await this._rag.clear();
+        }
+        async ask(question) {
+            await this._initialized;
+            const trackedKeys = this._rag.trackedKeys;
+            return this._rag.ask(question, {
+                documentKeys: trackedKeys.size > 0 ? [...trackedKeys] : undefined,
+                instructions: this._instructions,
+            });
+        }
+    };
+    return DocumentQA = _classThis;
+})();
+exports.DocumentQA = DocumentQA;
+const _filterSchema = {
+    type: "object",
+    properties: {
+        matching_appointments: { type: "array", items: { type: "string" } },
+        other_appointments: { type: "array", items: { type: "string" } },
+    },
+    required: ["matching_appointments", "other_appointments"],
+    additionalProperties: false,
+};
+let DatetimeFilter = (() => {
+    let _classDecorators = [telemetry_ts_1.telemetryClient.trackClass()];
+    let _classDescriptor;
+    let _classExtraInitializers = [];
+    let _classThis;
+    var DatetimeFilter = class {
+        static { _classThis = this; }
+        static {
+            const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(null) : void 0;
+            __esDecorate(null, _classDescriptor = { value: _classThis }, _classDecorators, { kind: "class", name: _classThis.name, metadata: _metadata }, null, _classExtraInitializers);
+            DatetimeFilter = _classThis = _classDescriptor.value;
+            if (_metadata) Object.defineProperty(_classThis, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
+            __runInitializers(_classThis, _classExtraInitializers);
+        }
+        _client;
+        _slotsStr;
+        constructor(sourceList) {
+            this._client = new client_ts_1.Client();
+            this._slotsStr = sourceList.join("\n");
+        }
+        async filter(query, { maxResults = 5 } = {}) {
+            const today = new Date().toLocaleDateString("en-US", {
+                month: "long",
+                day: "numeric",
+                year: "numeric",
+            });
+            const prompt = `Return datetime slots from the list that match the query.
+If none match, return close alternatives in other_appointments instead.
+Never return datetimes that are not in the list.
+
+Query: <query>${query}</query>
+Today's Date: ${today}
+Available slots:
+${this._slotsStr}
+
+Return at most ${maxResults} items per list.`;
+            const text = await _generate(this._client, prompt, _filterSchema);
+            const result = JSON.parse(text);
+            return [
+                result.matching_appointments.slice(0, maxResults),
+                result.other_appointments.slice(0, maxResults),
+            ];
+        }
+    };
+    return DatetimeFilter = _classThis;
+})();
+exports.DatetimeFilter = DatetimeFilter;
+let IntentRecognizer = (() => {
+    let _classDecorators = [telemetry_ts_1.telemetryClient.trackClass()];
+    let _classDescriptor;
+    let _classExtraInitializers = [];
+    let _classThis;
+    var IntentRecognizer = class {
+        static { _classThis = this; }
+        static {
+            const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(null) : void 0;
+            __esDecorate(null, _classDescriptor = { value: _classThis }, _classDecorators, { kind: "class", name: _classThis.name, metadata: _metadata }, null, _classExtraInitializers);
+            IntentRecognizer = _classThis = _classDescriptor.value;
+            if (_metadata) Object.defineProperty(_classThis, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
+            __runInitializers(_classThis, _classExtraInitializers);
+        }
+        _client;
+        _intentChoices;
+        _schema;
+        constructor(intentChoices) {
+            this._client = new client_ts_1.Client();
+            this._intentChoices = intentChoices;
+            const choiceList = Array.isArray(intentChoices) ? intentChoices : Object.keys(intentChoices);
+            this._schema = {
+                type: "object",
+                title: "ChoiceModel",
+                properties: {
+                    possible_matches: {
+                        type: "array",
+                        items: { type: "string", enum: choiceList },
+                        description: "Choices that could match the caller's intent, ordered by likelihood. Include all plausible matches.",
+                    },
+                },
+                required: ["possible_matches"],
+                additionalProperties: false,
+            };
+        }
+        async classify(intent) {
+            const choiceList = Array.isArray(this._intentChoices)
+                ? this._intentChoices
+                : Object.keys(this._intentChoices);
+            let prompt = `Classify the intent below into the most appropriate choice(s) from the list.
+
+Intent: <intent>${intent}</intent>
+Available Choices: ${JSON.stringify(choiceList)}
+
+Rules:
+- Default to returning a single choice — the one that best matches the intent.
+- Only return additional choices when the intent is genuinely ambiguous: a reasonable person reading it would be unable to decide which category it belongs to. Thematic overlap or partial relevance is NOT enough — do not include weakly or tangentially related choices.
+- Order matches by likelihood (most likely first).
+- If no choice plausibly matches, return an empty list.`;
+            if (!Array.isArray(this._intentChoices)) {
+                const descriptionString = Object.entries(this._intentChoices)
+                    .map(([key, val]) => `${key}: ${val}`)
+                    .join("\n  ");
+                prompt += `\n\nDetailed descriptions of each choice:\n  ${descriptionString}`;
+            }
+            const text = await _generate(this._client, prompt, this._schema);
+            const keys = JSON.parse(text).possible_matches;
+            if (!keys.length)
+                return null;
+            if (!Array.isArray(this._intentChoices)) {
+                const choices = this._intentChoices;
+                return keys.map((k) => ({ key: k, description: choices[k] }));
+            }
+            return keys.map((k) => ({ key: k }));
+        }
+    };
+    return IntentRecognizer = _classThis;
+})();
+exports.IntentRecognizer = IntentRecognizer;
 //# sourceMappingURL=llm.js.map

package/dist/src/helpers/llm.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm.js","sourceRoot":"","sources":["../../../src/helpers/llm.ts"],"names":[],"mappings":";;AAGA,8BAgBC;AAlBD,0CAA2C;AAEpC,KAAK,UAAU,SAAS,CAC7B,MAAc,EACd,MAAc,EACd,UAAmB;IAEnB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,iBAAiB,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;IAC7D,MAAM,IAAI,GAA4B,EAAE,MAAM,EAAE,CAAC;IACjD,IAAI,UAAU,KAAK,SAAS;QAAE,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAE5D,MAAM,QAAQ,GAAG,MAAM,IAAA,uBAAY,EAAC,GAAG,EAAE;QACvC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,GAAG,CAAC,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC5E,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;IAEH,OAAQ,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAsB,CAAC,IAAI,CAAC;AAC5D,CAAC"}
+{"version":3,"file":"llm.js","sourceRoot":"","sources":["../../../src/helpers/llm.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAMA,8BAiBC;AAtBD,4CAAsC;AACtC,kDAAkD;AAClD,0CAA2C;AAC3C,mDAA4C;AAErC,KAAK,UAAU,SAAS,CAC7B,MAAc,EACd,MAAc,EACd,UAAmB;IAEnB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,iBAAiB,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;IAC7D,MAAM,IAAI,GAA4B,EAAE,MAAM,EAAE,CAAC;IACjD,IAAI,UAAU,KAAK,SAAS;QAAE,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;IAE5D,MAAM,QAAQ,GAAG,MAAM,IAAA,uBAAY,EAAC,GAAG,EAAE;QACvC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,GAAG,CAAC,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC5E,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;QAC1B,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;KACpC,CAAC,CAAC;IAEH,OAAQ,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAsB,CAAC,IAAI,CAAC;AAC5D,CAAC;IAGY,UAAU;4BADtB,8BAAe,CAAC,UAAU,EAAE;;;;;;;;YAC7B,6KAuDC;;;YAvDY,uDAAU;;QACJ,IAAI,CAAY;QAChB,aAAa,CAAqB;QAClC,YAAY,CAAgB;QAE7C,YAAY,EACV,SAAS,EACT,GAAG,EACH,YAAY,EACZ,SAAS,MAMP,EAAE;YACJ,IAAI,CAAC,IAAI,GAAG,IAAI,yBAAS,CAAC,IAAI,kBAAM,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;YAElC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM,IAAI,GAAG,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACrE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACrD,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;YACxC,CAAC;QACH,CAAC;QAED,KAAK,CAAC,cAAc,CAAC,GAAW,EAAE,IAAY;YAC5C,MAAM,IAAI,CAAC,YAAY,CAAC;YACxB,MAAM,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC5C,CAAC;QAED,KAAK,CAAC,WAAW,CAAC,IAAY;YAC5B,MAAM,IAAI,CAAC,YAAY,CAAC;YACxB,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;QAED,KAAK,CAAC,cAAc,CAAC,GAAW;YAC9B,MAAM,IAAI,CAAC,YAAY,CAAC;YACxB,MAAM,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC;QAED,KAAK,CAAC,KAAK;YACT,MAAM,IAAI,CAAC,YAAY,CAAC;YACxB,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;QAED,KAAK,CAAC,GAAG,CAAC,QAAgB;YACxB,MAAM,IAAI,CAAC,YAAY,CAAC;YACxB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC;YAC1C,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE;gBAC7B,YAAY,EAAE,WAAW,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;gBACjE,YAAY,EAAE,IAAI,CAAC,aAAa;aACjC,CAAC,CAAC;QACL,CAAC;;;;AAtDU,gCAAU;AAyDvB,MAAM,aAAa,GAAG;IACpB,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,qBAAqB,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;QACnE,kBAAkB,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;KACjE;IACD,QAAQ,EAAE,CAAC,uBAAuB,EAAE,oBAAoB,CAAC;IACzD,oBAAoB,EAAE,KAAK;CAC5B,CAAC;IAGW,cAAc;4BAD1B,8BAAe,CAAC,UAAU,EAAE;;;;;;;;YAC7B,6KAwCC;;;YAxCY,uDAAc;;QACR,OAAO,CAAS;QAChB,SAAS,CAAS;QAEnC,YAAY,UAAoB;YAC9B,IAAI,CAAC,OAAO,GAAG,IAAI,kBAAM,EAAE,CAAC;YAC5B,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC;QAED,KAAK,CAAC,MAAM,CACV,KAAa,EACb,EAAE,UAAU,GAAG,CAAC,KAA8B,EAAE;YAEhD,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC,kBAAkB,CAAC,OAAO,EAAE;gBACnD,KAAK,EAAE,MAAM;gBACb,GAAG,EAAE,SAAS;gBACd,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG;;;;gBAIH,KAAK;gBACL,KAAK;;EAEnB,IAAI,CAAC,SAAS;;iBAEC,UAAU,kBAAkB,CAAC;YAE1C,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;YAClE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAG7B,CAAC;YACF,OAAO;gBACL,MAAM,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC;gBACjD,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC;aAC/C,CAAC;QACJ,CAAC;;;;AAvCU,wCAAc;IA2Cd,gBAAgB;4BAD5B,8BAAe,CAAC,UAAU,EAAE;;;;;;;;YAC7B,6KA2DC;;;YA3DY,uDAAgB;;QACV,OAAO,CAAS;QAChB,cAAc,CAAoC;QAClD,OAAO,CAAS;QAEjC,YAAY,aAAgD;YAC1D,IAAI,CAAC,OAAO,GAAG,IAAI,kBAAM,EAAE,CAAC;YAC5B,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;YACpC,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC7F,IAAI,CAAC,OAAO,GAAG;gBACb,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE,aAAa;gBACpB,UAAU,EAAE;oBACV,gBAAgB,EAAE;wBAChB,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE;wBAC3C,WAAW,EACT,qGAAqG;qBACxG;iBACF;gBACD,QAAQ,EAAE,CAAC,kBAAkB,CAAC;gBAC9B,oBAAoB,EAAE,KAAK;aAC5B,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,QAAQ,CAAC,MAAc;YAC3B,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC;gBACnD,CAAC,CAAC,IAAI,CAAC,cAAc;gBACrB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAErC,IAAI,MAAM,GAAG;;kBAEC,MAAM;qBACH,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;;;;;;wDAMS,CAAC;YAErD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;gBACxC,MAAM,iBAAiB,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC;qBAC1D,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,GAAG,EAAE,CAAC;qBACrC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAChB,MAAM,IAAI,gDAAgD,iBAAiB,EAAE,CAAC;YAChF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACjE,MAAM,IAAI,GAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAoC,CAAC,gBAAgB,CAAC;YAEnF,IAAI,CAAC,IAAI,CAAC,MAAM;gBAAE,OAAO,IAAI,CAAC;YAE9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;gBACxC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC;gBACpC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAChE,CAAC;YACD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACvC,CAAC;;;;AA1DU,4CAAgB"}

package/dist/src/helpers/server-rag.d.ts

@@ -0,0 +1,25 @@
+import type { Client } from "../client.ts";
+export declare function _contentKey(text: string): string;
+export declare function _prefixedKey(namespace: string | undefined, key: string): string;
+export declare class ServerRAG {
+    private readonly _client;
+    private readonly _namespace;
+    readonly trackedKeys: Set<string>;
+    constructor(client: Client, { namespace }?: {
+        namespace?: string;
+    });
+    uploadDocument(key: string, text: string): Promise<void>;
+    deleteDocument(key: string): Promise<void>;
+    listDocuments(): Promise<{
+        key: string;
+    }[]>;
+    ask(question: string, { documentKeys, instructions }?: {
+        documentKeys?: string[];
+        instructions?: string;
+    }): Promise<string>;
+    reconcile(documents: string[], ids?: string[]): Promise<void>;
+    upsertDocument(key: string, text: string): Promise<void>;
+    addDocument(text: string): Promise<void>;
+    removeDocument(key: string): Promise<void>;
+    clear(): Promise<void>;
+}

package/dist/src/helpers/server-rag.js

@@ -0,0 +1,119 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ServerRAG = void 0;
+exports._contentKey = _contentKey;
+exports._prefixedKey = _prefixedKey;
+const node_crypto_1 = require("node:crypto");
+const utils_ts_1 = require("../utils.js");
+function _contentKey(text) {
+    return (0, node_crypto_1.createHash)("sha256").update(text).digest("hex").slice(0, 16);
+}
+function _prefixedKey(namespace, key) {
+    return namespace ? `${namespace}.${key}` : key;
+}
+class ServerRAG {
+    _client;
+    _namespace;
+    trackedKeys = new Set();
+    constructor(client, { namespace } = {}) {
+        this._client = client;
+        this._namespace = namespace;
+    }
+    async uploadDocument(key, text) {
+        const url = new URL("v1/rag/documents", this._client.getHttpBase());
+        await (0, utils_ts_1.fetchOrThrow)(url, {
+            method: "POST",
+            headers: { ...(await this._client.headers()), "Content-Type": "application/json" },
+            body: JSON.stringify({ key, text }),
+            signal: AbortSignal.timeout(60_000),
+        });
+    }
+    async deleteDocument(key) {
+        const url = new URL(`v1/rag/documents/${key}`, this._client.getHttpBase());
+        await (0, utils_ts_1.fetchOrThrow)(url, {
+            method: "DELETE",
+            headers: await this._client.headers(),
+            signal: AbortSignal.timeout(30_000),
+        });
+    }
+    async listDocuments() {
+        const url = new URL("v1/rag/documents", this._client.getHttpBase());
+        const response = await (0, utils_ts_1.fetchOrThrow)(url, {
+            headers: await this._client.headers(),
+            signal: AbortSignal.timeout(30_000),
+        });
+        return (await response.json());
+    }
+    async ask(question, { documentKeys, instructions } = {}) {
+        const url = new URL("v1/rag/ask", this._client.getHttpBase());
+        const payload = { question };
+        if (documentKeys !== undefined)
+            payload.document_keys = documentKeys;
+        if (instructions !== undefined)
+            payload.instructions = instructions;
+        const response = await (0, utils_ts_1.fetchOrThrow)(url, {
+            method: "POST",
+            headers: { ...(await this._client.headers()), "Content-Type": "application/json" },
+            body: JSON.stringify(payload),
+            signal: AbortSignal.timeout(120_000),
+        });
+        const result = (await response.json());
+        if (result.warning)
+            console.warn(`Guava RAG: ${result.warning}`);
+        return result.answer;
+    }
+    async reconcile(documents, ids) {
+        if (ids !== undefined && ids.length !== documents.length) {
+            throw new Error(`ids length (${ids.length}) must match documents length (${documents.length})`);
+        }
+        const desired = new Map();
+        if (ids !== undefined) {
+            for (let i = 0; i < documents.length; i++) {
+                desired.set(_prefixedKey(this._namespace, ids[i]), documents[i]);
+            }
+        }
+        else {
+            for (const doc of documents) {
+                desired.set(_prefixedKey(this._namespace, _contentKey(doc)), doc);
+            }
+        }
+        const existing = await this.listDocuments();
+        const existingKeys = new Set(existing.map((d) => d.key));
+        const scopedKeys = this._namespace
+            ? new Set([...existingKeys].filter((k) => k.startsWith(`${this._namespace}.`)))
+            : existingKeys;
+        for (const [key, doc] of desired) {
+            if (ids !== undefined || !existingKeys.has(key)) {
+                await this.uploadDocument(key, doc);
+            }
+            this.trackedKeys.add(key);
+        }
+        const stale = [...scopedKeys].filter((k) => !desired.has(k));
+        for (const key of stale) {
+            await this.deleteDocument(key);
+        }
+    }
+    async upsertDocument(key, text) {
+        const fullKey = _prefixedKey(this._namespace, key);
+        await this.uploadDocument(fullKey, text);
+        this.trackedKeys.add(fullKey);
+    }
+    async addDocument(text) {
+        const fullKey = _prefixedKey(this._namespace, _contentKey(text));
+        await this.uploadDocument(fullKey, text);
+        this.trackedKeys.add(fullKey);
+    }
+    async removeDocument(key) {
+        const fullKey = _prefixedKey(this._namespace, key);
+        await this.deleteDocument(fullKey);
+        this.trackedKeys.delete(fullKey);
+    }
+    async clear() {
+        for (const key of this.trackedKeys) {
+            await this.deleteDocument(key);
+        }
+        this.trackedKeys.clear();
+    }
+}
+exports.ServerRAG = ServerRAG;
+//# sourceMappingURL=server-rag.js.map

package/dist/src/helpers/server-rag.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server-rag.js","sourceRoot":"","sources":["../../../src/helpers/server-rag.ts"],"names":[],"mappings":";;;AAIA,kCAEC;AAED,oCAEC;AAVD,6CAAyC;AAEzC,0CAA2C;AAE3C,SAAgB,WAAW,CAAC,IAAY;IACtC,OAAO,IAAA,wBAAU,EAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACtE,CAAC;AAED,SAAgB,YAAY,CAAC,SAA6B,EAAE,GAAW;IACrE,OAAO,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;AACjD,CAAC;AAED,MAAa,SAAS;IACH,OAAO,CAAS;IAChB,UAAU,CAAqB;IACvC,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IAEzC,YAAY,MAAc,EAAE,EAAE,SAAS,KAA6B,EAAE;QACpE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAAW,EAAE,IAAY;QAC5C,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,kBAAkB,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;QACpE,MAAM,IAAA,uBAAY,EAAC,GAAG,EAAE;YACtB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,GAAG,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAClF,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YACnC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;SACpC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAAW;QAC9B,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,oBAAoB,GAAG,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;QAC3E,MAAM,IAAA,uBAAY,EAAC,GAAG,EAAE;YACtB,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;YACrC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;SACpC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,kBAAkB,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;QACpE,MAAM,QAAQ,GAAG,MAAM,IAAA,uBAAY,EAAC,GAAG,EAAE;YACvC,OAAO,EAAE,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;YACrC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;SACpC,CAAC,CAAC;QACH,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAsB,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,GAAG,CACP,QAAgB,EAChB,EAAE,YAAY,EAAE,YAAY,KAAyD,EAAE;QAEvF,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;QAC9D,MAAM,OAAO,GAA4B,EAAE,QAAQ,EAAE,CAAC;QACtD,IAAI,YAAY,KAAK,SAAS;YAAE,OAAO,CAAC,aAAa,GAAG,YAAY,CAAC;QACrE,IAAI,YAAY,KAAK,SAAS;YAAE,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;QAEpE,MAAM,QAAQ,GAAG,MAAM,IAAA,uBAAY,EAAC,GAAG,EAAE;YACvC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,GAAG,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAClF,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAC7B,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC;SACrC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyC,CAAC;QAC/E,IAAI,MAAM,CAAC,OAAO;YAAE,OAAO,CAAC,IAAI,CAAC,cAAc,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QACjE,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAAmB,EAAE,GAAc;QACjD,IAAI,GAAG,KAAK,SAAS,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;YACzD,MAAM,IAAI,KAAK,CACb,eAAe,GAAG,CAAC,MAAM,kCAAkC,SAAS,CAAC,MAAM,GAAG,CAC/E,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC1C,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,CAAE,CAAC,EAAE,SAAS,CAAC,CAAC,CAAE,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;aAAM,CAAC;YACN,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;gBAC5B,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACpE,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC5C,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACzD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU;YAChC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;YAC/E,CAAC,CAAC,YAAY,CAAC;QAEjB,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,OAAO,EAAE,CAAC;YACjC,IAAI,GAAG,KAAK,SAAS,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAChD,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YACtC,CAAC;YACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,KAAK,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7D,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAAW,EAAE,IAAY;QAC5C,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QACnD,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY;QAC5B,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;QACjE,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAAW;QAC9B,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QACnD,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACnC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;CACF;AAvHD,8BAuHC"}

package/dist/src/socket/protocol.d.ts

@@ -1,9 +1,9 @@
 import { z } from "zod";
 export declare const CloseReason: z.ZodEnum<{
     unknown: "unknown";
-    done: "done";
     "authentication-failure": "authentication-failure";
     "state-lost": "state-lost";
+    done: "done";
     other: "other";
     "server-error": "server-error";
     "reconnection-failed": "reconnection-failed";
@@ -27,9 +27,9 @@ export declare const GuavaClose: z.ZodObject<{
     message_type: z.ZodLiteral<"close">;
     reason: z.ZodEnum<{
         unknown: "unknown";
-        done: "done";
         "authentication-failure": "authentication-failure";
         "state-lost": "state-lost";
+        done: "done";
         other: "other";
         "server-error": "server-error";
         "reconnection-failed": "reconnection-failed";
@@ -69,9 +69,9 @@ export declare const GuavaClientMessage: z.ZodDiscriminatedUnion<[z.ZodObject<{
     message_type: z.ZodLiteral<"close">;
     reason: z.ZodEnum<{
         unknown: "unknown";
-        done: "done";
         "authentication-failure": "authentication-failure";
         "state-lost": "state-lost";
+        done: "done";
         other: "other";
         "server-error": "server-error";
         "reconnection-failed": "reconnection-failed";
@@ -101,9 +101,9 @@ export declare const GuavaServerMessage: z.ZodDiscriminatedUnion<[z.ZodObject<{
     message_type: z.ZodLiteral<"close">;
     reason: z.ZodEnum<{
         unknown: "unknown";
-        done: "done";
         "authentication-failure": "authentication-failure";
         "state-lost": "state-lost";
+        done: "done";
         other: "other";
         "server-error": "server-error";
         "reconnection-failed": "reconnection-failed";

package/dist/src/testing/mocks.d.ts

@@ -1,6 +1,12 @@
 import { Call } from "../call.ts";
+import type { CallInfo } from "../socket/call-info.ts";
 import type { Command } from "../commands.ts";
 export declare class MockCall extends Call {
+    constructor(args?: {
+        callId?: string;
+        callInfo?: CallInfo;
+        variables?: Record<string, any>;
+    });
     get commands(): Command[];
     setField(fieldName: string, fieldValue: unknown): void;
 }