sigma-memory 0.1.0

package/src/qmd.ts

@@ -0,0 +1,180 @@
+import { execSync } from 'child_process';
+import type { MemoryConfig, SearchResult } from './types.js';
+
+export class QMDManager {
+  private config: MemoryConfig;
+  private qmdCommand: string;
+
+  constructor(config: MemoryConfig) {
+    this.config = config;
+    this.qmdCommand = config.qmdCommand || 'qmd';
+  }
+
+  /**
+   * Retourne true si QMD est installé et fonctionnel
+   */
+  isAvailable(): boolean {
+    if (!this.config.qmdEnabled) {
+      return false;
+    }
+
+    try {
+      // Teste si le binaire QMD est accessible
+      execSync(`${this.qmdCommand} --version`, { 
+        stdio: 'ignore',
+        timeout: 5000 
+      });
+      return true;
+    } catch (error) {
+      return false;
+    }
+  }
+
+  /**
+   * Lance `qmd query` et parse les résultats
+   */
+  async search(query: string, maxResults = 10): Promise<SearchResult[]> {
+    if (!this.isAvailable()) {
+      return [];
+    }
+
+    try {
+      const output = execSync(
+        `${this.qmdCommand} query "${query.replace(/"/g, '\\"')}" --limit ${maxResults} --format json`,
+        { 
+          encoding: 'utf8',
+          timeout: 30000,
+          cwd: this.config.memoryDir
+        }
+      );
+
+      if (!output.trim()) {
+        return [];
+      }
+
+      const result = JSON.parse(output);
+      
+      // Le format attendu est { results: [{file, line, content, score}] }
+      if (result.results && Array.isArray(result.results)) {
+        return result.results.map((item: any) => ({
+          file: item.file || '',
+          line: item.line || 0,
+          content: item.content || '',
+          score: item.score || 0
+        }));
+      }
+
+      return [];
+    } catch (error) {
+      console.error('QMD search error:', error);
+      return [];
+    }
+  }
+
+  /**
+   * Lance `qmd update` pour reindexer
+   */
+  async update(): Promise<boolean> {
+    if (!this.isAvailable()) {
+      return false;
+    }
+
+    try {
+      execSync(`${this.qmdCommand} update`, {
+        stdio: 'ignore',
+        timeout: 60000,
+        cwd: this.config.memoryDir
+      });
+      return true;
+    } catch (error) {
+      console.error('QMD update error:', error);
+      return false;
+    }
+  }
+
+  /**
+   * Lance `qmd embed` pour recalculer les embeddings
+   */
+  async embed(): Promise<boolean> {
+    if (!this.isAvailable()) {
+      return false;
+    }
+
+    try {
+      execSync(`${this.qmdCommand} embed`, {
+        stdio: 'ignore',
+        timeout: 300000, // 5 minutes pour les embeddings
+        cwd: this.config.memoryDir
+      });
+      return true;
+    } catch (error) {
+      console.error('QMD embed error:', error);
+      return false;
+    }
+  }
+
+  /**
+   * Liste les collections QMD
+   */
+  async collections(): Promise<string[]> {
+    if (!this.isAvailable()) {
+      return [];
+    }
+
+    try {
+      const output = execSync(`${this.qmdCommand} collections --format json`, {
+        encoding: 'utf8',
+        timeout: 10000,
+        cwd: this.config.memoryDir
+      });
+
+      if (!output.trim()) {
+        return [];
+      }
+
+      const result = JSON.parse(output);
+      
+      // Retourne la liste des noms de collections
+      if (result.collections && Array.isArray(result.collections)) {
+        return result.collections.map((col: any) => col.name || col);
+      }
+
+      return [];
+    } catch (error) {
+      console.error('QMD collections error:', error);
+      return [];
+    }
+  }
+
+  /**
+   * Retourne le statut (nb fichiers, nb chunks, dernière mise à jour)
+   */
+  async status(): Promise<{ files: number; chunks: number; lastUpdate: string | null } | null> {
+    if (!this.isAvailable()) {
+      return null;
+    }
+
+    try {
+      const output = execSync(`${this.qmdCommand} status --format json`, {
+        encoding: 'utf8',
+        timeout: 10000,
+        cwd: this.config.memoryDir
+      });
+
+      if (!output.trim()) {
+        return { files: 0, chunks: 0, lastUpdate: null };
+      }
+
+      const result = JSON.parse(output);
+      
+      return {
+        files: result.files || 0,
+        chunks: result.chunks || 0,
+        lastUpdate: result.lastUpdate || null
+      };
+    } catch (error) {
+      console.error('QMD status error:', error);
+      return { files: 0, chunks: 0, lastUpdate: null };
+    }
+  }
+}

package/src/types.ts

@@ -0,0 +1,96 @@
+export interface MemoryConfig {
+  memoryDir: string;        // ~/.phi/memory/
+  projectMemoryDir: string; // .phi/memory/ (dans le projet courant)
+  ontologyPath: string;     // ~/.phi/memory/ontology/graph.jsonl
+  qmdEnabled: boolean;
+  qmdCommand?: string;      // Chemin vers le binaire QMD
+}
+
+export interface SearchResult {
+  file: string;
+  line: number;
+  content: string;
+  score: number;
+}
+
+export interface OntologyEntity {
+  id: string;
+  type: 'Person' | 'Project' | 'Device' | 'Account' | 'Document' | 'Service' | 'Concept';
+  name: string;
+  properties: Record<string, string>;
+  createdAt: string;
+  updatedAt: string;
+}
+
+export interface OntologyRelation {
+  id: string;
+  from: string;      // entity ID
+  to: string;        // entity ID
+  type: string;      // 'owns' | 'uses' | 'deploys' | 'manages' | 'depends_on' | etc.
+  properties: Record<string, string>;
+  createdAt: string;
+}
+
+export interface Note {
+  file: string;
+  date: string;
+  content: string;
+}
+
+export interface UnifiedSearchResult {
+  source: 'notes' | 'ontology' | 'qmd';
+  type?: 'entity' | 'relation' | 'note' | 'file';
+  score: number;
+  data: any;
+}
+
+export interface MemoryStatus {
+  notes: {
+    count: number;
+    totalSize: number;
+    lastModified: string | null;
+  };
+  ontology: {
+    entities: number;
+    relations: number;
+    entitiesByType: Record<string, number>;
+    relationsByType: Record<string, number>;
+  };
+  qmd: {
+    available: boolean;
+    status?: {
+      files: number;
+      chunks: number;
+      lastUpdate: string | null;
+    };
+  };
+}
+
+// Types pour les entrées JSONL de l'ontologie
+export interface OntologyEntityEntry {
+  kind: 'entity';
+  id: string;
+  type: 'Person' | 'Project' | 'Device' | 'Account' | 'Document' | 'Service' | 'Concept';
+  name: string;
+  properties: Record<string, string>;
+  createdAt: string;
+  updatedAt: string;
+}
+
+export interface OntologyRelationEntry {
+  kind: 'relation';
+  id: string;
+  from: string;
+  to: string;
+  type: string;
+  properties: Record<string, string>;
+  createdAt: string;
+}
+
+export interface OntologyDeleteEntry {
+  kind: 'delete';
+  targetId: string;
+  deletedAt: string;
+}
+
+export type OntologyJSONLEntry = OntologyEntityEntry | OntologyRelationEntry | OntologyDeleteEntry;