convex-durable-agents 0.1.1 → 0.1.2

package/README.md

@@ -2,20 +2,27 @@
 
 [![npm version](https://badge.fury.io/js/convex-durable-agents.svg)](https://badge.fury.io/js/convex-durable-agents)
 
-A Convex component for building durable AI agents with an async tool loop. The goal of this component is to provide a way to build AI agents that can run indefinitely and survive failures and restarts. It provides some of the functionality of the [Convex Agents Component](https://www.convex.dev/components/agent) (such as persistent streaming), while deliberately leaving out some of the more advanced features (context management, RAG, rate limiting, etc.). The component is built on top of the [AI SDK v6](https://ai-sdk.dev/) and aims to expose its full `streamText` API with persistence and durable execution.
+A Convex component for building durable AI agents with an async tool loop. The goal of this component is to provide a
+way to build AI agents that can run indefinitely and survive failures and restarts. It provides some of the
+functionality of the [Convex Agents Component](https://www.convex.dev/components/agent) (such as persistent streaming),
+while deliberately leaving out some of the more advanced features (context management, RAG, rate limiting, etc.). The
+component is built on top of the [AI SDK v6](https://ai-sdk.dev/) SDK and aims to expose its full `streamText` API with
+persistence and durable execution.
 
-**Note:** This component is still in early development and is not yet ready for production use. The API will very likely change before a first stable release.
+**Note:** This component is still in early development and is not yet ready for production use. The API will very likely
+change before a first stable release.
 
 ## Features
 
 - **Async Execution**: Agent tool loop is executed asynchronously to avoid time limits of convex actions
 - **Tool Execution**: via convex actions - support for both sync and async tools
 - **Automatic Retries**: Failed tool calls are automatically retried
+- **Workpool Support**: Optionally route agent and tool execution through `@convex-dev/workpool` for parallelism control
+  and retry mechanisms
 
 ## Roadmap
 
 - **Durable Execution**: Agent tool loops survive crashes and dev server restarts
-- **Workpool/Workflow Support**: Support for workpool and workflow execution
 
 ## Installation
 
@@ -50,16 +57,12 @@ Create a chat handler with your AI model and tools:
 
 import { z } from "zod";
 import { components, internal } from "./_generated/api";
-import {
-  createActionTool,
-  defineAgentApi,
-  streamHandlerAction,
-} from "convex-durable-agents";
+import { createActionTool, defineAgentApi, streamHandlerAction } from "convex-durable-agents";
 import { openai } from "@ai-sdk/openai";
 
 // Define the stream handler with your model and tools
 export const chatAgentHandler = streamHandlerAction(components.durableAgents, {
-  model: 'anthropic/claude-haiku-4.5',
+  model: "anthropic/claude-haiku-4.5",
   system: "You are a helpful AI assistant.",
   tools: {
     get_weather: createActionTool({
@@ -84,12 +87,20 @@ export const {
   stopThread,
   addToolResult,
   addToolError,
-} = defineAgentApi(components.durableAgents, internal.chat.chatAgentHandler);
+} = defineAgentApi(components.durableAgents, internal.chat.chatAgentHandler, {
+  // Optional: Add authorization to protect thread access
+  authorizationCallback: async (ctx, threadId) => {
+    // Example: verify the user owns this thread
+    // const identity = await ctx.auth.getUserIdentity();
+    // if (!identity) throw new Error("Unauthorized");
+  },
+});
 ```
 
 #### Using Internal API
 
-If you want to restrict the agent API to only be callable from other Convex functions (not directly from clients), use `defineInternalAgentApi` instead:
+If you want to restrict the agent API to only be callable from other Convex functions (not directly from clients), use
+`defineInternalAgentApi` instead:
 
 ```ts
 // convex/chat.ts
@@ -105,6 +116,7 @@ export const {
 ```
 
 This is useful when you want to:
+
 - Add authentication/authorization checks before calling agent functions
 - Wrap agent functions with additional business logic
 - Prevent direct client access to the agent API
@@ -124,7 +136,7 @@ export const sendMessage = mutation({
   handler: async (ctx, args) => {
     const identity = await ctx.auth.getUserIdentity();
     if (!identity) throw new Error("Not authenticated");
-    
+
     // Call the internal agent API
     return ctx.runMutation(internal.chat.sendMessage, args);
   },
@@ -163,12 +175,12 @@ import { api } from "../convex/_generated/api";
 
 function ChatView({ threadId }: { threadId: string }) {
   const sendMessage = useMutation(api.chat.sendMessage);
-  
+
   const { messages, status, isRunning } = useThread(
     api.chat.listMessagesWithStreams,
     api.chat.getThread,
     { threadId },
-    { stream: true }
+    { stream: true },
   );
 
   return (
@@ -178,7 +190,7 @@ function ChatView({ threadId }: { threadId: string }) {
           <strong>{msg.role}:</strong> {msg.text}
         </div>
       ))}
-      
+
       <input
         onKeyPress={(e) => {
           if (e.key === "Enter" && !isRunning) {
@@ -196,7 +208,7 @@ function ChatView({ threadId }: { threadId: string }) {
 
 ### Client API
 
-#### `defineAgentApi(component, streamHandler)`
+#### `defineAgentApi(component, streamHandler, options?)`
 
 Creates the full agent API with **public** functions that can be called directly from clients:
 
@@ -212,20 +224,47 @@ Creates the full agent API with **public** functions that can be called directly
 - `addToolResult({ toolCallId, result })` - Add result for async tool
 - `addToolError({ toolCallId, error })` - Add error for async tool
 
-#### `defineInternalAgentApi(component, streamHandler)`
+**Options:**
+
+```ts
+type AgentApiOptions = {
+  authorizationCallback?: (ctx: QueryCtx | MutationCtx | ActionCtx, threadId: string) => Promise<void> | void;
+  workpoolEnqueueAction?: FunctionReference<"mutation", "internal">;
+  toolExecutionWorkpoolEnqueueAction?: FunctionReference<"mutation", "internal">;
+};
+```
 
-Creates the same agent API as `defineAgentApi`, but with **internal** functions that can only be called from other Convex functions (queries, mutations, actions). Use this when you want to add authentication, authorization, or other business logic before calling agent functions.
+- `authorizationCallback` - Called before any operation that accesses an existing thread. Use it to verify the user has
+  permission to access the thread. Throw an error to deny access.
+- `workpoolEnqueueAction` - Route agent and tool execution through a workpool for parallelism control
+- `toolExecutionWorkpoolEnqueueAction` - Override workpool for tool execution only (falls back to
+  `workpoolEnqueueAction` if not set)
+
+**Protected endpoints:** `sendMessage`, `resumeThread`, `stopThread`, `getThread`, `listMessages`,
+`listMessagesWithStreams`, `deleteThread`
+
+**Example with ownership check:**
 
 ```ts
-import { defineInternalAgentApi } from "convex-durable-agents";
+defineAgentApi(components.durableAgents, internal.chat.chatAgentHandler, {
+  authorizationCallback: async (ctx, threadId) => {
+    const identity = await ctx.auth.getUserIdentity();
+    if (!identity) throw new Error("Unauthorized");
 
-export const {
-  createThread,
-  sendMessage,
-  // ... all the same functions as defineAgentApi
-} = defineInternalAgentApi(components.durableAgents, internal.chat.chatAgentHandler);
+    // Query your threads table to verify ownership
+    const thread = await ctx.runQuery(api.threads.getOwner, { threadId });
+    if (thread?.userId !== identity.subject) {
+      throw new Error("Access denied");
+    }
+  },
+});
 ```
 
+#### `defineInternalAgentApi(component, streamHandler, options?)`
+
+Same as `defineAgentApi` but creates internal functions that can only be called from other Convex functions. Use this
+when you want to add authentication, authorization, or other business logic before calling agent functions.
+
 #### `streamHandlerAction(component, options)`
 
 Creates the stream handler action:
@@ -273,20 +312,15 @@ All-in-one hook for thread status and messages:
 
 ```ts
 const {
-  messages,    // UIMessage[]
-  thread,      // ThreadDoc | null
-  status,      // ThreadStatus
-  isLoading,   // boolean
-  isRunning,   // boolean
-  isComplete,  // boolean
-  isFailed,    // boolean
-  isStopped,   // boolean
-} = useThread(
-  api.chat.listMessagesWithStreams,
-  api.chat.getThread,
-  { threadId },
-  { stream: true }
-);
+  messages, // UIMessage[]
+  thread, // ThreadDoc | null
+  status, // ThreadStatus
+  isLoading, // boolean
+  isRunning, // boolean
+  isComplete, // boolean
+  isFailed, // boolean
+  isStopped, // boolean
+} = useThread(api.chat.listMessagesWithStreams, api.chat.getThread, { threadId }, { stream: true });
 ```
 
 #### `useSmoothText(text, options?)`
@@ -305,8 +339,9 @@ const [visibleText, { cursor, isStreaming }] = useSmoothText(text, {
 Subscribe to thread status changes:
 
 ```ts
-const { thread, status, isRunning, isComplete, isFailed, isStopped } = 
-  useThreadStatus(api.chat.getThread, { threadId });
+const { thread, status, isRunning, isComplete, isFailed, isStopped } = useThreadStatus(api.chat.getThread, {
+  threadId,
+});
 ```
 
 #### `useMessages(query, threadQuery, args)`
@@ -314,11 +349,7 @@ const { thread, status, isRunning, isComplete, isFailed, isStopped } =
 Fetch and transform messages:
 
 ```ts
-const { messages, isLoading, thread } = useMessages(
-  api.chat.listMessages,
-  api.chat.getThread,
-  { threadId }
-);
+const { messages, isLoading, thread } = useMessages(api.chat.listMessages, api.chat.getThread, { threadId });
 ```
 
 ## Thread Status
@@ -331,6 +362,80 @@ Threads can be in one of these states:
 - `failed` - An error occurred
 - `stopped` - User stopped the conversation
 
+## Workpool Integration
+
+For advanced use cases, you can route agent execution through the `@convex-dev/workpool` component. This provides:
+
+- **Parallelism Control**: Limit concurrent AI model calls and tool executions
+- **Retry Mechanisms**: Automatic retries with exponential backoff for failed actions
+- **Rate Limiting Protection**: Prevent overwhelming external APIs
+
+### Setup
+
+1. Install and configure the workpool component:
+
+```ts
+// convex/convex.config.ts
+import { defineApp } from "convex/server";
+import durableAgents from "convex-durable-agents/convex.config.js";
+import workpool from "@convex-dev/workpool/convex.config.js";
+
+const app = defineApp();
+app.use(durableAgents);
+app.use(workpool, { name: "agentWorkpool" });
+
+export default app;
+```
+
+2. Create the workpool bridge:
+
+```ts
+// convex/workpool.ts
+import { Workpool } from "@convex-dev/workpool";
+import { components } from "./_generated/api";
+import { createWorkpoolBridge } from "convex-durable-agents";
+
+const pool = new Workpool(components.agentWorkpool, {
+  maxParallelism: 5,
+});
+
+export const { enqueueWorkpoolAction } = createWorkpoolBridge(pool);
+```
+
+3. Pass the workpool to your agent API:
+
+```ts
+// convex/chat.ts
+export const {
+  createThread,
+  sendMessage,
+  // ...
+} = defineAgentApi(components.durableAgents, internal.chat.chatAgentHandler, {
+  workpoolEnqueueAction: internal.workpool.enqueueWorkpoolAction,
+});
+```
+
+### Separate Workpools for Tools
+
+You can use different workpools for the stream handler and tool execution:
+
+```ts
+// convex/workpool.ts
+const agentPool = new Workpool(components.agentWorkpool, { maxParallelism: 3 });
+const toolPool = new Workpool(components.toolWorkpool, { maxParallelism: 10 });
+
+export const { enqueueWorkpoolAction: enqueueAgentAction } = createWorkpoolBridge(agentPool);
+export const { enqueueWorkpoolAction: enqueueToolAction } = createWorkpoolBridge(toolPool);
+```
+
+```ts
+// convex/chat.ts
+defineAgentApi(components.durableAgents, internal.chat.chatAgentHandler, {
+  workpoolEnqueueAction: internal.workpool.enqueueAgentAction,
+  toolExecutionWorkpoolEnqueueAction: internal.workpool.enqueueToolAction,
+});
+```
+
 ## Architecture
 
 ```

package/dist/client/index.d.ts

@@ -111,6 +111,10 @@ type StreamHandlerArgs = Omit<Parameters<typeof streamText>[0], "tools" | "messa
     /** Optional: Save streaming deltas to the database for real-time client updates */
     saveStreamDeltas?: boolean | StreamingOptions;
     transformMessages?: (messages: ModelMessage[]) => ModelMessage[];
+    /** Optional: Function to enqueue actions via workpool (used for both stream handler and tools unless overridden) */
+    workpoolEnqueueAction?: FunctionReference<"mutation", "internal">;
+    /** Optional: Override workpool for tool execution only */
+    toolExecutionWorkpoolEnqueueAction?: FunctionReference<"mutation", "internal">;
 };
 export declare function streamHandlerAction(component: ComponentApi, { tools, saveStreamDeltas, transformMessages, ...streamTextArgs }: StreamHandlerArgs): RegisteredAction<"internal", {
     threadId: string;
@@ -189,17 +193,62 @@ export type AgentApi<V extends FunctionVisibility = "public"> = {
         error: string;
     }, null>;
 };
+export type AgentApiOptions = {
+    /** Optional authorization callback for thread access control */
+    authorizationCallback?: (ctx: QueryCtx | MutationCtx | ActionCtx, threadId: string) => Promise<void> | void;
+    /** Optional: Function to enqueue actions via workpool (used for both stream handler and tools unless overridden) */
+    workpoolEnqueueAction?: FunctionReference<"mutation", "internal">;
+    /** Optional: Override workpool for tool execution only */
+    toolExecutionWorkpoolEnqueueAction?: FunctionReference<"mutation", "internal">;
+};
 /**
  * Define a public agent API that can be called from clients.
  */
 export declare function defineAgentApi(component: ComponentApi, ref: FunctionReference<"action", "internal" | "public", {
     threadId: string;
-}>): AgentApi<"public">;
+}>, options?: AgentApiOptions): AgentApi<"public">;
 /**
  * Define an internal agent API that can only be called from other Convex functions.
  */
 export declare function defineInternalAgentApi(component: ComponentApi, ref: FunctionReference<"action", "internal" | "public", {
     threadId: string;
-}>): AgentApi<"internal">;
+}>, options?: AgentApiOptions): AgentApi<"internal">;
+/**
+ * Type for a Workpool instance that has an enqueueAction method.
+ * This is compatible with @convex-dev/workpool's Workpool class.
+ */
+type WorkpoolLike = {
+    enqueueAction: (ctx: GenericMutationCtx<GenericDataModel>, fn: FunctionReference<"action", FunctionVisibility, any, any>, fnArgs: any, options?: any) => Promise<any>;
+};
+/**
+ * Creates a workpool bridge mutation that can be used with defineAgentApi.
+ *
+ * This helper creates an internal mutation that forwards action execution to your workpool,
+ * allowing the agent to use workpool's parallelism controls and retry mechanisms.
+ *
+ * @example
+ * ```typescript
+ * // convex/workpool.ts
+ * import { Workpool } from "@convex-dev/workpool";
+ * import { components } from "./_generated/api";
+ * import { createWorkpoolBridge } from "convex-durable-agents";
+ *
+ * const pool = new Workpool(components.workpool, { maxParallelism: 5 });
+ * export const { enqueueWorkpoolAction } = createWorkpoolBridge(pool);
+ *
+ * // convex/chat.ts
+ * export const { createThread, sendMessage, ... } = defineAgentApi(
+ *   components.durable_agent,
+ *   internal.chat.chatAgentHandler,
+ *   { workpoolEnqueueAction: internal.workpool.enqueueWorkpoolAction }
+ * );
+ * ```
+ */
+export declare function createWorkpoolBridge(workpool: WorkpoolLike): {
+    enqueueWorkpoolAction: RegisteredMutation<"internal", {
+        action: string;
+        args: any;
+    }, Promise<null>>;
+};
 export {};
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,KAAK,YAAY,EAAE,UAAU,EAAmB,MAAM,IAAI,CAAC;AAChF,OAAO,EAGL,KAAK,iBAAiB,EACtB,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EACvB,KAAK,eAAe,EAMpB,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EACvB,KAAK,eAAe,EACrB,MAAM,eAAe,CAAC;AAEvB,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,sCAAsC,CAAC;AAM/D,MAAM,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,EAAE,UAAU,CAAC,CAAC;AAC3E,MAAM,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,EAAE,UAAU,GAAG,aAAa,CAAC,CAAC;AACjG,MAAM,MAAM,SAAS,GAAG,IAAI,CAC1B,gBAAgB,CAAC,gBAAgB,CAAC,EAClC,UAAU,GAAG,aAAa,GAAG,WAAW,GAAG,SAAS,GAAG,MAAM,GAAG,WAAW,CAC5E,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,EAAE,CAAC,EAAE,MAAM,CAAC;CACb,CAAC;AAEF,MAAM,MAAM,gBAAgB,CAAC,CAAC,IAAI;IAChC,IAAI,EAAE,CAAC,EAAE,CAAC;IACV,MAAM,EAAE,OAAO,CAAC;IAChB,cAAc,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG,WAAW,GAAG,uBAAuB,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,CAAC;AAEtG,MAAM,MAAM,SAAS,GAAG;IACtB,GAAG,EAAE,MAAM,CAAC;IACZ,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,YAAY,CAAC;IACrB,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,cAAc,EAAE,MAAM,CAAC;CACxB,CAAC;AAmBF,MAAM,MAAM,UAAU,GAAG;IACvB,GAAG,EAAE,MAAM,CAAC;IACZ,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE;QACP,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;QAC/C,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;KAClC,CAAC;CACH,CAAC;AAGF,MAAM,MAAM,QAAQ,CAAC,KAAK,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,IAAI;IACxD,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,OAAO,CAAC;IACpB,OAAO,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,CAAC,CAAC;IAC5D,UAAU,CAAC,EAAE,KAAK,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAGF,MAAM,MAAM,SAAS,CAAC,KAAK,GAAG,OAAO,IAAI;IACvC,IAAI,EAAE,OAAO,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,CAAC,CAAC;IAC7D,UAAU,CAAC,EAAE,KAAK,CAAC;CACpB,CAAC;AAGF,MAAM,MAAM,WAAW,CAAC,KAAK,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,IAAI,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;AAGxG,MAAM,MAAM,qBAAqB,CAAC,KAAK,GAAG,OAAO,IAAI;IACnD,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,KAAK,CAAC;CACb,CAAC;AAMF;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE;IACnD,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvB,OAAO,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,CAAC,CAAC;CAC7D,GAAG,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAW1B;AAED;;;GAGG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,GAAG,EAAE;IAC1C,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvB,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,CAAC,CAAC;CAC9D,GAAG,SAAS,CAAC,KAAK,CAAC,CAWnB;AAMD,MAAM,MAAM,gBAAgB,GAAG;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B,CAAC;AAuCF;;;;GAIG;AACH,qBAAa,aAAa;;aASN,SAAS,EAAE,YAAY;aACvB,GAAG,EAAE,SAAS;IAC9B,OAAO,CAAC,MAAM;IAKd,OAAO,CAAC,QAAQ;IAflB,QAAQ,EAAE,EAAE,CAAC,oBAAoB,CAAC,GAAG,SAAS,CAAC;IAK/C,eAAe,EAAE,eAAe,CAAC;gBAGf,SAAS,EAAE,YAAY,EACvB,GAAG,EAAE,SAAS,EACtB,MAAM,EAAE;QACd,UAAU,EAAE,MAAM,CAAC;QACnB,YAAY,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QAChD,WAAW,CAAC,EAAE,WAAW,CAAC;KAC3B,EACO,QAAQ,EAAE;QAChB,QAAQ,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,gBAAgB,GAAG,gBAAgB,GAAG,SAAS,CAAC;KACzD;IAoBG,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC;IAY9B,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAS9C,aAAa,CAAC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IA2C5D,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC;IAYvB,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAa1C;AAkDD,KAAK,iBAAiB,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,GAAG,UAAU,GAAG,QAAQ,CAAC,GAAG;IACjG,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;IACrD,mFAAmF;IACnF,gBAAgB,CAAC,EAAE,OAAO,GAAG,gBAAgB,CAAC;IAC9C,iBAAiB,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,KAAK,YAAY,EAAE,CAAC;CAClE,CAAC;AAEF,wBAAgB,mBAAmB,CACjC,SAAS,EAAE,YAAY,EACvB,EAAE,KAAK,EAAE,gBAAgB,EAAE,iBAA0C,EAAE,GAAG,cAAc,EAAE,EAAE,iBAAiB;;;kBAoM9G;AAkBD,MAAM,MAAM,UAAU,GAClB;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,GACrC;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,OAAO,EAAE,KAAK,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC;AAE7E,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,WAAW,GAAG,UAAU,GAAG,SAAS,CAAC;IAC7C,MAAM,CAAC,EAAE,gBAAgB,GAAG,gBAAgB,CAAC;IAC7C,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,QAAQ,EAAE,UAAU,EAAE,CAAC;IACvB,OAAO,CAAC,EACJ;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,aAAa,EAAE,CAAA;KAAE,GAC3C;QAAE,IAAI,EAAE,QAAQ,CAAC;QAAC,MAAM,EAAE,WAAW,EAAE,CAAA;KAAE,CAAC;CAC/C,CAAC;AAEF,MAAM,MAAM,QAAQ,CAAC,CAAC,SAAS,kBAAkB,GAAG,QAAQ,IAAI;IAC9D,YAAY,EAAE,gBAAgB,CAAC,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,MAAM,CAAC,CAAC;IAC/D,WAAW,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IAC/E,YAAY,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IACjF,UAAU,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IAC9D,SAAS,EAAE,eAAe,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,SAAS,GAAG,IAAI,CAAC,CAAC;IACtE,YAAY,EAAE,eAAe,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IACrE,uBAAuB,EAAE,eAAe,CACtC,CAAC,EACD;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,UAAU,CAAA;KAAE,EAC7C,yBAAyB,CAC1B,CAAC;IACF,WAAW,EAAE,eAAe,CAAC,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,SAAS,EAAE,CAAC,CAAC;IACjE,YAAY,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IAChE,aAAa,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,OAAO,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IACpF,YAAY,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;CAClF,CAAC;AAyOF;;GAEG;AACH,wBAAgB,cAAc,CAC5B,SAAS,EAAE,YAAY,EACvB,GAAG,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,GAC5E,QAAQ,CAAC,QAAQ,CAAC,CAQpB;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,SAAS,EAAE,YAAY,EACvB,GAAG,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,GAC5E,QAAQ,CAAC,UAAU,CAAC,CAQtB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/client/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,KAAK,YAAY,EAAE,UAAU,EAAmB,MAAM,IAAI,CAAC;AAChF,OAAO,EAGL,KAAK,iBAAiB,EACtB,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EACvB,KAAK,eAAe,EAMpB,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EACvB,KAAK,eAAe,EACrB,MAAM,eAAe,CAAC;AAEvB,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,sCAAsC,CAAC;AAM/D,MAAM,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,EAAE,UAAU,CAAC,CAAC;AAC3E,MAAM,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,EAAE,UAAU,GAAG,aAAa,CAAC,CAAC;AACjG,MAAM,MAAM,SAAS,GAAG,IAAI,CAC1B,gBAAgB,CAAC,gBAAgB,CAAC,EAClC,UAAU,GAAG,aAAa,GAAG,WAAW,GAAG,SAAS,GAAG,MAAM,GAAG,WAAW,CAC5E,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,EAAE,CAAC,EAAE,MAAM,CAAC;CACb,CAAC;AAEF,MAAM,MAAM,gBAAgB,CAAC,CAAC,IAAI;IAChC,IAAI,EAAE,CAAC,EAAE,CAAC;IACV,MAAM,EAAE,OAAO,CAAC;IAChB,cAAc,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG,WAAW,GAAG,uBAAuB,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,CAAC;AAEtG,MAAM,MAAM,SAAS,GAAG;IACtB,GAAG,EAAE,MAAM,CAAC;IACZ,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,YAAY,CAAC;IACrB,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,cAAc,EAAE,MAAM,CAAC;CACxB,CAAC;AAqBF,MAAM,MAAM,UAAU,GAAG;IACvB,GAAG,EAAE,MAAM,CAAC;IACZ,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE;QACP,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;QAC/C,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;KAClC,CAAC;CACH,CAAC;AAGF,MAAM,MAAM,QAAQ,CAAC,KAAK,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,IAAI;IACxD,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,OAAO,CAAC;IACpB,OAAO,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,CAAC,CAAC;IAC5D,UAAU,CAAC,EAAE,KAAK,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAGF,MAAM,MAAM,SAAS,CAAC,KAAK,GAAG,OAAO,IAAI;IACvC,IAAI,EAAE,OAAO,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,CAAC,CAAC;IAC7D,UAAU,CAAC,EAAE,KAAK,CAAC;CACpB,CAAC;AAGF,MAAM,MAAM,WAAW,CAAC,KAAK,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO,IAAI,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;AAGxG,MAAM,MAAM,qBAAqB,CAAC,KAAK,GAAG,OAAO,IAAI;IACnD,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,KAAK,CAAC;CACb,CAAC;AAMF;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE;IACnD,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvB,OAAO,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,CAAC,CAAC;CAC7D,GAAG,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAW1B;AAED;;;GAGG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,GAAG,EAAE;IAC1C,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvB,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,CAAC,CAAC;CAC9D,GAAG,SAAS,CAAC,KAAK,CAAC,CAWnB;AAMD,MAAM,MAAM,gBAAgB,GAAG;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B,CAAC;AAuCF;;;;GAIG;AACH,qBAAa,aAAa;;aASN,SAAS,EAAE,YAAY;aACvB,GAAG,EAAE,SAAS;IAC9B,OAAO,CAAC,MAAM;IAKd,OAAO,CAAC,QAAQ;IAflB,QAAQ,EAAE,EAAE,CAAC,oBAAoB,CAAC,GAAG,SAAS,CAAC;IAK/C,eAAe,EAAE,eAAe,CAAC;gBAGf,SAAS,EAAE,YAAY,EACvB,GAAG,EAAE,SAAS,EACtB,MAAM,EAAE;QACd,UAAU,EAAE,MAAM,CAAC;QACnB,YAAY,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QAChD,WAAW,CAAC,EAAE,WAAW,CAAC;KAC3B,EACO,QAAQ,EAAE;QAChB,QAAQ,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,gBAAgB,GAAG,gBAAgB,GAAG,SAAS,CAAC;KACzD;IAoBG,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC;IAY9B,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAS9C,aAAa,CAAC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgD5D,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC;IAYvB,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAa1C;AAkDD,KAAK,iBAAiB,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,GAAG,UAAU,GAAG,QAAQ,CAAC,GAAG;IACjG,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;IACrD,mFAAmF;IACnF,gBAAgB,CAAC,EAAE,OAAO,GAAG,gBAAgB,CAAC;IAC9C,iBAAiB,CAAC,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,KAAK,YAAY,EAAE,CAAC;IACjE,oHAAoH;IACpH,qBAAqB,CAAC,EAAE,iBAAiB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IAClE,0DAA0D;IAC1D,kCAAkC,CAAC,EAAE,iBAAiB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;CAChF,CAAC;AAEF,wBAAgB,mBAAmB,CACjC,SAAS,EAAE,YAAY,EACvB,EAAE,KAAK,EAAE,gBAAgB,EAAE,iBAA0C,EAAE,GAAG,cAAc,EAAE,EAAE,iBAAiB;;;kBA4M9G;AAkBD,MAAM,MAAM,UAAU,GAClB;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,GACrC;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,OAAO,EAAE,KAAK,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC;AAE7E,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,WAAW,GAAG,UAAU,GAAG,SAAS,CAAC;IAC7C,MAAM,CAAC,EAAE,gBAAgB,GAAG,gBAAgB,CAAC;IAC7C,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,QAAQ,EAAE,UAAU,EAAE,CAAC;IACvB,OAAO,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,aAAa,EAAE,CAAA;KAAE,GAAG;QAAE,IAAI,EAAE,QAAQ,CAAC;QAAC,MAAM,EAAE,WAAW,EAAE,CAAA;KAAE,CAAC;CACnG,CAAC;AAEF,MAAM,MAAM,QAAQ,CAAC,CAAC,SAAS,kBAAkB,GAAG,QAAQ,IAAI;IAC9D,YAAY,EAAE,gBAAgB,CAAC,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,MAAM,CAAC,CAAC;IAC/D,WAAW,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IAC/E,YAAY,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IACjF,UAAU,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IAC9D,SAAS,EAAE,eAAe,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,SAAS,GAAG,IAAI,CAAC,CAAC;IACtE,YAAY,EAAE,eAAe,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IACrE,uBAAuB,EAAE,eAAe,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,UAAU,CAAA;KAAE,EAAE,yBAAyB,CAAC,CAAC;IACtH,WAAW,EAAE,eAAe,CAAC,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,SAAS,EAAE,CAAC,CAAC;IACjE,YAAY,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IAChE,aAAa,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,OAAO,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;IACpF,YAAY,EAAE,kBAAkB,CAAC,CAAC,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,CAAC,CAAC;CAClF,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,gEAAgE;IAChE,qBAAqB,CAAC,EAAE,CAAC,GAAG,EAAE,QAAQ,GAAG,WAAW,GAAG,SAAS,EAAE,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC5G,oHAAoH;IACpH,qBAAqB,CAAC,EAAE,iBAAiB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IAClE,0DAA0D;IAC1D,kCAAkC,CAAC,EAAE,iBAAiB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;CAChF,CAAC;AAqSF;;GAEG;AACH,wBAAgB,cAAc,CAC5B,SAAS,EAAE,YAAY,EACvB,GAAG,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,EAC7E,OAAO,CAAC,EAAE,eAAe,GACxB,QAAQ,CAAC,QAAQ,CAAC,CAEpB;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,SAAS,EAAE,YAAY,EACvB,GAAG,EAAE,iBAAiB,CAAC,QAAQ,EAAE,UAAU,GAAG,QAAQ,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,EAC7E,OAAO,CAAC,EAAE,eAAe,GACxB,QAAQ,CAAC,UAAU,CAAC,CAStB;AAMD;;;GAGG;AAEH,KAAK,YAAY,GAAG;IAClB,aAAa,EAAE,CACb,GAAG,EAAE,kBAAkB,CAAC,gBAAgB,CAAC,EACzC,EAAE,EAAE,iBAAiB,CAAC,QAAQ,EAAE,kBAAkB,EAAE,GAAG,EAAE,GAAG,CAAC,EAC7D,MAAM,EAAE,GAAG,EACX,OAAO,CAAC,EAAE,GAAG,KACV,OAAO,CAAC,GAAG,CAAC,CAAC;CACnB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,YAAY;;;;;EAkB1D"}

package/dist/client/index.js

@@ -10,6 +10,8 @@ const vClientThreadDoc = v.object({
     stopSignal: v.boolean(),
     streamId: v.optional(v.union(v.string(), v.null())),
     streamFnHandle: v.string(),
+    workpoolEnqueueAction: v.optional(v.string()),
+    toolExecutionWorkpoolEnqueueAction: v.optional(v.string()),
 });
 // ============================================================================
 // Tool Definition Helpers
@@ -248,12 +250,16 @@ export function streamHandlerAction(component, { tools, saveStreamDeltas, transf
         },
         returns: v.null(),
         handler: async (ctx, args) => {
-            const thread = await ctx.runQuery(component.threads.get, { threadId: args.threadId });
+            const thread = await ctx.runQuery(component.threads.get, {
+                threadId: args.threadId,
+            });
             if (thread?.streamId !== args.streamId) {
                 throw new Error(`Thread ${args.threadId} streamId mismatch: ${thread?.streamId} !== ${args.streamId}`);
             }
             // Get the current message order for streaming
-            const messages = await ctx.runQuery(component.messages.list, { threadId: args.threadId });
+            const messages = await ctx.runQuery(component.messages.list, {
+                threadId: args.threadId,
+            });
             const currentOrder = messages.length > 0 ? Math.max(...messages.map((m) => m.order)) + 1 : 0;
             // Set up delta streamer if enabled
             let streamer;
@@ -433,7 +439,20 @@ async function checkThreadIsIdle(component, ctx, threadId) {
             throw new Error(`Thread ${threadId} status=${thread.status}, cannot resume`);
     }
 }
-function createAgentApi(component, ref, action, query, mutation) {
+async function serializeWorkpoolOptions(options) {
+    const result = {};
+    if (options?.workpoolEnqueueAction) {
+        const handle = await createFunctionHandle(options.workpoolEnqueueAction);
+        result.workpoolEnqueueAction = handle.toString();
+    }
+    if (options?.toolExecutionWorkpoolEnqueueAction) {
+        const handle = await createFunctionHandle(options.toolExecutionWorkpoolEnqueueAction);
+        result.toolExecutionWorkpoolEnqueueAction = handle.toString();
+    }
+    return result;
+}
+function createAgentApi(component, ref, action, query, mutation, options) {
+    const authorize = options?.authorizationCallback;
     return {
         createThread: action({
             args: {
@@ -443,8 +462,11 @@ function createAgentApi(component, ref, action, query, mutation) {
             handler: async (ctx, args) => {
                 // Create a function handle that can be scheduled from within the component
                 const handle = await createFunctionHandle(ref);
+                // Serialize workpool options
+                const serializedWorkpool = await serializeWorkpoolOptions(options);
                 const thread = await ctx.runMutation(component.threads.create, {
                     streamFnHandle: handle,
+                    ...serializedWorkpool,
                 });
                 if (args.prompt) {
                     await ctx.runMutation(component.messages.add, {
@@ -468,6 +490,8 @@ function createAgentApi(component, ref, action, query, mutation) {
             },
             returns: v.null(),
             handler: async (ctx, args) => {
+                if (authorize)
+                    await authorize(ctx, args.threadId);
                 await checkThreadIsIdle(component, ctx, args.threadId);
                 await ctx.runMutation(component.messages.add, {
                     threadId: args.threadId,
@@ -492,6 +516,8 @@ function createAgentApi(component, ref, action, query, mutation) {
             },
             returns: v.null(),
             handler: async (ctx, args) => {
+                if (authorize)
+                    await authorize(ctx, args.threadId);
                 const threadId = args.threadId;
                 await checkThreadIsIdle(component, ctx, threadId);
                 if (args.prompt) {
@@ -519,6 +545,8 @@ function createAgentApi(component, ref, action, query, mutation) {
             },
             returns: v.null(),
             handler: async (ctx, args) => {
+                if (authorize)
+                    await authorize(ctx, args.threadId);
                 await ctx.runMutation(component.threads.setStopSignal, {
                     threadId: args.threadId,
                     stopSignal: true,
@@ -532,7 +560,11 @@ function createAgentApi(component, ref, action, query, mutation) {
             },
             returns: v.union(vClientThreadDoc, v.null()),
             handler: async (ctx, args) => {
-                return ctx.runQuery(component.threads.get, { threadId: args.threadId });
+                if (authorize)
+                    await authorize(ctx, args.threadId);
+                return ctx.runQuery(component.threads.get, {
+                    threadId: args.threadId,
+                });
             },
         }),
         listMessages: query({
@@ -540,19 +572,30 @@ function createAgentApi(component, ref, action, query, mutation) {
                 threadId: v.string(),
             },
             handler: async (ctx, args) => {
-                return ctx.runQuery(component.messages.list, { threadId: args.threadId });
+                if (authorize)
+                    await authorize(ctx, args.threadId);
+                return ctx.runQuery(component.messages.list, {
+                    threadId: args.threadId,
+                });
             },
         }),
         listMessagesWithStreams: query({
             args: {
                 threadId: v.string(),
-                streamArgs: v.optional(v.union(v.object({ kind: v.literal("list"), startOrder: v.optional(v.number()) }), v.object({
+                streamArgs: v.optional(v.union(v.object({
+                    kind: v.literal("list"),
+                    startOrder: v.optional(v.number()),
+                }), v.object({
                     kind: v.literal("deltas"),
                     cursors: v.array(v.object({ streamId: v.string(), cursor: v.number() })),
                 }))),
             },
             handler: async (ctx, args) => {
-                const messages = await ctx.runQuery(component.messages.list, { threadId: args.threadId });
+                if (authorize)
+                    await authorize(ctx, args.threadId);
+                const messages = await ctx.runQuery(component.messages.list, {
+                    threadId: args.threadId,
+                });
                 if (!args.streamArgs) {
                     return { messages };
                 }
@@ -601,7 +644,11 @@ function createAgentApi(component, ref, action, query, mutation) {
             },
             returns: v.null(),
             handler: async (ctx, args) => {
-                await ctx.runMutation(component.threads.remove, { threadId: args.threadId });
+                if (authorize)
+                    await authorize(ctx, args.threadId);
+                await ctx.runMutation(component.threads.remove, {
+                    threadId: args.threadId,
+                });
                 return null;
             },
         }),
@@ -638,13 +685,52 @@ function createAgentApi(component, ref, action, query, mutation) {
 /**
  * Define a public agent API that can be called from clients.
  */
-export function defineAgentApi(component, ref) {
-    return createAgentApi(component, ref, actionGeneric, queryGeneric, mutationGeneric);
+export function defineAgentApi(component, ref, options) {
+    return createAgentApi(component, ref, actionGeneric, queryGeneric, mutationGeneric, options);
 }
 /**
  * Define an internal agent API that can only be called from other Convex functions.
  */
-export function defineInternalAgentApi(component, ref) {
-    return createAgentApi(component, ref, internalActionGeneric, internalQueryGeneric, internalMutationGeneric);
+export function defineInternalAgentApi(component, ref, options) {
+    return createAgentApi(component, ref, internalActionGeneric, internalQueryGeneric, internalMutationGeneric, options);
+}
+/**
+ * Creates a workpool bridge mutation that can be used with defineAgentApi.
+ *
+ * This helper creates an internal mutation that forwards action execution to your workpool,
+ * allowing the agent to use workpool's parallelism controls and retry mechanisms.
+ *
+ * @example
+ * ```typescript
+ * // convex/workpool.ts
+ * import { Workpool } from "@convex-dev/workpool";
+ * import { components } from "./_generated/api";
+ * import { createWorkpoolBridge } from "convex-durable-agents";
+ *
+ * const pool = new Workpool(components.workpool, { maxParallelism: 5 });
+ * export const { enqueueWorkpoolAction } = createWorkpoolBridge(pool);
+ *
+ * // convex/chat.ts
+ * export const { createThread, sendMessage, ... } = defineAgentApi(
+ *   components.durable_agent,
+ *   internal.chat.chatAgentHandler,
+ *   { workpoolEnqueueAction: internal.workpool.enqueueWorkpoolAction }
+ * );
+ * ```
+ */
+export function createWorkpoolBridge(workpool) {
+    return {
+        enqueueWorkpoolAction: internalMutationGeneric({
+            args: {
+                action: v.string(),
+                args: v.any(),
+            },
+            returns: v.null(),
+            handler: async (ctx, { action, args }) => {
+                await workpool.enqueueAction(ctx, action, args);
+                return null;
+            },
+        }),
+    };
 }
 //# sourceMappingURL=index.js.map