aimemory-core 1.0.1

package/.eslintrc.json

@@ -0,0 +1,22 @@
+{
+  "root": true,
+  "parser": "@typescript-eslint/parser",
+  "plugins": ["@typescript-eslint"],
+  "extends": [
+    "eslint:recommended",
+    "plugin:@typescript-eslint/recommended"
+  ],
+  "parserOptions": {
+    "ecmaVersion": 2022,
+    "sourceType": "module"
+  },
+  "env": {
+    "node": true,
+    "es2022": true
+  },
+  "rules": {
+    "@typescript-eslint/no-unused-vars": ["error", { "argsIgnorePattern": "^_" }],
+    "@typescript-eslint/explicit-function-return-type": "off",
+    "@typescript-eslint/no-explicit-any": "warn"
+  }
+}

package/.github/workflows/ci.yml

@@ -0,0 +1,57 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+      
+      - name: Use Node.js
+        uses: actions/setup-node@v4
+        with:
+          node-version: '20'
+          cache: 'npm'
+      
+      - name: Install dependencies
+        run: npm ci
+      
+      - name: Run linter
+        run: npm run lint
+      
+      - name: Run tests
+        run: npm test
+      
+      - name: Build
+        run: npm run build
+
+  publish:
+    needs: test
+    runs-on: ubuntu-latest
+    if: github.event_name == 'push' && github.ref == 'refs/heads/main'
+    
+    steps:
+      - uses: actions/checkout@v4
+      
+      - name: Use Node.js
+        uses: actions/setup-node@v4
+        with:
+          node-version: '20'
+          registry-url: 'https://registry.npmjs.org'
+      
+      - name: Install dependencies
+        run: npm ci
+      
+      - name: Build
+        run: npm run build
+      
+      - name: Publish to npm
+        run: npm publish
+        env:
+          NODE_AUTH_TOKEN: ${{ secrets.NPM_TOKEN }}

package/.prettierrc

@@ -0,0 +1,8 @@
+{
+  "semi": true,
+  "trailingComma": "es5",
+  "singleQuote": true,
+  "printWidth": 100,
+  "tabWidth": 2,
+  "useTabs": false
+}

package/README.md

@@ -0,0 +1,197 @@
+# AI Memory
+
+<p align="center">
+  <img src="https://img.shields.io/npm/v/ai-memory" alt="npm version">
+  <img src="https://img.shields.io/npm/dm/ai-memory" alt="npm downloads">
+  <img src="https://img.shields.io/github/license/ai-memory/ai-memory" alt="license">
+  <img src="https://img.shields.io/github/stars/ai-memory/ai-memory" alt="stars">
+</p>
+
+> Persistent memory system for LLM applications with semantic search, context management, and monetization-ready infrastructure.
+
+## Features
+
+- **Semantic Search** - Find relevant memories using vector embeddings and cosine similarity
+- **Multi-User Support** - Manage memories for multiple users with proper isolation
+- **Session Management** - Track conversations by session
+- **Context Builder** - Build optimized context for LLM prompts
+- **TypeScript Native** - Full TypeScript support with complete type definitions
+- **Rate Limiting** - Built-in rate limiting ready for monetization
+- **Usage Tracking** - Track API usage per user with configurable limits
+- **Flexible Storage** - In-memory, Redis, or PostgreSQL persistence
+
+## Installation
+
+```bash
+npm install aimemory-core
+```
+
+## Quick Start
+
+```typescript
+import { createAIMemory, EmbeddingFunction } from 'ai-memory';
+
+// Your embedding function (OpenAI, Cohere, etc.)
+const embeddingFunction: EmbeddingFunction = async (text) => {
+  return await openai.embeddings.create({
+    model: 'text-embedding-ada-002',
+    input: text,
+  }).then(res => res.data[0].embedding);
+};
+
+const memory = createAIMemory({
+  maxMemories: 1000,
+  context: { maxTokens: 2000 }
+});
+
+memory.setEmbeddingFunction(embeddingFunction);
+
+// Add memories
+await memory.rememberFact('The user prefers dark mode', {
+  userId: 'user-123',
+  tags: ['preferences', 'ui']
+});
+
+await memory.rememberPreference('User likes bullet point summaries', {
+  userId: 'user-123',
+  tags: ['communication']
+});
+
+// Search relevant context
+const results = await memory.recall('user preferences');
+console.log(`Found ${results.length} relevant memories`);
+
+// Build context for LLM
+const context = await memory.getContext('What does user prefer?');
+console.log(context.messages);
+```
+
+## Core Concepts
+
+### Memory Types
+
+AI Memory supports different types of memories:
+
+- **conversation** - User and assistant messages
+- **fact** - Factual information about users
+- **preference** - User preferences and likes
+- **instruction** - Instructions or rules to follow
+- **context** - General context information
+- **custom** - Custom memory types
+
+### Configuration
+
+```typescript
+const memory = createAIMemory({
+  maxMemories: 10000,           // Max memories per instance
+  defaultImportance: 0.5,        // Default importance (0-1)
+  embeddingDimension: 1536,      // Embedding vector dimension
+  context: {
+    maxTokens: 4000,             // Max tokens in context
+    relevanceThreshold: 0.7,     // Minimum relevance score
+    memoryTypes: ['fact', 'preference', 'instruction', 'conversation'],
+    maxMemories: 10,            // Max memories in context
+  },
+  autoCleanup: {
+    enabled: true,               // Auto cleanup old memories
+    maxAge: 90 * 24 * 60 * 60 * 1000, // 90 days
+    minImportance: 0.3,
+  }
+});
+```
+
+## Monetization
+
+AI Memory includes built-in support for usage limits and rate limiting:
+
+```typescript
+import { PlanManager, PLANS } from 'ai-memory';
+
+const planManager = new PlanManager();
+
+// Set user plan
+planManager.setUserPlan('user-123', 'pro');
+
+// Check rate limit
+const rateResult = await planManager.checkRateLimit('user-123');
+if (!rateResult.allowed) {
+  console.log(`Rate limit exceeded. Retry after ${rateResult.retryAfter}s`);
+}
+
+// Check usage limit
+const usageResult = await planManager.checkUsageLimit('user-123', 'api_calls');
+console.log(`Remaining: ${usageResult.remaining}/${usageResult.limit}`);
+```
+
+### Available Plans
+
+| Feature | Free | Pro | Enterprise |
+|---------|------|-----|------------|
+| Max Memories | 5,000 | 50,000 | Unlimited |
+| Searches/Day | 500 | 10,000 | Unlimited |
+| Persistence | In-Memory | Redis | PostgreSQL |
+| Support | Community | Email | Priority 24/7 |
+| Price | $0/mo | $29/mo | $99/mo |
+
+## API Reference
+
+### AIMemory Class
+
+| Method | Description |
+|--------|-------------|
+| `remember(content, metadata)` | Add a memory |
+| `rememberFact(content, metadata)` | Add factual memory |
+| `rememberPreference(content, metadata)` | Add preference |
+| `rememberInstruction(content, metadata)` | Add instruction |
+| `recall(query, options)` | Search memories |
+| `getContext(query)` | Build context for LLM |
+| `getMemory(id)` | Get memory by ID |
+| `updateMemory(id, updates)` | Update memory |
+| `forget(id)` | Delete memory |
+| `getRecentMemories(limit)` | Get recent memories |
+| `getStats()` | Get memory statistics |
+| `exportData()` | Export all memories |
+| `importData(memories)` | Import memories |
+
+## Storage Adapters
+
+### Redis (for Pro plans)
+
+```typescript
+import { createRedisRateLimiter } from 'ai-memory';
+
+const store = createRedisRateLimiter({
+  host: 'localhost',
+  port: 6379,
+  password: 'your-password',
+});
+```
+
+### PostgreSQL (for Enterprise plans)
+
+```typescript
+import { createPostgresUsageStore } from 'ai-memory';
+
+const store = createPostgresUsageStore({
+  host: 'localhost',
+  port: 5432,
+  database: 'ai_memory',
+  user: 'postgres',
+  password: 'your-password',
+});
+```
+
+## Use Cases
+
+- **AI Chatbots** - Maintain conversation context across sessions
+- **Virtual Assistants** - Remember user preferences and history
+- **Customer Support** - Track customer interactions and issues
+- **Analytics** - Analyze user behavior patterns over time
+
+## License
+
+MIT
+
+---
+
+<p align="center">Built with ❤️ for AI developers</p>

package/dist/config.d.ts

@@ -0,0 +1,20 @@
+import { ContextConfig } from './types.js';
+export interface AIMemoryConfig {
+    storagePath: string;
+    maxMemories: number;
+    defaultImportance: number;
+    embeddingDimension: number;
+    context: ContextConfig;
+    autoCleanup: {
+        enabled: boolean;
+        maxAge: number;
+        minImportance: number;
+    };
+    logging: {
+        enabled: boolean;
+        level: 'debug' | 'info' | 'warn' | 'error';
+    };
+}
+export declare const DEFAULT_CONFIG: AIMemoryConfig;
+export declare function validateConfig(config: Partial<AIMemoryConfig>): AIMemoryConfig;
+//# sourceMappingURL=config.d.ts.map

package/dist/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAc,MAAM,YAAY,CAAC;AAEvD,MAAM,WAAW,cAAc;IAC7B,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,OAAO,EAAE,aAAa,CAAC;IACvB,WAAW,EAAE;QACX,OAAO,EAAE,OAAO,CAAC;QACjB,MAAM,EAAE,MAAM,CAAC;QACf,aAAa,EAAE,MAAM,CAAC;KACvB,CAAC;IACF,OAAO,EAAE;QACP,OAAO,EAAE,OAAO,CAAC;QACjB,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;KAC5C,CAAC;CACH;AAED,eAAO,MAAM,cAAc,EAAE,cAuB5B,CAAC;AAEF,wBAAgB,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,cAAc,CAoB9E"}

package/dist/config.js

@@ -0,0 +1,41 @@
+export const DEFAULT_CONFIG = {
+    storagePath: './ai-memory-data',
+    maxMemories: 10000,
+    defaultImportance: 0.5,
+    embeddingDimension: 1536,
+    context: {
+        maxTokens: 4000,
+        includeSystemPrompt: true,
+        systemPrompt: 'You are a helpful AI assistant with memory of past conversations.',
+        relevanceThreshold: 0.7,
+        includeMetadata: false,
+        memoryTypes: ['conversation', 'fact', 'preference', 'context'],
+        maxMemories: 10,
+    },
+    autoCleanup: {
+        enabled: false,
+        maxAge: 90 * 24 * 60 * 60 * 1000,
+        minImportance: 0.3,
+    },
+    logging: {
+        enabled: true,
+        level: 'info',
+    },
+};
+export function validateConfig(config) {
+    const merged = { ...DEFAULT_CONFIG, ...config };
+    if (merged.maxMemories < 1) {
+        throw new Error('maxMemories must be at least 1');
+    }
+    if (merged.embeddingDimension < 1) {
+        throw new Error('embeddingDimension must be at least 1');
+    }
+    if (merged.context.maxTokens < 100) {
+        throw new Error('context.maxTokens must be at least 100');
+    }
+    if (merged.context.relevanceThreshold < 0 || merged.context.relevanceThreshold > 1) {
+        throw new Error('context.relevanceThreshold must be between 0 and 1');
+    }
+    return merged;
+}
+//# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAmBA,MAAM,CAAC,MAAM,cAAc,GAAmB;IAC5C,WAAW,EAAE,kBAAkB;IAC/B,WAAW,EAAE,KAAK;IAClB,iBAAiB,EAAE,GAAG;IACtB,kBAAkB,EAAE,IAAI;IACxB,OAAO,EAAE;QACP,SAAS,EAAE,IAAI;QACf,mBAAmB,EAAE,IAAI;QACzB,YAAY,EAAE,mEAAmE;QACjF,kBAAkB,EAAE,GAAG;QACvB,eAAe,EAAE,KAAK;QACtB,WAAW,EAAE,CAAC,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,SAAS,CAAiB;QAC9E,WAAW,EAAE,EAAE;KAChB;IACD,WAAW,EAAE;QACX,OAAO,EAAE,KAAK;QACd,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI;QAChC,aAAa,EAAE,GAAG;KACnB;IACD,OAAO,EAAE;QACP,OAAO,EAAE,IAAI;QACb,KAAK,EAAE,MAAM;KACd;CACF,CAAC;AAEF,MAAM,UAAU,cAAc,CAAC,MAA+B;IAC5D,MAAM,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IAEhD,IAAI,MAAM,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,IAAI,MAAM,CAAC,kBAAkB,GAAG,CAAC,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAC3D,CAAC;IAED,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,MAAM,CAAC,OAAO,CAAC,kBAAkB,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,kBAAkB,GAAG,CAAC,EAAE,CAAC;QACnF,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;IACxE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/contextBuilder.d.ts

@@ -0,0 +1,16 @@
+import { ContextConfig, ContextResult } from './types.js';
+import { MemoryManager } from './memoryManager.js';
+export declare class ContextBuilder {
+    private memoryManager;
+    private config;
+    constructor(memoryManager: MemoryManager, config?: Partial<ContextConfig>);
+    buildContext(query: string, additionalConfig?: Partial<ContextConfig>): Promise<ContextResult>;
+    private findRelevantMemories;
+    private buildContextContent;
+    private groupByType;
+    private estimateTokens;
+    updateConfig(config: Partial<ContextConfig>): void;
+    getConfig(): Readonly<ContextConfig>;
+}
+export declare function createContextBuilder(memoryManager: MemoryManager, config?: Partial<ContextConfig>): ContextBuilder;
+//# sourceMappingURL=contextBuilder.d.ts.map

package/dist/contextBuilder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextBuilder.d.ts","sourceRoot":"","sources":["../src/contextBuilder.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,aAAa,EACb,aAAa,EAGd,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,qBAAa,cAAc;IACzB,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,MAAM,CAAgB;gBAElB,aAAa,EAAE,aAAa,EAAE,MAAM,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC;IAcnE,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,gBAAgB,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,GAAG,OAAO,CAAC,aAAa,CAAC;YAoCtF,oBAAoB;IAmBlC,OAAO,CAAC,mBAAmB;IAuD3B,OAAO,CAAC,WAAW;IAoBnB,OAAO,CAAC,cAAc;IAWtB,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,GAAG,IAAI;IAIlD,SAAS,IAAI,QAAQ,CAAC,aAAa,CAAC;CAGrC;AAED,wBAAgB,oBAAoB,CAClC,aAAa,EAAE,aAAa,EAC5B,MAAM,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,GAC9B,cAAc,CAEhB"}

package/dist/contextBuilder.js

@@ -0,0 +1,139 @@
+export class ContextBuilder {
+    memoryManager;
+    config;
+    constructor(memoryManager, config) {
+        this.memoryManager = memoryManager;
+        this.config = {
+            maxTokens: 4000,
+            includeSystemPrompt: true,
+            systemPrompt: 'You are a helpful AI assistant with memory of past conversations and user preferences.',
+            relevanceThreshold: 0.7,
+            includeMetadata: false,
+            memoryTypes: ['conversation', 'fact', 'preference', 'context'],
+            maxMemories: 10,
+            ...config,
+        };
+    }
+    async buildContext(query, additionalConfig) {
+        const config = { ...this.config, ...additionalConfig };
+        const messages = [];
+        if (config.includeSystemPrompt) {
+            messages.push({
+                role: 'system',
+                content: config.systemPrompt,
+            });
+        }
+        const relevantMemories = await this.findRelevantMemories(query, config);
+        const contextContent = this.buildContextContent(relevantMemories, config);
+        if (contextContent) {
+            messages.push({
+                role: 'system',
+                content: contextContent,
+            });
+        }
+        const tokensEstimated = this.estimateTokens(messages);
+        const truncationApplied = tokensEstimated > config.maxTokens;
+        return {
+            messages,
+            memories: relevantMemories.map(r => r.memory),
+            stats: {
+                memoriesUsed: relevantMemories.length,
+                tokensEstimated,
+                truncationApplied,
+            },
+        };
+    }
+    async findRelevantMemories(query, config) {
+        const searchQuery = {
+            text: query,
+            filter: {
+                types: config.memoryTypes,
+                minImportance: config.relevanceThreshold,
+            },
+            limit: config.maxMemories,
+            threshold: config.relevanceThreshold,
+        };
+        const results = await this.memoryManager.search(searchQuery);
+        return results.filter(r => r.score >= config.relevanceThreshold);
+    }
+    buildContextContent(memories, config) {
+        if (memories.length === 0)
+            return '';
+        const sections = [];
+        const groupedByType = this.groupByType(memories);
+        if (groupedByType.fact && groupedByType.fact.length > 0) {
+            const facts = groupedByType.fact
+                .map(r => `- ${r.memory.content}`)
+                .join('\n');
+            sections.push(`Key Facts:\n${facts}`);
+        }
+        if (groupedByType.preference && groupedByType.preference.length > 0) {
+            const prefs = groupedByType.preference
+                .map(r => `- ${r.memory.content}`)
+                .join('\n');
+            sections.push(`User Preferences:\n${prefs}`);
+        }
+        if (groupedByType.conversation && groupedByType.conversation.length > 0) {
+            const recentConversations = groupedByType.conversation
+                .slice(0, 5)
+                .map(r => r.memory.content)
+                .join('\n');
+            sections.push(`Recent Conversation Context:\n${recentConversations}`);
+        }
+        if (groupedByType.context && groupedByType.context.length > 0) {
+            const context = groupedByType.context
+                .map(r => `- ${r.memory.content}`)
+                .join('\n');
+            sections.push(`Additional Context:\n${context}`);
+        }
+        if (groupedByType.instruction && groupedByType.instruction.length > 0) {
+            const instructions = groupedByType.instruction
+                .map(r => `- ${r.memory.content}`)
+                .join('\n');
+            sections.push(`Important Instructions:\n${instructions}`);
+        }
+        let content = sections.join('\n\n');
+        if (config.includeMetadata) {
+            const metadataList = memories
+                .map(r => `[${r.memory.metadata.type}] ${r.memory.metadata.tags.join(', ')}`)
+                .join('\n');
+            content += `\n\nMemory Metadata:\n${metadataList}`;
+        }
+        return content;
+    }
+    groupByType(memories) {
+        const grouped = {
+            conversation: [],
+            fact: [],
+            preference: [],
+            instruction: [],
+            context: [],
+            custom: [],
+        };
+        for (const result of memories) {
+            const type = result.memory.metadata.type;
+            if (grouped[type]) {
+                grouped[type].push(result);
+            }
+        }
+        return grouped;
+    }
+    estimateTokens(messages) {
+        let total = 0;
+        for (const message of messages) {
+            const words = message.content.split(/\s+/).length;
+            total += Math.ceil(words * 1.3);
+        }
+        return total;
+    }
+    updateConfig(config) {
+        this.config = { ...this.config, ...config };
+    }
+    getConfig() {
+        return { ...this.config };
+    }
+}
+export function createContextBuilder(memoryManager, config) {
+    return new ContextBuilder(memoryManager, config);
+}
+//# sourceMappingURL=contextBuilder.js.map

package/dist/contextBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextBuilder.js","sourceRoot":"","sources":["../src/contextBuilder.ts"],"names":[],"mappings":"AAUA,MAAM,OAAO,cAAc;IACjB,aAAa,CAAgB;IAC7B,MAAM,CAAgB;IAE9B,YAAY,aAA4B,EAAE,MAA+B;QACvE,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG;YACZ,SAAS,EAAE,IAAI;YACf,mBAAmB,EAAE,IAAI;YACzB,YAAY,EAAE,wFAAwF;YACtG,kBAAkB,EAAE,GAAG;YACvB,eAAe,EAAE,KAAK;YACtB,WAAW,EAAE,CAAC,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,SAAS,CAAC;YAC9D,WAAW,EAAE,EAAE;YACf,GAAG,MAAM;SACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAa,EAAE,gBAAyC;QACzE,MAAM,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAqB,EAAE,CAAC;QAEtC,IAAI,MAAM,CAAC,mBAAmB,EAAE,CAAC;YAC/B,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,MAAM,CAAC,YAAa;aAC9B,CAAC,CAAC;QACL,CAAC;QAED,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAExE,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;QAE1E,IAAI,cAAc,EAAE,CAAC;YACnB,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,cAAc;aACxB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,iBAAiB,GAAG,eAAe,GAAG,MAAM,CAAC,SAAS,CAAC;QAE7D,OAAO;YACL,QAAQ;YACR,QAAQ,EAAE,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;YAC7C,KAAK,EAAE;gBACL,YAAY,EAAE,gBAAgB,CAAC,MAAM;gBACrC,eAAe;gBACf,iBAAiB;aAClB;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,KAAa,EACb,MAAqB;QAErB,MAAM,WAAW,GAAgB;YAC/B,IAAI,EAAE,KAAK;YACX,MAAM,EAAE;gBACN,KAAK,EAAE,MAAM,CAAC,WAAW;gBACzB,aAAa,EAAE,MAAM,CAAC,kBAAkB;aACzC;YACD,KAAK,EAAE,MAAM,CAAC,WAAW;YACzB,SAAS,EAAE,MAAM,CAAC,kBAAkB;SACrC,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAE7D,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,MAAM,CAAC,kBAAkB,CAAC,CAAC;IACnE,CAAC;IAEO,mBAAmB,CAAC,QAAwB,EAAE,MAAqB;QACzE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAErC,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAEjD,IAAI,aAAa,CAAC,IAAI,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI;iBAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;iBACjC,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,QAAQ,CAAC,IAAI,CAAC,eAAe,KAAK,EAAE,CAAC,CAAC;QACxC,CAAC;QAED,IAAI,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpE,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU;iBACnC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;iBACjC,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,QAAQ,CAAC,IAAI,CAAC,sBAAsB,KAAK,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,aAAa,CAAC,YAAY,IAAI,aAAa,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxE,MAAM,mBAAmB,GAAG,aAAa,CAAC,YAAY;iBACnD,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;iBACX,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;iBAC1B,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,QAAQ,CAAC,IAAI,CAAC,iCAAiC,mBAAmB,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,aAAa,CAAC,OAAO,IAAI,aAAa,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9D,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO;iBAClC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;iBACjC,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,QAAQ,CAAC,IAAI,CAAC,wBAAwB,OAAO,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,aAAa,CAAC,WAAW,IAAI,aAAa,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtE,MAAM,YAAY,GAAG,aAAa,CAAC,WAAW;iBAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;iBACjC,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,QAAQ,CAAC,IAAI,CAAC,4BAA4B,YAAY,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEpC,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YAC3B,MAAM,YAAY,GAAG,QAAQ;iBAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;iBAC5E,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,OAAO,IAAI,yBAAyB,YAAY,EAAE,CAAC;QACrD,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,WAAW,CAAC,QAAwB;QAC1C,MAAM,OAAO,GAAuC;YAClD,YAAY,EAAE,EAAE;YAChB,IAAI,EAAE,EAAE;YACR,UAAU,EAAE,EAAE;YACd,WAAW,EAAE,EAAE;YACf,OAAO,EAAE,EAAE;YACX,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;YACzC,IAAI,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,cAAc,CAAC,QAA0B;QAC/C,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;YAClD,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC;QAClC,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,YAAY,CAAC,MAA8B;QACzC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;IAC9C,CAAC;IAED,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;CACF;AAED,MAAM,UAAU,oBAAoB,CAClC,aAA4B,EAC5B,MAA+B;IAE/B,OAAO,IAAI,cAAc,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;AACnD,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,45 @@
+import { Memory, MemoryMetadata, SearchResult, ContextResult, MemoryStats, EmbeddingFunction } from './types';
+import { AIMemoryConfig } from './config';
+import { MemoryManager } from './memoryManager';
+import { ContextBuilder } from './contextBuilder';
+export { Memory, MemoryMetadata, MemoryType, SearchResult, MemoryQuery, MemoryFilter, MemoryStats, ContextConfig, ContextMessage, ContextResult, EmbeddingFunction, StorageAdapter, VectorStoreAdapter, VectorEntry, SearchOptions, } from './types';
+export { AIMemoryConfig, DEFAULT_CONFIG, validateConfig, } from './config';
+export { InMemoryVectorStore, VectorStoreFactory, } from './vectorStore';
+export { MetadataStore, } from './metadataStore';
+export { MemoryManager, createMemoryManager, } from './memoryManager';
+export { ContextBuilder, createContextBuilder, } from './contextBuilder';
+export { RateLimiter, RateLimitResult, RateLimitConfig, RateLimitStore, InMemoryRateLimitStore, createRateLimiter, DEFAULT_RATE_LIMITS, } from './rateLimiter';
+export { UsageTracker, UsageLimits, UsageCheckResult, UsageMetric, UsageSummary, UsageStore, InMemoryUsageStore, createUsageTracker, DEFAULT_USAGE_LIMITS, } from './usageTracker';
+export { PlanManager, PlanManager as AIMemoryPro, Plan, ExtensionPlan, TeamPlan, ProductPlan, ProductType, PlanConfig, PlanFeatures, PlanLimits, PLANS, getPlan, getPlanLimits, getAvailablePlans, createPlanManager, } from './planManager';
+export { RedisRateLimitStore, PostgresUsageStore, RedisConfig, PostgresConfig, createRedisRateLimiter, createPostgresUsageStore, } from './storage';
+export { LicenseManager, AIMemoryLicense, LicenseConfig, LicenseValidation, createLicenseManager, getLicenseManager, } from './licensing';
+export declare class AIMemory {
+    private memoryManager;
+    private contextBuilder;
+    constructor(config?: Partial<AIMemoryConfig>);
+    setEmbeddingFunction(fn: EmbeddingFunction): void;
+    remember(content: string, metadata?: Partial<MemoryMetadata>): Promise<Memory>;
+    rememberConversation(content: string, role: 'user' | 'assistant', metadata?: Partial<MemoryMetadata>): Promise<Memory>;
+    rememberFact(content: string, metadata?: Partial<MemoryMetadata>): Promise<Memory>;
+    rememberPreference(content: string, metadata?: Partial<MemoryMetadata>): Promise<Memory>;
+    rememberInstruction(content: string, metadata?: Partial<MemoryMetadata>): Promise<Memory>;
+    recall(query: string, options?: {
+        limit?: number;
+        threshold?: number;
+    }): Promise<SearchResult[]>;
+    getContext(query: string): Promise<ContextResult>;
+    getMemory(id: string): Promise<Memory | null>;
+    updateMemory(id: string, updates: Partial<Memory>): Promise<Memory | null>;
+    forget(id: string): Promise<void>;
+    getRecentMemories(limit?: number): Promise<Memory[]>;
+    getStats(): Promise<MemoryStats>;
+    clear(): Promise<void>;
+    cleanup(): Promise<number>;
+    exportData(): Promise<Memory[]>;
+    importData(memories: Memory[]): Promise<void>;
+    getMemoryManager(): MemoryManager;
+    getContextBuilder(): ContextBuilder;
+}
+export declare function createAIMemory(config?: Partial<AIMemoryConfig>): AIMemory;
+export { createAIMemory as createAIBrain };
+//# 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,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAC9G,OAAO,EAAE,cAAc,EAAkB,MAAM,UAAU,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,EACL,MAAM,EACN,cAAc,EACd,UAAU,EACV,YAAY,EACZ,WAAW,EACX,YAAY,EACZ,WAAW,EACX,aAAa,EACb,cAAc,EACd,aAAa,EACb,iBAAiB,EACjB,cAAc,EACd,kBAAkB,EAClB,WAAW,EACX,aAAa,GACd,MAAM,SAAS,CAAC;AAEjB,OAAO,EACL,cAAc,EACd,cAAc,EACd,cAAc,GACf,MAAM,UAAU,CAAC;AAElB,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,aAAa,GACd,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EACL,aAAa,EACb,mBAAmB,GACpB,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EACL,cAAc,EACd,oBAAoB,GACrB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EACL,WAAW,EACX,eAAe,EACf,eAAe,EACf,cAAc,EACd,sBAAsB,EACtB,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,YAAY,EACZ,WAAW,EACX,gBAAgB,EAChB,WAAW,EACX,YAAY,EACZ,UAAU,EACV,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,GACrB,MAAM,gBAAgB,CAAC;AAExB,OAAO,EACL,WAAW,EACX,WAAW,IAAI,WAAW,EAC1B,IAAI,EACJ,aAAa,EACb,QAAQ,EACR,WAAW,EACX,WAAW,EACX,UAAU,EACV,YAAY,EACZ,UAAU,EACV,KAAK,EACL,OAAO,EACP,aAAa,EACb,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,WAAW,EACX,cAAc,EACd,sBAAsB,EACtB,wBAAwB,GACzB,MAAM,WAAW,CAAC;AAEnB,OAAO,EACL,cAAc,EACd,eAAe,EACf,aAAa,EACb,iBAAiB,EACjB,oBAAoB,EACpB,iBAAiB,GAClB,MAAM,aAAa,CAAC;AAErB,qBAAa,QAAQ;IACnB,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,cAAc,CAAiB;gBAE3B,MAAM,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC;IAS5C,oBAAoB,CAAC,EAAE,EAAE,iBAAiB,GAAG,IAAI;IAI3C,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAI9E,oBAAoB,CACxB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,GAAG,WAAW,EAC1B,QAAQ,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GACjC,OAAO,CAAC,MAAM,CAAC;IAIZ,YAAY,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAIlF,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAIxF,mBAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAIzF,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAQhG,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAIjD,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAI7C,YAAY,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAI1E,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjC,iBAAiB,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAIpD,QAAQ,IAAI,OAAO,CAAC,WAAW,CAAC;IAIhC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAItB,OAAO,IAAI,OAAO,CAAC,MAAM,CAAC;IAI1B,UAAU,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAI/B,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAInD,gBAAgB,IAAI,aAAa;IAIjC,iBAAiB,IAAI,cAAc;CAGpC;AAED,wBAAgB,cAAc,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,QAAQ,CAEzE;AAED,OAAO,EAAE,cAAc,IAAI,aAAa,EAAE,CAAC"}

package/dist/index.js

@@ -0,0 +1,88 @@
+import { validateConfig } from './config';
+import { MemoryManager } from './memoryManager';
+import { ContextBuilder } from './contextBuilder';
+export { DEFAULT_CONFIG, validateConfig, } from './config';
+export { InMemoryVectorStore, VectorStoreFactory, } from './vectorStore';
+export { MetadataStore, } from './metadataStore';
+export { MemoryManager, createMemoryManager, } from './memoryManager';
+export { ContextBuilder, createContextBuilder, } from './contextBuilder';
+export { RateLimiter, InMemoryRateLimitStore, createRateLimiter, DEFAULT_RATE_LIMITS, } from './rateLimiter';
+export { UsageTracker, InMemoryUsageStore, createUsageTracker, DEFAULT_USAGE_LIMITS, } from './usageTracker';
+export { PlanManager, PlanManager as AIMemoryPro, PLANS, getPlan, getPlanLimits, getAvailablePlans, createPlanManager, } from './planManager';
+export { RedisRateLimitStore, PostgresUsageStore, createRedisRateLimiter, createPostgresUsageStore, } from './storage';
+export { LicenseManager, createLicenseManager, getLicenseManager, } from './licensing';
+export class AIMemory {
+    memoryManager;
+    contextBuilder;
+    constructor(config) {
+        const validatedConfig = validateConfig(config || {});
+        this.memoryManager = new MemoryManager(validatedConfig);
+        this.contextBuilder = new ContextBuilder(this.memoryManager, validatedConfig.context);
+    }
+    setEmbeddingFunction(fn) {
+        this.memoryManager.setEmbeddingFunction(fn);
+    }
+    async remember(content, metadata) {
+        return this.memoryManager.addMemory(content, metadata);
+    }
+    async rememberConversation(content, role, metadata) {
+        return this.memoryManager.addConversation(content, role, metadata);
+    }
+    async rememberFact(content, metadata) {
+        return this.memoryManager.addFact(content, metadata);
+    }
+    async rememberPreference(content, metadata) {
+        return this.memoryManager.addPreference(content, metadata);
+    }
+    async rememberInstruction(content, metadata) {
+        return this.memoryManager.addInstruction(content, metadata);
+    }
+    async recall(query, options) {
+        return this.memoryManager.search({
+            text: query,
+            limit: options?.limit || 10,
+            threshold: options?.threshold,
+        });
+    }
+    async getContext(query) {
+        return this.contextBuilder.buildContext(query);
+    }
+    async getMemory(id) {
+        return this.memoryManager.getMemory(id);
+    }
+    async updateMemory(id, updates) {
+        return this.memoryManager.updateMemory(id, updates);
+    }
+    async forget(id) {
+        return this.memoryManager.deleteMemory(id);
+    }
+    async getRecentMemories(limit) {
+        return this.memoryManager.getRecentMemories(limit);
+    }
+    async getStats() {
+        return this.memoryManager.getStats();
+    }
+    async clear() {
+        return this.memoryManager.clear();
+    }
+    async cleanup() {
+        return this.memoryManager.cleanup();
+    }
+    async exportData() {
+        return this.memoryManager.exportData();
+    }
+    async importData(memories) {
+        return this.memoryManager.importData(memories);
+    }
+    getMemoryManager() {
+        return this.memoryManager;
+    }
+    getContextBuilder() {
+        return this.contextBuilder;
+    }
+}
+export function createAIMemory(config) {
+    return new AIMemory(config);
+}
+export { createAIMemory as createAIBrain };
+//# sourceMappingURL=index.js.map