@scenemesh/entity-engine-aimodule 1.0.0 → 1.0.1

package/dist/core/ai-provider.d.ts

@@ -0,0 +1,265 @@
+/**
+ * AI Provider and Model Management
+ *
+ * Manages custom providers, model registry, and global configuration.
+ * Includes model aliases, settings, and limitations.
+ */
+import type { ModelSettings, AISettingsManagement } from './ai-settings';
+import type { EmbeddingModelConfig, AIEmbeddingsIntegration } from './ai-embeddings';
+import { EventEmitter } from 'events';
+import { type Provider, customProvider, type ImageModel, wrapLanguageModel, type LanguageModel, type EmbeddingModel, createProviderRegistry, defaultSettingsMiddleware } from 'ai';
+/**
+ * Custom provider configuration
+ */
+export interface CustomProviderConfig {
+    languageModels?: Record<string, LanguageModel | {
+        model: LanguageModel;
+        middleware?: any;
+        settings?: {
+            maxOutputTokens?: number;
+            temperature?: number;
+            topP?: number;
+            topK?: number;
+            presencePenalty?: number;
+            frequencyPenalty?: number;
+            seed?: number;
+            providerOptions?: Record<string, any>;
+        };
+    }>;
+    embeddingModels?: Record<string, EmbeddingModel>;
+    imageModels?: Record<string, ImageModel>;
+    fallbackProvider?: Provider;
+}
+/**
+ * Provider registry configuration
+ */
+export interface ProviderRegistryConfig {
+    providers: Record<string, Provider>;
+    options?: {
+        separator?: string;
+    };
+}
+/**
+ * Model alias configuration
+ */
+export interface ModelAliasConfig {
+    alias: string;
+    providerId: string;
+    modelId: string;
+    settings?: any;
+    middleware?: any;
+}
+/**
+ * Global provider configuration
+ */
+export interface GlobalProviderConfig {
+    defaultProvider: Provider;
+    modelPrefixes?: Record<string, Provider>;
+}
+/**
+ * Provider health check results
+ */
+export interface ProviderHealthStatus {
+    providerId: string;
+    isHealthy: boolean;
+    latency?: number;
+    error?: string;
+    lastCheck: Date;
+}
+/**
+ * Model usage statistics
+ */
+export interface ModelUsageStats {
+    modelId: string;
+    providerId: string;
+    totalCalls: number;
+    totalTokens: number;
+    averageLatency: number;
+    errorRate: number;
+    lastUsed: Date;
+}
+/**
+ * AI Provider Management Integration Class
+ */
+export declare class AIProviderManagement extends EventEmitter {
+    private initialized;
+    private registry;
+    private customProviders;
+    private modelAliases;
+    private healthStatus;
+    private usageStats;
+    private defaultLanguageModel;
+    private defaultEmbeddingModel;
+    private globalProvider;
+    private embeddingModelConfigs;
+    private modelSettingsPresets;
+    private embeddingsIntegration;
+    private settingsManagement;
+    constructor();
+    /**
+     * 初始化
+     */
+    initialize(): Promise<void>;
+    /**
+     * 设置默认提供商 - 从环境变量加载配置
+     */
+    private setupDefaultProviders;
+    /**
+     * Create custom provider
+     */
+    createCustomProvider(id: string, config: CustomProviderConfig): Provider;
+    /**
+     * Create model alias
+     */
+    createModelAlias(aliasConfig: ModelAliasConfig): any;
+    /**
+     * Create model with predefined settings
+     */
+    createModelWithSettings(originalModel: LanguageModel, settings: {
+        maxOutputTokens?: number;
+        temperature?: number;
+        topP?: number;
+        topK?: number;
+        presencePenalty?: number;
+        frequencyPenalty?: number;
+        seed?: number;
+        providerOptions?: Record<string, any>;
+    }): any;
+    /**
+     * Create provider registry
+     */
+    createProviderRegistry(config: ProviderRegistryConfig): any;
+    /**
+     * Get language model from registry
+     */
+    getLanguageModel(modelId?: string): LanguageModel | null;
+    /**
+     * Get text embedding model from registry
+     */
+    getTextEmbeddingModel(modelId?: string): EmbeddingModel | null;
+    /**
+     * 从注册表获取图像模型 - 官方用法
+     */
+    getImageModel(modelId: string): ImageModel | null;
+    /**
+     * Set global default provider
+     */
+    setGlobalProvider(provider: Provider): void;
+    /**
+     * Get current global provider
+     */
+    getGlobalProvider(): Provider | null;
+    /**
+     * Check provider health status
+     */
+    checkProviderHealth(providerId: string): Promise<ProviderHealthStatus>;
+    /**
+     * Check health status of all providers
+     */
+    checkAllProvidersHealth(): Promise<ProviderHealthStatus[]>;
+    /**
+     * Record model usage statistics
+     */
+    private recordModelUsage;
+    /**
+     * Smart model selection based on health and load
+     */
+    selectBestModel(modelIds: string[], criteria?: {
+        prioritizeHealth?: boolean;
+        prioritizeLatency?: boolean;
+        prioritizeReliability?: boolean;
+    }): Promise<string | null>;
+    /**
+     * Create model chain with fallback
+     */
+    createModelChain(primaryModelId: string, fallbackModelIds: string[]): {
+        execute: <T>(operation: (model: LanguageModel) => Promise<T>) => Promise<T>;
+    };
+    /**
+     * Get all registered providers
+     */
+    getRegisteredProviders(): string[];
+    /**
+     * Get model aliases
+     */
+    getModelAliases(): Record<string, ModelAliasConfig>;
+    /**
+     * Get usage statistics
+     */
+    getUsageStats(): ModelUsageStats[];
+    /**
+     * Get health status report
+     */
+    getHealthReport(): {
+        totalProviders: number;
+        healthyProviders: number;
+        unhealthyProviders: number;
+        statuses: ProviderHealthStatus[];
+    };
+    /**
+     * Clean up resources
+     */
+    cleanup(): Promise<void>;
+    /**
+     * Check initialization status
+     */
+    isInitialized(): boolean;
+    /**
+     * Get statistics
+     */
+    getStats(): {
+        isInitialized: boolean;
+        customProviders: number;
+        modelAliases: number;
+        totalUsage: number;
+        hasRegistry: boolean;
+        hasGlobalProvider: boolean;
+    };
+    /**
+     * Get embeddings integration instance
+     */
+    getEmbeddingsIntegration(): AIEmbeddingsIntegration;
+    /**
+     * Configure embedding model parameters
+     */
+    configureEmbeddingModel(providerId: string, modelId: string, config: EmbeddingModelConfig): void;
+    /**
+     * Get embedding model configuration
+     */
+    getEmbeddingModelConfig(providerId: string, modelId: string): EmbeddingModelConfig | null;
+    /**
+     * Get all supported embedding models
+     */
+    getSupportedEmbeddingModels(): any;
+    /**
+     * Get embedding models by provider
+     */
+    getEmbeddingModelsByProvider(provider: string): any[];
+    /**
+     * Get settings management instance
+     */
+    getSettingsManagement(): AISettingsManagement;
+    /**
+     * Apply settings preset to model configuration
+     */
+    applySettingsPreset(modelId: string, presetName: string, additionalSettings?: Partial<ModelSettings>): ModelSettings;
+    /**
+     * Create model instance with custom settings
+     */
+    createModelWithCustomSettings(baseModel: LanguageModel, settings: ModelSettings): any;
+    /**
+     * Get model applied settings
+     */
+    getModelSettings(modelId: string): ModelSettings | null;
+    /**
+     * Create optimized model for specific use case
+     */
+    createOptimizedModel(baseModel: LanguageModel, useCase: 'creative' | 'analytical' | 'deterministic' | 'fast' | 'long-form' | 'code'): any;
+    /**
+     * Destroy instance - enhanced version
+     */
+    destroy(): Promise<void>;
+}
+export { type Provider, customProvider, type ImageModel, wrapLanguageModel, type LanguageModel, type EmbeddingModel, createProviderRegistry, defaultSettingsMiddleware };
+export default AIProviderManagement;
+//# sourceMappingURL=ai-provider.d.ts.map

package/dist/core/ai-provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-provider.d.ts","sourceRoot":"","sources":["../../src/core/ai-provider.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EACV,aAAa,EACb,oBAAoB,EACrB,MAAM,eAAe,CAAC;AAEvB,OAAO,KAAK,EACV,oBAAoB,EACpB,uBAAuB,EACxB,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EACL,KAAK,QAAQ,EACb,cAAc,EACd,KAAK,UAAU,EACf,iBAAiB,EACjB,KAAK,aAAa,EAClB,KAAK,cAAc,EACnB,sBAAsB,EACtB,yBAAyB,EAC1B,MAAM,IAAI,CAAC;AAIZ;;GAEG;AACH,MAAM,WAAW,oBAAoB;IAEnC,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,GAAG;QAC9C,KAAK,EAAE,aAAa,CAAC;QACrB,UAAU,CAAC,EAAE,GAAG,CAAC;QACjB,QAAQ,CAAC,EAAE;YACT,eAAe,CAAC,EAAE,MAAM,CAAC;YACzB,WAAW,CAAC,EAAE,MAAM,CAAC;YACrB,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,eAAe,CAAC,EAAE,MAAM,CAAC;YACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;YAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;SACvC,CAAC;KACH,CAAC,CAAC;IAGH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAGjD,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAGzC,gBAAgB,CAAC,EAAE,QAAQ,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACpC,OAAO,CAAC,EAAE;QACR,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,GAAG,CAAC;IACf,UAAU,CAAC,EAAE,GAAG,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,eAAe,EAAE,QAAQ,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;CAC1C;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,IAAI,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,IAAI,CAAC;CAChB;AAID;;GAEG;AACH,qBAAa,oBAAqB,SAAQ,YAAY;IACpD,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,eAAe,CAAoC;IAC3D,OAAO,CAAC,YAAY,CAA4C;IAChE,OAAO,CAAC,YAAY,CAAgD;IACpE,OAAO,CAAC,UAAU,CAA2C;IAC7D,OAAO,CAAC,oBAAoB,CAAoB;IAChD,OAAO,CAAC,qBAAqB,CAAoB;IACjD,OAAO,CAAC,cAAc,CAAyB;IAG/C,OAAO,CAAC,qBAAqB,CAAgD;IAC7E,OAAO,CAAC,oBAAoB,CAAyC;IACrE,OAAO,CAAC,qBAAqB,CAAM;IACnC,OAAO,CAAC,kBAAkB,CAAM;;IAShC;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAiCjC;;OAEG;YACW,qBAAqB;IA+DnC;;OAEG;IACH,oBAAoB,CAClB,EAAE,EAAE,MAAM,EACV,MAAM,EAAE,oBAAoB,GAC3B,QAAQ;IAqDX;;OAEG;IACH,gBAAgB,CACd,WAAW,EAAE,gBAAgB,GAC5B,GAAG;IAeN;;OAEG;IACH,uBAAuB,CACrB,aAAa,EAAE,aAAa,EAC5B,QAAQ,EAAE;QACR,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACvC,GACA,GAAG;IAWN;;OAEG;IACH,sBAAsB,CAAC,MAAM,EAAE,sBAAsB,GAAG,GAAG;IA8B3D;;OAEG;IACH,gBAAgB,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI;IAuDxD;;OAEG;IACH,qBAAqB,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,cAAc,GAAG,IAAI;IAiC9D;;OAEG;IACH,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI;IAoBjD;;OAEG;IACH,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI;IAU3C;;OAEG;IACH,iBAAiB,IAAI,QAAQ,GAAG,IAAI;IAMpC;;OAEG;IACG,mBAAmB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAoC5E;;OAEG;IACG,uBAAuB,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAehE;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAuBxB;;OAEG;IACG,eAAe,CACnB,QAAQ,EAAE,MAAM,EAAE,EAClB,QAAQ,GAAE;QACR,gBAAgB,CAAC,EAAE,OAAO,CAAC;QAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;QAC5B,qBAAqB,CAAC,EAAE,OAAO,CAAC;KAC5B,GACL,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IA8CzB;;OAEG;IACH,gBAAgB,CAAC,cAAc,EAAE,MAAM,EAAE,gBAAgB,EAAE,MAAM,EAAE,GAAG;QACpE,OAAO,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,KAAK,EAAE,aAAa,KAAK,OAAO,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;KAC7E;IA6CD;;OAEG;IACH,sBAAsB,IAAI,MAAM,EAAE;IAIlC;;OAEG;IACH,eAAe,IAAI,MAAM,CAAC,MAAM,EAAE,gBAAgB,CAAC;IAQnD;;OAEG;IACH,aAAa,IAAI,eAAe,EAAE;IAIlC;;OAEG;IACH,eAAe,IAAI;QACjB,cAAc,EAAE,MAAM,CAAC;QACvB,gBAAgB,EAAE,MAAM,CAAC;QACzB,kBAAkB,EAAE,MAAM,CAAC;QAC3B,QAAQ,EAAE,oBAAoB,EAAE,CAAC;KAClC;IAUD;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAgB9B;;OAEG;IACH,aAAa,IAAI,OAAO;IAIxB;;OAEG;IACH,QAAQ,IAAI;QACV,aAAa,EAAE,OAAO,CAAC;QACvB,eAAe,EAAE,MAAM,CAAC;QACxB,YAAY,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,OAAO,CAAC;QACrB,iBAAiB,EAAE,OAAO,CAAC;KAC5B;IAaD;;OAEG;IACH,wBAAwB,IAAI,uBAAuB;IAInD;;OAEG;IACH,uBAAuB,CACrB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,oBAAoB,GAC3B,IAAI;IAWP;;OAEG;IACH,uBAAuB,CACrB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,GACd,oBAAoB,GAAG,IAAI;IAK9B;;OAEG;IACH,2BAA2B,IAAI,GAAG;IAIlC;;OAEG;IACH,4BAA4B,CAAC,QAAQ,EAAE,MAAM,GAAG,GAAG,EAAE;IAMrD;;OAEG;IACH,qBAAqB,IAAI,oBAAoB;IAI7C;;OAEG;IACH,mBAAmB,CACjB,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,EAClB,kBAAkB,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,GAC1C,aAAa;IA0ChB;;OAEG;IACH,6BAA6B,CAC3B,SAAS,EAAE,aAAa,EACxB,QAAQ,EAAE,aAAa,GACtB,GAAG;IAmBN;;OAEG;IACH,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI;IAIvD;;OAEG;IACH,oBAAoB,CAClB,SAAS,EAAE,aAAa,EACxB,OAAO,EAAE,UAAU,GAAG,YAAY,GAAG,eAAe,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,GACnF,GAAG;IAoBN;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAsB/B;AAID,OAAO,EACL,KAAK,QAAQ,EACb,cAAc,EACd,KAAK,UAAU,EACf,iBAAiB,EAGjB,KAAK,aAAa,EAClB,KAAK,cAAc,EAEnB,sBAAsB,EACtB,yBAAyB,EAC1B,CAAC;AAGF,eAAe,oBAAoB,CAAC"}