@threaded/ai 1.0.29 → 1.1.0

package/dist/image.js

@@ -0,0 +1,120 @@
+import { getKey, parseModelName } from "./utils";
+const providerKeyEnvVars = {
+    openai: "OPENAI_API_KEY",
+    xai: "XAI_API_KEY",
+    google: "GEMINI_API_KEY",
+};
+const getApiKey = (provider) => {
+    try {
+        return getKey(provider);
+    }
+    catch {
+        const envVar = providerKeyEnvVars[provider];
+        const key = envVar ? process.env[envVar] || "" : "";
+        if (!key)
+            throw new Error(`No API key found for provider: ${provider}`);
+        return key;
+    }
+};
+const generateOpenAICompatible = async (endpoint, modelName, prompt, apiKey, config) => {
+    const isGptImage = modelName.startsWith("gpt-image");
+    const body = {
+        model: modelName,
+        prompt,
+    };
+    if (!isGptImage) {
+        body.response_format = config?.responseFormat || "b64_json";
+    }
+    if (config?.n)
+        body.n = config.n;
+    if (config?.size)
+        body.size = config.size;
+    if (config?.quality)
+        body.quality = config.quality;
+    if (config?.style && !isGptImage)
+        body.style = config.style;
+    if (isGptImage) {
+        if (config?.outputFormat)
+            body.output_format = config.outputFormat;
+        if (config?.outputCompression != null)
+            body.output_compression = config.outputCompression;
+        if (config?.background)
+            body.background = config.background;
+        if (config?.moderation)
+            body.moderation = config.moderation;
+    }
+    const response = await fetch(endpoint, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${apiKey}`,
+        },
+        body: JSON.stringify(body),
+    });
+    if (!response.ok) {
+        const error = await response.text();
+        throw new Error(`API error: ${error}`);
+    }
+    const data = (await response.json());
+    const image = data.data[0];
+    return {
+        data: image.b64_json || image.url,
+        revisedPrompt: image.revised_prompt,
+    };
+};
+const generateGoogle = async (modelName, prompt, apiKey, config) => {
+    const endpoint = `https://generativelanguage.googleapis.com/v1beta/models/${modelName}:generateContent`;
+    const body = {
+        contents: [{ parts: [{ text: prompt }] }],
+        generationConfig: {
+            responseModalities: ["TEXT", "IMAGE"],
+        },
+    };
+    const imageConfig = {};
+    if (config?.aspectRatio)
+        imageConfig.aspectRatio = config.aspectRatio;
+    if (config?.imageSize)
+        imageConfig.imageSize = config.imageSize;
+    if (Object.keys(imageConfig).length > 0) {
+        body.generationConfig.imageConfig = imageConfig;
+    }
+    const response = await fetch(endpoint, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            "x-goog-api-key": apiKey,
+        },
+        body: JSON.stringify(body),
+    });
+    if (!response.ok) {
+        const error = await response.text();
+        throw new Error(`Google API error: ${error}`);
+    }
+    const data = (await response.json());
+    const parts = data.candidates?.[0]?.content?.parts || [];
+    const imagePart = parts.find((p) => p.inlineData);
+    const textPart = parts.find((p) => p.text);
+    if (!imagePart?.inlineData?.data) {
+        throw new Error("No image data in response");
+    }
+    return {
+        data: imagePart.inlineData.data,
+        revisedPrompt: textPart?.text,
+    };
+};
+export const generateImage = async (model, prompt, config) => {
+    const { provider, model: modelName } = parseModelName(model);
+    const providerLower = provider.toLowerCase();
+    const apiKey = getApiKey(providerLower);
+    switch (providerLower) {
+        case "openai":
+            return generateOpenAICompatible("https://api.openai.com/v1/images/generations", modelName, prompt, apiKey, config);
+        case "xai":
+            return generateOpenAICompatible("https://api.x.ai/v1/images/generations", modelName, prompt, apiKey, config);
+        case "google":
+            return generateGoogle(modelName, prompt, apiKey, config);
+        default:
+            throw new Error(`Unsupported image generation provider: ${provider}`);
+    }
+};
+//# sourceMappingURL=image.js.map

package/dist/image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image.js","sourceRoot":"","sources":["../src/image.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAEjD,MAAM,kBAAkB,GAA2B;IACjD,MAAM,EAAE,gBAAgB;IACxB,GAAG,EAAE,aAAa;IAClB,MAAM,EAAE,gBAAgB;CACzB,CAAC;AAEF,MAAM,SAAS,GAAG,CAAC,QAAgB,EAAU,EAAE;IAC7C,IAAI,CAAC;QACH,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,MAAM,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACpD,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,QAAQ,EAAE,CAAC,CAAC;QACxE,OAAO,GAAG,CAAC;IACb,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,wBAAwB,GAAG,KAAK,EACpC,QAAgB,EAChB,SAAiB,EACjB,MAAc,EACd,MAAc,EACd,MAAoB,EACE,EAAE;IACxB,MAAM,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAErD,MAAM,IAAI,GAAwB;QAChC,KAAK,EAAE,SAAS;QAChB,MAAM;KACP,CAAC;IAEF,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,IAAI,CAAC,eAAe,GAAG,MAAM,EAAE,cAAc,IAAI,UAAU,CAAC;IAC9D,CAAC;IAED,IAAI,MAAM,EAAE,CAAC;QAAE,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;IACjC,IAAI,MAAM,EAAE,IAAI;QAAE,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;IAC1C,IAAI,MAAM,EAAE,OAAO;QAAE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IACnD,IAAI,MAAM,EAAE,KAAK,IAAI,CAAC,UAAU;QAAE,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IAE5D,IAAI,UAAU,EAAE,CAAC;QACf,IAAI,MAAM,EAAE,YAAY;YAAE,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,YAAY,CAAC;QACnE,IAAI,MAAM,EAAE,iBAAiB,IAAI,IAAI;YAAE,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,iBAAiB,CAAC;QAC1F,IAAI,MAAM,EAAE,UAAU;YAAE,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QAC5D,IAAI,MAAM,EAAE,UAAU;YAAE,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;IAC9D,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;QACrC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,MAAM,EAAE;SAClC;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,cAAc,KAAK,EAAE,CAAC,CAAC;IACzC,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAQ,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAE3B,OAAO;QACL,IAAI,EAAE,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,GAAG;QACjC,aAAa,EAAE,KAAK,CAAC,cAAc;KACpC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,cAAc,GAAG,KAAK,EAC1B,SAAiB,EACjB,MAAc,EACd,MAAc,EACd,MAAoB,EACE,EAAE;IACxB,MAAM,QAAQ,GAAG,2DAA2D,SAAS,kBAAkB,CAAC;IAExG,MAAM,IAAI,GAAwB;QAChC,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QACzC,gBAAgB,EAAE;YAChB,kBAAkB,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;SACtC;KACF,CAAC;IAEF,MAAM,WAAW,GAAwB,EAAE,CAAC;IAC5C,IAAI,MAAM,EAAE,WAAW;QAAE,WAAW,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IACtE,IAAI,MAAM,EAAE,SAAS;QAAE,WAAW,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;IAEhE,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxC,IAAI,CAAC,gBAAgB,CAAC,WAAW,GAAG,WAAW,CAAC;IAClD,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;QACrC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,gBAAgB,EAAE,MAAM;SACzB;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,qBAAqB,KAAK,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAQ,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;IAEzD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;IACvD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAEhD,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO;QACL,IAAI,EAAE,SAAS,CAAC,UAAU,CAAC,IAAI;QAC/B,aAAa,EAAE,QAAQ,EAAE,IAAI;KAC9B,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,KAAK,EAChC,KAAa,EACb,MAAc,EACd,MAAoB,EACE,EAAE;IACxB,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;IAC7D,MAAM,aAAa,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;IAC7C,MAAM,MAAM,GAAG,SAAS,CAAC,aAAa,CAAC,CAAC;IAExC,QAAQ,aAAa,EAAE,CAAC;QACtB,KAAK,QAAQ;YACX,OAAO,wBAAwB,CAC7B,8CAA8C,EAC9C,SAAS,EACT,MAAM,EACN,MAAM,EACN,MAAM,CACP,CAAC;QAEJ,KAAK,KAAK;YACR,OAAO,wBAAwB,CAC7B,wCAAwC,EACxC,SAAS,EACT,MAAM,EACN,MAAM,EACN,MAAM,CACP,CAAC;QAEJ,KAAK,QAAQ;YACX,OAAO,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAE3D;YACE,MAAM,IAAI,KAAK,CAAC,0CAA0C,QAAQ,EAAE,CAAC,CAAC;IAC1E,CAAC;AACH,CAAC,CAAC"}

package/dist/index.d.ts

@@ -1,350 +1,18 @@
-import { Client } from '@modelcontextprotocol/sdk/client/index';
-
-interface Message {
-    role: "system" | "user" | "assistant" | "tool";
-    content: string;
-    tool_call_id?: string;
-}
-interface ToolCall {
-    id: string;
-    function: {
-        name: string;
-        arguments: string;
-    };
-}
-interface ToolCallResult {
-    name: string;
-    inputs: any;
-    results: any;
-}
-interface ToolDefinition {
-    type: "function";
-    function: {
-        name: string;
-        description: string;
-        parameters: {
-            type: string;
-            properties: Record<string, any>;
-            required?: string[];
-        };
-    };
-}
-interface ToolConfig {
-    name: string;
-    description: string;
-    schema: Record<string, SchemaProperty> | StandardSchema;
-    execute: (args: any) => Promise<any> | any;
-    _maxCalls?: number;
-}
-interface SchemaProperty {
-    type: "string" | "number" | "boolean" | "array" | "object";
-    description?: string;
-    enum?: string[];
-    optional?: boolean;
-    items?: SchemaProperty;
-    properties?: Record<string, SchemaProperty>;
-}
-interface ToolExecutionConfig {
-    /** require user approval before executing tools */
-    requireApproval?: boolean;
-    /**
-     * custom callback to handle tool approval, return true to approve
-     *
-     * @example
-     * // simple callback
-     * approvalCallback: (call) => call.function.name !== 'dangerousTool'
-     *
-     * @example
-     * // event-driven (SSE): server sends approval request, waits for client POST
-     * approvalCallback: (call) => new Promise((resolve) => {
-     *   pendingApprovals.set(call.id, resolve);
-     *   res.write(`data: ${JSON.stringify({ type: 'approval_needed', call })}\n\n`);
-     * })
-     * // then: app.post('/approve/:id', (req) => pendingApprovals.get(id)(req.body.approved))
-     */
-    approvalCallback?: (call: ToolCall) => boolean | Promise<boolean>;
-    /** execute tools in parallel instead of sequentially */
-    parallel?: boolean;
-    /** number of times to retry failed tool executions */
-    retryCount?: number;
-    /** identifier for approval requests, useful for managing multiple approval flows */
-    approvalId?: string;
-    /** execute tools immediately upon approval instead of waiting for all approvals (default: false, only applies when requireApproval is true) */
-    executeOnApproval?: boolean;
-}
-type StreamEvent = {
-    type: 'content';
-    content: string;
-} | {
-    type: 'tool_calls_ready';
-    calls: ToolCall[];
-} | {
-    type: 'tool_executing';
-    call: ToolCall;
-} | {
-    type: 'tool_complete';
-    call: ToolCall;
-    result: any;
-} | {
-    type: 'tool_error';
-    call: ToolCall;
-    error: string;
-} | {
-    type: 'approval_requested';
-    call: ToolCall;
-    requestId: string;
-} | {
-    type: 'usage';
-    usage: TokenUsage;
-};
-interface ConversationContext {
-    history: Message[];
-    lastRequest?: Message;
-    lastResponse?: Message & {
-        tool_calls?: ToolCall[];
-    };
-    tools?: ToolDefinition[];
-    toolExecutors?: Record<string, Function>;
-    stream?: (event: StreamEvent) => void;
-    stopReason?: string;
-    toolCallCounts?: Record<string, number>;
-    toolLimits?: Record<string, number>;
-    toolConfig?: ToolExecutionConfig;
-    abortSignal?: AbortSignal;
-    usage?: TokenUsage;
-}
-declare enum Inherit {
-    Nothing = 0,
-    Conversation = 1,
-    Tools = 2,
-    All = 3
-}
-interface ScopeConfig {
-    inherit?: number;
-    tools?: ToolConfig[];
-    toolConfig?: ToolExecutionConfig;
-    system?: string;
-    silent?: boolean;
-    until?: (ctx: ConversationContext) => boolean;
-    stream?: (event: StreamEvent) => void;
-}
-type StepFunction = (ctx: ConversationContext) => Promise<ConversationContext>;
-type ComposedFunction = (ctxOrMessage: ConversationContext | string) => Promise<ConversationContext>;
-interface JsonSchema {
-    name: string;
-    schema: Record<string, any>;
-}
-interface StandardSchema {
-    "~standard": any;
-    [key: string]: any;
-}
-interface ProviderConfig {
-    model: string;
-    instructions?: string;
-    schema?: JsonSchema;
-    apiKey?: string;
-    baseUrl?: string;
-}
-interface ParsedModel {
-    provider: string;
-    model: string;
-}
-interface ApiKeys {
-    openai?: string;
-    anthropic?: string;
-    google?: string;
-    [provider: string]: string | undefined;
-}
-interface ThreadStore {
-    get(threadId: string): Promise<Message[]>;
-    set(threadId: string, messages: Message[]): Promise<void>;
-}
-interface Thread {
-    id: string;
-    store: ThreadStore;
-    generate(step: StepFunction): Promise<ConversationContext>;
-    message(content: string, workflow?: StepFunction, options?: {
-        abortSignal?: AbortSignal;
-    }): Promise<ConversationContext>;
-}
-interface RetryOptions {
-    times?: number;
-}
-interface ImageConfig {
-    n?: number;
-    size?: string;
-    quality?: "standard" | "hd" | "low" | "medium" | "high" | "auto";
-    style?: "vivid" | "natural";
-    responseFormat?: "url" | "b64_json";
-    aspectRatio?: string;
-    outputFormat?: "png" | "jpeg" | "webp";
-    outputCompression?: number;
-    background?: "transparent" | "opaque" | "auto";
-    moderation?: "auto" | "low";
-    imageSize?: "1K" | "2K";
-}
-interface ImageResult {
-    data: string;
-    revisedPrompt?: string;
-}
-interface TokenUsage {
-    promptTokens: number;
-    completionTokens: number;
-    totalTokens: number;
-}
-
-declare const createMCPTools: (client: Client) => Promise<ToolConfig[]>;
-
-declare const toolConfigToToolDefinition: (tool: ToolConfig) => ToolDefinition;
-declare const parseModelName: (model: string) => ParsedModel;
-declare const setKeys: (keys: ApiKeys) => void;
-declare const getKey: (provider: string) => string;
-declare const maxCalls: (toolConfig: ToolConfig, maxCalls: number) => ToolConfig;
-declare const addUsage: (existing: TokenUsage | undefined, promptTokens: number, completionTokens: number, totalTokens: number) => TokenUsage;
-
-/**
- * generates embeddings for text using openai or huggingface models
- *
- * openai models use the prefix "openai/" (e.g., "openai/text-embedding-3-small")
- * all other models use huggingface transformers
- *
- * @example
- * const vector = await embed("openai/text-embedding-3-small", "hello world");
- * const vector2 = await embed("Xenova/all-MiniLM-L6-v2", "hello world");
- */
-declare const embed: (model: string, text: string, config?: {
-    dimensions?: number;
-}) => Promise<number[]>;
-
-declare const generateImage: (model: string, prompt: string, config?: ImageConfig) => Promise<ImageResult>;
-
-interface ConfigOption {
-    values: string[];
-    default: string;
-    description: string;
-}
-interface ModelConfig {
-    [option: string]: ConfigOption;
-}
-interface ProviderModels {
-    [model: string]: ModelConfig;
-}
-interface ImageModelSchema {
-    [provider: string]: ProviderModels;
-}
-declare const IMAGE_MODEL_SCHEMA: ImageModelSchema;
-declare const IMAGE_EDIT_MODEL_SCHEMA: ImageModelSchema;
-declare function getModelConfig(provider: string, model: string): ModelConfig | null;
-declare function getDefaultConfig(provider: string, model: string): Record<string, string>;
-
-/**
- * @example
- * // in-memory (default)
- * const thread = getOrCreateThread('user-123');
- *
- * @example
- * // sqlite
- * const thread = getOrCreateThread('user-123', {
- *   async get(id) {
- *     const row = await db.get('SELECT messages FROM threads WHERE id = ?', id);
- *     return row ? JSON.parse(row.messages) : [];
- *   },
- *   async set(id, messages) {
- *     await db.run(
- *       'INSERT OR REPLACE INTO threads (id, messages, updated_at) VALUES (?, ?, ?)',
- *       id,
- *       JSON.stringify(messages),
- *       Date.now()
- *     );
- *   }
- * });
- */
-declare const getOrCreateThread: (id: string, store?: ThreadStore) => Thread;
-
-declare const isStandardSchema: (schema: any) => schema is StandardSchema;
-declare const convertStandardSchemaToJsonSchema: (standardSchema: StandardSchema, name?: string) => JsonSchema;
-declare const convertMCPSchemaToToolSchema: (mcpSchema: any) => Record<string, SchemaProperty>;
-declare function normalizeSchema(schema: JsonSchema | StandardSchema, name?: string): JsonSchema;
-declare const convertStandardSchemaToSchemaProperties: (standardSchema: StandardSchema) => Record<string, SchemaProperty>;
-
-/**
- * scope({ until: noToolsCalled() })
- */
-declare const noToolsCalled: () => (ctx: ConversationContext) => boolean;
-declare const everyNMessages: (n: number, step: StepFunction) => StepFunction;
-declare const everyNTokens: (n: number, step: StepFunction) => StepFunction;
-declare const appendToLastRequest: (content: string) => StepFunction;
-/**
- * toolNotUsedInNTurns({ toolName: "search_web", times: 10 }, appendToLastRequest("consider using web search..."))
- */
-declare const toolNotUsedInNTurns: ({ toolName, times }: {
-    toolName: string;
-    times: number;
-}, step: StepFunction) => StepFunction;
-declare const toolWasCalled: (name: string) => (ctx: ConversationContext) => boolean;
-
-interface ApprovalRequest {
-    id: string;
-    toolCall: ToolCall;
-    approvalId?: string;
-}
-interface ApprovalResponse {
-    id: string;
-    approved: boolean;
-    reason?: string;
-}
-declare const generateApprovalToken: () => string;
-declare const requestApproval: (toolCall: ToolCall, approvalId?: string) => Promise<ApprovalResponse>;
-declare const resolveApproval: (response: ApprovalResponse) => boolean;
-declare const onApprovalRequested: (listener: (request: ApprovalRequest) => void) => void;
-declare const onApprovalResolved: (listener: (response: ApprovalResponse) => void) => void;
-declare const removeApprovalListener: (event: "approvalRequested" | "approvalResolved", listener: (...args: any[]) => void) => void;
-
-declare const tap: (fn: (ctx: ConversationContext) => Promise<void> | void) => StepFunction;
-
-declare const when: (condition: (ctx: ConversationContext) => boolean, action: StepFunction) => StepFunction;
-
-declare const model: ({ model, schema, system, apiKey, baseUrl, }?: {
-    model?: string;
-    schema?: JsonSchema | StandardSchema;
-    system?: string | ((ctx: ConversationContext) => string);
-    apiKey?: string;
-    baseUrl?: string;
-}) => ComposedFunction;
-
-/**
- * scope({}, retry({ times: 2 }, model(...)))
- */
-declare const retry: ({ times }: RetryOptions | undefined, step: StepFunction) => StepFunction;
-
-declare const scope: (config: ScopeConfig, ...steps: StepFunction[]) => StepFunction;
-
-type RateLimitConfig = {
-    rps: number;
-    burst: number;
-    concurrency: number;
-};
-/**
- * creates a rate limiter that wraps async functions with burst, rate, and concurrency controls
- *
- * @param config - rate limit configuration
- * @param config.rps - maximum requests per second
- * @param config.burst - maximum burst size (initial token bucket capacity)
- * @param config.concurrency - maximum concurrent in-flight requests
- *
- * @example
- * const limiter = rateLimited({ rps: 10, burst: 20, concurrency: 5 });
- *
- * const workflow = limiter(
- *   compose(
- *     scope({ tools: [searchTool] }, model())
- *   )
- * );
- *
- * await workflow("hello");
- */
-declare const rateLimited: (config: RateLimitConfig) => <T extends (...args: any[]) => Promise<any>>(fn: T) => T;
-
-declare const compose: (...steps: StepFunction[]) => ComposedFunction;
-
-export { type ApiKeys, type ApprovalRequest, type ApprovalResponse, type ComposedFunction, type ConfigOption, type ConversationContext, IMAGE_EDIT_MODEL_SCHEMA, IMAGE_MODEL_SCHEMA, type ImageConfig, type ImageModelSchema, type ImageResult, Inherit, type JsonSchema, type Message, type ModelConfig, type ParsedModel, type ProviderConfig, type ProviderModels, type RetryOptions, type SchemaProperty, type ScopeConfig, type StandardSchema, type StepFunction, type StreamEvent, type Thread, type ThreadStore, type TokenUsage, type ToolCall, type ToolCallResult, type ToolConfig, type ToolDefinition, type ToolExecutionConfig, addUsage, appendToLastRequest, compose, convertMCPSchemaToToolSchema, convertStandardSchemaToJsonSchema, convertStandardSchemaToSchemaProperties, createMCPTools, embed, everyNMessages, everyNTokens, generateApprovalToken, generateImage, getDefaultConfig, getKey, getModelConfig, getOrCreateThread, isStandardSchema, maxCalls, model, noToolsCalled, normalizeSchema, onApprovalRequested, onApprovalResolved, parseModelName, rateLimited, removeApprovalListener, requestApproval, resolveApproval, retry, scope, setKeys, tap, toolConfigToToolDefinition, toolNotUsedInNTurns, toolWasCalled, when };
+export * from "./mcp";
+export * from "./types";
+export * from "./utils";
+export * from "./embed";
+export * from "./image";
+export * from "./image-model-schema";
+export * from "./thread";
+export * from "./schema";
+export * from "./helpers";
+export * from "./approval";
+export * from "./composition/tap";
+export * from "./composition/when";
+export * from "./composition/model";
+export * from "./composition/retry";
+export * from "./composition/scope";
+export * from "./utils/rateLimited";
+export * from "./composition/compose";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,OAAO,CAAC;AACtB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,sBAAsB,CAAC;AACrC,cAAc,UAAU,CAAC;AACzB,cAAc,UAAU,CAAC;AACzB,cAAc,WAAW,CAAC;AAC1B,cAAc,YAAY,CAAC;AAC3B,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,uBAAuB,CAAC"}