@yun-zero/claw-memory 0.1.0

package/src/tools/memory.ts

@@ -0,0 +1,137 @@
+import type { Database } from 'better-sqlite3';
+import { v4 as uuidv4 } from 'uuid';
+import * as path from 'path';
+import * as fs from 'fs';
+import { homedir } from 'os';
+
+export function registerMemoryTools(
+  tools: any,
+  db: Database,
+  dataDir: string
+) {
+  // 解析 dataDir 中的 ~
+  const resolvedDataDir = dataDir.replace(/^~/, homedir());
+
+  // memory_save tool
+  tools.register({
+    name: 'memory_save',
+    description: 'Save conversation to memory with metadata',
+    schema: {
+      type: 'object',
+      properties: {
+        content: { type: 'string', description: 'Conversation content' },
+        metadata: {
+          type: 'object',
+          properties: {
+            tags: { type: 'array', items: { type: 'string' } },
+            keywords: { type: 'array', items: { type: 'string' } },
+            importance: { type: 'number', minimum: 0, maximum: 1 }
+          }
+        }
+      },
+      required: ['content']
+    },
+    handler: async (params: any) => {
+      const { content, metadata = {} } = params;
+      const memoryId = uuidv4();
+
+      const memoriesDir = path.join(resolvedDataDir, 'memories');
+      if (!fs.existsSync(memoriesDir)) {
+        fs.mkdirSync(memoriesDir, { recursive: true });
+      }
+
+      const contentPath = path.join(memoriesDir, `${memoryId}.md`);
+      fs.writeFileSync(contentPath, content, 'utf-8');
+
+      db.prepare(`
+        INSERT INTO memories (id, content_path, summary, importance, created_at)
+        VALUES (?, ?, ?, ?, datetime('now'))
+      `).run(
+        memoryId,
+        contentPath,
+        metadata.summary || content.substring(0, 200),
+        metadata.importance || 0.5
+      );
+
+      return { success: true, memory_id: memoryId };
+    }
+  });
+
+  // memory_search tool
+  tools.register({
+    name: 'memory_search',
+    description: 'Search memories by query',
+    schema: {
+      type: 'object',
+      properties: {
+        query: { type: 'string' },
+        limit: { type: 'number', default: 10 }
+      },
+      required: ['query']
+    },
+    handler: async (params: any) => {
+      const { query, limit = 10 } = params;
+      const memories = db.prepare(`
+        SELECT id, summary, importance, created_at
+        FROM memories
+        WHERE summary LIKE ?
+        ORDER BY importance DESC
+        LIMIT ?
+      `).all(`%${query}%`, limit);
+
+      return { memories };
+    }
+  });
+
+  // memory_summary tool
+  tools.register({
+    name: 'memory_summary',
+    description: 'Get memory summary for time period',
+    schema: {
+      type: 'object',
+      properties: {
+        period: { type: 'string', enum: ['day', 'week', 'month'] },
+        date: { type: 'string' }
+      },
+      required: ['period']
+    },
+    handler: async (params: any) => {
+      const { period } = params;
+      const today = new Date();
+      let startDate: Date;
+
+      switch (period) {
+        case 'day':
+          startDate = today;
+          break;
+        case 'week':
+          startDate = new Date(today);
+          startDate.setDate(today.getDate() - today.getDay());
+          break;
+        case 'month':
+          startDate = new Date(today.getFullYear(), today.getMonth(), 1);
+          break;
+        default:
+          startDate = today;
+      }
+
+      const startDateStr = startDate.toISOString().split('T')[0];
+
+      const memories = db.prepare(`
+        SELECT summary, importance, created_at
+        FROM memories
+        WHERE date(created_at) >= date(?)
+        ORDER BY importance DESC
+        LIMIT 20
+      `).all(startDateStr);
+
+      const count = Array.isArray(memories) ? memories.length : 0;
+
+      return {
+        period,
+        count,
+        memories: Array.isArray(memories) ? memories.slice(0, 10) : []
+      };
+    }
+  });
+}

package/src/types.ts

@@ -0,0 +1,139 @@
+/**
+ * Claw-Memory Type Definitions
+ * Lightweight AI memory system for OpenClaw and Claude Code
+ */
+
+export interface IntegratedSummary {
+  active_areas: string[];
+  key_topics: string[];
+  recent_summary: string;
+}
+
+export interface Memory {
+  id: string;
+  contentPath: string;
+  summary: string | null;
+  integratedSummary: IntegratedSummary | null;
+  createdAt: Date;
+  updatedAt: Date;
+  tokenCount: number;
+  importance: number;
+  accessCount: number;
+  lastAccessedAt: Date | null;
+  isArchived: boolean;
+  isDuplicate: boolean;
+  duplicateOf: string | null;
+}
+
+export interface Entity {
+  id: string;
+  name: string;
+  type: 'keyword' | 'tag' | 'subject' | 'person' | 'project';
+  parentId: string | null;
+  level: number;
+  embedding: Buffer | null;
+  metadata: Record<string, unknown> | null;
+  createdAt: Date;
+}
+
+export interface MemoryEntity {
+  memoryId: string;
+  entityId: string;
+  relevance: number;
+  source: 'auto' | 'manual';
+  createdAt: Date;
+}
+
+export interface EntityRelation {
+  id: string;
+  sourceId: string;
+  targetId: string;
+  relationType: 'related' | 'parent' | 'similar' | 'co_occur';
+  weight: number;
+  evidenceCount: number;
+  createdAt: Date;
+}
+
+export interface TimeBucket {
+  date: string; // YYYY-MM-DD
+  memoryCount: number;
+  summary: string | null;
+  summaryGeneratedAt: Date | null;
+  keyTopics: string[] | null;
+  createdAt: Date;
+}
+
+export interface SaveMemoryInput {
+  content: string;
+  metadata: {
+    tags?: string[];
+    subjects?: string[];
+    keywords?: string[];
+    importance?: number;
+    summary?: string;
+  };
+  userId?: string;
+}
+
+export interface SearchMemoryInput {
+  query: string;
+  timeRange?: 'today' | 'week' | 'month' | 'year' | 'all';
+  tags?: string[];
+  limit?: number;
+  maxTokens?: number;
+}
+
+export interface GetContextInput {
+  query: string;
+  maxTokens?: number;
+}
+
+export interface GetSummaryInput {
+  period: 'day' | 'week' | 'month';
+  date?: string;
+}
+
+export interface WeeklyReport {
+  // 1. 基础统计
+  period: { start: string; end: string };
+  basic: {
+    totalMemories: number;
+    totalTokens: number;
+    avgImportance: number;
+  };
+
+  // 2. 标签分布维度
+  tags: {
+    topTags: { name: string; count: number }[];
+    tagDistribution: Record<string, number>;
+  };
+
+  // 3. 主题/关键词维度
+  topics: {
+    keywords: { word: string; count: number }[];
+    keyTopics: string[];
+  };
+
+  // 4. 重要性维度
+  importance: {
+    highPriority: Pick<Memory, 'id' | 'summary' | 'importance'>[];
+    mediumPriority: Pick<Memory, 'id' | 'summary' | 'importance'>[];
+    lowPriority: Pick<Memory, 'id' | 'summary' | 'importance'>[];
+  };
+
+  // 5. 访问模式维度
+  access: {
+    mostAccessed: Pick<Memory, 'id' | 'summary' | 'accessCount'>[];
+    recentlyCreated: Pick<Memory, 'id' | 'summary' | 'createdAt'>[];
+    recentlyAccessed: Pick<Memory, 'id' | 'summary' | 'lastAccessedAt'>[];
+  };
+
+  // 6. 实体关系维度
+  entities: {
+    relatedGroups: { entity: string; related: string[] }[];
+    coOccurringTags: [string, string][];
+  };
+
+  // 7. LLM 智能总结
+  summary: string;
+}

package/tsconfig.json

@@ -0,0 +1,20 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "lib": ["ES2022"],
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true,
+    "resolveJsonModule": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist", "tests"]
+}

package/vitest.config.ts

@@ -0,0 +1,16 @@
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    globals: true,
+    environment: 'node',
+    include: ['test/**/*.test.ts'],
+    testTimeout: 60000,
+    pool: 'forks',
+    poolOptions: {
+      forks: {
+        singleFork: true,
+      },
+    },
+  },
+});