@astro-minimax/ai 0.2.0

package/dist/provider-manager/manager.d.ts

@@ -0,0 +1,18 @@
+import type { ProviderAdapter, ProviderManagerEnv, ProviderManagerOptions, ProviderStatus, StreamTextOptions, StreamTextResult } from './types.js';
+export declare class ProviderManager {
+    private providers;
+    private mockAdapter;
+    private options;
+    constructor(env: ProviderManagerEnv, options?: ProviderManagerOptions);
+    private initializeProviders;
+    private createAdapter;
+    getAvailableProvider(): Promise<ProviderAdapter | null>;
+    streamText(options: StreamTextOptions): Promise<StreamTextResult>;
+    getProviderStatus(): ProviderStatus[];
+    hasProviders(): boolean;
+    getProviderCount(): number;
+    getAvailableAdapter(): Promise<ProviderAdapter | null>;
+}
+export declare function getProviderManager(env: ProviderManagerEnv, options?: ProviderManagerOptions): ProviderManager;
+export declare function resetProviderManager(): void;
+//# sourceMappingURL=manager.d.ts.map

package/dist/provider-manager/manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/provider-manager/manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EAEf,kBAAkB,EAClB,sBAAsB,EACtB,cAAc,EACd,iBAAiB,EACjB,gBAAgB,EAGjB,MAAM,YAAY,CAAC;AAMpB,qBAAa,eAAe;IAC1B,OAAO,CAAC,SAAS,CAAyB;IAC1C,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,OAAO,CAA0L;gBAE7L,GAAG,EAAE,kBAAkB,EAAE,OAAO,CAAC,EAAE,sBAAsB;IAcrE,OAAO,CAAC,mBAAmB;IAyB3B,OAAO,CAAC,aAAa;IAWf,oBAAoB,IAAI,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;IASvD,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAqCvE,iBAAiB,IAAI,cAAc,EAAE;IAWrC,YAAY,IAAI,OAAO;IAIvB,gBAAgB,IAAI,MAAM;IAIpB,mBAAmB,IAAI,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;CAG7D;AAID,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,kBAAkB,EAAE,OAAO,CAAC,EAAE,sBAAsB,GAAG,eAAe,CAK7G;AAED,wBAAgB,oBAAoB,IAAI,IAAI,CAE3C"}

package/dist/provider-manager/manager.js

@@ -0,0 +1,121 @@
+import { parseProviderConfigs, validateProviderConfig } from './config.js';
+import { OpenAIAdapter } from './openai.js';
+import { WorkersAIAdapter } from './workers.js';
+import { MockAdapter } from './mock.js';
+export class ProviderManager {
+    providers = [];
+    mockAdapter;
+    options;
+    constructor(env, options) {
+        this.options = {
+            unhealthyThreshold: options?.unhealthyThreshold ?? 3,
+            healthRecoveryTTL: options?.healthRecoveryTTL ?? 60000,
+            enableMockFallback: options?.enableMockFallback ?? true,
+            onProviderSwitch: options?.onProviderSwitch,
+            onStreamError: options?.onStreamError,
+            onHealthChange: options?.onHealthChange,
+        };
+        this.mockAdapter = new MockAdapter();
+        this.initializeProviders(env);
+    }
+    initializeProviders(env) {
+        const configs = parseProviderConfigs(env);
+        for (const config of configs) {
+            if (config.enabled === false)
+                continue;
+            const validationError = validateProviderConfig(config);
+            if (validationError) {
+                console.warn(`[ProviderManager] Skipping invalid config: ${validationError}`);
+                continue;
+            }
+            try {
+                const adapter = this.createAdapter(config, env);
+                if (adapter) {
+                    this.providers.push(adapter);
+                }
+            }
+            catch (error) {
+                console.warn(`[ProviderManager] Failed to create adapter for ${config.id}:`, error);
+            }
+        }
+        this.providers.sort((a, b) => b.weight - a.weight);
+    }
+    createAdapter(config, env) {
+        switch (config.type) {
+            case 'openai':
+                return new OpenAIAdapter(config);
+            case 'workers':
+                return new WorkersAIAdapter(config, env);
+            default:
+                return null;
+        }
+    }
+    async getAvailableProvider() {
+        for (const provider of this.providers) {
+            if (await provider.isAvailable()) {
+                return provider;
+            }
+        }
+        return null;
+    }
+    async streamText(options) {
+        let lastProviderId = null;
+        let lastError = null;
+        for (const provider of this.providers) {
+            const isAvailable = await provider.isAvailable();
+            if (!isAvailable)
+                continue;
+            try {
+                const result = await provider.streamText(options);
+                provider.recordSuccess();
+                if (lastProviderId && lastProviderId !== provider.id) {
+                    this.options.onProviderSwitch?.(lastProviderId, provider.id, 'fallback success');
+                }
+                return result;
+            }
+            catch (error) {
+                lastError = error instanceof Error ? error : new Error(String(error));
+                lastProviderId = provider.id;
+                provider.recordFailure(lastError);
+                this.options.onStreamError?.(provider.id, lastError);
+                if (!provider.getHealth().healthy) {
+                    this.options.onHealthChange?.(provider.id, false);
+                }
+            }
+        }
+        if (this.options.enableMockFallback) {
+            this.options.onProviderSwitch?.(lastProviderId, 'mock', 'all providers failed');
+            return this.mockAdapter.streamText(options);
+        }
+        throw lastError || new Error('No providers available');
+    }
+    getProviderStatus() {
+        return this.providers.map(provider => ({
+            id: provider.id,
+            type: provider.type,
+            weight: provider.weight,
+            enabled: true,
+            health: provider.getHealth(),
+            model: provider.model,
+        }));
+    }
+    hasProviders() {
+        return this.providers.length > 0;
+    }
+    getProviderCount() {
+        return this.providers.length;
+    }
+    async getAvailableAdapter() {
+        return this.getAvailableProvider();
+    }
+}
+let managerInstance = null;
+export function getProviderManager(env, options) {
+    if (!managerInstance) {
+        managerInstance = new ProviderManager(env, options);
+    }
+    return managerInstance;
+}
+export function resetProviderManager() {
+    managerInstance = null;
+}

package/dist/provider-manager/mock.d.ts

@@ -0,0 +1,18 @@
+import type { StreamTextOptions, StreamTextResult } from './types.js';
+import { BaseProviderAdapter } from './base.js';
+export declare class MockAdapter extends BaseProviderAdapter {
+    readonly id: "mock";
+    readonly type: "mock";
+    readonly weight = 0;
+    readonly model = "mock";
+    readonly keywordModel = "mock";
+    readonly evidenceModel = "mock";
+    readonly timeout = 0;
+    constructor();
+    isAvailable(): Promise<boolean>;
+    streamText(options: StreamTextOptions): Promise<StreamTextResult>;
+    getProvider(): {
+        chatModel: (model: string) => unknown;
+    };
+}
+//# sourceMappingURL=mock.d.ts.map

package/dist/provider-manager/mock.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock.d.ts","sourceRoot":"","sources":["../../src/provider-manager/mock.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AACtE,OAAO,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAMhD,qBAAa,WAAY,SAAQ,mBAAmB;IAClD,QAAQ,CAAC,EAAE,EAAG,MAAM,CAAU;IAC9B,QAAQ,CAAC,IAAI,EAAG,MAAM,CAAU;IAChC,QAAQ,CAAC,MAAM,KAAe;IAC9B,QAAQ,CAAC,KAAK,UAAU;IACxB,QAAQ,CAAC,YAAY,UAAU;IAC/B,QAAQ,CAAC,aAAa,UAAU;IAChC,QAAQ,CAAC,OAAO,KAAK;;IAMf,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IAI/B,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAqDvE,WAAW,IAAI;QAAE,SAAS,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAA;KAAE;CAGzD"}

package/dist/provider-manager/mock.js

@@ -0,0 +1,56 @@
+import { BaseProviderAdapter } from './base.js';
+import { getMockResponse } from '../providers/mock.js';
+const MOCK_WEIGHT = 0;
+const CHAR_DELAY_MS = 15;
+export class MockAdapter extends BaseProviderAdapter {
+    id = 'mock';
+    type = 'mock';
+    weight = MOCK_WEIGHT;
+    model = 'mock';
+    keywordModel = 'mock';
+    evidenceModel = 'mock';
+    timeout = 0;
+    constructor() {
+        super({ unhealthyThreshold: 999 });
+    }
+    async isAvailable() {
+        return true;
+    }
+    async streamText(options) {
+        const { userQuestion = '', lang = 'zh' } = options;
+        const text = getMockResponse(userQuestion, lang);
+        const partId = `mock-${Date.now()}`;
+        const encoder = new TextEncoder();
+        const stream = new ReadableStream({
+            async start(controller) {
+                const write = (event) => controller.enqueue(encoder.encode(`data: ${JSON.stringify(event)}\n\n`));
+                write({ type: 'text-start', id: partId });
+                for (let i = 0; i < text.length;) {
+                    const chunkSize = Math.random() < 0.3 ? 3 : Math.random() < 0.5 ? 2 : 1;
+                    const chunk = text.slice(i, i + chunkSize);
+                    i += chunkSize;
+                    controller.enqueue(encoder.encode(`data: ${JSON.stringify({ type: 'text-delta', id: partId, delta: chunk })}\n\n`));
+                    await new Promise(r => setTimeout(r, CHAR_DELAY_MS + Math.random() * 20));
+                }
+                controller.enqueue(encoder.encode(`data: ${JSON.stringify({ type: 'text-end', id: partId })}\n\n`));
+                controller.enqueue(encoder.encode(`data: ${JSON.stringify({ type: 'finish', finishReason: 'stop' })}\n\n`));
+                controller.close();
+            },
+        });
+        const response = new Response(stream, {
+            headers: {
+                'Content-Type': 'text/event-stream; charset=utf-8',
+                'Cache-Control': 'no-cache, no-store',
+                'Access-Control-Allow-Origin': '*',
+            },
+        });
+        return {
+            toUIMessageStreamResponse: () => response,
+            providerId: this.id,
+            isMock: true,
+        };
+    }
+    getProvider() {
+        throw new Error('Mock provider does not support chatModel interface');
+    }
+}

package/dist/provider-manager/openai.d.ts

@@ -0,0 +1,20 @@
+import type { OpenAIProviderConfig, StreamTextOptions, StreamTextResult } from './types.js';
+import { BaseProviderAdapter } from './base.js';
+export declare class OpenAIAdapter extends BaseProviderAdapter {
+    readonly id: string;
+    readonly type: "openai";
+    readonly weight: number;
+    readonly model: string;
+    readonly keywordModel: string;
+    readonly evidenceModel: string;
+    readonly timeout: number;
+    private provider;
+    private config;
+    constructor(config: OpenAIProviderConfig);
+    streamText(options: StreamTextOptions): Promise<StreamTextResult>;
+    getConfig(): OpenAIProviderConfig;
+    getProvider(): {
+        chatModel: (model: string) => unknown;
+    };
+}
+//# sourceMappingURL=openai.d.ts.map

package/dist/provider-manager/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/provider-manager/openai.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC5F,OAAO,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAUhD,qBAAa,aAAc,SAAQ,mBAAmB;IACpD,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAG,QAAQ,CAAU;IAClC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAEzB,OAAO,CAAC,QAAQ,CAA4C;IAC5D,OAAO,CAAC,MAAM,CAAuB;gBAEzB,MAAM,EAAE,oBAAoB;IAwBlC,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAuCvE,SAAS,IAAI,oBAAoB;IAIjC,WAAW,IAAI;QAAE,SAAS,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAA;KAAE;CAGzD"}

package/dist/provider-manager/openai.js

@@ -0,0 +1,83 @@
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
+import { streamText, convertToModelMessages } from 'ai';
+import { ProxyAgent, fetch as undiciFetch } from 'undici';
+import { BaseProviderAdapter } from './base.js';
+function createProxyFetch() {
+    const proxyUrl = process.env.https_proxy || process.env.HTTPS_PROXY || process.env.http_proxy || process.env.HTTP_PROXY;
+    if (!proxyUrl)
+        return undefined;
+    const agent = new ProxyAgent(proxyUrl);
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    return ((url, init) => undiciFetch(url, { ...init, dispatcher: agent }));
+}
+export class OpenAIAdapter extends BaseProviderAdapter {
+    id;
+    type = 'openai';
+    weight;
+    model;
+    keywordModel;
+    evidenceModel;
+    timeout;
+    provider;
+    config;
+    constructor(config) {
+        super({
+            unhealthyThreshold: config.maxRetries ? config.maxRetries + 2 : 3,
+        });
+        this.id = config.id;
+        this.weight = config.weight ?? 100;
+        this.model = config.model;
+        this.keywordModel = config.keywordModel ?? config.model;
+        this.evidenceModel = config.evidenceModel ?? this.keywordModel;
+        this.timeout = config.timeout ?? 30000;
+        this.config = config;
+        const proxyFetch = createProxyFetch();
+        this.provider = createOpenAICompatible({
+            name: `openai-${config.id}`,
+            baseURL: config.baseURL,
+            apiKey: config.apiKey,
+            includeUsage: true,
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            ...(proxyFetch && { fetch: proxyFetch }),
+        });
+    }
+    async streamText(options) {
+        const { system, messages, temperature = 0.7, maxOutputTokens, topP, abortSignal, onError } = options;
+        const abortController = new AbortController();
+        const timeoutId = setTimeout(() => abortController.abort(), this.timeout);
+        if (abortSignal) {
+            abortSignal.addEventListener('abort', () => abortController.abort());
+        }
+        try {
+            const result = streamText({
+                model: this.provider.chatModel(this.model),
+                system,
+                messages: await convertToModelMessages(messages),
+                temperature,
+                maxOutputTokens,
+                topP,
+                abortSignal: abortController.signal,
+                onError: ({ error }) => {
+                    onError?.(error instanceof Error ? error : new Error(String(error)));
+                },
+            });
+            const streamResult = {
+                toUIMessageStreamResponse: (responseOptions) => result.toUIMessageStreamResponse(responseOptions),
+                providerId: this.id,
+                isMock: false,
+            };
+            clearTimeout(timeoutId);
+            return streamResult;
+        }
+        catch (error) {
+            clearTimeout(timeoutId);
+            throw error;
+        }
+    }
+    getConfig() {
+        return { ...this.config };
+    }
+    getProvider() {
+        return this.provider;
+    }
+}

package/dist/provider-manager/types.d.ts

@@ -0,0 +1,217 @@
+/**
+ * Type definitions for the AI Provider Manager.
+ *
+ * Supports multiple provider types with priority-based fallback.
+ */
+import type { UIMessage } from 'ai';
+/**
+ * Base configuration shared by all provider types.
+ */
+export interface BaseProviderConfig {
+    /** Unique identifier for this provider instance */
+    id: string;
+    /** Provider type */
+    type: 'openai' | 'workers';
+    /** Priority weight (higher = more preferred). Default: 100 */
+    weight?: number;
+    /** Default model for chat */
+    model: string;
+    /** Model for keyword extraction (optional, defaults to model) */
+    keywordModel?: string;
+    /** Model for evidence analysis (optional, defaults to keywordModel) */
+    evidenceModel?: string;
+    /** Request timeout in milliseconds. Default: 30000 */
+    timeout?: number;
+    /** Max retries on failure. Default: 0 (no retry, fallback to next provider) */
+    maxRetries?: number;
+    /** Enable this provider. Default: true */
+    enabled?: boolean;
+}
+/**
+ * OpenAI-compatible provider configuration.
+ * Supports DeepSeek, Moonshot, Qwen, OpenAI, and any OpenAI-compatible API.
+ */
+export interface OpenAIProviderConfig extends BaseProviderConfig {
+    type: 'openai';
+    /** API base URL (e.g., https://api.deepseek.com/v1) */
+    baseURL: string;
+    /** API key */
+    apiKey: string;
+}
+/**
+ * Cloudflare Workers AI provider configuration.
+ * Uses AI binding directly from the Cloudflare environment.
+ */
+export interface WorkersAIProviderConfig extends BaseProviderConfig {
+    type: 'workers';
+    /** AI binding name in Cloudflare environment. Default: 'minimaxAI' */
+    bindingName: string;
+}
+/**
+ * Union type for all provider configurations.
+ */
+export type ProviderConfig = OpenAIProviderConfig | WorkersAIProviderConfig;
+/**
+ * Environment variable interface for provider configuration.
+ * Used to build ProviderConfig from environment variables.
+ */
+export interface ProviderManagerEnv {
+    [key: string]: unknown;
+    /** JSON string containing array of ProviderConfig */
+    AI_PROVIDERS?: string;
+    AI_BASE_URL?: string;
+    AI_API_KEY?: string;
+    AI_MODEL?: string;
+    AI_KEYWORD_MODEL?: string;
+    AI_EVIDENCE_MODEL?: string;
+    AI_BINDING_NAME?: string;
+    AI_WORKERS_MODEL?: string;
+}
+/**
+ * Health status of a provider instance.
+ */
+export interface ProviderHealth {
+    /** Whether the provider is currently healthy */
+    healthy: boolean;
+    /** Number of consecutive failures */
+    consecutiveFailures: number;
+    /** Total number of requests */
+    totalRequests: number;
+    /** Total number of successful requests */
+    successfulRequests: number;
+    /** Last error message (if any) */
+    lastError?: string;
+    /** Timestamp of last error */
+    lastErrorTime?: number;
+    /** Timestamp of last successful request */
+    lastSuccessTime?: number;
+    /** Timestamp when health status was last updated */
+    lastChecked: number;
+}
+/**
+ * Runtime status of a provider.
+ */
+export interface ProviderStatus {
+    /** Provider ID */
+    id: string;
+    /** Provider type */
+    type: 'openai' | 'workers' | 'mock';
+    /** Priority weight */
+    weight: number;
+    /** Whether the provider is enabled */
+    enabled: boolean;
+    /** Current health status */
+    health: ProviderHealth;
+    /** Model name */
+    model: string;
+}
+/**
+ * Options for streaming text generation.
+ */
+export interface StreamTextOptions {
+    /** System prompt */
+    system?: string;
+    /** Conversation messages */
+    messages: UIMessage[];
+    /** Temperature (0-1). Default: 0.7 */
+    temperature?: number;
+    /** Maximum output tokens */
+    maxOutputTokens?: number;
+    /** Top-p sampling */
+    topP?: number;
+    /** Abort signal for cancellation */
+    abortSignal?: AbortSignal;
+    /** Language for mock responses (zh/en) */
+    lang?: string;
+    /** User question (for mock responses) */
+    userQuestion?: string;
+    /** Callback for errors during streaming */
+    onError?: (error: Error) => void;
+}
+/**
+ * Result from streamText operation.
+ * Compatible with AI SDK's streamText result.
+ */
+export interface StreamTextResult {
+    /** Convert to UI Message Stream Response for AI SDK v6 */
+    toUIMessageStreamResponse: (options?: {
+        headers?: HeadersInit;
+    }) => Response;
+    /** The text content (for non-streaming usage) */
+    text?: Promise<string>;
+    /** Provider ID that handled the request */
+    providerId: string;
+    /** Whether this is a mock response */
+    isMock: boolean;
+}
+/**
+ * Result from provider health check.
+ */
+export interface HealthCheckResult {
+    providerId: string;
+    healthy: boolean;
+    latency?: number;
+    error?: string;
+}
+/**
+ * Options for creating a ProviderManager.
+ */
+export interface ProviderManagerOptions {
+    /** Number of consecutive failures before marking unhealthy. Default: 3 */
+    unhealthyThreshold?: number;
+    /** Time in ms before retrying an unhealthy provider. Default: 60000 (1 min) */
+    healthRecoveryTTL?: number;
+    /** Enable mock provider as final fallback. Default: true */
+    enableMockFallback?: boolean;
+    /** Callback when provider switch occurs */
+    onProviderSwitch?: (fromId: string | null, toId: string, reason: string) => void;
+    /** Callback when streaming error occurs */
+    onStreamError?: (providerId: string, error: Error) => void;
+    /** Callback when health status changes */
+    onHealthChange?: (providerId: string, healthy: boolean) => void;
+}
+/**
+ * Abstract interface for provider adapters.
+ * All provider types must implement this interface.
+ */
+export interface ProviderAdapter {
+    /** Provider identifier */
+    readonly id: string;
+    /** Provider type */
+    readonly type: 'openai' | 'workers' | 'mock';
+    /** Priority weight */
+    readonly weight: number;
+    /** Default model */
+    readonly model: string;
+    /** Model for keyword extraction */
+    readonly keywordModel: string;
+    /** Model for evidence analysis */
+    readonly evidenceModel: string;
+    /** Request timeout in ms */
+    readonly timeout: number;
+    /**
+     * Check if this provider is currently available and healthy.
+     */
+    isAvailable(): Promise<boolean>;
+    /**
+     * Stream text generation.
+     */
+    streamText(options: StreamTextOptions): Promise<StreamTextResult>;
+    /**
+     * Get current health status.
+     */
+    getHealth(): ProviderHealth;
+    /**
+     * Record a successful request.
+     */
+    recordSuccess(): void;
+    /**
+     * Record a failed request.
+     */
+    recordFailure(error: Error): void;
+    getProvider(): {
+        chatModel: (model: string) => unknown;
+    };
+    dispose?(): void;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/provider-manager/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/provider-manager/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAMpC;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,mDAAmD;IACnD,EAAE,EAAE,MAAM,CAAC;IACX,oBAAoB;IACpB,IAAI,EAAE,QAAQ,GAAG,SAAS,CAAC;IAC3B,8DAA8D;IAC9D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,6BAA6B;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,iEAAiE;IACjE,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,uEAAuE;IACvE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,sDAAsD;IACtD,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,+EAA+E;IAC/E,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,0CAA0C;IAC1C,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED;;;GAGG;AACH,MAAM,WAAW,oBAAqB,SAAQ,kBAAkB;IAC9D,IAAI,EAAE,QAAQ,CAAC;IACf,uDAAuD;IACvD,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc;IACd,MAAM,EAAE,MAAM,CAAC;CAChB;AAED;;;GAGG;AACH,MAAM,WAAW,uBAAwB,SAAQ,kBAAkB;IACjE,IAAI,EAAE,SAAS,CAAC;IAChB,sEAAsE;IACtE,WAAW,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,oBAAoB,GAAG,uBAAuB,CAAC;AAE5E;;;GAGG;AACH,MAAM,WAAW,kBAAkB;IACjC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;IACvB,qDAAqD;IACrD,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAE3B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAMD;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,gDAAgD;IAChD,OAAO,EAAE,OAAO,CAAC;IACjB,qCAAqC;IACrC,mBAAmB,EAAE,MAAM,CAAC;IAC5B,+BAA+B;IAC/B,aAAa,EAAE,MAAM,CAAC;IACtB,0CAA0C;IAC1C,kBAAkB,EAAE,MAAM,CAAC;IAC3B,kCAAkC;IAClC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,8BAA8B;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,2CAA2C;IAC3C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,oDAAoD;IACpD,WAAW,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,kBAAkB;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,oBAAoB;IACpB,IAAI,EAAE,QAAQ,GAAG,SAAS,GAAG,MAAM,CAAC;IACpC,sBAAsB;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,sCAAsC;IACtC,OAAO,EAAE,OAAO,CAAC;IACjB,4BAA4B;IAC5B,MAAM,EAAE,cAAc,CAAC;IACvB,iBAAiB;IACjB,KAAK,EAAE,MAAM,CAAC;CACf;AAMD;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,oBAAoB;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,4BAA4B;IAC5B,QAAQ,EAAE,SAAS,EAAE,CAAC;IACtB,sCAAsC;IACtC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,4BAA4B;IAC5B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,qBAAqB;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,oCAAoC;IACpC,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,0CAA0C;IAC1C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,yCAAyC;IACzC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,2CAA2C;IAC3C,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAClC;AAED;;;GAGG;AACH,MAAM,WAAW,gBAAgB;IAC/B,0DAA0D;IAC1D,yBAAyB,EAAE,CAAC,OAAO,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,WAAW,CAAA;KAAE,KAAK,QAAQ,CAAC;IAC7E,iDAAiD;IACjD,IAAI,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACvB,2CAA2C;IAC3C,UAAU,EAAE,MAAM,CAAC;IACnB,sCAAsC;IACtC,MAAM,EAAE,OAAO,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAMD;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,0EAA0E;IAC1E,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,+EAA+E;IAC/E,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,4DAA4D;IAC5D,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,2CAA2C;IAC3C,gBAAgB,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,IAAI,CAAC;IACjF,2CAA2C;IAC3C,aAAa,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IAC3D,0CAA0C;IAC1C,cAAc,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;CACjE;AAMD;;;GAGG;AACH,MAAM,WAAW,eAAe;IAC9B,0BAA0B;IAC1B,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,oBAAoB;IACpB,QAAQ,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,GAAG,MAAM,CAAC;IAC7C,sBAAsB;IACtB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,oBAAoB;IACpB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,mCAAmC;IACnC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,kCAAkC;IAClC,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,4BAA4B;IAC5B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAEzB;;OAEG;IACH,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IAEhC;;OAEG;IACH,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAElE;;OAEG;IACH,SAAS,IAAI,cAAc,CAAC;IAE5B;;OAEG;IACH,aAAa,IAAI,IAAI,CAAC;IAEtB;;OAEG;IACH,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC;IAElC,WAAW,IAAI;QAAE,SAAS,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAA;KAAE,CAAC;IAEzD,OAAO,CAAC,IAAI,IAAI,CAAC;CAClB"}

package/dist/provider-manager/types.js

@@ -0,0 +1,6 @@
+/**
+ * Type definitions for the AI Provider Manager.
+ *
+ * Supports multiple provider types with priority-based fallback.
+ */
+export {};

package/dist/provider-manager/workers.d.ts

@@ -0,0 +1,20 @@
+import type { WorkersAIProviderConfig, StreamTextOptions, StreamTextResult, ProviderManagerEnv } from './types.js';
+import { BaseProviderAdapter } from './base.js';
+export declare class WorkersAIAdapter extends BaseProviderAdapter {
+    readonly id: string;
+    readonly type: "workers";
+    readonly weight: number;
+    readonly model: string;
+    readonly keywordModel: string;
+    readonly evidenceModel: string;
+    readonly timeout: number;
+    private provider;
+    private config;
+    constructor(config: WorkersAIProviderConfig, env: ProviderManagerEnv);
+    streamText(options: StreamTextOptions): Promise<StreamTextResult>;
+    getConfig(): WorkersAIProviderConfig;
+    getProvider(): {
+        chatModel: (model: string) => unknown;
+    };
+}
+//# sourceMappingURL=workers.d.ts.map

package/dist/provider-manager/workers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workers.d.ts","sourceRoot":"","sources":["../../src/provider-manager/workers.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AACnH,OAAO,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAEhD,qBAAa,gBAAiB,SAAQ,mBAAmB;IACvD,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAG,SAAS,CAAU;IACnC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,aAAa,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAGzB,OAAO,CAAC,QAAQ,CAAM;IACtB,OAAO,CAAC,MAAM,CAA0B;gBAE5B,MAAM,EAAE,uBAAuB,EAAE,GAAG,EAAE,kBAAkB;IAqB9D,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAyCvE,SAAS,IAAI,uBAAuB;IAIpC,WAAW,IAAI;QAAE,SAAS,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAA;KAAE;CAKzD"}

package/dist/provider-manager/workers.js

@@ -0,0 +1,74 @@
+import { createWorkersAI } from 'workers-ai-provider';
+import { streamText, convertToModelMessages } from 'ai';
+import { BaseProviderAdapter } from './base.js';
+export class WorkersAIAdapter extends BaseProviderAdapter {
+    id;
+    type = 'workers';
+    weight;
+    model;
+    keywordModel;
+    evidenceModel;
+    timeout;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    provider;
+    config;
+    constructor(config, env) {
+        super({
+            unhealthyThreshold: config.maxRetries ? config.maxRetries + 2 : 3,
+        });
+        this.id = config.id;
+        this.weight = config.weight ?? 90;
+        this.model = config.model;
+        this.keywordModel = config.keywordModel ?? config.model;
+        this.evidenceModel = config.evidenceModel ?? this.keywordModel;
+        this.timeout = config.timeout ?? 30000;
+        this.config = config;
+        const binding = env[config.bindingName];
+        if (!binding) {
+            throw new Error(`Workers AI binding '${config.bindingName}' not found in environment`);
+        }
+        this.provider = createWorkersAI({ binding: binding });
+    }
+    async streamText(options) {
+        const { system, messages, temperature = 0.7, maxOutputTokens, topP, abortSignal, onError } = options;
+        const abortController = new AbortController();
+        const timeoutId = setTimeout(() => abortController.abort(), this.timeout);
+        if (abortSignal) {
+            abortSignal.addEventListener('abort', () => abortController.abort());
+        }
+        try {
+            const model = this.provider(this.model, { safePrompt: true });
+            const result = streamText({
+                model,
+                system,
+                messages: await convertToModelMessages(messages),
+                temperature,
+                maxOutputTokens,
+                topP,
+                abortSignal: abortController.signal,
+                onError: ({ error }) => {
+                    onError?.(error instanceof Error ? error : new Error(String(error)));
+                },
+            });
+            const streamResult = {
+                toUIMessageStreamResponse: (responseOptions) => result.toUIMessageStreamResponse(responseOptions),
+                providerId: this.id,
+                isMock: false,
+            };
+            clearTimeout(timeoutId);
+            return streamResult;
+        }
+        catch (error) {
+            clearTimeout(timeoutId);
+            throw error;
+        }
+    }
+    getConfig() {
+        return { ...this.config };
+    }
+    getProvider() {
+        return {
+            chatModel: (modelId) => this.provider(modelId, { safePrompt: true }),
+        };
+    }
+}

package/dist/providers/index.d.ts

@@ -0,0 +1,2 @@
+export { getMockResponse, createMockStream } from './mock.js';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC"}