@convex-dev/agent 0.5.0-alpha.1 → 0.6.0-alpha.0

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, } 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 { docsToModelMessages, toModelMessage } 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,85 +36,157 @@ 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) {
-        const targetMessage = contextMessages.find((m) => m._id === args.upToAndIncludingMessageId)?.message;
-        const messagesToSearch = targetMessage ? [targetMessage] : args.messages;
+    if ((opts.searchOptions?.textSearch || opts.searchOptions?.vectorSearch) &&
+        opts.searchOptions?.limit) {
         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;
+                const effectiveModel = embeddingFields.embeddingModel ?? embeddingFields.textEmbeddingModel;
+                embeddingModel = effectiveModel
+                    ? getModelName(effectiveModel)
+                    : undefined;
+                // TODO: if the text matches the target message, save the embedding
+                // for the target message and return the embeddingId on the message.
+            }
         }
-        const embeddingFields = opts.searchOptions?.vectorSearch
-            ? 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.userId ??
                     (args.threadId &&
                         (await ctx.runQuery(component.threads.getThread, {
                             threadId: args.threadId,
-                        }))?.userId)
+                        }))?.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?.embeddingModel,
+            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(contextMessages.sort((a, b) => 
-    // Sort the raw MessageDocs by order and stepOrder
-    a.order === b.order ? a.stepOrder - b.stepOrder : a.order - b.order));
+    return { recentMessages, searchMessages };
 }
 /**
  * Filter out tool messages that don't have both a tool call and response.
+ * For the approval workflow, tool calls with approval responses (but no tool-results yet)
+ * should also be kept.
  * @param docs The messages to filter.
  * @returns The filtered messages.
  */
 export function filterOutOrphanedToolMessages(docs) {
     const toolCallIds = new Set();
+    const toolResultIds = new Set();
+    // Track approval workflow: toolCallId → approvalId
+    const approvalRequestsByToolCallId = new Map();
+    // Track which approvalIds have responses
+    const approvalResponseIds = new Set();
     const result = [];
     for (const doc of docs) {
-        if (doc.message?.role === "assistant" &&
-            Array.isArray(doc.message.content)) {
+        if (doc.message && Array.isArray(doc.message.content)) {
             for (const content of doc.message.content) {
                 if (content.type === "tool-call") {
                     toolCallIds.add(content.toolCallId);
                 }
+                else if (content.type === "tool-result") {
+                    toolResultIds.add(content.toolCallId);
+                }
+                else if (content.type === "tool-approval-request") {
+                    const approvalRequest = content;
+                    approvalRequestsByToolCallId.set(approvalRequest.toolCallId, approvalRequest.approvalId);
+                }
+                else if (content.type === "tool-approval-response") {
+                    const approvalResponse = content;
+                    approvalResponseIds.add(approvalResponse.approvalId);
+                }
             }
-            result.push(doc);
         }
-        else if (doc.message?.role === "tool") {
-            if (doc.message.content.every((c) => toolCallIds.has(c.toolCallId))) {
-                result.push(doc);
+    }
+    // Helper: check if tool call has a corresponding approval response
+    const hasApprovalResponse = (toolCallId) => {
+        const approvalId = approvalRequestsByToolCallId.get(toolCallId);
+        return approvalId !== undefined && approvalResponseIds.has(approvalId);
+    };
+    for (const doc of docs) {
+        if (doc.message?.role === "assistant" &&
+            Array.isArray(doc.message.content)) {
+            const content = doc.message.content.filter((p) => p.type !== "tool-call" ||
+                toolResultIds.has(p.toolCallId) ||
+                hasApprovalResponse(p.toolCallId));
+            if (content.length) {
+                result.push({
+                    ...doc,
+                    message: {
+                        ...doc.message,
+                        content,
+                    },
+                });
             }
-            else {
-                console.debug("Filtering out orphaned tool message", doc);
+        }
+        else if (doc.message?.role === "tool") {
+            const content = doc.message.content.filter((c) => {
+                // tool-result parts have toolCallId
+                if (c.type === "tool-result") {
+                    return toolCallIds.has(c.toolCallId);
+                }
+                // tool-approval-response parts don't have toolCallId, so include them
+                return true;
+            });
+            if (content.length) {
+                result.push({
+                    ...doc,
+                    message: {
+                        ...doc.message,
+                        content,
+                    },
+                });
             }
         }
         else {
@@ -111,4 +195,231 @@ export function filterOutOrphanedToolMessages(docs) {
     }
     return result;
 }
+/**
+ * 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) {
+    const textEmbeddingModel = options.embeddingModel ?? options.textEmbeddingModel;
+    if (!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(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, args) {
+    const { userId, threadId, values, abortSignal, headers, agentName, usageHandler, textEmbeddingModel, embeddingModel, callSettings, } = args;
+    const effectiveEmbeddingModel = embeddingModel ?? textEmbeddingModel;
+    assert(effectiveEmbeddingModel, "an embeddingModel (or textEmbeddingModel) is required to be set for vector search");
+    const result = await embedMany_({
+        ...callSettings,
+        model: effectiveEmbeddingModel,
+        values,
+        abortSignal,
+        headers,
+    });
+    if (usageHandler && result.usage) {
+        await usageHandler(ctx, {
+            userId,
+            threadId,
+            agentName,
+            model: getModelName(effectiveEmbeddingModel),
+            provider: getProviderName(effectiveEmbeddingModel),
+            providerMetadata: undefined,
+            usage: {
+                inputTokens: result.usage.tokens,
+                outputTokens: 0,
+                totalTokens: result.usage.tokens,
+                // These detail fields are required by LanguageModelUsage type but we don't
+                // have the granular data, so we provide objects with undefined values.
+                inputTokenDetails: {
+                    cacheReadTokens: undefined,
+                    cacheWriteTokens: undefined,
+                    noCacheTokens: undefined,
+                },
+                outputTokenDetails: {
+                    textTokens: undefined,
+                    reasoningTokens: undefined,
+                },
+            },
+        });
+    }
+    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 effectiveEmbeddingModel = args.embeddingModel ?? args.textEmbeddingModel;
+    if (!effectiveEmbeddingModel) {
+        throw new Error("an embeddingModel (or textEmbeddingModel) is required to generate and save embeddings");
+    }
+    const toEmbed = messages.filter((m) => !m.embeddingId && m.message);
+    if (toEmbed.length === 0) {
+        return;
+    }
+    const embeddings = await embedMessages(ctx, { ...args, embeddingModel: effectiveEmbeddingModel }, 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),
+        });
+    }
+}
+/**
+ * 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, embeddingModel } = args;
+    const effectiveEmbeddingModel = embeddingModel ?? textEmbeddingModel;
+    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(effectiveEmbeddingModel, "An embeddingModel (or 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],
+                    embeddingModel: effectiveEmbeddingModel,
+                })).embeddings[0],
+                embeddingModel: effectiveEmbeddingModel,
+            };
+        },
+    });
+    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 && effectiveEmbeddingModel) {
+            // 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,
+                embeddingModel: effectiveEmbeddingModel,
+            }, [promptMessage]);
+        }
+    }
+    const search = docsToModelMessages(searchMessages);
+    const recent = docsToModelMessages(prePromptDocs);
+    const inputMessages = messages.map(toModelMessage);
+    const inputPrompt = promptArray.map(toModelMessage);
+    const existingResponses = docsToModelMessages(existingResponseDocs);
+    const allMessages = [
+        ...search,
+        ...recent,
+        ...inputMessages,
+        ...inputPrompt,
+        ...existingResponses,
+    ];
+    let processedMessages = args.contextHandler
+        ? await args.contextHandler(ctx, {
+            allMessages,
+            search,
+            recent,
+            inputMessages,
+            inputPrompt,
+            existingResponses,
+            userId,
+            threadId,
+        })
+        : allMessages;
+    // 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,GACZ,MAAM,cAAc,CAAC;AAGtB,MAAM,8BAA8B,GAAG,GAAG,CAAC;AAO3C;;;;;;;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,IAAI,IAAI,CAAC,aAAa,EAAE,UAAU,IAAI,IAAI,CAAC,aAAa,EAAE,YAAY,EAAE,CAAC;QACvE,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,GAAG,IAAI,CAAC,aAAa,EAAE,YAAY;YACtD,CAAC,CAAC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC;YACjC,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,SAAS,CACxC,SAAS,CAAC,QAAQ,CAAC,cAAc,EACjC;YACE,0BAA0B,EAAE,IAAI,EAAE,kBAAkB;gBAClD,CAAC,CAAC,IAAI,CAAC,MAAM;oBACX,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;gBACd,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,cAAc;SAChD,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,CAClC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;IAC5B,kDAAkD;IAClD,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CACpE,CACF,CAAC;AACJ,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"}
+{"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;AAWtB,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpE,MAAM,8BAA8B,GAAG,GAAG,CAAC;AAC3C,4EAA4E;AAC5E,sCAAsC;AACtC,MAAM,yBAAyB,GAAG,MAAM,CAAC;AAiBzC;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,GAAuC,EACvC,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,GAAuC,EACvC,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,MAAM,cAAc,GAClB,eAAe,CAAC,cAAc,IAAI,eAAe,CAAC,kBAAkB,CAAC;gBACvE,cAAc,GAAG,cAAc;oBAC7B,CAAC,CAAC,YAAY,CAAC,cAAc,CAAC;oBAC9B,CAAC,CAAC,SAAS,CAAC;gBACd,mEAAmE;gBACnE,oEAAoE;YACtE,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;;;;;;GAMG;AACH,MAAM,UAAU,6BAA6B,CAAC,IAAkB;IAC9D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IACxC,mDAAmD;IACnD,MAAM,4BAA4B,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC/D,yCAAyC;IACzC,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAU,CAAC;IAE9C,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACtD,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;qBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBAC1C,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACxC,CAAC;qBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;oBACpD,MAAM,eAAe,GAAG,OAIvB,CAAC;oBACF,4BAA4B,CAAC,GAAG,CAC9B,eAAe,CAAC,UAAU,EAC1B,eAAe,CAAC,UAAU,CAC3B,CAAC;gBACJ,CAAC;qBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,wBAAwB,EAAE,CAAC;oBACrD,MAAM,gBAAgB,GAAG,OAGxB,CAAC;oBACF,mBAAmB,CAAC,GAAG,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,MAAM,mBAAmB,GAAG,CAAC,UAAkB,EAAE,EAAE;QACjD,MAAM,UAAU,GAAG,4BAA4B,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChE,OAAO,UAAU,KAAK,SAAS,IAAI,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACzE,CAAC,CAAC;IAEF,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,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CACxC,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,KAAK,WAAW;gBACtB,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC/B,mBAAmB,CAAC,CAAC,CAAC,UAAU,CAAC,CACpC,CAAC;YACF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,MAAM,CAAC,IAAI,CAAC;oBACV,GAAG,GAAG;oBACN,OAAO,EAAE;wBACP,GAAG,GAAG,CAAC,OAAO;wBACd,OAAO;qBACR;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;aAAM,IAAI,GAAG,CAAC,OAAO,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YACxC,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC/C,oCAAoC;gBACpC,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBAC7B,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;gBACvC,CAAC;gBACD,sEAAsE;gBACtE,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,MAAM,CAAC,IAAI,CAAC;oBACV,GAAG,GAAG;oBACN,OAAO,EAAE;wBACP,GAAG,GAAG,CAAC,OAAO;wBACd,OAAO;qBACR;iBACF,CAAC,CAAC;YACL,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,GAAc,EACd,EACE,MAAM,EACN,QAAQ,EACR,GAAG,OAAO,EAQX,EACD,QAAoC;IASpC,MAAM,kBAAkB,GACtB,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,kBAAkB,CAAC;IACvD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,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,kBAAkB,CAAC,CAAC;QAC/C,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,GAAc,EACd,IAUC;IAED,MAAM,EACJ,MAAM,EACN,QAAQ,EACR,MAAM,EACN,WAAW,EACX,OAAO,EACP,SAAS,EACT,YAAY,EACZ,kBAAkB,EAClB,cAAc,EACd,YAAY,GACb,GAAG,IAAI,CAAC;IACT,MAAM,uBAAuB,GAAG,cAAc,IAAI,kBAAkB,CAAC;IACrE,MAAM,CACJ,uBAAuB,EACvB,mFAAmF,CACpF,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC;QAC9B,GAAG,YAAY;QACf,KAAK,EAAE,uBAAuB;QAC9B,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,uBAAuB,CAAC;YAC5C,QAAQ,EAAE,eAAe,CAAC,uBAAuB,CAAC;YAClD,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;gBAChC,2EAA2E;gBAC3E,uEAAuE;gBACvE,iBAAiB,EAAE;oBACjB,eAAe,EAAE,SAAS;oBAC1B,gBAAgB,EAAE,SAAS;oBAC3B,aAAa,EAAE,SAAS;iBACzB;gBACD,kBAAkB,EAAE;oBAClB,UAAU,EAAE,SAAS;oBACrB,eAAe,EAAE,SAAS;iBAC3B;aACF;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,GAAc,EACd,SAAyB,EACzB,IASiD,EACjD,QAAsB;IAEtB,MAAM,uBAAuB,GAC3B,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,kBAAkB,CAAC;IACjD,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CACb,uFAAuF,CACxF,CAAC;IACJ,CAAC;IACD,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,EAAE,GAAG,IAAI,EAAE,cAAc,EAAE,uBAAuB,EAAE,EACpD,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,GAAc,EACd,SAAyB,EACzB,IAQQ;IAMR,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,kBAAkB,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC;IACtE,MAAM,uBAAuB,GAAG,cAAc,IAAI,kBAAkB,CAAC;IAErE,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,uBAAuB,EACvB,mHAAmH,CACpH,CAAC;YACF,OAAO;gBACL,SAAS,EAAE,CACT,MAAM,SAAS,CAAC,GAAG,EAAE;oBACnB,GAAG,IAAI;oBACP,MAAM;oBACN,MAAM,EAAE,CAAC,IAAI,CAAC;oBACd,cAAc,EAAE,uBAAuB;iBACxC,CAAC,CACH,CAAC,UAAU,CAAC,CAAC,CAAC;gBACf,cAAc,EAAE,uBAAuB;aACxC,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,uBAAuB,EAAE,CAAC;YAC1D,wEAAwE;YACxE,yEAAyE;YACzE,kCAAkC;YAClC,MAAM,yBAAyB,CAC7B,GAAG,EACH,SAAS,EACT;gBACE,GAAG,IAAI;gBACP,MAAM;gBACN,cAAc,EAAE,uBAAuB;aACxC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,mBAAmB,CAAC,cAAc,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,mBAAmB,CAAC,aAAa,CAAC,CAAC;IAClD,MAAM,aAAa,GAAG,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACnD,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACpD,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,oBAAoB,CAAC,CAAC;IAEpE,MAAM,WAAW,GAAG;QAClB,GAAG,MAAM;QACT,GAAG,MAAM;QACT,GAAG,aAAa;QAChB,GAAG,WAAW;QACd,GAAG,iBAAiB;KACrB,CAAC;IACF,IAAI,iBAAiB,GAAG,IAAI,CAAC,cAAc;QACzC,CAAC,CAAC,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE;YAC7B,WAAW;YACX,MAAM;YACN,MAAM;YACN,aAAa;YACb,WAAW;YACX,iBAAiB;YACjB,MAAM;YACN,QAAQ;SACT,CAAC;QACJ,CAAC,CAAC,WAAW,CAAC;IAEhB,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,84 @@
+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 startGeneration<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[];
+        prompt?: never;
+        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,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAC7E,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;AAKxC,wBAAsB,eAAe,CACnC,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,CAAC,EAAE,aAAa,CAAC;IAC9B,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,MAAM,CAAC,EAAE,KAAK,CAAC;QACf,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,CAgMD"}