@convex-dev/agent 0.2.6-alpha.1 → 0.2.6

package/dist/client/search.js

@@ -1,6 +1,13 @@
+import { embedMany as embedMany_, } from "ai";
 import { assert } from "convex-helpers";
-import { DEFAULT_MESSAGE_RANGE, DEFAULT_RECENT_MESSAGES, extractText, sorted, } from "../shared.js";
+import { validateVectorDimension, } from "../component/vector/tables.js";
+import { DEFAULT_MESSAGE_RANGE, DEFAULT_RECENT_MESSAGES, extractText, getModelName, getProviderName, isTool, sorted, } from "../shared.js";
+import { inlineMessagesFiles } from "./files.js";
+import { deserializeMessage } from "../mapping.js";
 const DEFAULT_VECTOR_SCORE_THRESHOLD = 0.0;
+// 10k characters should be more than enough for most cases, and stays under
+// the 8k token limit for some models.
+const MAX_EMBEDDING_TEXT_LENGTH = 10_000;
 /**
  * Fetch the context messages for a thread.
  * @param ctx Either a query, mutation, or action ctx.
@@ -10,13 +17,18 @@ const DEFAULT_VECTOR_SCORE_THRESHOLD = 0.0;
  * @returns
  */
 export async function fetchContextMessages(ctx, component, args) {
+    const { recentMessages, searchMessages } = await fetchRecentAndSearchMessages(ctx, component, args);
+    return [...searchMessages, ...recentMessages];
+}
+export async function fetchRecentAndSearchMessages(ctx, component, args) {
     assert(args.userId || args.threadId, "Specify userId or threadId");
     const opts = args.contextOptions;
     // Fetch the latest messages from the thread
     let included;
-    const contextMessages = [];
-    if (args.threadId &&
-        (opts.recentMessages !== 0 || args.upToAndIncludingMessageId)) {
+    let recentMessages = [];
+    let searchMessages = [];
+    const targetMessageId = args.targetMessageId ?? args.upToAndIncludingMessageId;
+    if (args.threadId && opts.recentMessages !== 0) {
         const { page } = await ctx.runQuery(component.messages.listMessagesByThreadId, {
             threadId: args.threadId,
             excludeToolMessages: opts.excludeToolMessages,
@@ -24,34 +36,53 @@ export async function fetchContextMessages(ctx, component, args) {
                 numItems: opts.recentMessages ?? DEFAULT_RECENT_MESSAGES,
                 cursor: null,
             },
-            upToAndIncludingMessageId: args.upToAndIncludingMessageId,
+            upToAndIncludingMessageId: targetMessageId,
             order: "desc",
             statuses: ["success"],
         });
         included = new Set(page.map((m) => m._id));
-        contextMessages.push(
-        // Reverse since we fetched in descending order
-        ...page.reverse());
+        recentMessages = filterOutOrphanedToolMessages(sorted(page));
     }
     if ((opts.searchOptions?.textSearch || opts.searchOptions?.vectorSearch) &&
         opts.searchOptions?.limit) {
-        const targetMessage = contextMessages.find((m) => m._id === args.upToAndIncludingMessageId)?.message;
-        const messagesToSearch = targetMessage ? [targetMessage] : args.messages;
         if (!("runAction" in ctx)) {
             throw new Error("searchUserMessages only works in an action");
         }
-        const lastMessage = messagesToSearch.at(-1);
-        assert(lastMessage, "No messages to search");
-        const text = extractText(lastMessage);
-        assert(text, `No text to search in message ${JSON.stringify(lastMessage)}`);
-        assert(!args.contextOptions?.searchOptions?.vectorSearch || "runAction" in ctx, "You must do vector search from an action");
-        if (opts.searchOptions?.vectorSearch && !args.getEmbedding) {
-            throw new Error("You must provide an embedding and embeddingModel to use vector search");
+        let text = args.searchText;
+        let embedding;
+        let embeddingModel;
+        if (!text) {
+            if (targetMessageId) {
+                const targetMessage = recentMessages.find((m) => m._id === targetMessageId);
+                if (targetMessage) {
+                    text = targetMessage.text;
+                }
+                else {
+                    const targetSearchFields = await ctx.runQuery(component.messages.getMessageSearchFields, {
+                        messageId: targetMessageId,
+                    });
+                    text = targetSearchFields.text;
+                    embedding = targetSearchFields.embedding;
+                    embeddingModel = targetSearchFields.embeddingModel;
+                }
+                assert(text, "Target message has no text for searching");
+            }
+            else if (args.messages?.length) {
+                text = extractText(args.messages.at(-1));
+                assert(text, "Final context message has no text to search");
+            }
+            assert(text, "No text to search");
+        }
+        if (opts.searchOptions?.vectorSearch) {
+            if (!embedding && args.getEmbedding) {
+                const embeddingFields = await args.getEmbedding(text);
+                embedding = embeddingFields.embedding;
+                embeddingModel = embeddingFields.textEmbeddingModel
+                    ? getModelName(embeddingFields.textEmbeddingModel)
+                    : undefined;
+            }
         }
-        const embeddingFields = opts.searchOptions?.vectorSearch && text
-            ? await args.getEmbedding?.(text)
-            : undefined;
-        const searchMessages = await ctx.runAction(component.messages.searchMessages, {
+        const searchResults = await ctx.runAction(component.messages.searchMessages, {
             searchAllMessagesForUserId: opts?.searchOtherThreads
                 ? (args.userId ??
                     (args.threadId &&
@@ -60,25 +91,25 @@ export async function fetchContextMessages(ctx, component, args) {
                         }))?.userId))
                 : undefined,
             threadId: args.threadId,
-            beforeMessageId: args.upToAndIncludingMessageId,
+            targetMessageId,
             limit: opts.searchOptions?.limit ?? 10,
             messageRange: {
                 ...DEFAULT_MESSAGE_RANGE,
                 ...opts.searchOptions?.messageRange,
             },
             text,
+            textSearch: opts.searchOptions?.textSearch,
+            vectorSearch: opts.searchOptions?.vectorSearch,
             vectorScoreThreshold: opts.searchOptions?.vectorScoreThreshold ??
                 DEFAULT_VECTOR_SCORE_THRESHOLD,
-            embedding: embeddingFields?.embedding,
-            embeddingModel: embeddingFields?.textEmbeddingModel
-                ? getModelName(embeddingFields.textEmbeddingModel)
-                : undefined,
+            embedding,
+            embeddingModel,
         });
         // TODO: track what messages we used for context
-        contextMessages.unshift(...searchMessages.filter((m) => !included?.has(m._id)));
+        searchMessages = filterOutOrphanedToolMessages(sorted(searchResults.filter((m) => !included?.has(m._id))));
     }
     // Ensure we don't include tool messages without a corresponding tool call
-    return filterOutOrphanedToolMessages(sorted(contextMessages));
+    return { recentMessages, searchMessages };
 }
 /**
  * Filter out tool messages that don't have both a tool call and response.
@@ -112,19 +143,220 @@ export function filterOutOrphanedToolMessages(docs) {
     }
     return result;
 }
-export function getModelName(embeddingModel) {
-    if (typeof embeddingModel === "string") {
-        if (embeddingModel.includes("/")) {
-            return embeddingModel.split("/").slice(1).join("/");
-        }
-        return embeddingModel;
+/**
+ * Embed a list of messages, including calling any usage handler.
+ * This will not save the embeddings to the database.
+ */
+export async function embedMessages(ctx, { userId, threadId, ...options }, messages) {
+    if (!options.textEmbeddingModel) {
+        return undefined;
+    }
+    let embeddings;
+    const messageTexts = messages.map((m) => !isTool(m) && extractText(m));
+    // Find the indexes of the messages that have text.
+    const textIndexes = messageTexts
+        .map((t, i) => (t ? i : undefined))
+        .filter((i) => i !== undefined);
+    if (textIndexes.length === 0) {
+        return undefined;
+    }
+    const values = messageTexts
+        .map((t) => t && t.trim().slice(0, MAX_EMBEDDING_TEXT_LENGTH))
+        .filter((t) => !!t);
+    // Then embed those messages.
+    const textEmbeddings = await embedMany(ctx, {
+        ...options,
+        userId,
+        threadId,
+        values,
+    });
+    // Then assemble the embeddings into a single array with nulls for the messages without text.
+    const embeddingsOrNull = Array(messages.length).fill(null);
+    textIndexes.forEach((i, j) => {
+        embeddingsOrNull[i] = textEmbeddings.embeddings[j];
+    });
+    if (textEmbeddings.embeddings.length > 0) {
+        const dimension = textEmbeddings.embeddings[0].length;
+        validateVectorDimension(dimension);
+        const model = getModelName(options.textEmbeddingModel);
+        embeddings = { vectors: embeddingsOrNull, dimension, model };
+    }
+    return embeddings;
+}
+/**
+ * Embeds many strings, calling any usage handler.
+ * @param ctx The ctx parameter to an action.
+ * @param args Arguments to AI SDK's embedMany, and context for the embedding,
+ *   passed to the usage handler.
+ * @returns The embeddings for the strings, matching the order of the values.
+ */
+export async function embedMany(ctx, { userId, threadId, values, abortSignal, headers, agentName, usageHandler, textEmbeddingModel, callSettings, }) {
+    const embeddingModel = textEmbeddingModel;
+    assert(embeddingModel, "a textEmbeddingModel is required to be set for vector search");
+    const result = await embedMany_({
+        ...callSettings,
+        model: embeddingModel,
+        values,
+        abortSignal,
+        headers,
+    });
+    if (usageHandler && result.usage) {
+        await usageHandler(ctx, {
+            userId,
+            threadId,
+            agentName,
+            model: getModelName(embeddingModel),
+            provider: getProviderName(embeddingModel),
+            providerMetadata: undefined,
+            usage: {
+                inputTokens: result.usage.tokens,
+                outputTokens: 0,
+                totalTokens: result.usage.tokens,
+            },
+        });
+    }
+    return { embeddings: result.embeddings };
+}
+/**
+ * Embed a list of messages, and save the embeddings to the database.
+ * @param ctx The ctx parameter to an action.
+ * @param component The agent component, usually components.agent.
+ * @param args The context for the embedding, passed to the usage handler.
+ * @param messages The messages to embed, in the Agent MessageDoc format.
+ */
+export async function generateAndSaveEmbeddings(ctx, component, args, messages) {
+    const toEmbed = messages.filter((m) => !m.embeddingId && m.message);
+    if (toEmbed.length === 0) {
+        return;
+    }
+    const embeddings = await embedMessages(ctx, args, toEmbed.map((m) => m.message));
+    if (embeddings && embeddings.vectors.some((v) => v !== null)) {
+        await ctx.runMutation(component.vector.index.insertBatch, {
+            vectorDimension: embeddings.dimension,
+            vectors: toEmbed
+                .map((m, i) => ({
+                messageId: m._id,
+                model: embeddings.model,
+                table: "messages",
+                userId: m.userId,
+                threadId: m.threadId,
+                vector: embeddings.vectors[i],
+            }))
+                .filter((v) => v.vector !== null),
+        });
     }
-    return embeddingModel.modelId;
 }
-export function getProviderName(embeddingModel) {
-    if (typeof embeddingModel === "string") {
-        return embeddingModel.split("/").at(0);
+/**
+ * Similar to fetchContextMessages, but also combines the input messages,
+ * with search context, recent messages, input messages, then prompt messages.
+ * If there is a promptMessageId and prompt message(s) provided, it will splice
+ * the prompt messages into the history to replace the promptMessageId message,
+ * but still be followed by any existing messages that were in response to the
+ * promptMessageId message.
+ */
+export async function fetchContextWithPrompt(ctx, component, args) {
+    const { threadId, userId, textEmbeddingModel } = args;
+    const promptArray = getPromptArray(args.prompt);
+    const searchText = promptArray.length
+        ? extractText(promptArray.at(-1))
+        : args.promptMessageId
+            ? undefined
+            : args.messages?.at(-1)
+                ? extractText(args.messages.at(-1))
+                : undefined;
+    // If only a messageId is provided, this will add that message to the end.
+    const { recentMessages, searchMessages } = await fetchRecentAndSearchMessages(ctx, component, {
+        userId,
+        threadId,
+        targetMessageId: args.promptMessageId,
+        searchText,
+        contextOptions: args.contextOptions ?? {},
+        getEmbedding: async (text) => {
+            assert(textEmbeddingModel, "A textEmbeddingModel is required to be set on the Agent that you're doing vector search with");
+            return {
+                embedding: (await embedMany(ctx, {
+                    ...args,
+                    userId,
+                    values: [text],
+                    textEmbeddingModel,
+                })).embeddings[0],
+                textEmbeddingModel,
+            };
+        },
+    });
+    const promptMessageIndex = args.promptMessageId
+        ? recentMessages.findIndex((m) => m._id === args.promptMessageId)
+        : -1;
+    const promptMessage = promptMessageIndex !== -1 ? recentMessages[promptMessageIndex] : undefined;
+    let prePromptDocs = recentMessages;
+    const messages = args.messages ?? [];
+    let existingResponseDocs = [];
+    if (promptMessage) {
+        prePromptDocs = recentMessages.slice(0, promptMessageIndex);
+        existingResponseDocs = recentMessages.slice(promptMessageIndex + 1);
+        if (promptArray.length === 0) {
+            // If they didn't override the prompt, use the existing prompt message.
+            if (promptMessage.message) {
+                promptArray.push(promptMessage.message);
+            }
+        }
+        if (!promptMessage.embeddingId && textEmbeddingModel) {
+            // Lazily generate embeddings for the prompt message, if it doesn't have
+            // embeddings yet. This can happen if the message was saved in a mutation
+            // where the LLM is not available.
+            await generateAndSaveEmbeddings(ctx, component, {
+                ...args,
+                userId,
+                textEmbeddingModel,
+            }, [promptMessage]);
+        }
     }
-    return embeddingModel.provider;
+    const search = searchMessages
+        .map((m) => m.message)
+        .filter((m) => !!m)
+        .map(deserializeMessage);
+    const recent = prePromptDocs
+        .map((m) => m.message)
+        .filter((m) => !!m)
+        .map(deserializeMessage);
+    const inputMessages = messages.map(deserializeMessage);
+    const inputPrompt = promptArray.map(deserializeMessage);
+    const existingResponses = existingResponseDocs
+        .map((m) => m.message)
+        .filter((m) => !!m)
+        .map(deserializeMessage);
+    let processedMessages = args.contextHandler
+        ? await args.contextHandler(ctx, {
+            search,
+            recent,
+            inputMessages,
+            inputPrompt,
+            existingResponses,
+            userId,
+            threadId,
+        })
+        : [
+            ...search,
+            ...recent,
+            ...inputMessages,
+            ...inputPrompt,
+            ...existingResponses,
+        ];
+    // Process messages to inline localhost files (if not, file urls pointing to localhost will be sent to LLM providers)
+    if (process.env.CONVEX_CLOUD_URL?.startsWith("http://127.0.0.1")) {
+        processedMessages = await inlineMessagesFiles(processedMessages);
+    }
+    return {
+        messages: processedMessages,
+        order: promptMessage?.order,
+        stepOrder: promptMessage?.stepOrder,
+    };
+}
+export function getPromptArray(prompt) {
+    return !prompt
+        ? []
+        : Array.isArray(prompt)
+            ? prompt
+            : [{ role: "user", content: prompt }];
 }
 //# sourceMappingURL=search.js.map

package/dist/client/search.js.map

@@ -1 +1 @@
-{"version":3,"file":"search.js","sourceRoot":"","sources":["../../src/client/search.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EACL,qBAAqB,EACrB,uBAAuB,EACvB,WAAW,EACX,MAAM,GACP,MAAM,cAAc,CAAC;AAGtB,MAAM,8BAA8B,GAAG,GAAG,CAAC;AAS3C;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,GAA+B,EAC/B,SAAyB,EACzB,IAYC;IAED,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,4BAA4B,CAAC,CAAC;IACnE,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC;IACjC,4CAA4C;IAC5C,IAAI,QAAiC,CAAC;IACtC,MAAM,eAAe,GAAiB,EAAE,CAAC;IACzC,IACE,IAAI,CAAC,QAAQ;QACb,CAAC,IAAI,CAAC,cAAc,KAAK,CAAC,IAAI,IAAI,CAAC,yBAAyB,CAAC,EAC7D,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,GAAG,CAAC,QAAQ,CACjC,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EACzC;YACE,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;YAC7C,cAAc,EAAE;gBACd,QAAQ,EAAE,IAAI,CAAC,cAAc,IAAI,uBAAuB;gBACxD,MAAM,EAAE,IAAI;aACb;YACD,yBAAyB,EAAE,IAAI,CAAC,yBAAyB;YACzD,KAAK,EAAE,MAAM;YACb,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB,CACF,CAAC;QACF,QAAQ,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3C,eAAe,CAAC,IAAI;QAClB,+CAA+C;QAC/C,GAAG,IAAI,CAAC,OAAO,EAAE,CAClB,CAAC;IACJ,CAAC;IACD,IACE,CAAC,IAAI,CAAC,aAAa,EAAE,UAAU,IAAI,IAAI,CAAC,aAAa,EAAE,YAAY,CAAC;QACpE,IAAI,CAAC,aAAa,EAAE,KAAK,EACzB,CAAC;QACD,MAAM,aAAa,GAAG,eAAe,CAAC,IAAI,CACxC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,CAAC,yBAAyB,CAChD,EAAE,OAAO,CAAC;QACX,MAAM,gBAAgB,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC;QACzE,IAAI,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QACD,MAAM,WAAW,GAAG,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;QAC7C,MAAM,CAAC,WAAW,EAAE,uBAAuB,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,EAAE,gCAAgC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC5E,MAAM,CACJ,CAAC,IAAI,CAAC,cAAc,EAAE,aAAa,EAAE,YAAY,IAAI,WAAW,IAAI,GAAG,EACvE,0CAA0C,CAC3C,CAAC;QACF,IAAI,IAAI,CAAC,aAAa,EAAE,YAAY,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YAC3D,MAAM,IAAI,KAAK,CACb,uEAAuE,CACxE,CAAC;QACJ,CAAC;QACD,MAAM,eAAe,GACnB,IAAI,CAAC,aAAa,EAAE,YAAY,IAAI,IAAI;YACtC,CAAC,CAAC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC;YACjC,CAAC,CAAC,SAAS,CAAC;QAChB,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,SAAS,CACxC,SAAS,CAAC,QAAQ,CAAC,cAAc,EACjC;YACE,0BAA0B,EAAE,IAAI,EAAE,kBAAkB;gBAClD,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM;oBACZ,CAAC,IAAI,CAAC,QAAQ;wBACZ,CACE,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE;4BAC9C,QAAQ,EAAE,IAAI,CAAC,QAAQ;yBACxB,CAAC,CACH,EAAE,MAAM,CAAC,CAAC;gBACf,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,eAAe,EAAE,IAAI,CAAC,yBAAyB;YAC/C,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;YACtC,YAAY,EAAE;gBACZ,GAAG,qBAAqB;gBACxB,GAAG,IAAI,CAAC,aAAa,EAAE,YAAY;aACpC;YACD,IAAI;YACJ,oBAAoB,EAClB,IAAI,CAAC,aAAa,EAAE,oBAAoB;gBACxC,8BAA8B;YAChC,SAAS,EAAE,eAAe,EAAE,SAAS;YACrC,cAAc,EAAE,eAAe,EAAE,kBAAkB;gBACjD,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,kBAAkB,CAAC;gBAClD,CAAC,CAAC,SAAS;SACd,CACF,CAAC;QACF,gDAAgD;QAChD,eAAe,CAAC,OAAO,CACrB,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CACvD,CAAC;IACJ,CAAC;IACD,0EAA0E;IAC1E,OAAO,6BAA6B,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC;AAChE,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,6BAA6B,CAAC,IAAkB;IAC9D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IACE,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,WAAW;YACjC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAClC,CAAC;YACD,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC1C,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACjC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACtC,CAAC;YACH,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;aAAM,IAAI,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YACxC,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;gBACpE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,GAAG,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,cAA+D;IAE/D,IAAI,OAAO,cAAc,KAAK,QAAQ,EAAE,CAAC;QACvC,IAAI,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACjC,OAAO,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,cAAc,CAAC;IACxB,CAAC;IACD,OAAO,cAAc,CAAC,OAAO,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,cAA+D;IAE/D,IAAI,OAAO,cAAc,KAAK,QAAQ,EAAE,CAAC;QACvC,OAAO,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,CAAC;IAC1C,CAAC;IACD,OAAO,cAAc,CAAC,QAAQ,CAAC;AACjC,CAAC"}
+{"version":3,"file":"search.js","sourceRoot":"","sources":["../../src/client/search.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,IAAI,UAAU,GAGxB,MAAM,IAAI,CAAC;AACZ,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAExC,OAAO,EACL,uBAAuB,GAExB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,qBAAqB,EACrB,uBAAuB,EACvB,WAAW,EACX,YAAY,EACZ,eAAe,EACf,MAAM,EACN,MAAM,GACP,MAAM,cAAc,CAAC;AAUtB,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD,MAAM,8BAA8B,GAAG,GAAG,CAAC;AAC3C,4EAA4E;AAC5E,sCAAsC;AACtC,MAAM,yBAAyB,GAAG,MAAM,CAAC;AAOzC;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,GAA+B,EAC/B,SAAyB,EACzB,IAuBC;IAED,MAAM,EAAE,cAAc,EAAE,cAAc,EAAE,GAAG,MAAM,4BAA4B,CAC3E,GAAG,EACH,SAAS,EACT,IAAI,CACL,CAAC;IACF,OAAO,CAAC,GAAG,cAAc,EAAE,GAAG,cAAc,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,GAA+B,EAC/B,SAAyB,EACzB,IAuBC;IAED,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,4BAA4B,CAAC,CAAC;IACnE,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC;IACjC,4CAA4C;IAC5C,IAAI,QAAiC,CAAC;IACtC,IAAI,cAAc,GAAiB,EAAE,CAAC;IACtC,IAAI,cAAc,GAAiB,EAAE,CAAC;IACtC,MAAM,eAAe,GACnB,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,yBAAyB,CAAC;IACzD,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,cAAc,KAAK,CAAC,EAAE,CAAC;QAC/C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,GAAG,CAAC,QAAQ,CACjC,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EACzC;YACE,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;YAC7C,cAAc,EAAE;gBACd,QAAQ,EAAE,IAAI,CAAC,cAAc,IAAI,uBAAuB;gBACxD,MAAM,EAAE,IAAI;aACb;YACD,yBAAyB,EAAE,eAAe;YAC1C,KAAK,EAAE,MAAM;YACb,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB,CACF,CAAC;QACF,QAAQ,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC3C,cAAc,GAAG,6BAA6B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/D,CAAC;IACD,IACE,CAAC,IAAI,CAAC,aAAa,EAAE,UAAU,IAAI,IAAI,CAAC,aAAa,EAAE,YAAY,CAAC;QACpE,IAAI,CAAC,aAAa,EAAE,KAAK,EACzB,CAAC;QACD,IAAI,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC;QAC3B,IAAI,SAA+B,CAAC;QACpC,IAAI,cAAkC,CAAC;QACvC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,aAAa,GAAG,cAAc,CAAC,IAAI,CACvC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,eAAe,CACjC,CAAC;gBACF,IAAI,aAAa,EAAE,CAAC;oBAClB,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC;gBAC5B,CAAC;qBAAM,CAAC;oBACN,MAAM,kBAAkB,GAAG,MAAM,GAAG,CAAC,QAAQ,CAC3C,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EACzC;wBACE,SAAS,EAAE,eAAe;qBAC3B,CACF,CAAC;oBACF,IAAI,GAAG,kBAAkB,CAAC,IAAI,CAAC;oBAC/B,SAAS,GAAG,kBAAkB,CAAC,SAAS,CAAC;oBACzC,cAAc,GAAG,kBAAkB,CAAC,cAAc,CAAC;gBACrD,CAAC;gBACD,MAAM,CAAC,IAAI,EAAE,0CAA0C,CAAC,CAAC;YAC3D,CAAC;iBAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;gBACjC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC;gBAC1C,MAAM,CAAC,IAAI,EAAE,6CAA6C,CAAC,CAAC;YAC9D,CAAC;YACD,MAAM,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,YAAY,EAAE,CAAC;YACrC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpC,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;gBACtD,SAAS,GAAG,eAAe,CAAC,SAAS,CAAC;gBACtC,cAAc,GAAG,eAAe,CAAC,kBAAkB;oBACjD,CAAC,CAAC,YAAY,CAAC,eAAe,CAAC,kBAAkB,CAAC;oBAClD,CAAC,CAAC,SAAS,CAAC;YAChB,CAAC;QACH,CAAC;QACD,MAAM,aAAa,GAAG,MAAM,GAAG,CAAC,SAAS,CACvC,SAAS,CAAC,QAAQ,CAAC,cAAc,EACjC;YACE,0BAA0B,EAAE,IAAI,EAAE,kBAAkB;gBAClD,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM;oBACZ,CAAC,IAAI,CAAC,QAAQ;wBACZ,CACE,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE;4BAC9C,QAAQ,EAAE,IAAI,CAAC,QAAQ;yBACxB,CAAC,CACH,EAAE,MAAM,CAAC,CAAC;gBACf,CAAC,CAAC,SAAS;YACb,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,eAAe;YACf,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;YACtC,YAAY,EAAE;gBACZ,GAAG,qBAAqB;gBACxB,GAAG,IAAI,CAAC,aAAa,EAAE,YAAY;aACpC;YACD,IAAI;YACJ,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,UAAU;YAC1C,YAAY,EAAE,IAAI,CAAC,aAAa,EAAE,YAAY;YAC9C,oBAAoB,EAClB,IAAI,CAAC,aAAa,EAAE,oBAAoB;gBACxC,8BAA8B;YAChC,SAAS;YACT,cAAc;SACf,CACF,CAAC;QACF,gDAAgD;QAChD,cAAc,GAAG,6BAA6B,CAC5C,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAC3D,CAAC;IACJ,CAAC;IACD,0EAA0E;IAC1E,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,CAAC;AAC5C,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,6BAA6B,CAAC,IAAkB;IAC9D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IACE,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,WAAW;YACjC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAClC,CAAC;YACD,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC1C,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACjC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACtC,CAAC;YACH,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;aAAM,IAAI,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YACxC,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;gBACpE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,GAAG,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,GAAiB,EACjB,EACE,MAAM,EACN,QAAQ,EACR,GAAG,OAAO,EAK4D,EACxE,QAAoC;IASpC,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChC,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,UAMS,CAAC;IACd,MAAM,YAAY,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;IACvE,mDAAmD;IACnD,MAAM,WAAW,GAAG,YAAY;SAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;SAClC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IAClC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,MAAM,GAAG,YAAY;SACxB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,yBAAyB,CAAC,CAAC;SAC7D,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,6BAA6B;IAC7B,MAAM,cAAc,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE;QAC1C,GAAG,OAAO;QACV,MAAM;QACN,QAAQ;QACR,MAAM;KACP,CAAC,CAAC;IACH,6FAA6F;IAC7F,MAAM,gBAAgB,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3D,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QAC3B,gBAAgB,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IACH,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QACtD,uBAAuB,CAAC,SAAS,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACvD,UAAU,GAAG,EAAE,OAAO,EAAE,gBAAgB,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;IAC/D,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,GAAiB,EACjB,EACE,MAAM,EACN,QAAQ,EACR,MAAM,EACN,WAAW,EACX,OAAO,EACP,SAAS,EACT,YAAY,EACZ,kBAAkB,EAClB,YAAY,GAQ0D;IAExE,MAAM,cAAc,GAAG,kBAAkB,CAAC;IAC1C,MAAM,CACJ,cAAc,EACd,8DAA8D,CAC/D,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC;QAC9B,GAAG,YAAY;QACf,KAAK,EAAE,cAAc;QACrB,MAAM;QACN,WAAW;QACX,OAAO;KACR,CAAC,CAAC;IACH,IAAI,YAAY,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACjC,MAAM,YAAY,CAAC,GAAG,EAAE;YACtB,MAAM;YACN,QAAQ;YACR,SAAS;YACT,KAAK,EAAE,YAAY,CAAC,cAAc,CAAC;YACnC,QAAQ,EAAE,eAAe,CAAC,cAAc,CAAC;YACzC,gBAAgB,EAAE,SAAS;YAC3B,KAAK,EAAE;gBACL,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;gBAChC,YAAY,EAAE,CAAC;gBACf,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;aACjC;SACF,CAAC,CAAC;IACL,CAAC;IACD,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC;AAC3C,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,GAAiB,EACjB,SAAyB,EACzB,IAKiD,EACjD,QAAsB;IAEtB,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC;IACpE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO;IACT,CAAC;IACD,MAAM,UAAU,GAAG,MAAM,aAAa,CACpC,GAAG,EACH,IAAI,EACJ,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAQ,CAAC,CAC/B,CAAC;IACF,IAAI,UAAU,IAAI,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,EAAE,CAAC;QAC7D,MAAM,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE;YACxD,eAAe,EAAE,UAAU,CAAC,SAAS;YACrC,OAAO,EAAE,OAAO;iBACb,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACd,SAAS,EAAE,CAAC,CAAC,GAAG;gBAChB,KAAK,EAAE,UAAU,CAAC,KAAK;gBACvB,KAAK,EAAE,UAAU;gBACjB,MAAM,EAAE,CAAC,CAAC,MAAM;gBAChB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC,CAAE;aAC/B,CAAC,CAAC;iBACF,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,IAAI,CAAC;SACpC,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,GAAiB,EACjB,SAAyB,EACzB,IAQQ;IAMR,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC;IAEtD,MAAM,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAEhD,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM;QACnC,CAAC,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;QAClC,CAAC,CAAC,IAAI,CAAC,eAAe;YACpB,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;gBACrB,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;gBACpC,CAAC,CAAC,SAAS,CAAC;IAClB,0EAA0E;IAC1E,MAAM,EAAE,cAAc,EAAE,cAAc,EAAE,GAAG,MAAM,4BAA4B,CAC3E,GAAG,EACH,SAAS,EACT;QACE,MAAM;QACN,QAAQ;QACR,eAAe,EAAE,IAAI,CAAC,eAAe;QACrC,UAAU;QACV,cAAc,EAAE,IAAI,CAAC,cAAc,IAAI,EAAE;QACzC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,CACJ,kBAAkB,EAClB,8FAA8F,CAC/F,CAAC;YACF,OAAO;gBACL,SAAS,EAAE,CACT,MAAM,SAAS,CAAC,GAAG,EAAE;oBACnB,GAAG,IAAI;oBACP,MAAM;oBACN,MAAM,EAAE,CAAC,IAAI,CAAC;oBACd,kBAAkB;iBACnB,CAAC,CACH,CAAC,UAAU,CAAC,CAAC,CAAC;gBACf,kBAAkB;aACnB,CAAC;QACJ,CAAC;KACF,CACF,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAI,CAAC,eAAe;QAC7C,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,CAAC,eAAe,CAAC;QACjE,CAAC,CAAC,CAAC,CAAC,CAAC;IACP,MAAM,aAAa,GACjB,kBAAkB,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7E,IAAI,aAAa,GAAG,cAAc,CAAC;IACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;IACrC,IAAI,oBAAoB,GAAiB,EAAE,CAAC;IAC5C,IAAI,aAAa,EAAE,CAAC;QAClB,aAAa,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,kBAAkB,CAAC,CAAC;QAC5D,oBAAoB,GAAG,cAAc,CAAC,KAAK,CAAC,kBAAkB,GAAG,CAAC,CAAC,CAAC;QACpE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,uEAAuE;YACvE,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;gBAC1B,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,WAAW,IAAI,kBAAkB,EAAE,CAAC;YACrD,wEAAwE;YACxE,yEAAyE;YACzE,kCAAkC;YAClC,MAAM,yBAAyB,CAC7B,GAAG,EACH,SAAS,EACT;gBACE,GAAG,IAAI;gBACP,MAAM;gBACN,kBAAkB;aACnB,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,cAAc;SAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;SACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;SAClB,GAAG,CAAC,kBAAkB,CAAC,CAAC;IAC3B,MAAM,MAAM,GAAG,aAAa;SACzB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;SACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;SAClB,GAAG,CAAC,kBAAkB,CAAC,CAAC;IAC3B,MAAM,aAAa,GAAG,QAAQ,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;IACvD,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;IACxD,MAAM,iBAAiB,GAAG,oBAAoB;SAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;SACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;SAClB,GAAG,CAAC,kBAAkB,CAAC,CAAC;IAE3B,IAAI,iBAAiB,GAAG,IAAI,CAAC,cAAc;QACzC,CAAC,CAAC,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE;YAC7B,MAAM;YACN,MAAM;YACN,aAAa;YACb,WAAW;YACX,iBAAiB;YACjB,MAAM;YACN,QAAQ;SACT,CAAC;QACJ,CAAC,CAAC;YACE,GAAG,MAAM;YACT,GAAG,MAAM;YACT,GAAG,aAAa;YAChB,GAAG,WAAW;YACd,GAAG,iBAAiB;SACrB,CAAC;IAEN,qHAAqH;IACrH,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;QACjE,iBAAiB,GAAG,MAAM,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;IACnE,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,iBAAiB;QAC3B,KAAK,EAAE,aAAa,EAAE,KAAK;QAC3B,SAAS,EAAE,aAAa,EAAE,SAAS;KACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,MAAuD;IAEvD,OAAO,CAAC,MAAM;QACZ,CAAC,CAAC,EAAE;QACJ,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YACrB,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;AAC5C,CAAC"}

package/dist/client/start.d.ts

@@ -0,0 +1,83 @@
+import { type CallSettings, type GenerateObjectResult, type IdGenerator, type LanguageModel, type ModelMessage, type StepResult, type StopCondition, type ToolSet } from "ai";
+import type { ActionCtx, AgentComponent, Config, Options } from "./types.js";
+import type { Message, MessageDoc } from "../validators.js";
+import { type ModelOrMetadata } from "../shared.js";
+import type { Agent } from "./index.js";
+export declare function start<T, Tools extends ToolSet = ToolSet, CustomCtx extends object = object>(ctx: ActionCtx & CustomCtx, component: AgentComponent, 
+/**
+ * These are the arguments you'll pass to the LLM call such as
+ * `generateText` or `streamText`. This function will look up the context
+ * and provide functions to save the steps, abort the generation, and more.
+ * The type of the arguments returned infers from the type of the arguments
+ * you pass here.
+ */
+args: T & {
+    /**
+     * If provided, this message will be used as the "prompt" for the LLM call,
+     * instead of the prompt or messages.
+     * This is useful if you want to first save a user message, then use it as
+     * the prompt for the LLM call in another call.
+     */
+    promptMessageId?: string;
+    /**
+     * The model to use for the LLM calls. This will override the model specified
+     * in the Agent constructor.
+     */
+    model?: LanguageModel;
+    /**
+     * The tools to use for the tool calls. This will override tools specified
+     * in the Agent constructor or createThread / continueThread.
+     */
+    tools?: Tools;
+    /**
+     * The single prompt message to use for the LLM call. This will be the
+     * last message in the context. If it's a string, it will be a user role.
+     */
+    prompt?: string | (ModelMessage | Message)[];
+    /**
+     * If provided alongside prompt, the ordering will be:
+     * 1. system prompt
+     * 2. search context
+     * 3. recent messages
+     * 4. these messages
+     * 5. prompt messages, including those already on the same `order` as
+     *   the promptMessageId message, if provided.
+     */
+    messages?: (ModelMessage | Message)[];
+    /**
+     * The abort signal to be passed to the LLM call. If triggered, it will
+     * mark the pending message as failed. If the generation is asynchronously
+     * aborted, it will trigger this signal when detected.
+     */
+    abortSignal?: AbortSignal;
+    stopWhen?: StopCondition<Tools> | Array<StopCondition<Tools>>;
+    _internal?: {
+        generateId?: IdGenerator;
+    };
+}, { threadId, ...opts }: Options & Config & {
+    userId?: string | null;
+    threadId?: string;
+    languageModel: LanguageModel;
+    agentName: string;
+    agentForToolCtx?: Agent;
+}): Promise<{
+    args: T & {
+        system?: string;
+        model: LanguageModel;
+        messages: ModelMessage[];
+        tools?: Tools;
+    } & CallSettings;
+    order: number;
+    stepOrder: number;
+    userId: string | undefined;
+    promptMessageId: string | undefined;
+    updateModel: (model: ModelOrMetadata | undefined) => void;
+    save: <TOOLS extends ToolSet>(toSave: {
+        step: StepResult<TOOLS>;
+    } | {
+        object: GenerateObjectResult<unknown>;
+    }, createPendingMessage?: boolean) => Promise<void>;
+    fail: (reason: string) => Promise<void>;
+    getSavedMessages: () => MessageDoc[];
+}>;
+//# sourceMappingURL=start.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"start.d.ts","sourceRoot":"","sources":["../../src/client/start.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,YAAY,EACjB,KAAK,oBAAoB,EACzB,KAAK,WAAW,EAChB,KAAK,aAAa,EAClB,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,KAAK,aAAa,EAClB,KAAK,OAAO,EACb,MAAM,IAAI,CAAC;AAMZ,OAAO,KAAK,EACV,SAAS,EACT,cAAc,EACd,MAAM,EACN,OAAO,EAER,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC5D,OAAO,EAGL,KAAK,eAAe,EACrB,MAAM,cAAc,CAAC;AAEtB,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAIxC,wBAAsB,KAAK,CACzB,CAAC,EACD,KAAK,SAAS,OAAO,GAAG,OAAO,EAC/B,SAAS,SAAS,MAAM,GAAG,MAAM,EAEjC,GAAG,EAAE,SAAS,GAAG,SAAS,EAC1B,SAAS,EAAE,cAAc;AACzB;;;;;;GAMG;AACH,IAAI,EAAE,CAAC,GAAG;IACR;;;;;OAKG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;;OAGG;IACH,KAAK,CAAC,EAAE,aAAa,CAAC;IACtB;;;OAGG;IACH,KAAK,CAAC,EAAE,KAAK,CAAC;IACd;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,CAAC,YAAY,GAAG,OAAO,CAAC,EAAE,CAAC;IAC7C;;;;;;;;OAQG;IACH,QAAQ,CAAC,EAAE,CAAC,YAAY,GAAG,OAAO,CAAC,EAAE,CAAC;IACtC;;;;OAIG;IACH,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,QAAQ,CAAC,EAAE,aAAa,CAAC,KAAK,CAAC,GAAG,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IAC9D,SAAS,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,WAAW,CAAA;KAAE,CAAC;CAC1C,EACD,EACE,QAAQ,EACR,GAAG,IAAI,EACR,EAAE,OAAO,GACR,MAAM,GAAG;IACP,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,aAAa,CAAC;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,KAAK,CAAC;CACzB,GACF,OAAO,CAAC;IACT,IAAI,EAAE,CAAC,GAAG;QACR,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,aAAa,CAAC;QACrB,QAAQ,EAAE,YAAY,EAAE,CAAC;QACzB,KAAK,CAAC,EAAE,KAAK,CAAC;KACf,GAAG,YAAY,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,eAAe,EAAE,MAAM,GAAG,SAAS,CAAC;IACpC,WAAW,EAAE,CAAC,KAAK,EAAE,eAAe,GAAG,SAAS,KAAK,IAAI,CAAC;IAC1D,IAAI,EAAE,CAAC,KAAK,SAAS,OAAO,EAC1B,MAAM,EACF;QAAE,IAAI,EAAE,UAAU,CAAC,KAAK,CAAC,CAAA;KAAE,GAC3B;QAAE,MAAM,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAA;KAAE,EAC7C,oBAAoB,CAAC,EAAE,OAAO,KAC3B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,gBAAgB,EAAE,MAAM,UAAU,EAAE,CAAC;CACtC,CAAC,CA8LD"}

package/dist/client/start.js

@@ -0,0 +1,171 @@
+import { stepCountIs, } from "ai";
+import { serializeNewMessagesInStep, serializeObjectResult, } from "../mapping.js";
+import { embedMessages, fetchContextWithPrompt } from "./search.js";
+import { saveMessages } from "./messages.js";
+import { getModelName, getProviderName, } from "../shared.js";
+import { wrapTools } from "./createTool.js";
+import { omit } from "convex-helpers";
+import { saveInputMessages } from "./saveInputMessages.js";
+export async function start(ctx, component, 
+/**
+ * These are the arguments you'll pass to the LLM call such as
+ * `generateText` or `streamText`. This function will look up the context
+ * and provide functions to save the steps, abort the generation, and more.
+ * The type of the arguments returned infers from the type of the arguments
+ * you pass here.
+ */
+args, { threadId, ...opts }) {
+    const userId = opts.userId ??
+        (threadId &&
+            (await ctx.runQuery(component.threads.getThread, { threadId }))
+                ?.userId) ??
+        undefined;
+    const context = await fetchContextWithPrompt(ctx, component, {
+        ...opts,
+        userId,
+        threadId,
+        messages: args.messages,
+        prompt: args.prompt,
+        promptMessageId: args.promptMessageId,
+    });
+    const saveMessages = opts.storageOptions?.saveMessages ?? "promptAndOutput";
+    const { promptMessageId, pendingMessage, savedMessages } = threadId && saveMessages !== "none"
+        ? await saveInputMessages(ctx, component, {
+            ...opts,
+            userId,
+            threadId,
+            prompt: args.prompt,
+            messages: args.messages,
+            promptMessageId: args.promptMessageId,
+            storageOptions: { saveMessages },
+        })
+        : {
+            promptMessageId: args.promptMessageId,
+            pendingMessage: undefined,
+            savedMessages: [],
+        };
+    const order = pendingMessage?.order ?? context.order;
+    const stepOrder = pendingMessage?.stepOrder ?? context.stepOrder;
+    let pendingMessageId = pendingMessage?._id;
+    const model = args.model ?? opts.languageModel;
+    let activeModel = model;
+    const fail = async (reason) => {
+        if (pendingMessageId) {
+            await ctx.runMutation(component.messages.finalizeMessage, {
+                messageId: pendingMessageId,
+                result: { status: "failed", error: reason },
+            });
+        }
+    };
+    if (args.abortSignal) {
+        const abortSignal = args.abortSignal;
+        abortSignal.addEventListener("abort", async () => {
+            await fail(abortSignal.reason?.toString() ?? "abortSignal");
+        }, { once: true });
+    }
+    const toolCtx = {
+        ...ctx,
+        userId,
+        threadId,
+        promptMessageId,
+        agent: opts.agentForToolCtx,
+    };
+    const tools = wrapTools(toolCtx, args.tools);
+    const aiArgs = {
+        ...opts.callSettings,
+        providerOptions: opts.providerOptions,
+        ...omit(args, ["promptMessageId", "messages", "prompt"]),
+        model,
+        messages: context.messages,
+        stopWhen: args.stopWhen ?? (opts.maxSteps ? stepCountIs(opts.maxSteps) : undefined),
+        tools,
+    };
+    if (pendingMessageId) {
+        if (!aiArgs._internal?.generateId) {
+            aiArgs._internal = {
+                ...aiArgs._internal,
+                generateId: pendingMessageId
+                    ? () => pendingMessageId ?? crypto.randomUUID()
+                    : undefined,
+            };
+        }
+    }
+    return {
+        args: aiArgs,
+        order: order ?? 0,
+        stepOrder: stepOrder ?? 0,
+        userId,
+        promptMessageId,
+        getSavedMessages: () => savedMessages,
+        updateModel: (model) => {
+            if (model) {
+                activeModel = model;
+            }
+        },
+        fail,
+        save: async (toSave, createPendingMessage) => {
+            if (threadId && saveMessages !== "none") {
+                const serialized = "object" in toSave
+                    ? await serializeObjectResult(ctx, component, toSave.object, activeModel)
+                    : await serializeNewMessagesInStep(ctx, component, toSave.step, activeModel);
+                const embeddings = await embedMessages(ctx, { threadId, ...opts, userId }, serialized.messages.map((m) => m.message));
+                if (createPendingMessage) {
+                    serialized.messages.push({
+                        message: { role: "assistant", content: [] },
+                        status: "pending",
+                    });
+                    embeddings?.vectors.push(null);
+                }
+                const saved = await ctx.runMutation(component.messages.addMessages, {
+                    userId,
+                    threadId,
+                    agentName: opts.agentName,
+                    promptMessageId,
+                    pendingMessageId,
+                    messages: serialized.messages,
+                    embeddings,
+                    failPendingSteps: false,
+                });
+                const lastMessage = saved.messages.at(-1);
+                if (createPendingMessage) {
+                    if (lastMessage.status === "failed") {
+                        pendingMessageId = undefined;
+                        savedMessages.push(...saved.messages);
+                        await fail(lastMessage.error ??
+                            "Aborting - the pending message was marked as failed");
+                    }
+                    else {
+                        pendingMessageId = lastMessage._id;
+                        savedMessages.push(...saved.messages.slice(0, -1));
+                    }
+                }
+                else {
+                    pendingMessageId = undefined;
+                    savedMessages.push(...saved.messages);
+                }
+            }
+            const output = "object" in toSave ? toSave.object : toSave.step;
+            if (opts.rawRequestResponseHandler) {
+                await opts.rawRequestResponseHandler(ctx, {
+                    userId,
+                    threadId,
+                    agentName: opts.agentName,
+                    request: output.request,
+                    response: output.response,
+                });
+            }
+            if (opts.usageHandler && output.usage) {
+                await opts.usageHandler(ctx, {
+                    userId,
+                    threadId,
+                    agentName: opts.agentName,
+                    model: getModelName(activeModel),
+                    provider: getProviderName(activeModel),
+                    usage: output.usage,
+                    providerMetadata: output.providerMetadata,
+                });
+            }
+        },
+    };
+}
+//# sourceMappingURL=start.js.map