@convex-dev/agent 0.1.10 → 0.1.12

package/src/client/index.ts

@@ -417,6 +417,8 @@ export class Agent<AgentTools extends ToolSet> {
               userId,
               threadId,
               promptMessageId: messageId,
+              model: aiArgs.model.modelId,
+              provider: aiArgs.model.provider,
               step,
             });
           }
@@ -579,6 +581,8 @@ export class Agent<AgentTools extends ToolSet> {
           const saved = await this.saveStep(ctx, {
             userId,
             threadId,
+            model: aiArgs.model.modelId,
+            provider: aiArgs.model.provider,
             promptMessageId: messageId,
             step,
           });
@@ -662,6 +666,8 @@ export class Agent<AgentTools extends ToolSet> {
           promptMessageId: messageId,
           result,
           userId,
+          model: aiArgs.model.modelId,
+          provider: aiArgs.model.provider,
         });
       }
       result.messageId = messageId;
@@ -760,6 +766,8 @@ export class Agent<AgentTools extends ToolSet> {
               logprobs: undefined,
               toJsonResponse: stream.toTextStreamResponse,
             },
+            model: aiArgs.model.modelId,
+            provider: aiArgs.model.provider,
           });
         }
         if (trackUsage && result.usage) {
@@ -1071,9 +1079,7 @@ export class Agent<AgentTools extends ToolSet> {
       const targetMessage = contextMessages.find(
         (m) => m._id === args.upToAndIncludingMessageId
       )?.message;
-      const messagesToSearch = targetMessage
-        ? [targetMessage, ...args.messages]
-        : args.messages;
+      const messagesToSearch = targetMessage ? [targetMessage] : args.messages;
       if (!("runAction" in ctx)) {
         throw new Error("searchUserMessages only works in an action");
       }
@@ -1352,6 +1358,8 @@ export class Agent<AgentTools extends ToolSet> {
       userId: string | undefined;
       threadId: string;
       promptMessageId: string;
+      model: string | undefined;
+      provider: string | undefined;
       result: GenerateObjectResult<unknown>;
       metadata?: Omit<MessageWithMetadata, "message">;
     }
@@ -1361,8 +1369,8 @@ export class Agent<AgentTools extends ToolSet> {
       this.component,
       args.result,
       {
-        model: this.options.chat.modelId,
-        provider: this.options.chat.provider,
+        model: args.model ?? this.options.chat.modelId,
+        provider: args.provider ?? this.options.chat.provider,
       }
     );
     const embeddings = await this.generateEmbeddings(
@@ -1443,18 +1451,16 @@ export class Agent<AgentTools extends ToolSet> {
   }> {
     contextOptions ||= this.options.contextOptions;
     storageOptions ||= this.options.storageOptions;
-    // If only a messageId is provided, this will be empty.
-    const messages = args.promptMessageId
-      ? []
-      : promptOrMessagesToCoreMessages(args);
+    // If only a promptMessageId is provided, this will be empty.
+    const messages = promptOrMessagesToCoreMessages(args);
     const userId =
       argsUserId ??
       (threadId &&
         (await ctx.runQuery(this.component.threads.getThread, { threadId }))
           ?.userId);
     assert(
-      !args.promptMessageId || !(args.prompt || args.messages),
-      "you can't specify a prompt or message if you specify a promptMessageId"
+      !args.promptMessageId || !args.prompt,
+      "you can't specify a prompt if you specify a promptMessageId"
     );
     // If only a messageId is provided, this will add that message to the end.
     const contextMessages = await this.fetchContextMessages(ctx, {
@@ -1464,29 +1470,33 @@ export class Agent<AgentTools extends ToolSet> {
       messages,
       contextOptions,
     });
-    // 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.
-    if (
-      args.promptMessageId &&
-      !contextMessages.at(-1)?.embeddingId &&
-      this.options.textEmbedding
-    ) {
-      await this.generateAndSaveEmbeddings(ctx, {
-        messageIds: [args.promptMessageId],
-      });
+    // If it was a promptMessageId, pop it off context messages
+    // and add to the end of messages.
+    const promptMessage =
+      !!args.promptMessageId &&
+      contextMessages.at(-1)?._id === args.promptMessageId
+        ? contextMessages.pop()
+        : undefined;
+    if (promptMessage?.message) {
+      messages.push(deserializeMessage(promptMessage.message));
+      // 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.
+      if (!promptMessage.embeddingId && this.options.textEmbedding) {
+        await this.generateAndSaveEmbeddings(ctx, {
+          messageIds: [promptMessage._id],
+        });
+      }
     }
-    let messageId = args.promptMessageId;
-    let order = args.promptMessageId
-      ? contextMessages.at(-1)?.order
-      : undefined;
-    let stepOrder = args.promptMessageId
-      ? contextMessages.at(-1)?.stepOrder
-      : undefined;
+    let messageId = promptMessage?._id;
+    let order = promptMessage?.order;
+    let stepOrder = promptMessage?.stepOrder;
     if (
       threadId &&
       messages.length &&
       storageOptions?.saveMessages !== "none" &&
+      // If it was a promptMessageId, we don't want to save it again.
+      (!args.promptMessageId || storageOptions?.saveMessages === "all") &&
       storageOptions?.saveAnyInputMessages !== false
     ) {
       const saveAll = storageOptions?.saveMessages === "all";

package/src/mapping.ts

@@ -334,9 +334,17 @@ function encodeBase64(data: ArrayBuffer): string {
 export function promptOrMessagesToCoreMessages(args: {
   prompt?: string;
   messages?: CoreMessage[] | AIMessageWithoutId[];
+  promptMessageId?: string;
 }): CoreMessage[] {
   const messages: CoreMessage[] = [];
-  assert(args.prompt || args.messages, "messages or prompt is required");
+  assert(
+    args.prompt || args.messages || args.promptMessageId,
+    "messages or prompt or promptMessageId is required"
+  );
+  assert(
+    !args.promptMessageId || !args.prompt,
+    "you can't specify a prompt if you specify a promptMessageId. use messages instead."
+  );
   if (args.messages) {
     if (
       args.messages.some(
@@ -357,6 +365,5 @@ export function promptOrMessagesToCoreMessages(args: {
   if (args.prompt) {
     messages.push({ role: "user", content: args.prompt });
   }
-  assert(messages.length > 0, "Messages must contain at least one message");
   return messages;
 }

package/src/react/useSmoothText.ts

@@ -17,19 +17,26 @@ export function useSmoothText(
   text: string,
   {
     charsPerSec = 256,
+    startStreaming = false,
   }: {
     /**
      * The number of characters to display per second.
      */
     charsPerSec?: number;
+    /**
+     * Whether to initially start streaming.
+     * If this later turns to false, it'll continue streaming.
+     * This will start streaming the first value it sees.
+     */
+    startStreaming?: boolean;
   } = {}
 ): [string, { cursor: number; isStreaming: boolean }] {
-  const [visibleText, setVisibleText] = useState(text);
+  const [visibleText, setVisibleText] = useState(startStreaming ? "" : text);
   const smoothState = useRef({
-    tick: Date.now() + (text.length * 1000) / charsPerSec,
-    cursor: text.length,
+    tick: Date.now() + (visibleText.length * 1000) / charsPerSec,
+    cursor: visibleText.length,
     start: Date.now(),
-    initialLength: text.length,
+    initialLength: visibleText.length,
     charsPerMs: charsPerSec / 1000,
   });
 

package/dist/commonjs/client/_generated/_ignore.d.ts

@@ -1 +0,0 @@
-//# sourceMappingURL=_ignore.d.ts.map

package/dist/commonjs/client/_generated/_ignore.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"_ignore.d.ts","sourceRoot":"","sources":["../../../../src/client/_generated/_ignore.ts"],"names":[],"mappings":""}

package/dist/commonjs/client/_generated/_ignore.js

@@ -1,3 +0,0 @@
-"use strict";
-// This is only here so convex-test can detect a _generated folder
-//# sourceMappingURL=_ignore.js.map

package/dist/commonjs/client/_generated/_ignore.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"_ignore.js","sourceRoot":"","sources":["../../../../src/client/_generated/_ignore.ts"],"names":[],"mappings":";AAAA,kEAAkE"}

package/dist/commonjs/client/createTool.d.ts

@@ -1,47 +0,0 @@
-import type { Schema, Tool, ToolExecutionOptions, ToolSet } from "ai";
-import { z } from "zod";
-import type { Agent } from "./index.js";
-import type { RunActionCtx } from "./types.js";
-export type ToolCtx<TOOLS extends ToolSet = ToolSet> = RunActionCtx & {
-    agent: Agent<TOOLS>;
-    userId?: string;
-    threadId?: string;
-    messageId?: string;
-};
-/**
- * This is a wrapper around the ai.tool function that adds extra context to the
- * tool call, including the action context, userId, threadId, and messageId.
- * @param tool The tool. See https://sdk.vercel.ai/docs/ai-sdk-core/tools-and-tool-calling
- * but swap parameters for args and handler for execute.
- * @returns A tool to be used with the AI SDK.
- */
-export declare function createTool<PARAMETERS extends ToolParameters, RESULT>(t: {
-    /**
-    An optional description of what the tool does.
-    Will be used by the language model to decide whether to use the tool.
-    Not used for provider-defined tools.
-       */
-    description?: string;
-    /**
-    The schema of the input that the tool expects. The language model will use this to generate the input.
-    It is also used to validate the output of the language model.
-    Use descriptions to make the input understandable for the language model.
-       */
-    args: PARAMETERS;
-    /**
-    An async function that is called with the arguments from the tool call and produces a result.
-    If not provided, the tool will not be executed automatically.
-  
-    @args is the input of the tool call.
-    @options.abortSignal is a signal that can be used to abort the tool call.
-       */
-    handler: (ctx: ToolCtx, args: inferParameters<PARAMETERS>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
-    ctx?: ToolCtx;
-}): Tool<PARAMETERS, RESULT> & {
-    execute: (args: inferParameters<PARAMETERS>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
-};
-export declare function wrapTools(ctx: ToolCtx, ...toolSets: (ToolSet | undefined)[]): ToolSet;
-type ToolParameters = z.ZodTypeAny | Schema<any>;
-type inferParameters<PARAMETERS extends ToolParameters> = PARAMETERS extends Schema<any> ? PARAMETERS["_type"] : PARAMETERS extends z.ZodTypeAny ? z.infer<PARAMETERS> : never;
-export {};
-//# sourceMappingURL=createTool.d.ts.map

package/dist/commonjs/client/createTool.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"createTool.d.ts","sourceRoot":"","sources":["../../../src/client/createTool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,oBAAoB,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAEtE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,MAAM,OAAO,CAAC,KAAK,SAAS,OAAO,GAAG,OAAO,IAAI,YAAY,GAAG;IACpE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF;;;;;;GAMG;AACH,wBAAgB,UAAU,CAAC,UAAU,SAAS,cAAc,EAAE,MAAM,EAAE,CAAC,EAAE;IACvE;;;;SAIK;IACL,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;SAIK;IACL,IAAI,EAAE,UAAU,CAAC;IACjB;;;;;;SAMK;IACL,OAAO,EAAE,CACP,GAAG,EAAE,OAAO,EACZ,IAAI,EAAE,eAAe,CAAC,UAAU,CAAC,EACjC,OAAO,EAAE,oBAAoB,KAC1B,WAAW,CAAC,MAAM,CAAC,CAAC;IACzB,GAAG,CAAC,EAAE,OAAO,CAAC;CACf,GAAG,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,GAAG;IAC7B,OAAO,EAAE,CACP,IAAI,EAAE,eAAe,CAAC,UAAU,CAAC,EACjC,OAAO,EAAE,oBAAoB,KAC1B,WAAW,CAAC,MAAM,CAAC,CAAC;CAC1B,CAqBA;AAED,wBAAgB,SAAS,CACvB,GAAG,EAAE,OAAO,EACZ,GAAG,QAAQ,EAAE,CAAC,OAAO,GAAG,SAAS,CAAC,EAAE,GACnC,OAAO,CAiBT;AAID,KAAK,cAAc,GAAG,CAAC,CAAC,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;AACjD,KAAK,eAAe,CAAC,UAAU,SAAS,cAAc,IAEpD,UAAU,SAAS,MAAM,CAAC,GAAG,CAAC,GAC1B,UAAU,CAAC,OAAO,CAAC,GACnB,UAAU,SAAS,CAAC,CAAC,UAAU,GAC7B,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,GACnB,KAAK,CAAC"}

package/dist/commonjs/client/createTool.js

@@ -1,49 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createTool = createTool;
-exports.wrapTools = wrapTools;
-const ai_1 = require("ai");
-/**
- * This is a wrapper around the ai.tool function that adds extra context to the
- * tool call, including the action context, userId, threadId, and messageId.
- * @param tool The tool. See https://sdk.vercel.ai/docs/ai-sdk-core/tools-and-tool-calling
- * but swap parameters for args and handler for execute.
- * @returns A tool to be used with the AI SDK.
- */
-function createTool(t) {
-    const args = {
-        __acceptsCtx: true,
-        ctx: t.ctx,
-        description: t.description,
-        parameters: t.args,
-        async execute(args, options) {
-            if (!this.ctx) {
-                throw new Error("To use a Convex tool, you must either provide the ctx" +
-                    " at definition time (dynamically in an action), or use the Agent to" +
-                    " call it (which injects the ctx, userId and threadId)");
-            }
-            return t.handler(this.ctx, args, options);
-        },
-    };
-    return (0, ai_1.tool)(args);
-}
-function wrapTools(ctx, ...toolSets) {
-    const output = {};
-    for (const toolSet of toolSets) {
-        if (!toolSet) {
-            continue;
-        }
-        for (const [name, tool] of Object.entries(toolSet)) {
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            if (!tool.__acceptsCtx) {
-                output[name] = tool;
-            }
-            else {
-                const out = { ...tool, ctx };
-                output[name] = out;
-            }
-        }
-    }
-    return output;
-}
-//# sourceMappingURL=createTool.js.map

package/dist/commonjs/client/createTool.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"createTool.js","sourceRoot":"","sources":["../../../src/client/createTool.ts"],"names":[],"mappings":";;AAoBA,gCAoDC;AAED,8BAoBC;AA7FD,2BAA0B;AAY1B;;;;;;GAMG;AACH,SAAgB,UAAU,CAA4C,CA0BrE;IAMC,MAAM,IAAI,GAAG;QACX,YAAY,EAAE,IAAI;QAClB,GAAG,EAAE,CAAC,CAAC,GAAG;QACV,WAAW,EAAE,CAAC,CAAC,WAAW;QAC1B,UAAU,EAAE,CAAC,CAAC,IAAI;QAClB,KAAK,CAAC,OAAO,CACX,IAAiC,EACjC,OAA6B;YAE7B,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CACb,uDAAuD;oBACrD,qEAAqE;oBACrE,uDAAuD,CAC1D,CAAC;YACJ,CAAC;YACD,OAAO,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAC5C,CAAC;KACF,CAAC;IACF,OAAO,IAAA,SAAI,EAAC,IAAI,CAAC,CAAC;AACpB,CAAC;AAED,SAAgB,SAAS,CACvB,GAAY,EACZ,GAAG,QAAiC;IAEpC,MAAM,MAAM,GAAG,EAAa,CAAC;IAC7B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,SAAS;QACX,CAAC;QACD,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACnD,8DAA8D;YAC9D,IAAI,CAAE,IAAY,CAAC,YAAY,EAAE,CAAC;gBAChC,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,GAAG,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/commonjs/client/files.d.ts

@@ -1,56 +0,0 @@
-import type { FilePart, ImagePart } from "ai";
-import type { Id } from "../component/_generated/dataModel.js";
-import type { ActionCtx, AgentComponent, QueryCtx, RunMutationCtx } from "./types.js";
-export declare const MAX_FILE_SIZE: number;
-type File = {
-    url: string;
-    fileId: string;
-    storageId: Id<"_storage">;
-    hash: string;
-    filename: string | undefined;
-};
-/**
- * Store a file in the file storage and return the URL and fileId.
- * @param ctx A ctx object from an action.
- * @param component The agent component.
- * @param blob The blob to store.
- * @param filename The filename to store.
- * @param sha256 The sha256 hash of the file. If not provided, it will be
- *   computed. However, to ensure no corruption during transfer, you can
- *   calculate this on the client to enforce integrity.
- * @returns The URL, fileId, and storageId of the stored file.
- */
-export declare function storeFile(ctx: ActionCtx | RunMutationCtx, component: AgentComponent, blob: Blob, filename?: string, sha256?: string): Promise<{
-    file: File;
-    filePart: FilePart;
-    imagePart: ImagePart | undefined;
-}>;
-/**
- * Get file metadata from the component.
- * This also returns filePart (and imagePart if the file is an image),
- * which are useful to construct a CoreMessage like
- * ```ts
- * const { filePart, imagePart } = await getFile(ctx, components.agent, fileId);
- * const message: UserMessage = {
- *   role: "user",
- *   content: [imagePart ?? filePart],
- * };
- * ```
- * @param ctx A ctx object from an action or query.
- * @param component The agent component, usually `components.agent`.
- * @param fileId The fileId of the file to get.
- * @returns The file metadata and content parts.
- */
-export declare function getFile(ctx: ActionCtx | QueryCtx, component: AgentComponent, fileId: string): Promise<{
-    file: {
-        fileId: string;
-        url: string;
-        storageId: Id<"_storage">;
-        hash: string;
-        filename: string | undefined;
-    };
-    filePart: FilePart;
-    imagePart: ImagePart | undefined;
-}>;
-export {};
-//# sourceMappingURL=files.d.ts.map

package/dist/commonjs/client/files.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"files.d.ts","sourceRoot":"","sources":["../../../src/client/files.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,sCAAsC,CAAC;AAC/D,OAAO,KAAK,EACV,SAAS,EACT,cAAc,EACd,QAAQ,EACR,cAAc,EACf,MAAM,YAAY,CAAC;AAEpB,eAAO,MAAM,aAAa,QAAY,CAAC;AAEvC,KAAK,IAAI,GAAG;IACV,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B,CAAC;AAEF;;;;;;;;;;GAUG;AACH,wBAAsB,SAAS,CAC7B,GAAG,EAAE,SAAS,GAAG,cAAc,EAC/B,SAAS,EAAE,cAAc,EACzB,IAAI,EAAE,IAAI,EACV,QAAQ,CAAC,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,MAAM,GACd,OAAO,CAAC;IACT,IAAI,EAAE,IAAI,CAAC;IACX,QAAQ,EAAE,QAAQ,CAAC;IACnB,SAAS,EAAE,SAAS,GAAG,SAAS,CAAC;CAClC,CAAC,CAiED;AAED;;;;;;;;;;;;;;;GAeG;AACH,wBAAsB,OAAO,CAC3B,GAAG,EAAE,SAAS,GAAG,QAAQ,EACzB,SAAS,EAAE,cAAc,EACzB,MAAM,EAAE,MAAM;;;;mBAemB,EAAE,CAAC,UAAU,CAAC;;;;cAYrC,QAAQ;eACP,SAAS,GAAG,SAAS;GARjC"}

package/dist/commonjs/client/files.js

@@ -1,129 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MAX_FILE_SIZE = void 0;
-exports.storeFile = storeFile;
-exports.getFile = getFile;
-exports.MAX_FILE_SIZE = 1024 * 64;
-/**
- * Store a file in the file storage and return the URL and fileId.
- * @param ctx A ctx object from an action.
- * @param component The agent component.
- * @param blob The blob to store.
- * @param filename The filename to store.
- * @param sha256 The sha256 hash of the file. If not provided, it will be
- *   computed. However, to ensure no corruption during transfer, you can
- *   calculate this on the client to enforce integrity.
- * @returns The URL, fileId, and storageId of the stored file.
- */
-async function storeFile(ctx, component, blob, filename, sha256) {
-    if (!("runAction" in ctx)) {
-        throw new Error("You're trying to save a file that's too large in a mutation. " +
-            "You can store the file in file storage from an action first, then pass a URL instead. " +
-            "To have the agent component track the file, you can use `saveFile` from an action then use the fileId with getFile in the mutation. " +
-            "Read more in the docs.");
-    }
-    const hash = sha256 ||
-        Array.from(new Uint8Array(await crypto.subtle.digest("SHA-256", await blob.arrayBuffer())))
-            .map((b) => b.toString(16).padStart(2, "0"))
-            .join("");
-    const reused = await ctx.runMutation(component.files.useExistingFile, {
-        hash,
-        filename,
-    });
-    if (reused) {
-        const url = (await ctx.storage.getUrl(reused.storageId));
-        return {
-            ...getParts(url, blob.type, filename),
-            file: {
-                url,
-                fileId: reused.fileId,
-                storageId: reused.storageId,
-                hash,
-                filename,
-            },
-        };
-    }
-    const newStorageId = await ctx.storage.store(blob);
-    if (sha256) {
-        const metadata = await ctx.storage.getMetadata(newStorageId);
-        if (metadata?.sha256 !== sha256) {
-            throw new Error("Hash mismatch: " + metadata?.sha256 + " != " + sha256);
-        }
-    }
-    const { fileId, storageId } = await ctx.runMutation(component.files.addFile, {
-        storageId: newStorageId,
-        hash,
-        filename,
-        mimeType: blob.type,
-    });
-    const url = (await ctx.storage.getUrl(storageId));
-    if (storageId !== newStorageId) {
-        // We're re-using another file's storageId
-        // Because we try to reuse the file above, this should be very very rare
-        // and only in the case of racing to check then store the file.
-        await ctx.storage.delete(newStorageId);
-    }
-    return {
-        ...getParts(url, blob.type, filename),
-        file: {
-            url,
-            fileId,
-            storageId: storageId,
-            hash,
-            filename,
-        },
-    };
-}
-/**
- * Get file metadata from the component.
- * This also returns filePart (and imagePart if the file is an image),
- * which are useful to construct a CoreMessage like
- * ```ts
- * const { filePart, imagePart } = await getFile(ctx, components.agent, fileId);
- * const message: UserMessage = {
- *   role: "user",
- *   content: [imagePart ?? filePart],
- * };
- * ```
- * @param ctx A ctx object from an action or query.
- * @param component The agent component, usually `components.agent`.
- * @param fileId The fileId of the file to get.
- * @returns The file metadata and content parts.
- */
-async function getFile(ctx, component, fileId) {
-    const file = await ctx.runQuery(component.files.get, { fileId });
-    if (!file) {
-        throw new Error(`File not found in component: ${fileId}`);
-    }
-    const url = await ctx.storage.getUrl(file.storageId);
-    if (!url) {
-        throw new Error(`File not found in storage: ${file.storageId}`);
-    }
-    return {
-        ...getParts(url, file.mimeType, file.filename),
-        file: {
-            fileId,
-            url,
-            storageId: file.storageId,
-            hash: file.hash,
-            filename: file.filename,
-        },
-    };
-}
-function getParts(url, mimeType, filename) {
-    const filePart = {
-        type: "file",
-        data: new URL(url),
-        mimeType,
-        filename,
-    };
-    const imagePart = mimeType.startsWith("image/")
-        ? {
-            type: "image",
-            image: new URL(url),
-            mimeType,
-        }
-        : undefined;
-    return { filePart, imagePart };
-}
-//# sourceMappingURL=files.js.map

package/dist/commonjs/client/files.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"files.js","sourceRoot":"","sources":["../../../src/client/files.ts"],"names":[],"mappings":";;;AA8BA,8BA2EC;AAkBD,0BAuBC;AAzIY,QAAA,aAAa,GAAG,IAAI,GAAG,EAAE,CAAC;AAUvC;;;;;;;;;;GAUG;AACI,KAAK,UAAU,SAAS,CAC7B,GAA+B,EAC/B,SAAyB,EACzB,IAAU,EACV,QAAiB,EACjB,MAAe;IAMf,IAAI,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,+DAA+D;YAC7D,wFAAwF;YACxF,sIAAsI;YACtI,wBAAwB,CAC3B,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,GACR,MAAM;QACN,KAAK,CAAC,IAAI,CACR,IAAI,UAAU,CACZ,MAAM,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC,CAChE,CACF;aACE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;aAC3C,IAAI,CAAC,EAAE,CAAC,CAAC;IACd,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,EAAE;QACpE,IAAI;QACJ,QAAQ;KACT,CAAC,CAAC;IACH,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAE,CAAC;QAC1D,OAAO;YACL,GAAG,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC;YACrC,IAAI,EAAE;gBACJ,GAAG;gBACH,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,SAAS,EAAE,MAAM,CAAC,SAA2B;gBAC7C,IAAI;gBACJ,QAAQ;aACT;SACF,CAAC;IACJ,CAAC;IACD,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnD,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QAC7D,IAAI,QAAQ,EAAE,MAAM,KAAK,MAAM,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,iBAAiB,GAAG,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IACD,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE;QAC3E,SAAS,EAAE,YAAY;QACvB,IAAI;QACJ,QAAQ;QACR,QAAQ,EAAE,IAAI,CAAC,IAAI;KACpB,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAA2B,CAAC,CAAE,CAAC;IACrE,IAAI,SAAS,KAAK,YAAY,EAAE,CAAC;QAC/B,0CAA0C;QAC1C,wEAAwE;QACxE,+DAA+D;QAC/D,MAAM,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC;IACD,OAAO;QACL,GAAG,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC;QACrC,IAAI,EAAE;YACJ,GAAG;YACH,MAAM;YACN,SAAS,EAAE,SAA2B;YACtC,IAAI;YACJ,QAAQ;SACT;KACF,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACI,KAAK,UAAU,OAAO,CAC3B,GAAyB,EACzB,SAAyB,EACzB,MAAc;IAEd,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IACjE,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,SAA2B,CAAC,CAAC;IACvE,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CAAC,8BAA8B,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;IAClE,CAAC;IACD,OAAO;QACL,GAAG,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC;QAC9C,IAAI,EAAE;YACJ,MAAM;YACN,GAAG;YACH,SAAS,EAAE,IAAI,CAAC,SAA2B;YAC3C,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB;KACF,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CACf,GAAW,EACX,QAAgB,EAChB,QAA4B;IAK5B,MAAM,QAAQ,GAAa;QACzB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,IAAI,GAAG,CAAC,GAAG,CAAC;QAClB,QAAQ;QACR,QAAQ;KACT,CAAC;IACF,MAAM,SAAS,GAA0B,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC;QACpE,CAAC,CAAC;YACE,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,IAAI,GAAG,CAAC,GAAG,CAAC;YACnB,QAAQ;SACT;QACH,CAAC,CAAC,SAAS,CAAC;IACd,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;AACjC,CAAC"}