@convex-dev/agent 0.6.0 → 0.6.1

package/dist/client/mockModel.d.ts

@@ -0,0 +1,42 @@
+import type { LanguageModelV3, LanguageModelV3Content } from "@ai-sdk/provider";
+import { type ProviderMetadata } from "ai";
+export declare const DEFAULT_TEXT = "\nA A A A A A A A A A A A A A A\nB B B B B B B B B B B B B B B\nC C C C C C C C C C C C C C C\nD D D D D D D D D D D D D D D\n";
+export type MockModelArgs = {
+    provider?: LanguageModelV3["provider"];
+    modelId?: LanguageModelV3["modelId"];
+    supportedUrls?: LanguageModelV3["supportedUrls"] | (() => LanguageModelV3["supportedUrls"]);
+    chunkDelayInMs?: number;
+    initialDelayInMs?: number;
+    /** A list of the responses for multiple steps.
+     * For tool calls, the first list would include a tool call part,
+     * then the next list would be after the tool response or another tool call.
+     * Tool responses come from actual tool calls!
+     */
+    contentSteps?: LanguageModelV3Content[][];
+    /** A single list of content responded from each step.
+     * Provide contentSteps instead if you want to do multi-step responses with
+     * tool calls.
+     */
+    content?: LanguageModelV3Content[];
+    doGenerate?: LanguageModelV3["doGenerate"];
+    doStream?: LanguageModelV3["doStream"];
+    providerMetadata?: ProviderMetadata;
+    fail?: boolean | {
+        probability?: number;
+        error?: string;
+    };
+};
+export declare function mockModel(args?: MockModelArgs): LanguageModelV3;
+export declare class MockLanguageModel implements LanguageModelV3 {
+    readonly specificationVersion = "v3";
+    private _supportedUrls;
+    readonly provider: LanguageModelV3["provider"];
+    readonly modelId: LanguageModelV3["modelId"];
+    doGenerate: LanguageModelV3["doGenerate"];
+    doStream: LanguageModelV3["doStream"];
+    doGenerateCalls: Parameters<LanguageModelV3["doGenerate"]>[0][];
+    doStreamCalls: Parameters<LanguageModelV3["doStream"]>[0][];
+    constructor(args: MockModelArgs);
+    get supportedUrls(): Record<string, RegExp[]> | PromiseLike<Record<string, RegExp[]>>;
+}
+//# sourceMappingURL=mockModel.d.ts.map

package/dist/client/mockModel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mockModel.d.ts","sourceRoot":"","sources":["../../src/client/mockModel.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,sBAAsB,EAEvB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAA0B,KAAK,gBAAgB,EAAE,MAAM,IAAI,CAAC;AAGnE,eAAO,MAAM,YAAY,mIAKxB,CAAC;AAgBF,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,CAAC,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IACvC,OAAO,CAAC,EAAE,eAAe,CAAC,SAAS,CAAC,CAAC;IACrC,aAAa,CAAC,EACV,eAAe,CAAC,eAAe,CAAC,GAChC,CAAC,MAAM,eAAe,CAAC,eAAe,CAAC,CAAC,CAAC;IAC7C,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;;OAIG;IACH,YAAY,CAAC,EAAE,sBAAsB,EAAE,EAAE,CAAC;IAC1C;;;OAGG;IACH,OAAO,CAAC,EAAE,sBAAsB,EAAE,CAAC;IAEnC,UAAU,CAAC,EAAE,eAAe,CAAC,YAAY,CAAC,CAAC;IAC3C,QAAQ,CAAC,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IACvC,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC,IAAI,CAAC,EACD,OAAO,GACP;QACE,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;CACP,CAAC;AAMF,wBAAgB,SAAS,CAAC,IAAI,CAAC,EAAE,aAAa,GAAG,eAAe,CAE/D;AAED,qBAAa,iBAAkB,YAAW,eAAe;IACvD,QAAQ,CAAC,oBAAoB,QAAQ;IAErC,OAAO,CAAC,cAAc,CAAyC;IAE/D,QAAQ,CAAC,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAC/C,QAAQ,CAAC,OAAO,EAAE,eAAe,CAAC,SAAS,CAAC,CAAC;IAE7C,UAAU,EAAE,eAAe,CAAC,YAAY,CAAC,CAAC;IAC1C,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;IAEtC,eAAe,EAAE,UAAU,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAM;IACrE,aAAa,EAAE,UAAU,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAM;gBAErD,IAAI,EAAE,aAAa;IAsK/B,IAAI,aAAa,qEAEhB;CACF"}

package/dist/client/mockModel.js

@@ -0,0 +1,182 @@
+import { simulateReadableStream } from "ai";
+import { assert, pick } from "convex-helpers";
+export const DEFAULT_TEXT = `
+A A A A A A A A A A A A A A A
+B B B B B B B B B B B B B B B
+C C C C C C C C C C C C C C C
+D D D D D D D D D D D D D D D
+`;
+const DEFAULT_USAGE = {
+    outputTokens: 10,
+    inputTokens: 3,
+    totalTokens: 13,
+    inputTokenDetails: {
+        noCacheTokens: 3,
+        cacheReadTokens: 0,
+        cacheWriteTokens: 0,
+    },
+    outputTokenDetails: {
+        textTokens: 10,
+        reasoningTokens: 0,
+    },
+};
+function atMostOneOf(...args) {
+    return args.filter(Boolean).length <= 1;
+}
+export function mockModel(args) {
+    return new MockLanguageModel(args ?? {});
+}
+export class MockLanguageModel {
+    specificationVersion = "v3";
+    _supportedUrls;
+    provider;
+    modelId;
+    doGenerate;
+    doStream;
+    doGenerateCalls = [];
+    doStreamCalls = [];
+    constructor(args) {
+        assert(atMostOneOf(args.content, args.contentSteps, args.doGenerate && args.doStream), "Expected only one of content, contentSteps, or doGenerate and doStream");
+        this.provider = args.provider || "mock-provider";
+        this.modelId = args.modelId || "mock-model-id";
+        const { content = [{ type: "text", text: DEFAULT_TEXT }], contentSteps = [content], chunkDelayInMs = 0, initialDelayInMs = 0, supportedUrls = {}, } = args;
+        const fail = args.fail &&
+            (args.fail === true ||
+                !args.fail.probability ||
+                Math.random() < args.fail.probability);
+        const error = (typeof args.fail === "object" && args.fail.error) ||
+            "Mock error message";
+        const metadata = pick(args, ["providerMetadata"]);
+        const chunkResponses = contentSteps.map((content) => {
+            const chunks = [
+                { type: "stream-start", warnings: [] },
+            ];
+            chunks.push(...content.flatMap((c, ci) => {
+                if (c.type !== "text" && c.type !== "reasoning") {
+                    return [c];
+                }
+                const metadata = pick(c, ["providerMetadata"]);
+                const deltas = c.text.split(" ");
+                const parts = [];
+                if (c.type === "reasoning") {
+                    parts.push({
+                        type: "reasoning-start",
+                        id: `reasoning-${ci}`,
+                        ...metadata,
+                    });
+                    parts.push(...deltas.map((delta, di) => ({
+                        type: "reasoning-delta",
+                        delta: (di ? " " : "") + delta,
+                        id: `reasoning-${ci}`,
+                        ...metadata,
+                    })));
+                    parts.push({
+                        type: "reasoning-end",
+                        id: `reasoning-${ci}`,
+                        ...metadata,
+                    });
+                }
+                else if (c.type === "text") {
+                    parts.push({
+                        type: "text-start",
+                        id: `txt-${ci}`,
+                        ...metadata,
+                    });
+                    parts.push(...deltas.map((delta, di) => ({
+                        type: "text-delta",
+                        delta: (di ? " " : "") + delta,
+                        id: `txt-${ci}`,
+                        ...metadata,
+                    })));
+                    parts.push({
+                        type: "text-end",
+                        id: `txt-${ci}`,
+                        ...metadata,
+                    });
+                }
+                return parts;
+            }));
+            if (fail) {
+                chunks.push({
+                    type: "error",
+                    error,
+                });
+            }
+            chunks.push({
+                type: "finish",
+                finishReason: fail ? "error" : "stop",
+                usage: DEFAULT_USAGE,
+                ...metadata,
+            });
+            return chunks;
+        });
+        let callIndex = 0;
+        this.doGenerate = async (options) => {
+            this.doGenerateCalls.push(options);
+            if (fail) {
+                throw new Error(error);
+            }
+            if (typeof args.doGenerate === "function") {
+                return args.doGenerate(options);
+            }
+            else if (Array.isArray(args.doGenerate)) {
+                return args.doGenerate[this.doGenerateCalls.length];
+            }
+            else if (contentSteps.length) {
+                const result = {
+                    content: contentSteps[callIndex % contentSteps.length],
+                    finishReason: "stop",
+                    usage: DEFAULT_USAGE,
+                    ...metadata,
+                    warnings: [],
+                };
+                callIndex++;
+                return result;
+            }
+            else {
+                throw new Error("Unexpected: no content or doGenerate");
+            }
+        };
+        this.doStream = async (options) => {
+            this.doStreamCalls.push(options);
+            if (typeof args.doStream === "function") {
+                return args.doStream(options);
+            }
+            else if (Array.isArray(args.doStream)) {
+                return args.doStream[this.doStreamCalls.length];
+            }
+            else if (contentSteps) {
+                const stream = simulateReadableStream({
+                    chunks: chunkResponses[callIndex % chunkResponses.length],
+                    initialDelayInMs,
+                    chunkDelayInMs,
+                });
+                callIndex++;
+                if (options.abortSignal) {
+                    options.abortSignal.addEventListener("abort", () => {
+                        console.warn("abortSignal in mock model not supported");
+                    });
+                }
+                return {
+                    stream,
+                    request: { body: {} },
+                    response: { headers: {} },
+                };
+            }
+            else if (args.doStream) {
+                return args.doStream;
+            }
+            else {
+                throw new Error("Provide either content or doStream");
+            }
+        };
+        this._supportedUrls =
+            typeof supportedUrls === "function"
+                ? supportedUrls
+                : async () => supportedUrls;
+    }
+    get supportedUrls() {
+        return this._supportedUrls();
+    }
+}
+//# sourceMappingURL=mockModel.js.map

package/dist/client/mockModel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mockModel.js","sourceRoot":"","sources":["../../src/client/mockModel.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,sBAAsB,EAAyB,MAAM,IAAI,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAE9C,MAAM,CAAC,MAAM,YAAY,GAAG;;;;;CAK3B,CAAC;AACF,MAAM,aAAa,GAAG;IACpB,YAAY,EAAE,EAAE;IAChB,WAAW,EAAE,CAAC;IACd,WAAW,EAAE,EAAE;IACf,iBAAiB,EAAE;QACjB,aAAa,EAAE,CAAC;QAChB,eAAe,EAAE,CAAC;QAClB,gBAAgB,EAAE,CAAC;KACpB;IACD,kBAAkB,EAAE;QAClB,UAAU,EAAE,EAAE;QACd,eAAe,EAAE,CAAC;KACnB;CACF,CAAC;AAiCF,SAAS,WAAW,CAAC,GAAG,IAAe;IACrC,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,IAAoB;IAC5C,OAAO,IAAI,iBAAiB,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,OAAO,iBAAiB;IACnB,oBAAoB,GAAG,IAAI,CAAC;IAE7B,cAAc,CAAyC;IAEtD,QAAQ,CAA8B;IACtC,OAAO,CAA6B;IAE7C,UAAU,CAAgC;IAC1C,QAAQ,CAA8B;IAEtC,eAAe,GAAmD,EAAE,CAAC;IACrE,aAAa,GAAiD,EAAE,CAAC;IAEjE,YAAY,IAAmB;QAC7B,MAAM,CACJ,WAAW,CACT,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,QAAQ,CACjC,EACD,wEAAwE,CACzE,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,eAAe,CAAC;QACjD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,eAAe,CAAC;QAC/C,MAAM,EACJ,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,EAChD,YAAY,GAAG,CAAC,OAAO,CAAC,EACxB,cAAc,GAAG,CAAC,EAClB,gBAAgB,GAAG,CAAC,EACpB,aAAa,GAAG,EAAE,GACnB,GAAG,IAAI,CAAC;QACT,MAAM,IAAI,GACR,IAAI,CAAC,IAAI;YACT,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI;gBACjB,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW;gBACtB,IAAI,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,KAAK,GACT,CAAC,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;YAClD,oBAAoB,CAAC;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAElD,MAAM,cAAc,GAAkC,YAAY,CAAC,GAAG,CACpE,CAAC,OAAO,EAAE,EAAE;YACV,MAAM,MAAM,GAAgC;gBAC1C,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,EAAE,EAAE;aACvC,CAAC;YACF,MAAM,CAAC,IAAI,CACT,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,EAA+B,EAAE;gBACxD,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAChD,OAAO,CAAC,CAAC,CAAC,CAAC;gBACb,CAAC;gBACD,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBAC/C,MAAM,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBACjC,MAAM,KAAK,GAAgC,EAAE,CAAC;gBAC9C,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC3B,KAAK,CAAC,IAAI,CAAC;wBACT,IAAI,EAAE,iBAAiB;wBACvB,EAAE,EAAE,aAAa,EAAE,EAAE;wBACrB,GAAG,QAAQ;qBACZ,CAAC,CAAC;oBACH,KAAK,CAAC,IAAI,CACR,GAAG,MAAM,CAAC,GAAG,CACX,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CACZ,CAAC;wBACC,IAAI,EAAE,iBAAiB;wBACvB,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK;wBAC9B,EAAE,EAAE,aAAa,EAAE,EAAE;wBACrB,GAAG,QAAQ;qBACZ,CAAqC,CACzC,CACF,CAAC;oBACF,KAAK,CAAC,IAAI,CAAC;wBACT,IAAI,EAAE,eAAe;wBACrB,EAAE,EAAE,aAAa,EAAE,EAAE;wBACrB,GAAG,QAAQ;qBACZ,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC7B,KAAK,CAAC,IAAI,CAAC;wBACT,IAAI,EAAE,YAAY;wBAClB,EAAE,EAAE,OAAO,EAAE,EAAE;wBACf,GAAG,QAAQ;qBACZ,CAAC,CAAC;oBACH,KAAK,CAAC,IAAI,CACR,GAAG,MAAM,CAAC,GAAG,CACX,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CACZ,CAAC;wBACC,IAAI,EAAE,YAAY;wBAClB,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK;wBAC9B,EAAE,EAAE,OAAO,EAAE,EAAE;wBACf,GAAG,QAAQ;qBACZ,CAAqC,CACzC,CACF,CAAC;oBACF,KAAK,CAAC,IAAI,CAAC;wBACT,IAAI,EAAE,UAAU;wBAChB,EAAE,EAAE,OAAO,EAAE,EAAE;wBACf,GAAG,QAAQ;qBACZ,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CACH,CAAC;YACF,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,OAAO;oBACb,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;YACD,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;gBACrC,KAAK,EAAE,aAAa;gBACpB,GAAI,QAAgB;aACrB,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC,CACF,CAAC;QACF,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,IAAI,CAAC,UAAU,GAAG,KAAK,EAAE,OAAO,EAAE,EAAE;YAClC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEnC,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC;YACzB,CAAC;YACD,IAAI,OAAO,IAAI,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBAC1C,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YAClC,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1C,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YACtD,CAAC;iBAAM,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;gBAC/B,MAAM,MAAM,GAAG;oBACb,OAAO,EAAE,YAAY,CAAC,SAAS,GAAG,YAAY,CAAC,MAAM,CAAC;oBACtD,YAAY,EAAE,MAAe;oBAC7B,KAAK,EAAE,aAAa;oBACpB,GAAI,QAAgB;oBACpB,QAAQ,EAAE,EAAE;iBACb,CAAC;gBACF,SAAS,EAAE,CAAC;gBACZ,OAAO,MAAM,CAAC;YAChB,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,KAAK,EAAE,OAAO,EAAE,EAAE;YAChC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEjC,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAChC,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YAClD,CAAC;iBAAM,IAAI,YAAY,EAAE,CAAC;gBACxB,MAAM,MAAM,GAAG,sBAAsB,CAAC;oBACpC,MAAM,EAAE,cAAc,CAAC,SAAS,GAAG,cAAc,CAAC,MAAM,CAAC;oBACzD,gBAAgB;oBAChB,cAAc;iBACf,CAAC,CAAC;gBACH,SAAS,EAAE,CAAC;gBAEZ,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;oBACxB,OAAO,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;wBACjD,OAAO,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;oBAC1D,CAAC,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO;oBACL,MAAM;oBACN,OAAO,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;oBACrB,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;iBAC1B,CAAC;YACJ,CAAC;iBAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC,QAAQ,CAAC;YACvB,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;YACxD,CAAC;QACH,CAAC,CAAC;QACF,IAAI,CAAC,cAAc;YACjB,OAAO,aAAa,KAAK,UAAU;gBACjC,CAAC,CAAC,aAAa;gBACf,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,aAAa,CAAC;IAClC,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;IAC/B,CAAC;CACF"}

package/dist/client/saveInputMessages.d.ts

@@ -0,0 +1,20 @@
+import { type ModelMessage } from "ai";
+import type { MessageDoc } from "../validators.js";
+import type { ActionCtx, AgentComponent, Config, MutationCtx } from "./types.js";
+import type { Message } from "../validators.js";
+export declare function saveInputMessages(ctx: MutationCtx | ActionCtx, component: AgentComponent, { threadId, userId, prompt, messages, ...args }: {
+    prompt: string | (ModelMessage | Message)[] | undefined;
+    messages: (ModelMessage | Message)[] | undefined;
+    promptMessageId: string | undefined;
+    userId: string | undefined;
+    threadId: string;
+    agentName?: string;
+    storageOptions?: {
+        saveMessages?: "all" | "promptAndOutput";
+    };
+} & Pick<Config, "usageHandler" | "textEmbeddingModel" | "embeddingModel" | "callSettings">): Promise<{
+    promptMessageId: string | undefined;
+    pendingMessage: MessageDoc;
+    savedMessages: MessageDoc[];
+}>;
+//# sourceMappingURL=saveInputMessages.d.ts.map

package/dist/client/saveInputMessages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"saveInputMessages.d.ts","sourceRoot":"","sources":["../../src/client/saveInputMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,IAAI,CAAC;AACvC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,KAAK,EACV,SAAS,EACT,cAAc,EACd,MAAM,EACN,WAAW,EACZ,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAIhD,wBAAsB,iBAAiB,CACrC,GAAG,EAAE,WAAW,GAAG,SAAS,EAC5B,SAAS,EAAE,cAAc,EACzB,EACE,QAAQ,EACR,MAAM,EACN,MAAM,EACN,QAAQ,EACR,GAAG,IAAI,EACR,EAAE;IACD,MAAM,EAAE,MAAM,GAAG,CAAC,YAAY,GAAG,OAAO,CAAC,EAAE,GAAG,SAAS,CAAC;IACxD,QAAQ,EAAE,CAAC,YAAY,GAAG,OAAO,CAAC,EAAE,GAAG,SAAS,CAAC;IACjD,eAAe,EAAE,MAAM,GAAG,SAAS,CAAC;IACpC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE;QACf,YAAY,CAAC,EAAE,KAAK,GAAG,iBAAiB,CAAC;KAC1C,CAAC;CACH,GAAG,IAAI,CAAC,MAAM,EAAE,cAAc,GAAG,oBAAoB,GAAG,gBAAgB,GAAG,cAAc,CAAC,GAC1F,OAAO,CAAC;IACT,eAAe,EAAE,MAAM,GAAG,SAAS,CAAC;IACpC,cAAc,EAAE,UAAU,CAAC;IAC3B,aAAa,EAAE,UAAU,EAAE,CAAC;CAC7B,CAAC,CA8DD"}

package/dist/client/saveInputMessages.js

@@ -0,0 +1,58 @@
+import {} from "ai";
+import { embedMessages, getPromptArray } from "./search.js";
+import { saveMessages } from "./messages.js";
+import { assert } from "convex-helpers";
+export async function saveInputMessages(ctx, component, { threadId, userId, prompt, messages, ...args }) {
+    const shouldSave = args.storageOptions?.saveMessages ?? "promptAndOutput";
+    // If only a promptMessageId is provided, this will be empty.
+    const promptArray = getPromptArray(prompt);
+    const toSave = [];
+    if (args.promptMessageId) {
+        // We don't save any inputs if a promptMessageId is provided.
+        // It's unclear where they'd want to save the new messages.
+    }
+    else if (shouldSave === "all") {
+        if (messages)
+            toSave.push(...messages);
+        toSave.push(...promptArray);
+    }
+    else {
+        if (promptArray.length) {
+            // We treat the whole promptArray as the prompt message to save.
+            toSave.push(...promptArray);
+        }
+        else if (messages) {
+            // Otherwise, treat the last message as the prompt message to save.
+            toSave.push(...messages.slice(-1));
+        }
+    }
+    let embeddings;
+    if ((args.embeddingModel ?? args.textEmbeddingModel) && toSave.length) {
+        assert("runAction" in ctx, "You must be in an action context to generate embeddings");
+        embeddings = await embedMessages(ctx, { ...args, userId: userId ?? undefined, threadId }, toSave);
+        if (embeddings) {
+            // for the pending message
+            embeddings.vectors.push(null);
+        }
+    }
+    const saved = await saveMessages(ctx, component, {
+        threadId,
+        userId,
+        messages: [...toSave, { role: "assistant", content: [] }],
+        metadata: [
+            ...Array.from({ length: toSave.length }, () => ({})),
+            { status: "pending" },
+        ],
+        failPendingSteps: !!args.promptMessageId,
+        promptMessageId: args.promptMessageId,
+        embeddings,
+    });
+    return {
+        promptMessageId: toSave.length
+            ? saved.messages.at(-2)._id
+            : args.promptMessageId,
+        pendingMessage: saved.messages.at(-1),
+        savedMessages: saved.messages.slice(0, -1),
+    };
+}
+//# sourceMappingURL=saveInputMessages.js.map

package/dist/client/saveInputMessages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"saveInputMessages.js","sourceRoot":"","sources":["../../src/client/saveInputMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,MAAM,IAAI,CAAC;AAEvC,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAO5D,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE7C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAGxC,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,GAA4B,EAC5B,SAAyB,EACzB,EACE,QAAQ,EACR,MAAM,EACN,MAAM,EACN,QAAQ,EACR,GAAG,IAAI,EAWkF;IAM3F,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,EAAE,YAAY,IAAI,iBAAiB,CAAC;IAC1E,6DAA6D;IAC7D,MAAM,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IAE3C,MAAM,MAAM,GAA+B,EAAE,CAAC;IAC9C,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;QACzB,6DAA6D;QAC7D,2DAA2D;IAC7D,CAAC;SAAM,IAAI,UAAU,KAAK,KAAK,EAAE,CAAC;QAChC,IAAI,QAAQ;YAAE,MAAM,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;QACvC,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;IAC9B,CAAC;SAAM,CAAC;QACN,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;YACvB,gEAAgE;YAChE,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;QAC9B,CAAC;aAAM,IAAI,QAAQ,EAAE,CAAC;YACpB,mEAAmE;YACnE,MAAM,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IACD,IAAI,UAMS,CAAC;IACd,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,kBAAkB,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QACtE,MAAM,CACJ,WAAW,IAAI,GAAG,EAClB,yDAAyD,CAC1D,CAAC;QACF,UAAU,GAAG,MAAM,aAAa,CAC9B,GAAG,EACH,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,MAAM,IAAI,SAAS,EAAE,QAAQ,EAAE,EAClD,MAAM,CACP,CAAC;QACF,IAAI,UAAU,EAAE,CAAC;YACf,0BAA0B;YAC1B,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IACD,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,GAAG,EAAE,SAAS,EAAE;QAC/C,QAAQ;QACR,MAAM;QACN,QAAQ,EAAE,CAAC,GAAG,MAAM,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACzD,QAAQ,EAAE;YACR,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACpD,EAAE,MAAM,EAAE,SAAS,EAAE;SACtB;QACD,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe;QACxC,eAAe,EAAE,IAAI,CAAC,eAAe;QACrC,UAAU;KACX,CAAC,CAAC;IACH,OAAO;QACL,eAAe,EAAE,MAAM,CAAC,MAAM;YAC5B,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,GAAG;YAC5B,CAAC,CAAC,IAAI,CAAC,eAAe;QACxB,cAAc,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE;QACtC,aAAa,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;KAC3C,CAAC;AACJ,CAAC"}