ocerebro 0.3.1 → 0.3.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ocerebro",
-  "version": "0.3.1",
+  "version": "0.3.3",
   "description": "OCerebro - Sistema de Memoria para Agentes (Claude Code/MCP)",
   "main": "bin/ocerebro.js",
   "bin": {

package/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ocerebro"
-version = "0.3.1"
+version = "0.3.3"
 description = "OCerebro - Sistema de Memoria para Agentes (Claude Code/MCP)"
 readme = "README.md"
 requires-python = ">=3.10"

package/src/index/entities_db.py

@@ -50,6 +50,7 @@ class EntitiesDB:
                 memory_id TEXT,
                 entity_name TEXT,
                 entity_type TEXT,
+                source TEXT DEFAULT 'content',
                 confidence REAL DEFAULT 1.0,
                 span_start INTEGER,
                 span_end INTEGER,
@@ -75,6 +76,11 @@ class EntitiesDB:
             ON entities(memory_id)
         """)
 
+        conn.execute("""
+            CREATE INDEX IF NOT EXISTS idx_entities_source
+            ON entities(source)
+        """)
+
         # Tabela de cache de hash (para evitar reprocessamento)
         conn.execute("""
             CREATE TABLE IF NOT EXISTS entity_cache (
@@ -128,7 +134,8 @@ class EntitiesDB:
         confidence: float = 1.0,
         span_start: int = 0,
         span_end: int = 0,
-        context_snippet: str = ""
+        context_snippet: str = "",
+        source: str = "content"
     ) -> str:
         """
         Insere uma entidade.
@@ -141,6 +148,7 @@ class EntitiesDB:
             span_start: Posição inicial no texto
             span_end: Posição final no texto
             context_snippet: Contexto ao redor da entidade
+            source: Origem da entidade ('frontmatter' ou 'content')
 
         Returns:
             ID da entidade
@@ -150,13 +158,14 @@ class EntitiesDB:
         conn = self._connect()
         conn.execute("""
             INSERT OR REPLACE INTO entities
-            (id, memory_id, entity_name, entity_type, confidence, span_start, span_end, context_snippet)
-            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+            (id, memory_id, entity_name, entity_type, source, confidence, span_start, span_end, context_snippet)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
         """, (
             entity_id,
             memory_id,
             entity_name,
             entity_type,
+            source,
             confidence,
             span_start,
             span_end,
@@ -207,7 +216,7 @@ class EntitiesDB:
 
     def delete_entities_by_memory(self, memory_id: str) -> int:
         """
-        Remove entidades de uma memória.
+        Remove todas as entidades de uma memória.
 
         Args:
             memory_id: ID da memória
@@ -232,6 +241,27 @@ class EntitiesDB:
         conn.close()
         return deleted
 
+    def delete_entities_by_source(self, memory_id: str, source: str) -> int:
+        """
+        Remove entidades de uma memória por fonte (frontmatter ou content).
+
+        Args:
+            memory_id: ID da memória
+            source: Fonte das entidades ('frontmatter' ou 'content')
+
+        Returns:
+            Número de entidades removidas
+        """
+        conn = self._connect()
+        cursor = conn.execute(
+            "DELETE FROM entities WHERE memory_id = ? AND source = ?",
+            (memory_id, source)
+        )
+        deleted = cursor.rowcount
+        conn.commit()
+        conn.close()
+        return deleted
+
     # ========================================================================
     # OPERAÇÕES DE RELACIONAMENTOS
     # ========================================================================
@@ -521,7 +551,8 @@ class EntitiesDB:
                 memory_id,
                 f"TYPE:{frontmatter['type']}",
                 "META",
-                confidence=1.0
+                confidence=1.0,
+                source="frontmatter"
             )
             entity_ids.append(eid)
 
@@ -531,7 +562,8 @@ class EntitiesDB:
                 memory_id,
                 project,
                 "PROJECT",
-                confidence=1.0
+                confidence=1.0,
+                source="frontmatter"
             )
             entity_ids.append(eid)
 
@@ -544,7 +576,8 @@ class EntitiesDB:
                         memory_id,
                         f"TAG:{tag}",
                         "TAG",
-                        confidence=1.0
+                        confidence=1.0,
+                        source="frontmatter"
                     )
                     entity_ids.append(eid)
             elif isinstance(tags, list):
@@ -553,7 +586,8 @@ class EntitiesDB:
                         memory_id,
                         f"TAG:{tag}",
                         "TAG",
-                        confidence=1.0
+                        confidence=1.0,
+                        source="frontmatter"
                     )
                     entity_ids.append(eid)
 
@@ -580,14 +614,18 @@ class EntitiesDB:
         cached_hash = self.get_cached_hash(memory_id)
         current_hash = self._compute_hash(content)
 
+        skip_delete = False
         if cached_hash == current_hash:
             # Conteúdo igual, verifica se já tem entidades
             existing = self.get_entities_by_memory(memory_id)
             if existing:
                 return []  # Já processado, sem mudanças
+            # Cache bate mas entidades vazias (ex: banco limpo) - reprocessa sem delete
+            skip_delete = True
 
-        # Conteúdo novo ou mudou - remove entidades antigas e reprocessa
-        self.delete_entities_by_memory(memory_id)
+        # Conteúdo novo ou mudou - remove apenas entidades de conteúdo e reprocessa
+        if not skip_delete:
+            self.delete_entities_by_source(memory_id, "content")
 
         entity_ids = []
 

package/src/mcp/server.py

@@ -693,6 +693,7 @@ Uma chamada por memória. O sistema salva e indexa automaticamente.
     def _capture_memory(self, args: Dict[str, Any]) -> str:
         """Salva uma memória no diretório nativo do Claude Code."""
         import re
+        import yaml
         from datetime import datetime
         from src.core.paths import get_memory_index
 
@@ -711,15 +712,36 @@ Uma chamada por memória. O sistema salva e indexa automaticamente.
         file_path = mem_dir / f"{mem_name}.md"
         file_path.write_text(content, encoding="utf-8")
 
-        desc_match = re.search(r'description:\s*(.*)', content)
-        type_match = re.search(r'type:\s*(.*)', content)
-        project_match = re.search(r'project:\s*(.*)', content)
-        tags_match = re.search(r'tags:\s*(.*)', content)
-
-        desc = desc_match.group(1).strip() if desc_match else "sem descrição"
-        m_type = type_match.group(1).strip() if type_match else "project"
-        project = project_match.group(1).strip() if project_match else "unknown"
-        tags = tags_match.group(1).strip() if tags_match else ""
+        # Parse frontmatter uma única vez com yaml.safe_load
+        frontmatter_match = re.match(r'^---\n(.*?)\n---\n(.*)$', content, re.DOTALL)
+        frontmatter = {}
+        body_content = ""
+        if frontmatter_match:
+            try:
+                frontmatter = yaml.safe_load(frontmatter_match.group(1)) or {}
+                body_content = frontmatter_match.group(2)
+            except Exception:
+                pass  # Fallback para regex se yaml falhar
+
+        # Extrai variáveis do frontmatter parseado (fallback para regex se necessário)
+        m_type = frontmatter.get('type', '')
+        project = frontmatter.get('project', '')
+        tags = frontmatter.get('tags', '')
+        desc = frontmatter.get('description', '')
+
+        # Fallback via regex se frontmatter parsing falhou
+        if not m_type:
+            type_match = re.search(r'type:\s*(.*)', content)
+            m_type = type_match.group(1).strip() if type_match else "project"
+        if not project:
+            project_match = re.search(r'project:\s*(.*)', content)
+            project = project_match.group(1).strip() if project_match else "unknown"
+        if not tags:
+            tags_match = re.search(r'tags:\s*(.*)', content)
+            tags = tags_match.group(1).strip() if tags_match else ""
+        if not desc:
+            desc_match = re.search(r'description:\s*(.*)', content)
+            desc = desc_match.group(1).strip() if desc_match else "sem descrição"
 
         ts = datetime.now().strftime("%Y-%m-%d")
         entry = f"- [{m_type}] {mem_name}.md ({ts}): {desc}\n"
@@ -733,28 +755,25 @@ Uma chamada por memória. O sistema salva e indexa automaticamente.
         else:
             index_path.write_text(f"# Memórias do Projeto\n\n{entry}", encoding="utf-8")
 
-        # BUG FIX: Registrar entidades no grafo (frontmatter + conteúdo)
-        if self.entities_db:
-            import yaml
-            frontmatter_match = re.match(r'^---\n(.*?)\n---\n(.*)$', content, re.DOTALL)
-            if frontmatter_match:
-                try:
-                    frontmatter = yaml.safe_load(frontmatter_match.group(1))
-                    # Extrai entidades do frontmatter
-                    self.entities_db.extract_from_frontmatter(
-                        memory_id=mem_name,
-                        frontmatter=frontmatter or {},
-                        project=project
-                    )
-                    # Extrai entidades do conteúdo (spaCy NER)
-                    body_content = frontmatter_match.group(2)
-                    self.entities_db.extract_from_content(
-                        memory_id=mem_name,
-                        content=body_content,
-                        use_spacy=True
-                    )
-                except Exception as e:
-                    pass  # Falha silenciosa se frontmatter inválido
+        # Registrar entidades no grafo (frontmatter + conteúdo)
+        # ORDEM IMPORTANTE: content primeiro, frontmatter depois
+        # extract_from_content() deleta apenas entidades 'content', preservando 'frontmatter'
+        if self.entities_db and frontmatter_match:
+            try:
+                # 1. Extrai entidades do conteúdo (spaCy NER)
+                self.entities_db.extract_from_content(
+                    memory_id=mem_name,
+                    content=body_content,
+                    use_spacy=True
+                )
+                # 2. Extrai entidades do frontmatter - preservadas
+                self.entities_db.extract_from_frontmatter(
+                    memory_id=mem_name,
+                    frontmatter=frontmatter,
+                    project=project
+                )
+            except Exception:
+                pass  # Falha silenciosa se frontmatter inválido
 
         return f"✅ Memória '{mem_name}' salva em {file_path}"