@hazeljs/ai 0.2.0-alpha.1

package/dist/ai-enhanced.types.d.ts

@@ -0,0 +1,277 @@
+/**
+ * AI Provider types
+ */
+export type AIProvider = 'openai' | 'anthropic' | 'gemini' | 'cohere' | 'ollama' | 'huggingface';
+/**
+ * AI model configuration
+ */
+export interface AIModelConfig {
+    /**
+     * Provider name
+     */
+    provider: AIProvider;
+    /**
+     * Model name
+     */
+    model: string;
+    /**
+     * API key
+     */
+    apiKey?: string;
+    /**
+     * Temperature (0-1)
+     */
+    temperature?: number;
+    /**
+     * Max tokens
+     */
+    maxTokens?: number;
+    /**
+     * Top P
+     */
+    topP?: number;
+    /**
+     * Enable streaming
+     */
+    streaming?: boolean;
+    /**
+     * Custom endpoint
+     */
+    endpoint?: string;
+}
+/**
+ * AI message role
+ */
+export type AIMessageRole = 'system' | 'user' | 'assistant' | 'function' | 'tool';
+/**
+ * AI message
+ */
+export interface AIToolCall {
+    id: string;
+    type: 'function';
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+export interface AIMessage {
+    role: AIMessageRole;
+    content: string;
+    name?: string;
+    toolCallId?: string;
+    functionCall?: {
+        name: string;
+        arguments: string;
+    };
+    toolCalls?: AIToolCall[];
+}
+/**
+ * AI completion request
+ */
+export interface AICompletionRequest {
+    messages: AIMessage[];
+    model?: string;
+    temperature?: number;
+    maxTokens?: number;
+    topP?: number;
+    stream?: boolean;
+    functions?: AIFunction[];
+    functionCall?: 'auto' | 'none' | {
+        name: string;
+    };
+}
+/**
+ * AI completion response
+ */
+export interface AICompletionResponse {
+    id: string;
+    content: string;
+    role: AIMessageRole;
+    model: string;
+    usage?: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+    functionCall?: {
+        name: string;
+        arguments: string;
+    };
+    toolCalls?: AIToolCall[];
+    finishReason?: string;
+}
+/**
+ * AI streaming chunk
+ */
+export interface AIStreamChunk {
+    id: string;
+    content: string;
+    delta: string;
+    done: boolean;
+    usage?: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+}
+/**
+ * AI function definition
+ */
+export interface AIFunction {
+    name: string;
+    description: string;
+    parameters: {
+        type: 'object';
+        properties: Record<string, {
+            type: string;
+            description?: string;
+            enum?: string[];
+        }>;
+        required?: string[];
+    };
+}
+/**
+ * AI embedding request
+ */
+export interface AIEmbeddingRequest {
+    input: string | string[];
+    model?: string;
+}
+/**
+ * AI embedding response
+ */
+export interface AIEmbeddingResponse {
+    embeddings: number[][];
+    model: string;
+    usage?: {
+        promptTokens: number;
+        totalTokens: number;
+    };
+}
+/**
+ * AI provider interface
+ */
+export interface IAIProvider {
+    /**
+     * Provider name
+     */
+    readonly name: AIProvider;
+    /**
+     * Generate completion
+     */
+    complete(request: AICompletionRequest): Promise<AICompletionResponse>;
+    /**
+     * Generate streaming completion
+     */
+    streamComplete(request: AICompletionRequest): AsyncGenerator<AIStreamChunk>;
+    /**
+     * Generate embeddings
+     */
+    embed(request: AIEmbeddingRequest): Promise<AIEmbeddingResponse>;
+    /**
+     * Check if provider is available
+     */
+    isAvailable(): Promise<boolean>;
+    /**
+     * Generate speech from text (TTS). Optional; supported by OpenAI and compatible providers.
+     */
+    speech?(input: string, options?: {
+        voice?: string;
+        model?: string;
+        format?: 'mp3' | 'opus';
+    }): Promise<Buffer>;
+}
+/**
+ * Vector database types
+ */
+export type VectorDatabase = 'pinecone' | 'weaviate' | 'qdrant' | 'chroma';
+/**
+ * Vector store configuration
+ */
+export interface VectorStoreConfig {
+    database: VectorDatabase;
+    apiKey?: string;
+    endpoint?: string;
+    index?: string;
+    namespace?: string;
+}
+/**
+ * Vector document
+ */
+export interface VectorDocument {
+    id: string;
+    content: string;
+    embedding?: number[];
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Vector search request
+ */
+export interface VectorSearchRequest {
+    query: string;
+    topK?: number;
+    filter?: Record<string, unknown>;
+    namespace?: string;
+}
+/**
+ * Vector search result
+ */
+export interface VectorSearchResult {
+    id: string;
+    content: string;
+    score: number;
+    metadata?: Record<string, unknown>;
+}
+/**
+ * AI context manager
+ */
+export interface AIContext {
+    messages: AIMessage[];
+    maxTokens: number;
+    currentTokens: number;
+    addMessage(message: AIMessage): void;
+    getMessages(): AIMessage[];
+    clear(): void;
+    trimToLimit(): void;
+}
+/**
+ * Token usage tracker
+ */
+export interface TokenUsage {
+    userId?: string;
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+    cost?: number;
+    timestamp: number;
+}
+/**
+ * Token limit configuration
+ */
+export interface TokenLimitConfig {
+    maxTokensPerRequest?: number;
+    maxTokensPerDay?: number;
+    maxTokensPerMonth?: number;
+    costPerToken?: number;
+}
+/**
+ * AI function decorator options
+ */
+export interface AIFunctionOptions {
+    provider: AIProvider;
+    model: string;
+    streaming?: boolean;
+    temperature?: number;
+    maxTokens?: number;
+    systemPrompt?: string;
+}
+/**
+ * AI validation options
+ */
+export interface AIValidationOptions {
+    provider: AIProvider;
+    model?: string;
+    instruction: string;
+    failOnInvalid?: boolean;
+}
+//# sourceMappingURL=ai-enhanced.types.d.ts.map

package/dist/ai-enhanced.types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-enhanced.types.d.ts","sourceRoot":"","sources":["../src/ai-enhanced.types.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,GAAG,aAAa,CAAC;AAEjG;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,QAAQ,EAAE,UAAU,CAAC;IAErB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IAEpB;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,MAAM,aAAa,GAAG,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,UAAU,GAAG,MAAM,CAAC;AAElF;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,aAAa,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,SAAS,EAAE,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;CACnD;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,aAAa,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,YAAY,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE;QACV,IAAI,EAAE,QAAQ,CAAC;QACf,UAAU,EAAE,MAAM,CAChB,MAAM,EACN;YACE,IAAI,EAAE,MAAM,CAAC;YACb,WAAW,CAAC,EAAE,MAAM,CAAC;YACrB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;SACjB,CACF,CAAC;QACF,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;KACrB,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,UAAU,EAAE,MAAM,EAAE,EAAE,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;IAE1B;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC;IAEtE;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,mBAAmB,GAAG,cAAc,CAAC,aAAa,CAAC,CAAC;IAE5E;;OAEG;IACH,KAAK,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAEjE;;OAEG;IACH,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IAEhC;;OAEG;IACH,MAAM,CAAC,CACL,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,KAAK,GAAG,MAAM,CAAA;KAAE,GACpE,OAAO,CAAC,MAAM,CAAC,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,UAAU,GAAG,UAAU,GAAG,QAAQ,GAAG,QAAQ,CAAC;AAE3E;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,cAAc,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,QAAQ,EAAE,SAAS,EAAE,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IAEtB,UAAU,CAAC,OAAO,EAAE,SAAS,GAAG,IAAI,CAAC;IACrC,WAAW,IAAI,SAAS,EAAE,CAAC;IAC3B,KAAK,IAAI,IAAI,CAAC;IACd,WAAW,IAAI,IAAI,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,UAAU,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,UAAU,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB"}

package/dist/ai-enhanced.types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/ai.decorator.d.ts

@@ -0,0 +1,4 @@
+import { AITaskConfig } from './ai.types';
+import 'reflect-metadata';
+export declare function AITask(config: AITaskConfig): MethodDecorator;
+//# sourceMappingURL=ai.decorator.d.ts.map

package/dist/ai.decorator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.decorator.d.ts","sourceRoot":"","sources":["../src/ai.decorator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC1C,OAAO,kBAAkB,CAAC;AAM1B,wBAAgB,MAAM,CAAC,MAAM,EAAE,YAAY,GAAG,eAAe,CA2D5D"}

package/dist/ai.decorator.js

@@ -0,0 +1,57 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AITask = AITask;
+require("reflect-metadata");
+const core_1 = __importDefault(require("@hazeljs/core"));
+const AI_TASK_METADATA_KEY = 'hazel:ai-task';
+function AITask(config) {
+    return (target, propertyKey, descriptor) => {
+        // Store the AI task configuration in metadata
+        Reflect.defineMetadata(AI_TASK_METADATA_KEY, config, target, propertyKey);
+        // Replace the method with our AI-powered version
+        descriptor.value = async function (...args) {
+            try {
+                // Get the AI task configuration from metadata
+                const taskConfig = Reflect.getMetadata(AI_TASK_METADATA_KEY, target, propertyKey);
+                if (!taskConfig) {
+                    throw new Error('AI task configuration not found');
+                }
+                // Get the AI service from the container
+                const aiService = this.aiService;
+                if (!aiService) {
+                    throw new Error('AI service not found. Make sure to inject AIService in the constructor.');
+                }
+                core_1.default.debug('Executing AI task with config:', {
+                    name: taskConfig.name,
+                    stream: taskConfig.stream,
+                });
+                // Execute the AI task
+                const result = await aiService.executeTask(taskConfig, args[0]);
+                // If streaming is enabled and a stream is returned, return the stream directly
+                if (taskConfig.stream && result.stream) {
+                    core_1.default.debug('Returning stream from AI task');
+                    return result.stream;
+                }
+                // If there's an error, throw it
+                if (result.error) {
+                    core_1.default.error('AI task error:', result.error);
+                    throw new Error(result.error);
+                }
+                // For non-streaming responses, return the data
+                core_1.default.debug('Returning data from AI task:', result.data);
+                return result.data;
+            }
+            catch (error) {
+                core_1.default.error('AI task execution failed:', error);
+                if (error instanceof Error) {
+                    throw new Error(`AI task execution failed: ${error.message}`);
+                }
+                throw new Error('AI task execution failed: Unknown error');
+            }
+        };
+        return descriptor;
+    };
+}

package/dist/ai.decorator.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ai.decorator.test.d.ts.map

package/dist/ai.decorator.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.decorator.test.d.ts","sourceRoot":"","sources":["../src/ai.decorator.test.ts"],"names":[],"mappings":""}

package/dist/ai.decorator.test.js

@@ -0,0 +1,189 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const ai_decorator_1 = require("./ai.decorator");
+const ai_service_1 = require("./ai.service");
+const core_1 = require("@hazeljs/core");
+const core_2 = require("@hazeljs/core");
+// Mock the AIService
+jest.mock('./ai.service', () => {
+    const mockAIService = jest.fn().mockImplementation(() => ({
+        executeTask: jest.fn().mockImplementation(async (config, input) => ({
+            data: `Processed: ${input}`,
+        })),
+    }));
+    return {
+        AIService: mockAIService,
+    };
+});
+describe('AITask Decorator', () => {
+    let container;
+    let mockAIService;
+    beforeEach(() => {
+        container = core_1.Container.getInstance();
+        container.clear();
+        // Create a mock instance
+        mockAIService = new ai_service_1.AIService();
+        // Register AIService with the mock instance
+        container.register(ai_service_1.AIService, mockAIService);
+    });
+    afterEach(() => {
+        container.clear();
+        jest.clearAllMocks();
+    });
+    it('should inject AIService', () => {
+        let TestClass = class TestClass {
+            constructor(aiService) {
+                this.aiService = aiService;
+            }
+            async testMethod(input) {
+                return input;
+            }
+        };
+        __decorate([
+            (0, ai_decorator_1.AITask)({
+                name: 'test-task',
+                prompt: 'Test prompt',
+                provider: 'openai',
+                model: 'gpt-3.5-turbo',
+                outputType: 'string',
+            }),
+            __metadata("design:type", Function),
+            __metadata("design:paramtypes", [String]),
+            __metadata("design:returntype", Promise)
+        ], TestClass.prototype, "testMethod", null);
+        TestClass = __decorate([
+            (0, core_2.Injectable)(),
+            __metadata("design:paramtypes", [ai_service_1.AIService])
+        ], TestClass);
+        // Register the test class
+        container.register(TestClass, new TestClass(mockAIService));
+        const instance = container.resolve(TestClass);
+        expect(instance.aiService).toBeDefined();
+        expect(instance.aiService).toBe(mockAIService);
+    });
+    it('should work with multiple dependencies', () => {
+        let OtherService = class OtherService {
+        };
+        OtherService = __decorate([
+            (0, core_2.Injectable)()
+        ], OtherService);
+        let TestClass = class TestClass {
+            constructor(aiService, otherService) {
+                this.aiService = aiService;
+                this.otherService = otherService;
+            }
+            async testMethod(input) {
+                return input;
+            }
+        };
+        __decorate([
+            (0, ai_decorator_1.AITask)({
+                name: 'test-task',
+                prompt: 'Test prompt',
+                provider: 'openai',
+                model: 'gpt-3.5-turbo',
+                outputType: 'string',
+            }),
+            __metadata("design:type", Function),
+            __metadata("design:paramtypes", [String]),
+            __metadata("design:returntype", Promise)
+        ], TestClass.prototype, "testMethod", null);
+        TestClass = __decorate([
+            (0, core_2.Injectable)(),
+            __metadata("design:paramtypes", [ai_service_1.AIService,
+                OtherService])
+        ], TestClass);
+        // Register both services
+        const otherService = new OtherService();
+        container.register(OtherService, otherService);
+        container.register(TestClass, new TestClass(mockAIService, otherService));
+        const instance = container.resolve(TestClass);
+        expect(instance.aiService).toBeDefined();
+        expect(instance.aiService).toBe(mockAIService);
+        expect(instance.otherService).toBeDefined();
+        expect(instance.otherService).toBe(otherService);
+    });
+    it('should throw error if AIService is not registered', async () => {
+        let TestClass = class TestClass {
+            constructor(aiService) {
+                this.aiService = aiService;
+            }
+            async testMethod(input) {
+                return input;
+            }
+        };
+        __decorate([
+            (0, ai_decorator_1.AITask)({
+                name: 'test-task',
+                prompt: 'Test prompt',
+                provider: 'openai',
+                model: 'gpt-3.5-turbo',
+                outputType: 'string',
+            }),
+            __metadata("design:type", Function),
+            __metadata("design:paramtypes", [String]),
+            __metadata("design:returntype", Promise)
+        ], TestClass.prototype, "testMethod", null);
+        TestClass = __decorate([
+            (0, core_2.Injectable)(),
+            __metadata("design:paramtypes", [ai_service_1.AIService])
+        ], TestClass);
+        container.clear();
+        // Don't register AIService
+        const instance = new TestClass(null);
+        container.register(TestClass, instance);
+        const resolved = container.resolve(TestClass);
+        // The error should be thrown when trying to execute the decorated method
+        await expect(resolved.testMethod('test')).rejects.toThrow('AI task execution failed: AI service not found. Make sure to inject AIService in the constructor.');
+    });
+    it('should execute AI task', async () => {
+        let TestClass = class TestClass {
+            constructor(aiService) {
+                this.aiService = aiService;
+            }
+            async testMethod(input) {
+                return input;
+            }
+        };
+        __decorate([
+            (0, ai_decorator_1.AITask)({
+                name: 'test-task',
+                prompt: 'Test prompt',
+                provider: 'openai',
+                model: 'gpt-4',
+                temperature: 0.7,
+                outputType: 'string',
+            }),
+            __metadata("design:type", Function),
+            __metadata("design:paramtypes", [String]),
+            __metadata("design:returntype", Promise)
+        ], TestClass.prototype, "testMethod", null);
+        TestClass = __decorate([
+            (0, core_2.Injectable)(),
+            __metadata("design:paramtypes", [ai_service_1.AIService])
+        ], TestClass);
+        // Register the test class with AIService
+        container.register(TestClass, new TestClass(mockAIService));
+        const instance = container.resolve(TestClass);
+        const result = await instance.testMethod('test input');
+        expect(result).toBeDefined();
+        expect(result).toBe('Processed: test input');
+        expect(mockAIService.executeTask).toHaveBeenCalledWith(expect.objectContaining({
+            name: 'test-task',
+            provider: 'openai',
+            model: 'gpt-4',
+            temperature: 0.7,
+            prompt: 'Test prompt',
+            outputType: 'string',
+        }), 'test input');
+    });
+});

package/dist/ai.module.d.ts

@@ -0,0 +1,12 @@
+import { AIProvider } from './ai-enhanced.types';
+export interface AIModuleOptions {
+    defaultProvider?: AIProvider;
+    providers?: AIProvider[];
+    apiKeys?: Partial<Record<AIProvider, string>>;
+}
+export declare class AIModule {
+    private static options;
+    static register(options: AIModuleOptions): typeof AIModule;
+    static getOptions(): AIModuleOptions;
+}
+//# sourceMappingURL=ai.module.d.ts.map

package/dist/ai.module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.module.d.ts","sourceRoot":"","sources":["../src/ai.module.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,MAAM,WAAW,eAAe;IAC9B,eAAe,CAAC,EAAE,UAAU,CAAC;IAC7B,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;IACzB,OAAO,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;CAC/C;AAED,qBAIa,QAAQ;IACnB,OAAO,CAAC,MAAM,CAAC,OAAO,CAAuB;IAE7C,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,QAAQ;IAqB1D,MAAM,CAAC,UAAU,IAAI,eAAe;CAGrC"}

package/dist/ai.module.js

@@ -0,0 +1,44 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var AIModule_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIModule = void 0;
+const core_1 = require("@hazeljs/core");
+const ai_service_1 = require("./ai.service");
+const ai_enhanced_service_1 = require("./ai-enhanced.service");
+let AIModule = AIModule_1 = class AIModule {
+    static register(options) {
+        AIModule_1.options = options;
+        // Set API keys in environment if provided (allows runtime configuration)
+        if (options.apiKeys) {
+            const keyMap = {
+                openai: 'OPENAI_API_KEY',
+                anthropic: 'ANTHROPIC_API_KEY',
+                gemini: 'GEMINI_API_KEY',
+                cohere: 'COHERE_API_KEY',
+            };
+            for (const [provider, key] of Object.entries(options.apiKeys)) {
+                if (key && keyMap[provider]) {
+                    process.env[keyMap[provider]] = key;
+                }
+            }
+        }
+        return AIModule_1;
+    }
+    static getOptions() {
+        return AIModule_1.options;
+    }
+};
+exports.AIModule = AIModule;
+AIModule.options = {};
+exports.AIModule = AIModule = AIModule_1 = __decorate([
+    (0, core_1.HazelModule)({
+        providers: [ai_service_1.AIService, ai_enhanced_service_1.AIEnhancedService],
+        exports: [ai_service_1.AIService, ai_enhanced_service_1.AIEnhancedService],
+    })
+], AIModule);

package/dist/ai.module.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ai.module.test.d.ts.map

package/dist/ai.module.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.module.test.d.ts","sourceRoot":"","sources":["../src/ai.module.test.ts"],"names":[],"mappings":""}

package/dist/ai.module.test.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const ai_module_1 = require("./ai.module");
+const ai_service_1 = require("./ai.service");
+const core_1 = require("@hazeljs/core");
+describe('AIModule', () => {
+    it('should be defined', () => {
+        expect(ai_module_1.AIModule).toBeDefined();
+    });
+    it('should provide AIService', () => {
+        const app = new core_1.HazelApp(ai_module_1.AIModule);
+        const container = app.getContainer();
+        const aiService = container.resolve(ai_service_1.AIService);
+        expect(aiService).toBeInstanceOf(ai_service_1.AIService);
+    });
+    it('should provide AIService as singleton', () => {
+        const app = new core_1.HazelApp(ai_module_1.AIModule);
+        const container = app.getContainer();
+        const service1 = container.resolve(ai_service_1.AIService);
+        const service2 = container.resolve(ai_service_1.AIService);
+        expect(service1).toBe(service2);
+    });
+});

package/dist/ai.service.d.ts

@@ -0,0 +1,11 @@
+import { AITaskConfig, AITaskResult } from './ai.types';
+import './prompts/task.prompt';
+export declare class AIService {
+    private providers;
+    constructor();
+    private initializeProviders;
+    private formatPrompt;
+    private parseResponse;
+    executeTask(config: AITaskConfig, input: unknown): Promise<AITaskResult>;
+}
+//# sourceMappingURL=ai.service.d.ts.map

package/dist/ai.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.service.d.ts","sourceRoot":"","sources":["../src/ai.service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAiB,YAAY,EAAE,MAAM,YAAY,CAAC;AAIvE,OAAO,uBAAuB,CAAC;AAW/B,qBACa,SAAS;IACpB,OAAO,CAAC,SAAS,CAAsC;;IAOvD,OAAO,CAAC,mBAAmB;IAoL3B,OAAO,CAAC,YAAY;IAmBpB,OAAO,CAAC,aAAa;IAkBf,WAAW,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC;CAkC/E"}