tribalmemory 0.1.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

tribalmemory/services/vector_store.py

@@ -252,7 +252,63 @@ class LanceDBVectorStore(IVectorStore):
     async def count(self, filters: Optional[dict] = None) -> int:
         entries = await self.list(limit=100000, filters=filters)
         return len(entries)
-    
+
+    async def get_stats(self) -> dict:
+        """Compute stats natively over LanceDB rows.
+
+        Iterates rows in pages to avoid loading all embeddings into
+        RAM. Only the metadata columns are read.
+        """
+        await self._ensure_initialized()
+
+        by_source: dict[str, int] = {}
+        by_instance: dict[str, int] = {}
+        by_tag: dict[str, int] = {}
+        total = 0
+        corrections = 0
+
+        page_size = 1000
+        offset = 0
+        while True:
+            rows = (
+                self._table.search()
+                .where("deleted = false")
+                .select(["source_type", "source_instance", "tags",
+                         "supersedes"])
+                .limit(page_size + offset)
+                .to_list()
+            )
+            page = rows[offset:]
+            if not page:
+                break
+
+            for row in page:
+                total += 1
+                src = row.get("source_type", "unknown")
+                by_source[src] = by_source.get(src, 0) + 1
+
+                inst = row.get("source_instance", "unknown")
+                by_instance[inst] = by_instance.get(inst, 0) + 1
+
+                tags = json.loads(row.get("tags", "[]"))
+                for tag in tags:
+                    by_tag[tag] = by_tag.get(tag, 0) + 1
+
+                if row.get("supersedes"):
+                    corrections += 1
+
+            if len(page) < page_size:
+                break
+            offset += page_size
+
+        return {
+            "total_memories": total,
+            "by_source_type": by_source,
+            "by_tag": by_tag,
+            "by_instance": by_instance,
+            "corrections": corrections,
+        }
+
     def _row_to_entry(self, row: dict) -> MemoryEntry:
         return MemoryEntry(
             id=row["id"],
@@ -358,3 +414,32 @@ class InMemoryVectorStore(IVectorStore):
     async def count(self, filters: Optional[dict] = None) -> int:
         entries = await self.list(limit=100000, filters=filters)
         return len(entries)
+
+    async def get_stats(self) -> dict:
+        """Compute stats in a single pass over in-memory entries."""
+        by_source: dict[str, int] = {}
+        by_instance: dict[str, int] = {}
+        by_tag: dict[str, int] = {}
+        total = 0
+        corrections = 0
+
+        for entry in self._store.values():
+            if entry.id in self._deleted:
+                continue
+            total += 1
+            src = entry.source_type.value
+            by_source[src] = by_source.get(src, 0) + 1
+            inst = entry.source_instance
+            by_instance[inst] = by_instance.get(inst, 0) + 1
+            for tag in entry.tags:
+                by_tag[tag] = by_tag.get(tag, 0) + 1
+            if entry.supersedes:
+                corrections += 1
+
+        return {
+            "total_memories": total,
+            "by_source_type": by_source,
+            "by_tag": by_tag,
+            "by_instance": by_instance,
+            "corrections": corrections,
+        }

{tribalmemory-0.1.1.dist-info → tribalmemory-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tribalmemory
-Version: 0.1.1
+Version: 0.3.0
 Summary: Shared memory infrastructure for multi-instance AI agents
 Author-email: Joe <joe@example.com>
 License: Apache-2.0

{tribalmemory-0.1.1.dist-info → tribalmemory-0.3.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 tribalmemory/__init__.py,sha256=DNgC_ZT0lrhxsPdhXu4oeG_UdrLstYQHeKwR-U2toeY,104
-tribalmemory/cli.py,sha256=1QDcRxBBNBm1ysUv06FgwMCBHMctdJyUARuy4y0XHSs,10292
-tribalmemory/interfaces.py,sha256=W2BNYyYlY-RHTbnJn8-u1wbhfcUK5egzjUzrGexxyB0,9966
+tribalmemory/cli.py,sha256=wbVsgMst4NEuWkwA2dily_qC8AP6jvAHKwHgE3B-TG8,15960
+tribalmemory/interfaces.py,sha256=hs32jxImd3AXa0EfysGhP-goDAMQ90fqVQV4PbTpgQQ,12287
 tribalmemory/utils.py,sha256=aei7xR6OVMGkllPySA2boeHyI3D1JsHTUX1YeaZdkMY,696
 tribalmemory/a21/__init__.py,sha256=u1793uKzbWGKwiAVmCxEO9_3rdTL7QKTLhQQB8Umsl4,1181
 tribalmemory/a21/system.py,sha256=gGFVWBTckSTFv8ZciEUc05HYjwxZP22UpIqbxXD6giM,9185
@@ -19,7 +19,7 @@ tribalmemory/a21/providers/openai.py,sha256=MxFJXph8rVFAqkVMCS3vxdqwBB8_MhoqqGg6
 tribalmemory/a21/providers/timestamp.py,sha256=T1sJkaQSixRip2C0AUPnljWP0u5w4iHoIcVRmG8FgPo,2994
 tribalmemory/mcp/__init__.py,sha256=r7CzTwzAnSgox8_cBNZgtXDjMGxC7o8polptSvB-AvY,262
 tribalmemory/mcp/__main__.py,sha256=GX6mNmM_Lpq4EyuTr__awVTf1btCi7TpEkNtMTkgas0,115
-tribalmemory/mcp/server.py,sha256=otmDaFe2sXlHNyPorLgUIvkaM6F01EQ-1OdiyhqGDoU,15458
+tribalmemory/mcp/server.py,sha256=URosl0xbxqn0m57bkO52y-pKIBzgkP-f8wTm_MRWzd8,24708
 tribalmemory/performance/__init__.py,sha256=truSiOqk2WuzatOrzVQICwF8pxLxMlCD8R9O-uqiX3I,55
 tribalmemory/performance/benchmarks.py,sha256=2MVfML04Y-05YpmHCsU-SLtS05-H38oJ7a6DCk2qGIc,8985
 tribalmemory/performance/corpus_generator.py,sha256=ovln1d-7JGd5fJbdSRsdxlA0uaqLCVM3Lo_1SDGRkA0,5894
@@ -27,25 +27,29 @@ tribalmemory/portability/__init__.py,sha256=_USwXZyUKn9idsItv94AAzxKv1wqURYf68Iz
 tribalmemory/portability/embedding_metadata.py,sha256=uT_f9jc_vpemY2WxK_UhP3kOAWgZLMRShKKxh3OzzPM,10478
 tribalmemory/server/__init__.py,sha256=2YYwVr1IEr93MVa_7BKCX43bC82-ySnAfwypUw_nQJQ,238
 tribalmemory/server/__main__.py,sha256=Uk2_8MH-aQ65QWDg_XMe5i-hdaQB-yJApLT8YUaIdEs,116
-tribalmemory/server/app.py,sha256=P4QrOaw35t8NQvrJjYCXMEEFMuH2MrIibCyB2uwpvVU,4836
-tribalmemory/server/config.py,sha256=4BCRUjTnkVCEGPNUK5UiOFXyO91BxwxXsCuEG0lnVh4,3552
-tribalmemory/server/models.py,sha256=bY6IBn6_AW7PnGunQGa0W7LgKym0-KW05lGMztLbhY0,5652
-tribalmemory/server/routes.py,sha256=8-Ie0Ib7t_tUT_kZvsb1QUuNQZ2Dd6rOTlI0d2CoiAk,11403
+tribalmemory/server/app.py,sha256=Ku2UOJ2cnh-RIhtLBaKuZvOx8tjNVbOplhn1U2ePbBM,6796
+tribalmemory/server/config.py,sha256=UHiUIdonPdnsjdZh1vE0C6aOVbP4gS5Bxh7YGJgHh2U,5484
+tribalmemory/server/models.py,sha256=-Y8l5uJ7ij_oiwNtAnLRbmXE9oQlwcBsrCRgnYp14N0,7719
+tribalmemory/server/routes.py,sha256=O2qYpiFYFXY5akTAYcHYmv6AZNqyz3NwnxGPx7GPHFw,13842
 tribalmemory/services/__init__.py,sha256=htv8HuG_r_lYJwP5Q1bwO-oir436U0NfJrOxk9mB7kU,468
 tribalmemory/services/deduplication.py,sha256=E8PaIDB6g24H28tKHrB5rMBJaKGGT3pFTDepXQThvcc,3679
 tribalmemory/services/embeddings.py,sha256=0kY1uPyCg81AlRTNg5QhXbRLDv8hN9khKR4JDGF2sik,10005
+tribalmemory/services/fts_store.py,sha256=5-SBGmzDeQR0-8aDMMO-zZuo8M7VK_tlKYUVDNAitV4,8424
+tribalmemory/services/graph_store.py,sha256=zekWCdmqvZR9qH09FFqp6L7QKSI1z368CKLUDH1osys,24376
 tribalmemory/services/import_export.py,sha256=KfEl5EXAFcuyDmhOYJZfjiIRtJYY4qQlvxv2ePJQHaA,15095
-tribalmemory/services/memory.py,sha256=UDypOGf8fAfN4iepkncrJn9v3QWvQx5lmItSZfwLUGg,9438
-tribalmemory/services/vector_store.py,sha256=8oq1V-pedIW4A7FXvqHY91L5DK56YdGlZUC3qjDtugo,12703
+tribalmemory/services/memory.py,sha256=uWw-Eg8cJYmzwMsXBxmkGzPRM66Ew13OuRDKBAI2lng,27472
+tribalmemory/services/reranker.py,sha256=0RSvQFHB609aWcxBl92fiwxIbAYILTqNL9MaAp0lQ74,8720
+tribalmemory/services/session_store.py,sha256=wkVF9pNJOqkVXIYOkyvABSaRxaLoHDldT8KTZGThDU0,13818
+tribalmemory/services/vector_store.py,sha256=fL8YgnHiCLPqxqV64pQ_rMLIzdJc6ohN2c4wgGz0buw,15364
 tribalmemory/testing/__init__.py,sha256=XVS3uy0BjABCErgZohaqtj5MF20NKmj1KmtJfRiI_XI,524
 tribalmemory/testing/embedding_utils.py,sha256=E40lSAU7cz-ow2hzKLSaZmafIlOP8d9gakP8R2DTlGM,3705
 tribalmemory/testing/fixtures.py,sha256=_zDyUVm6CQqXK1Us8CN6A95tJcmo1D7LFDktIvjOmwM,3584
 tribalmemory/testing/metrics.py,sha256=X1n84dJDNQXsfGV-i-MzhsWKnFgqHWIcIaQB-BUp0e0,8711
 tribalmemory/testing/mocks.py,sha256=sjLy-pq3D_T21rEvWWKM_bqw7xnchRPGLARZNfKkpGU,19788
 tribalmemory/testing/semantic_expansions.py,sha256=AbbJIXYN4EJT8WKJ7UmIlNRlv63VejbcnbzBy2z2Ofk,2953
-tribalmemory-0.1.1.dist-info/licenses/LICENSE,sha256=M8D9Xf3B6C6DFiCgAAhKcXeTscaC4cw1fhr3LHUrALU,10774
-tribalmemory-0.1.1.dist-info/METADATA,sha256=IyhoTSK64QZKLTu0CjFRnmbfbvlkUGNtFNxvx2R21Ag,9239
-tribalmemory-0.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-tribalmemory-0.1.1.dist-info/entry_points.txt,sha256=9Pep7JNCk9ifdFP4WbeCugDOjMrLVegGJ5iuvbcZ9e8,103
-tribalmemory-0.1.1.dist-info/top_level.txt,sha256=kX36ZpH4W7EWcInV4MrIudicusdz5hfkezKMZ3HCMQs,13
-tribalmemory-0.1.1.dist-info/RECORD,,
+tribalmemory-0.3.0.dist-info/licenses/LICENSE,sha256=M8D9Xf3B6C6DFiCgAAhKcXeTscaC4cw1fhr3LHUrALU,10774
+tribalmemory-0.3.0.dist-info/METADATA,sha256=05lUAOlIf1Zei-I2USroVXkuq2gCijCrCn52B_IqR6Y,9239
+tribalmemory-0.3.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+tribalmemory-0.3.0.dist-info/entry_points.txt,sha256=9Pep7JNCk9ifdFP4WbeCugDOjMrLVegGJ5iuvbcZ9e8,103
+tribalmemory-0.3.0.dist-info/top_level.txt,sha256=kX36ZpH4W7EWcInV4MrIudicusdz5hfkezKMZ3HCMQs,13
+tribalmemory-0.3.0.dist-info/RECORD,,