@upstash/workflow 0.2.5 → 0.2.6

package/astro.d.mts

@@ -1,6 +1,7 @@
 import { APIContext, APIRoute } from 'astro';
-import { b as WorkflowContext, j as PublicServeOptions } from './types-CalpUeFX.mjs';
+import { b as WorkflowContext, j as PublicServeOptions } from './types-Cuqlx2Cr.mjs';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/astro.d.ts

@@ -1,6 +1,7 @@
 import { APIContext, APIRoute } from 'astro';
-import { b as WorkflowContext, j as PublicServeOptions } from './types-CalpUeFX.js';
+import { b as WorkflowContext, j as PublicServeOptions } from './types-Cuqlx2Cr.js';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/astro.js

@@ -1574,11 +1574,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1638,8 +1640,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -1911,8 +1916,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2667,7 +2674,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/astro.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // platforms/astro.ts
 function serve(routeFunction, options) {

package/{chunk-42MM2EPQ.mjs → chunk-OLNSY3BB.mjs}

@@ -1584,14 +1584,16 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 };
 
 // src/context/api/openai.ts
-import { openai } from "@upstash/qstash";
+import { custom, openai } from "@upstash/qstash";
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? openai({ token, organization }) : custom({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: openai({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1651,8 +1653,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return createOpenAI({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -1924,8 +1929,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2680,7 +2687,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/cloudflare.d.mts

@@ -1,5 +1,6 @@
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.mjs';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/cloudflare.d.ts

@@ -1,5 +1,6 @@
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.js';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/cloudflare.js

@@ -1574,11 +1574,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1638,8 +1640,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -1911,8 +1916,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2667,7 +2674,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/cloudflare.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // platforms/cloudflare.ts
 var getArgs = (args) => {

package/express.d.mts

@@ -1,6 +1,7 @@
-import { R as RouteFunction, W as WorkflowServeOptions } from './types-CalpUeFX.mjs';
+import { R as RouteFunction, W as WorkflowServeOptions } from './types-Cuqlx2Cr.mjs';
 import { Router } from 'express';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/express.d.ts

@@ -1,6 +1,7 @@
-import { R as RouteFunction, W as WorkflowServeOptions } from './types-CalpUeFX.js';
+import { R as RouteFunction, W as WorkflowServeOptions } from './types-Cuqlx2Cr.js';
 import { Router } from 'express';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/express.js

@@ -25246,11 +25246,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -25310,8 +25312,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -25583,8 +25588,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -26339,7 +26346,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/express.mjs

@@ -4,7 +4,7 @@ import {
   __require,
   __toESM,
   serveBase
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // node_modules/depd/index.js
 var require_depd = __commonJS({

package/h3.d.mts

@@ -1,6 +1,7 @@
 import * as h3 from 'h3';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.mjs';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/h3.d.ts

@@ -1,6 +1,7 @@
 import * as h3 from 'h3';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.js';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/h3.js

@@ -1886,11 +1886,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1950,8 +1952,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -2223,8 +2228,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2979,7 +2986,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/h3.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // node_modules/defu/dist/defu.mjs
 function isPlainObject(value) {

package/hono.d.mts

@@ -1,7 +1,8 @@
 import { Context } from 'hono';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.mjs';
 import { Variables } from 'hono/types';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/hono.d.ts

@@ -1,7 +1,8 @@
 import { Context } from 'hono';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.js';
 import { Variables } from 'hono/types';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/hono.js

@@ -1574,11 +1574,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1638,8 +1640,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -1911,8 +1916,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2667,7 +2674,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/hono.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // platforms/hono.ts
 var serve = (routeFunction, options) => {

package/index.d.mts

@@ -1,6 +1,7 @@
-import { R as RouteFunction, W as WorkflowServeOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-CalpUeFX.mjs';
-export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-CalpUeFX.mjs';
+import { R as RouteFunction, W as WorkflowServeOptions, E as ExclusiveValidationOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-Cuqlx2Cr.mjs';
+export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-Cuqlx2Cr.mjs';
 import { Client as Client$1, QstashError } from '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 
@@ -12,7 +13,7 @@ import '@ai-sdk/openai';
  * @param options - Options including the client, onFinish callback, and initialPayloadParser.
  * @returns An async method that consumes incoming requests and runs the workflow.
  */
-declare const serve: <TInitialPayload = unknown, TRequest extends Request = Request, TResponse extends Response = Response>(routeFunction: RouteFunction<TInitialPayload>, options?: Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "useJSONContent">) => {
+declare const serve: <TInitialPayload = unknown, TRequest extends Request = Request, TResponse extends Response = Response>(routeFunction: RouteFunction<TInitialPayload>, options?: Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "useJSONContent" | "schema" | "initialPayloadParser"> & ExclusiveValidationOptions<TInitialPayload>) => {
     handler: (request: TRequest) => Promise<TResponse>;
 };
 
@@ -191,4 +192,4 @@ declare class WorkflowAbort extends Error {
     constructor(stepName: string, stepInfo?: Step, cancelWorkflow?: boolean);
 }
 
-export { Client, NotifyResponse, RouteFunction, Step, Waiter, WorkflowAbort, WorkflowError, WorkflowServeOptions, serve };
+export { Client, ExclusiveValidationOptions, NotifyResponse, RouteFunction, Step, Waiter, WorkflowAbort, WorkflowError, WorkflowServeOptions, serve };

package/index.d.ts

@@ -1,6 +1,7 @@
-import { R as RouteFunction, W as WorkflowServeOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-CalpUeFX.js';
-export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-CalpUeFX.js';
+import { R as RouteFunction, W as WorkflowServeOptions, E as ExclusiveValidationOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-Cuqlx2Cr.js';
+export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-Cuqlx2Cr.js';
 import { Client as Client$1, QstashError } from '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 
@@ -12,7 +13,7 @@ import '@ai-sdk/openai';
  * @param options - Options including the client, onFinish callback, and initialPayloadParser.
  * @returns An async method that consumes incoming requests and runs the workflow.
  */
-declare const serve: <TInitialPayload = unknown, TRequest extends Request = Request, TResponse extends Response = Response>(routeFunction: RouteFunction<TInitialPayload>, options?: Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "useJSONContent">) => {
+declare const serve: <TInitialPayload = unknown, TRequest extends Request = Request, TResponse extends Response = Response>(routeFunction: RouteFunction<TInitialPayload>, options?: Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "useJSONContent" | "schema" | "initialPayloadParser"> & ExclusiveValidationOptions<TInitialPayload>) => {
     handler: (request: TRequest) => Promise<TResponse>;
 };
 
@@ -191,4 +192,4 @@ declare class WorkflowAbort extends Error {
     constructor(stepName: string, stepInfo?: Step, cancelWorkflow?: boolean);
 }
 
-export { Client, NotifyResponse, RouteFunction, Step, Waiter, WorkflowAbort, WorkflowError, WorkflowServeOptions, serve };
+export { Client, ExclusiveValidationOptions, NotifyResponse, RouteFunction, Step, Waiter, WorkflowAbort, WorkflowError, WorkflowServeOptions, serve };

package/index.js

@@ -1587,11 +1587,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1651,8 +1653,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -1924,8 +1929,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2680,7 +2687,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/index.mjs

@@ -9,7 +9,7 @@ import {
   makeNotifyRequest,
   serve,
   triggerFirstInvocation
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // src/client/index.ts
 import { Client as QStashClient } from "@upstash/qstash";

package/nextjs.d.mts

@@ -1,6 +1,7 @@
 import { NextApiHandler } from 'next';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.mjs';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/nextjs.d.ts

@@ -1,6 +1,7 @@
 import { NextApiHandler } from 'next';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.js';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/nextjs.js

@@ -1575,11 +1575,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1639,8 +1641,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -1912,8 +1917,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2668,7 +2675,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/nextjs.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // platforms/nextjs.ts
 var serve = (routeFunction, options) => {

package/package.json

@@ -1 +1 @@
-{"name":"@upstash/workflow","version":"v0.2.5","description":"Durable, Reliable and Performant Serverless Functions","main":"./index.js","module":"./index.mjs","types":"./index.d.ts","files":["./*"],"exports":{".":{"import":"./index.mjs","require":"./index.js"},"./dist/nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./h3":{"import":"./h3.mjs","require":"./h3.js"},"./svelte":{"import":"./svelte.mjs","require":"./svelte.js"},"./solidjs":{"import":"./solidjs.mjs","require":"./solidjs.js"},"./workflow":{"import":"./workflow.mjs","require":"./workflow.js"},"./hono":{"import":"./hono.mjs","require":"./hono.js"},"./cloudflare":{"import":"./cloudflare.mjs","require":"./cloudflare.js"},"./astro":{"import":"./astro.mjs","require":"./astro.js"},"./express":{"import":"./express.mjs","require":"./express.js"}},"scripts":{"build":"tsup && cp README.md ./dist/ && cp package.json ./dist/ && cp LICENSE ./dist/","test":"bun test src","fmt":"prettier --write .","lint":"tsc && eslint \"{src,platforms}/**/*.{js,ts,tsx}\" --quiet --fix","check-exports":"bun run build && cd dist && attw -P"},"repository":{"type":"git","url":"git+https://github.com/upstash/workflow-ts.git"},"keywords":["upstash","qstash","workflow","serverless"],"author":"Cahid Arda Oz","license":"MIT","bugs":{"url":"https://github.com/upstash/workflow-ts/issues"},"homepage":"https://github.com/upstash/workflow-ts#readme","devDependencies":{"@commitlint/cli":"^19.5.0","@commitlint/config-conventional":"^19.5.0","@eslint/js":"^9.11.1","@solidjs/start":"^1.0.8","@sveltejs/kit":"^2.6.1","@types/bun":"^1.1.10","@types/express":"^5.0.0","astro":"^4.16.7","eslint":"^9.11.1","eslint-plugin-unicorn":"^55.0.0","express":"^4.21.1","globals":"^15.10.0","h3":"^1.12.0","hono":"^4.6.3","husky":"^9.1.6","next":"^14.2.14","prettier":"3.3.3","tsup":"^8.3.0","typescript":"^5.7.2","typescript-eslint":"^8.18.0"},"dependencies":{"@ai-sdk/openai":"^1.0.15","@upstash/qstash":"^2.7.20","ai":"^4.0.30","zod":"^3.24.1"},"directories":{"example":"examples"}}
+{"name":"@upstash/workflow","version":"v0.2.6","description":"Durable, Reliable and Performant Serverless Functions","main":"./index.js","module":"./index.mjs","types":"./index.d.ts","files":["./*"],"exports":{".":{"import":"./index.mjs","require":"./index.js"},"./dist/nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./h3":{"import":"./h3.mjs","require":"./h3.js"},"./svelte":{"import":"./svelte.mjs","require":"./svelte.js"},"./solidjs":{"import":"./solidjs.mjs","require":"./solidjs.js"},"./workflow":{"import":"./workflow.mjs","require":"./workflow.js"},"./hono":{"import":"./hono.mjs","require":"./hono.js"},"./cloudflare":{"import":"./cloudflare.mjs","require":"./cloudflare.js"},"./astro":{"import":"./astro.mjs","require":"./astro.js"},"./express":{"import":"./express.mjs","require":"./express.js"}},"scripts":{"build":"tsup && cp README.md ./dist/ && cp package.json ./dist/ && cp LICENSE ./dist/","test":"bun test src","fmt":"prettier --write .","lint":"tsc && eslint \"{src,platforms}/**/*.{js,ts,tsx}\" --quiet --fix","check-exports":"bun run build && cd dist && attw -P"},"repository":{"type":"git","url":"git+https://github.com/upstash/workflow-ts.git"},"keywords":["upstash","qstash","workflow","serverless"],"author":"Cahid Arda Oz","license":"MIT","bugs":{"url":"https://github.com/upstash/workflow-ts/issues"},"homepage":"https://github.com/upstash/workflow-ts#readme","devDependencies":{"@commitlint/cli":"^19.5.0","@commitlint/config-conventional":"^19.5.0","@eslint/js":"^9.11.1","@solidjs/start":"^1.0.8","@sveltejs/kit":"^2.6.1","@types/bun":"^1.1.10","@types/express":"^5.0.0","astro":"^4.16.7","eslint":"^9.11.1","eslint-plugin-unicorn":"^55.0.0","express":"^4.21.1","globals":"^15.10.0","h3":"^1.12.0","hono":"^4.6.3","husky":"^9.1.6","next":"^14.2.14","prettier":"3.3.3","tsup":"^8.3.0","typescript":"^5.7.2","typescript-eslint":"^8.18.0"},"dependencies":{"@ai-sdk/openai":"^1.0.15","@upstash/qstash":"^2.7.20","ai":"^4.0.30","zod":"^3.24.1"},"directories":{"example":"examples"}}

package/solidjs.d.mts

@@ -1,6 +1,7 @@
 import { APIEvent } from '@solidjs/start/server';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.mjs';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/solidjs.d.ts

@@ -1,6 +1,7 @@
 import { APIEvent } from '@solidjs/start/server';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.js';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/solidjs.js

@@ -1574,11 +1574,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1638,8 +1640,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -1911,8 +1916,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2667,7 +2674,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/solidjs.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // platforms/solidjs.ts
 var serve = (routeFunction, options) => {

package/svelte.d.mts

@@ -1,6 +1,7 @@
 import { RequestHandler } from '@sveltejs/kit';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.mjs';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.mjs';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/svelte.d.ts

@@ -1,6 +1,7 @@
 import { RequestHandler } from '@sveltejs/kit';
-import { R as RouteFunction, j as PublicServeOptions } from './types-CalpUeFX.js';
+import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.js';
 import '@upstash/qstash';
+import 'zod';
 import 'ai';
 import '@ai-sdk/openai';
 

package/svelte.js

@@ -1574,11 +1574,13 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
-    const { token, organization, operation, ...parameters } = settings;
+    const { token, organization, operation, baseURL, ...parameters } = settings;
+    const useOpenAI = baseURL === void 0;
+    const provider = useOpenAI ? (0, import_qstash6.openai)({ token, organization }) : (0, import_qstash6.custom)({ baseUrl: baseURL, token });
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash6.openai)({ token, organization })
+        provider
       },
       ...parameters
     });
@@ -1638,8 +1640,11 @@ you need from that agent.
 `;
 
 // src/agents/adapters.ts
-var createWorkflowOpenAI = (context) => {
+var createWorkflowOpenAI = (context, config) => {
+  const { baseURL, apiKey } = config ?? {};
   return (0, import_openai2.createOpenAI)({
+    baseURL,
+    apiKey,
     compatibility: "strict",
     fetch: async (input, init) => {
       try {
@@ -1911,8 +1916,10 @@ var WorkflowAgents = class {
    * creates an openai model for agents
    */
   openai(...params) {
-    const openai2 = createWorkflowOpenAI(this.context);
-    return openai2(...params);
+    const [model, settings] = params;
+    const { baseURL, apiKey, ...otherSettings } = settings ?? {};
+    const openai2 = createWorkflowOpenAI(this.context, { baseURL, apiKey });
+    return openai2(model, otherSettings);
   }
 };
 
@@ -2667,7 +2674,8 @@ var processOptions = (options) => {
         return void 0;
       }
       try {
-        return JSON.parse(initialRequest);
+        const parsed = JSON.parse(initialRequest);
+        return options?.schema ? options.schema.parse(parsed) : parsed;
       } catch (error) {
         if (error instanceof SyntaxError) {
           return initialRequest;

package/svelte.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-42MM2EPQ.mjs";
+} from "./chunk-OLNSY3BB.mjs";
 
 // platforms/svelte.ts
 var serve = (routeFunction, options) => {

package/{types-CalpUeFX.d.mts → types-Cuqlx2Cr.d.mts}

@@ -1,4 +1,5 @@
 import { PublishRequest, Client, Receiver, HTTPMethods as HTTPMethods$1 } from '@upstash/qstash';
+import { z } from 'zod';
 import * as ai from 'ai';
 import { CoreTool, generateText } from 'ai';
 import * as _ai_sdk_openai from '@ai-sdk/openai';
@@ -349,6 +350,7 @@ declare class OpenAIAPI extends BaseWorkflowApi {
     call<TResult = ChatCompletion, TBody = CreateChatCompletion>(stepName: string, settings: ApiCallSettings<TBody, {
         token: string;
         organization?: string;
+        baseURL?: string;
         operation: "chat.completions.create";
     }>): Promise<CallResponse<TResult>>;
 }
@@ -391,45 +393,16 @@ declare class WorkflowApi extends BaseWorkflowApi {
 }
 
 /**
- * An Agent which utilizes the model and tools available to it
- * to achieve a given task
+ * creates an AI SDK openai client with a custom
+ * fetch implementation which uses context.call.
  *
- * @param name Name of the agent
- * @param background Background of the agent
- * @param model LLM model to use
- * @param tools tools available to the agent
- * @param maxSteps number of times the agent can call the LLM at most. If
- *   the agent abruptly stops execution after calling tools, you may need
- *   to increase maxSteps
- * @param temparature temparature used when calling the LLM
+ * @param context workflow context
+ * @returns ai sdk openai
  */
-declare class Agent {
-    readonly name: AgentParameters["name"];
-    readonly tools: AgentParameters["tools"];
-    readonly maxSteps: AgentParameters["maxSteps"];
-    readonly background: AgentParameters["background"];
-    readonly model: AgentParameters["model"];
-    readonly temparature: AgentParameters["temparature"];
-    private readonly context;
-    constructor({ tools, maxSteps, background, name, model, temparature }: AgentParameters, context: WorkflowContext);
-    /**
-     * Trigger the agent by passing a prompt
-     *
-     * @param prompt task to assign to the agent
-     * @returns Response as `{ text: string }`
-     */
-    call({ prompt }: {
-        prompt: string;
-    }): Promise<{
-        text: string;
-    }>;
-    /**
-     * Convert the agent to a tool which can be used by other agents.
-     *
-     * @returns the agent as a tool
-     */
-    asTool(): AISDKTool;
-}
+declare const createWorkflowOpenAI: (context: WorkflowContext, config?: {
+    baseURL?: string;
+    apiKey?: string;
+}) => _ai_sdk_openai.OpenAIProvider;
 
 type AISDKTool = CoreTool;
 type LangchainTool = {
@@ -501,15 +474,53 @@ type MultiAgentTaskParams = TaskParams & {
      */
     background?: string;
 };
+type ModelParams = Parameters<ReturnType<typeof createWorkflowOpenAI>>;
+type CustomModelSettings = ModelParams["1"] & {
+    baseURL?: string;
+    apiKey?: string;
+};
+type CustomModelParams = [ModelParams[0], CustomModelSettings?];
 
 /**
- * creates an AI SDK openai client with a custom
- * fetch implementation which uses context.call.
+ * An Agent which utilizes the model and tools available to it
+ * to achieve a given task
  *
- * @param context workflow context
- * @returns ai sdk openai
+ * @param name Name of the agent
+ * @param background Background of the agent
+ * @param model LLM model to use
+ * @param tools tools available to the agent
+ * @param maxSteps number of times the agent can call the LLM at most. If
+ *   the agent abruptly stops execution after calling tools, you may need
+ *   to increase maxSteps
+ * @param temparature temparature used when calling the LLM
  */
-declare const createWorkflowOpenAI: (context: WorkflowContext) => _ai_sdk_openai.OpenAIProvider;
+declare class Agent {
+    readonly name: AgentParameters["name"];
+    readonly tools: AgentParameters["tools"];
+    readonly maxSteps: AgentParameters["maxSteps"];
+    readonly background: AgentParameters["background"];
+    readonly model: AgentParameters["model"];
+    readonly temparature: AgentParameters["temparature"];
+    private readonly context;
+    constructor({ tools, maxSteps, background, name, model, temparature }: AgentParameters, context: WorkflowContext);
+    /**
+     * Trigger the agent by passing a prompt
+     *
+     * @param prompt task to assign to the agent
+     * @returns Response as `{ text: string }`
+     */
+    call({ prompt }: {
+        prompt: string;
+    }): Promise<{
+        text: string;
+    }>;
+    /**
+     * Convert the agent to a tool which can be used by other agents.
+     *
+     * @returns the agent as a tool
+     */
+    asTool(): AISDKTool;
+}
 
 /**
  * An Agent Task
@@ -621,7 +632,7 @@ declare class WorkflowAgents {
     /**
      * creates an openai model for agents
      */
-    openai(...params: Parameters<ReturnType<typeof createWorkflowOpenAI>>): ai.LanguageModelV1;
+    openai(...params: CustomModelParams): ai.LanguageModelV1;
 }
 
 /**
@@ -1026,7 +1037,7 @@ type StepFunction<TResult> = AsyncStepFunction<TResult> | SyncStepFunction<TResu
 type ParallelCallState = "first" | "partial" | "discard" | "last";
 type RouteFunction<TInitialPayload> = (context: WorkflowContext<TInitialPayload>) => Promise<void>;
 type FinishCondition = "success" | "duplicate-step" | "fromCallback" | "auth-fail" | "failure-callback" | "workflow-already-ended";
-type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload = unknown> = {
+type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload = unknown> = ValidationOptions<TInitialPayload> & {
     /**
      * QStash client
      */
@@ -1038,10 +1049,6 @@ type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload
      * @returns response
      */
     onStepFinish?: (workflowRunId: string, finishCondition: FinishCondition) => TResponse;
-    /**
-     * Function to parse the initial payload passed by the user
-     */
-    initialPayloadParser?: (initialPayload: string) => TInitialPayload;
     /**
      * Url of the endpoint where the workflow is set up.
      *
@@ -1122,6 +1129,17 @@ type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload
      * Set `disableTelemetry` to disable this behavior.
      */
     disableTelemetry?: boolean;
+} & ValidationOptions<TInitialPayload>;
+type ValidationOptions<TInitialPayload> = {
+    schema?: z.ZodType<TInitialPayload>;
+    initialPayloadParser?: (initialPayload: string) => TInitialPayload;
+};
+type ExclusiveValidationOptions<TInitialPayload> = {
+    schema?: ValidationOptions<TInitialPayload>["schema"];
+    initialPayloadParser?: never;
+} | {
+    schema?: never;
+    initialPayloadParser?: ValidationOptions<TInitialPayload>["initialPayloadParser"];
 };
 type Telemetry = {
     /**
@@ -1137,7 +1155,7 @@ type Telemetry = {
      */
     runtime?: string;
 };
-type PublicServeOptions<TInitialPayload = unknown, TResponse extends Response = Response> = Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "onStepFinish" | "useJSONContent">;
+type PublicServeOptions<TInitialPayload = unknown, TResponse extends Response = Response> = Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "onStepFinish" | "useJSONContent" | "schema" | "initialPayloadParser"> & ExclusiveValidationOptions<TInitialPayload>;
 /**
  * Payload passed as body in failureFunction
  */
@@ -1295,4 +1313,4 @@ type HeaderParams = {
     callTimeout?: never;
 });
 
-export { type AsyncStepFunction as A, type CallResponse as C, type Duration as D, type FinishCondition as F, type HeaderParams as H, type LogLevel as L, type NotifyResponse as N, type ParallelCallState as P, type RouteFunction as R, type Step as S, type Telemetry as T, type WorkflowServeOptions as W, type Waiter as a, WorkflowContext as b, type WorkflowClient as c, type WorkflowReceiver as d, StepTypes as e, type StepType as f, type RawStep as g, type SyncStepFunction as h, type StepFunction as i, type PublicServeOptions as j, type FailureFunctionPayload as k, type RequiredExceptFields as l, type WaitRequest as m, type WaitStepResponse as n, type NotifyStepResponse as o, type WaitEventOptions as p, type CallSettings as q, type WorkflowLoggerOptions as r, WorkflowLogger as s };
+export { type AsyncStepFunction as A, type CallResponse as C, type Duration as D, type ExclusiveValidationOptions as E, type FinishCondition as F, type HeaderParams as H, type LogLevel as L, type NotifyResponse as N, type ParallelCallState as P, type RouteFunction as R, type Step as S, type Telemetry as T, type WorkflowServeOptions as W, type Waiter as a, WorkflowContext as b, type WorkflowClient as c, type WorkflowReceiver as d, StepTypes as e, type StepType as f, type RawStep as g, type SyncStepFunction as h, type StepFunction as i, type PublicServeOptions as j, type FailureFunctionPayload as k, type RequiredExceptFields as l, type WaitRequest as m, type WaitStepResponse as n, type NotifyStepResponse as o, type WaitEventOptions as p, type CallSettings as q, type WorkflowLoggerOptions as r, WorkflowLogger as s };

package/{types-CalpUeFX.d.ts → types-Cuqlx2Cr.d.ts}

@@ -1,4 +1,5 @@
 import { PublishRequest, Client, Receiver, HTTPMethods as HTTPMethods$1 } from '@upstash/qstash';
+import { z } from 'zod';
 import * as ai from 'ai';
 import { CoreTool, generateText } from 'ai';
 import * as _ai_sdk_openai from '@ai-sdk/openai';
@@ -349,6 +350,7 @@ declare class OpenAIAPI extends BaseWorkflowApi {
     call<TResult = ChatCompletion, TBody = CreateChatCompletion>(stepName: string, settings: ApiCallSettings<TBody, {
         token: string;
         organization?: string;
+        baseURL?: string;
         operation: "chat.completions.create";
     }>): Promise<CallResponse<TResult>>;
 }
@@ -391,45 +393,16 @@ declare class WorkflowApi extends BaseWorkflowApi {
 }
 
 /**
- * An Agent which utilizes the model and tools available to it
- * to achieve a given task
+ * creates an AI SDK openai client with a custom
+ * fetch implementation which uses context.call.
  *
- * @param name Name of the agent
- * @param background Background of the agent
- * @param model LLM model to use
- * @param tools tools available to the agent
- * @param maxSteps number of times the agent can call the LLM at most. If
- *   the agent abruptly stops execution after calling tools, you may need
- *   to increase maxSteps
- * @param temparature temparature used when calling the LLM
+ * @param context workflow context
+ * @returns ai sdk openai
  */
-declare class Agent {
-    readonly name: AgentParameters["name"];
-    readonly tools: AgentParameters["tools"];
-    readonly maxSteps: AgentParameters["maxSteps"];
-    readonly background: AgentParameters["background"];
-    readonly model: AgentParameters["model"];
-    readonly temparature: AgentParameters["temparature"];
-    private readonly context;
-    constructor({ tools, maxSteps, background, name, model, temparature }: AgentParameters, context: WorkflowContext);
-    /**
-     * Trigger the agent by passing a prompt
-     *
-     * @param prompt task to assign to the agent
-     * @returns Response as `{ text: string }`
-     */
-    call({ prompt }: {
-        prompt: string;
-    }): Promise<{
-        text: string;
-    }>;
-    /**
-     * Convert the agent to a tool which can be used by other agents.
-     *
-     * @returns the agent as a tool
-     */
-    asTool(): AISDKTool;
-}
+declare const createWorkflowOpenAI: (context: WorkflowContext, config?: {
+    baseURL?: string;
+    apiKey?: string;
+}) => _ai_sdk_openai.OpenAIProvider;
 
 type AISDKTool = CoreTool;
 type LangchainTool = {
@@ -501,15 +474,53 @@ type MultiAgentTaskParams = TaskParams & {
      */
     background?: string;
 };
+type ModelParams = Parameters<ReturnType<typeof createWorkflowOpenAI>>;
+type CustomModelSettings = ModelParams["1"] & {
+    baseURL?: string;
+    apiKey?: string;
+};
+type CustomModelParams = [ModelParams[0], CustomModelSettings?];
 
 /**
- * creates an AI SDK openai client with a custom
- * fetch implementation which uses context.call.
+ * An Agent which utilizes the model and tools available to it
+ * to achieve a given task
  *
- * @param context workflow context
- * @returns ai sdk openai
+ * @param name Name of the agent
+ * @param background Background of the agent
+ * @param model LLM model to use
+ * @param tools tools available to the agent
+ * @param maxSteps number of times the agent can call the LLM at most. If
+ *   the agent abruptly stops execution after calling tools, you may need
+ *   to increase maxSteps
+ * @param temparature temparature used when calling the LLM
  */
-declare const createWorkflowOpenAI: (context: WorkflowContext) => _ai_sdk_openai.OpenAIProvider;
+declare class Agent {
+    readonly name: AgentParameters["name"];
+    readonly tools: AgentParameters["tools"];
+    readonly maxSteps: AgentParameters["maxSteps"];
+    readonly background: AgentParameters["background"];
+    readonly model: AgentParameters["model"];
+    readonly temparature: AgentParameters["temparature"];
+    private readonly context;
+    constructor({ tools, maxSteps, background, name, model, temparature }: AgentParameters, context: WorkflowContext);
+    /**
+     * Trigger the agent by passing a prompt
+     *
+     * @param prompt task to assign to the agent
+     * @returns Response as `{ text: string }`
+     */
+    call({ prompt }: {
+        prompt: string;
+    }): Promise<{
+        text: string;
+    }>;
+    /**
+     * Convert the agent to a tool which can be used by other agents.
+     *
+     * @returns the agent as a tool
+     */
+    asTool(): AISDKTool;
+}
 
 /**
  * An Agent Task
@@ -621,7 +632,7 @@ declare class WorkflowAgents {
     /**
      * creates an openai model for agents
      */
-    openai(...params: Parameters<ReturnType<typeof createWorkflowOpenAI>>): ai.LanguageModelV1;
+    openai(...params: CustomModelParams): ai.LanguageModelV1;
 }
 
 /**
@@ -1026,7 +1037,7 @@ type StepFunction<TResult> = AsyncStepFunction<TResult> | SyncStepFunction<TResu
 type ParallelCallState = "first" | "partial" | "discard" | "last";
 type RouteFunction<TInitialPayload> = (context: WorkflowContext<TInitialPayload>) => Promise<void>;
 type FinishCondition = "success" | "duplicate-step" | "fromCallback" | "auth-fail" | "failure-callback" | "workflow-already-ended";
-type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload = unknown> = {
+type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload = unknown> = ValidationOptions<TInitialPayload> & {
     /**
      * QStash client
      */
@@ -1038,10 +1049,6 @@ type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload
      * @returns response
      */
     onStepFinish?: (workflowRunId: string, finishCondition: FinishCondition) => TResponse;
-    /**
-     * Function to parse the initial payload passed by the user
-     */
-    initialPayloadParser?: (initialPayload: string) => TInitialPayload;
     /**
      * Url of the endpoint where the workflow is set up.
      *
@@ -1122,6 +1129,17 @@ type WorkflowServeOptions<TResponse extends Response = Response, TInitialPayload
      * Set `disableTelemetry` to disable this behavior.
      */
     disableTelemetry?: boolean;
+} & ValidationOptions<TInitialPayload>;
+type ValidationOptions<TInitialPayload> = {
+    schema?: z.ZodType<TInitialPayload>;
+    initialPayloadParser?: (initialPayload: string) => TInitialPayload;
+};
+type ExclusiveValidationOptions<TInitialPayload> = {
+    schema?: ValidationOptions<TInitialPayload>["schema"];
+    initialPayloadParser?: never;
+} | {
+    schema?: never;
+    initialPayloadParser?: ValidationOptions<TInitialPayload>["initialPayloadParser"];
 };
 type Telemetry = {
     /**
@@ -1137,7 +1155,7 @@ type Telemetry = {
      */
     runtime?: string;
 };
-type PublicServeOptions<TInitialPayload = unknown, TResponse extends Response = Response> = Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "onStepFinish" | "useJSONContent">;
+type PublicServeOptions<TInitialPayload = unknown, TResponse extends Response = Response> = Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "onStepFinish" | "useJSONContent" | "schema" | "initialPayloadParser"> & ExclusiveValidationOptions<TInitialPayload>;
 /**
  * Payload passed as body in failureFunction
  */
@@ -1295,4 +1313,4 @@ type HeaderParams = {
     callTimeout?: never;
 });
 
-export { type AsyncStepFunction as A, type CallResponse as C, type Duration as D, type FinishCondition as F, type HeaderParams as H, type LogLevel as L, type NotifyResponse as N, type ParallelCallState as P, type RouteFunction as R, type Step as S, type Telemetry as T, type WorkflowServeOptions as W, type Waiter as a, WorkflowContext as b, type WorkflowClient as c, type WorkflowReceiver as d, StepTypes as e, type StepType as f, type RawStep as g, type SyncStepFunction as h, type StepFunction as i, type PublicServeOptions as j, type FailureFunctionPayload as k, type RequiredExceptFields as l, type WaitRequest as m, type WaitStepResponse as n, type NotifyStepResponse as o, type WaitEventOptions as p, type CallSettings as q, type WorkflowLoggerOptions as r, WorkflowLogger as s };
+export { type AsyncStepFunction as A, type CallResponse as C, type Duration as D, type ExclusiveValidationOptions as E, type FinishCondition as F, type HeaderParams as H, type LogLevel as L, type NotifyResponse as N, type ParallelCallState as P, type RouteFunction as R, type Step as S, type Telemetry as T, type WorkflowServeOptions as W, type Waiter as a, WorkflowContext as b, type WorkflowClient as c, type WorkflowReceiver as d, StepTypes as e, type StepType as f, type RawStep as g, type SyncStepFunction as h, type StepFunction as i, type PublicServeOptions as j, type FailureFunctionPayload as k, type RequiredExceptFields as l, type WaitRequest as m, type WaitStepResponse as n, type NotifyStepResponse as o, type WaitEventOptions as p, type CallSettings as q, type WorkflowLoggerOptions as r, WorkflowLogger as s };