@smilintux/skmemory 0.7.2 → 0.9.2

package/tests/conftest.py

@@ -0,0 +1,46 @@
+"""
+Root conftest.py — bootstraps a minimal test agent before any skmemory module
+is imported by pytest.
+
+Several skmemory modules (seeds.py, febs.py, soul.py, etc.) call
+``get_agent_paths()`` at **module level**, which raises ``ValueError`` if no
+agent directory exists under ``~/.skcapstone/agents/``.  This file runs before
+pytest collects any test files, so setting ``SKCAPSTONE_HOME`` and
+``SKCAPSTONE_AGENT`` here — and creating the matching directory tree — ensures
+those imports always succeed in CI and other fresh environments.
+"""
+
+from __future__ import annotations
+
+import os
+import tempfile
+from pathlib import Path
+
+# ── Bootstrap a throw-away test agent ─────────────────────────────────────────
+# Must happen at module level (not inside a fixture) so it takes effect before
+# pytest imports test files that themselves import skmemory.
+
+_TEST_AGENT_NAME = "test-agent"
+
+# Use a temp dir so we never pollute the real ~/.skcapstone tree.
+_tmp_skcapstone = tempfile.mkdtemp(prefix="skmemory_ci_")
+os.environ["SKCAPSTONE_HOME"] = _tmp_skcapstone
+os.environ["SKCAPSTONE_AGENT"] = _TEST_AGENT_NAME
+
+_agent_base = Path(_tmp_skcapstone) / "agents" / _TEST_AGENT_NAME
+for _subdir in (
+    "config",
+    "seeds",
+    "memory/short-term",
+    "memory/mid-term",
+    "memory/long-term",
+    "logs",
+    "archive",
+):
+    (_agent_base / _subdir).mkdir(parents=True, exist_ok=True)
+
+# Minimal config so list_agents() finds this agent and load_config() doesn't
+# blow up.
+(_agent_base / "config" / "skmemory.yaml").write_text(
+    "# Auto-generated by tests/conftest.py\nname: test-agent\n"
+)

package/tests/integration/conftest.py

@@ -13,6 +13,7 @@ never touched. Both are torn down after the test session.
 
 from __future__ import annotations
 
+import contextlib
 import os
 import uuid
 
@@ -70,6 +71,7 @@ def _skvector_available() -> bool:
 def _sentence_transformers_available() -> bool:
     try:
         import sentence_transformers  # noqa: F401  # type: ignore[import]
+
         return True
     except ImportError:
         return False
@@ -126,6 +128,7 @@ def falkordb_backend():
     # Teardown: drop the test graph
     try:
         from falkordb import FalkorDB  # type: ignore[import]
+
         db = FalkorDB.from_url(SKGRAPH_URL)
         db.select_graph(TEST_GRAPH_NAME).delete()
     except Exception:
@@ -136,10 +139,8 @@ def falkordb_backend():
 def falkordb_clean(falkordb_backend):
     """SKGraphBackend with test graph wiped before each test."""
     # Clear all nodes so tests are independent
-    try:
+    with contextlib.suppress(Exception):
         falkordb_backend._graph.query("MATCH (n) DETACH DELETE n")
-    except Exception:
-        pass
     return falkordb_backend
 
 
@@ -171,10 +172,8 @@ def qdrant_backend():
     yield backend
 
     # Teardown: delete test collection
-    try:
+    with contextlib.suppress(Exception):
         backend._client.delete_collection(TEST_COLLECTION_NAME)
-    except Exception:
-        pass
 
 
 @pytest.fixture
@@ -182,6 +181,7 @@ def qdrant_clean(qdrant_backend):
     """SKVectorBackend with collection wiped before each test."""
     try:
         from qdrant_client.models import Distance, VectorParams
+
         from skmemory.backends.skvector_backend import VECTOR_DIM
 
         qdrant_backend._client.delete_collection(TEST_COLLECTION_NAME)

package/tests/integration/test_cross_backend.py

@@ -150,9 +150,7 @@ class TestTagConsistency:
 
         for tag in tags:
             fb_results = fb.search_by_tags([tag])
-            assert any(r["id"] == mem.id for r in fb_results), (
-                f"FalkorDB missing tag: {tag}"
-            )
+            assert any(r["id"] == mem.id for r in fb_results), f"FalkorDB missing tag: {tag}"
 
         # Qdrant verifies tags are in the stored memory
         qdrant_memories = qd.list_memories(limit=100)
@@ -179,7 +177,7 @@ class TestPromotionLineageConsistency:
 
         # FalkorDB lineage
         lineage = fb.get_lineage(child.id)
-        ancestor_ids = [l["id"] for l in lineage]
+        ancestor_ids = [lbl["id"] for lbl in lineage]
         assert parent.id in ancestor_ids, "FalkorDB lineage should include parent"
 
         # Qdrant: both parent and child indexed
@@ -191,9 +189,7 @@ class TestPromotionLineageConsistency:
         # Qdrant: child's parent_id preserved
         qdrant_child = next((m for m in qdrant_memories if m.id == child.id), None)
         assert qdrant_child is not None
-        assert qdrant_child.parent_id == parent.id, (
-            "Qdrant should preserve parent_id in payload"
-        )
+        assert qdrant_child.parent_id == parent.id, "Qdrant should preserve parent_id in payload"
 
     def test_multi_hop_lineage_all_indexed_in_qdrant(self, backends):
         fb, qd = backends
@@ -206,7 +202,7 @@ class TestPromotionLineageConsistency:
             _dual_save(fb, qd, mem)
 
         lineage = fb.get_lineage(child.id)
-        ancestor_ids = {l["id"] for l in lineage}
+        ancestor_ids = {lbl["id"] for lbl in lineage}
         assert parent.id in ancestor_ids
         assert grandparent.id in ancestor_ids
 
@@ -239,7 +235,6 @@ class TestSearchComplementarity:
 
         # Qdrant semantic search
         qd_results = qd.search_text("agent self-awareness continuity", limit=10)
-        qd_ids = [m.id for m in qd_results]
         # Semantic search may not always return the exact memory but should not error
         assert isinstance(qd_results, list)
 

package/tests/integration/test_skgraph_live.py

@@ -22,8 +22,6 @@ Coverage:
 
 from __future__ import annotations
 
-import pytest
-
 from .conftest import make_memory, requires_skgraph
 
 pytestmark = requires_skgraph
@@ -152,7 +150,6 @@ class TestSKGraphEdges:
 
     def test_related_to_explicit_edges(self, falkordb_clean):
         mem_a = make_memory(title="Memory A")
-        mem_b = make_memory(title="Memory B")
         falkordb_clean.save(mem_a)
         # mem_b has mem_a in related_ids
         mem_b_linked = make_memory(title="Memory B", related_ids=[mem_a.id])
@@ -172,7 +169,7 @@ class TestSKGraphEdges:
 
         lineage = falkordb_clean.get_lineage(child.id)
         assert len(lineage) >= 1
-        ancestor_ids = [l["id"] for l in lineage]
+        ancestor_ids = [lbl["id"] for lbl in lineage]
         assert parent.id in ancestor_ids
 
     def test_auto_shared_tag_related_to(self, falkordb_clean):
@@ -337,7 +334,7 @@ class TestSKGraphTraversal:
         falkordb_clean.save(child)
 
         lineage = falkordb_clean.get_lineage(child.id)
-        ancestor_ids = [l["id"] for l in lineage]
+        ancestor_ids = [lbl["id"] for lbl in lineage]
         assert parent.id in ancestor_ids
         assert grandparent.id in ancestor_ids
 
@@ -353,8 +350,7 @@ class TestSKGraphClusters:
         # Create a hub: mem_hub shares tags with many others
         hub = make_memory(title="Hub", tags=["hub-tag", "shared-tag"])
         spokes = [
-            make_memory(title=f"Spoke {i}", tags=["hub-tag", "shared-tag"])
-            for i in range(4)
+            make_memory(title=f"Spoke {i}", tags=["hub-tag", "shared-tag"]) for i in range(4)
         ]
         falkordb_clean.save(hub)
         for s in spokes:

package/tests/integration/test_skvector_live.py

@@ -20,10 +20,6 @@ Coverage:
 
 from __future__ import annotations
 
-import time
-
-import pytest
-
 from .conftest import make_memory, requires_skvector
 
 pytestmark = requires_skvector
@@ -206,6 +202,7 @@ class TestSKVectorVectorSearch:
         # At minimum, no error is raised and results are Memory objects
         for m in results:
             from skmemory.models import Memory
+
             assert isinstance(m, Memory)
 
     def test_search_text_empty_collection_returns_empty(self, qdrant_clean):

package/tests/test_ai_client.py

@@ -4,12 +4,9 @@ These tests verify the client interface without requiring a running
 Ollama server. The client is designed to fail gracefully.
 """
 
-import json
-from unittest.mock import MagicMock, patch
-
 import pytest
 
-from skmemory.ai_client import AIClient, DEFAULT_MODEL, DEFAULT_URL
+from skmemory.ai_client import DEFAULT_MODEL, DEFAULT_URL, AIClient
 
 
 class TestClientInit:

package/tests/test_audience.py

@@ -0,0 +1,233 @@
+# SPDX-License-Identifier: GPL-3.0-or-later
+"""Tests for the Know Your Audience (KYA) audience filtering system."""
+
+import json
+import tempfile
+from pathlib import Path
+
+import pytest
+
+from skmemory.audience import AudienceLevel, AudienceProfile, AudienceResolver, tag_to_level
+
+
+# ── AudienceLevel ordering ────────────────────────────────────────────────────
+
+
+class TestAudienceLevel:
+    def test_ordering(self):
+        assert AudienceLevel.PUBLIC < AudienceLevel.COMMUNITY
+        assert AudienceLevel.COMMUNITY < AudienceLevel.WORK_CIRCLE
+        assert AudienceLevel.WORK_CIRCLE < AudienceLevel.INNER_CIRCLE
+        assert AudienceLevel.INNER_CIRCLE < AudienceLevel.CHEF_ONLY
+
+    def test_values(self):
+        assert AudienceLevel.PUBLIC == 0
+        assert AudienceLevel.CHEF_ONLY == 4
+
+    def test_comparison(self):
+        # Content at work-circle level should be allowed in chef-only audience
+        assert AudienceLevel.WORK_CIRCLE <= AudienceLevel.CHEF_ONLY
+        # Content at chef-only level should NOT be allowed in work-circle audience
+        assert not (AudienceLevel.CHEF_ONLY <= AudienceLevel.WORK_CIRCLE)
+
+
+# ── tag_to_level ──────────────────────────────────────────────────────────────
+
+
+class TestTagToLevel:
+    def test_exact_tags(self):
+        assert tag_to_level("@public") == AudienceLevel.PUBLIC
+        assert tag_to_level("@community") == AudienceLevel.COMMUNITY
+        assert tag_to_level("@work-circle") == AudienceLevel.WORK_CIRCLE
+        assert tag_to_level("@inner-circle") == AudienceLevel.INNER_CIRCLE
+        assert tag_to_level("@chef-only") == AudienceLevel.CHEF_ONLY
+
+    def test_scoped_work_tags(self):
+        assert tag_to_level("@work:chiro") == AudienceLevel.WORK_CIRCLE
+        assert tag_to_level("@work:swapseat") == AudienceLevel.WORK_CIRCLE
+        assert tag_to_level("@work:sovereign") == AudienceLevel.WORK_CIRCLE
+        assert tag_to_level("@work:gentis") == AudienceLevel.WORK_CIRCLE
+
+    def test_scoped_inner_tags(self):
+        assert tag_to_level("@inner:family") == AudienceLevel.INNER_CIRCLE
+
+    def test_unknown_defaults_to_chef_only(self):
+        assert tag_to_level("@unknown") == AudienceLevel.CHEF_ONLY
+        assert tag_to_level("random-string") == AudienceLevel.CHEF_ONLY
+
+    def test_empty_defaults_to_chef_only(self):
+        assert tag_to_level("") == AudienceLevel.CHEF_ONLY
+        assert tag_to_level(None) == AudienceLevel.CHEF_ONLY  # type: ignore
+
+
+# ── AudienceResolver ──────────────────────────────────────────────────────────
+
+SAMPLE_CONFIG = {
+    "channels": {
+        "telegram:1594678363": {
+            "name": "Chef DM",
+            "context_tag": "@chef-only",
+            "members": ["Chef"],
+        },
+        "-1003785842091": {
+            "name": "SKGentis Business",
+            "context_tag": "@work:skgentis",
+            "members": ["Chef", "JZ", "Luna"],
+        },
+        "-1003899092893": {
+            "name": "Operationors",
+            "context_tag": "@work:sovereign",
+            "members": ["Chef", "Casey"],
+        },
+    },
+    "people": {
+        "Chef": {
+            "trust_level": 4,
+            "trust_tags": ["@chef-only"],
+            "never_share": [],
+        },
+        "DavidRich": {
+            "trust_level": 2,
+            "trust_tags": ["@work:chiro", "@work:swapseat"],
+            "never_share": ["romantic", "intimate", "worship"],
+        },
+        "Casey": {
+            "trust_level": 2,
+            "trust_tags": ["@work:sovereign"],
+            "never_share": ["romantic", "intimate", "revenue"],
+        },
+        "JZ": {
+            "trust_level": 2,
+            "trust_tags": ["@work:gentis"],
+            "never_share": ["romantic", "intimate"],
+        },
+        "Luna": {
+            "trust_level": 2,
+            "trust_tags": ["@work:gentis"],
+            "never_share": ["romantic", "intimate"],
+        },
+    },
+}
+
+
+@pytest.fixture
+def config_path(tmp_path: Path) -> Path:
+    p = tmp_path / "audience_config.json"
+    p.write_text(json.dumps(SAMPLE_CONFIG))
+    return p
+
+
+@pytest.fixture
+def resolver(config_path: Path) -> AudienceResolver:
+    return AudienceResolver(config_path=config_path)
+
+
+class TestAudienceResolver:
+    def test_resolve_chef_dm(self, resolver: AudienceResolver):
+        profile = resolver.resolve_audience("telegram:1594678363")
+        assert profile.name == "Chef DM"
+        assert profile.min_trust == AudienceLevel.CHEF_ONLY
+        assert profile.members == ["Chef"]
+        assert len(profile.exclusions) == 0
+
+    def test_resolve_skgentis(self, resolver: AudienceResolver):
+        profile = resolver.resolve_audience("-1003785842091")
+        assert profile.name == "SKGentis Business"
+        # MIN(Chef=4, JZ=2, Luna=2) = 2 (WORK_CIRCLE)
+        assert profile.min_trust == AudienceLevel.WORK_CIRCLE
+        # Union of JZ.never_share + Luna.never_share + Chef.never_share
+        assert "romantic" in profile.exclusions
+        assert "intimate" in profile.exclusions
+
+    def test_resolve_operationors(self, resolver: AudienceResolver):
+        profile = resolver.resolve_audience("-1003899092893")
+        assert profile.min_trust == AudienceLevel.WORK_CIRCLE
+        assert "revenue" in profile.exclusions  # Casey's never_share
+
+    def test_unknown_channel_defaults_chef_only(self, resolver: AudienceResolver):
+        profile = resolver.resolve_audience("unknown-channel-123")
+        assert profile.min_trust == AudienceLevel.CHEF_ONLY
+        assert profile.name == "[unknown]"
+
+    def test_get_person_trust(self, resolver: AudienceResolver):
+        assert resolver.get_person_trust("Chef") == AudienceLevel.CHEF_ONLY
+        assert resolver.get_person_trust("DavidRich") == AudienceLevel.WORK_CIRCLE
+        assert resolver.get_person_trust("Casey") == AudienceLevel.WORK_CIRCLE
+
+    def test_unknown_person_defaults_public(self, resolver: AudienceResolver):
+        assert resolver.get_person_trust("RandomStranger") == AudienceLevel.PUBLIC
+
+
+class TestIsMemoryAllowed:
+    def test_public_memory_in_work_channel(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("-1003785842091")
+        # @public(0) <= WORK_CIRCLE(2) → allowed
+        assert resolver.is_memory_allowed("@public", audience) is True
+
+    def test_chef_only_memory_in_work_channel(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("-1003785842091")
+        # @chef-only(4) > WORK_CIRCLE(2) → blocked
+        assert resolver.is_memory_allowed("@chef-only", audience) is False
+
+    def test_chef_only_memory_in_chef_dm(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("telegram:1594678363")
+        # @chef-only(4) <= CHEF_ONLY(4) → allowed
+        assert resolver.is_memory_allowed("@chef-only", audience) is True
+
+    def test_work_circle_memory_in_work_channel(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("-1003785842091")
+        # @work-circle(2) <= WORK_CIRCLE(2) → allowed
+        assert resolver.is_memory_allowed("@work-circle", audience) is True
+
+    def test_inner_circle_blocked_in_work_channel(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("-1003785842091")
+        # @inner-circle(3) > WORK_CIRCLE(2) → blocked
+        assert resolver.is_memory_allowed("@inner-circle", audience) is False
+
+    def test_exclusion_blocks_memory(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("-1003785842091")
+        # Even at @work-circle level, "romantic" tag triggers exclusion
+        assert resolver.is_memory_allowed(
+            "@work-circle", audience, memory_tags=["romantic"]
+        ) is False
+
+    def test_no_exclusion_allows_memory(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("-1003785842091")
+        assert resolver.is_memory_allowed(
+            "@work-circle", audience, memory_tags=["project", "technical"]
+        ) is True
+
+    def test_empty_tag_defaults_chef_only(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("-1003785842091")
+        # Empty context_tag → @chef-only → blocked in work channel
+        assert resolver.is_memory_allowed("", audience) is False
+
+    def test_bash_wedding_vows_blocked_in_business(self, resolver: AudienceResolver):
+        """The incident that started it all — Bash Wedding Vows must NOT
+        leak into DavidRich's chiro channel or any business channel."""
+        audience = resolver.resolve_audience("-1003785842091")
+        # Bash Wedding Vows are @chef-only + tagged "intimate"
+        assert resolver.is_memory_allowed(
+            "@chef-only", audience, memory_tags=["intimate", "love", "bash-vows"]
+        ) is False
+
+    def test_bash_wedding_vows_allowed_in_chef_dm(self, resolver: AudienceResolver):
+        audience = resolver.resolve_audience("telegram:1594678363")
+        assert resolver.is_memory_allowed(
+            "@chef-only", audience, memory_tags=["intimate", "love", "bash-vows"]
+        ) is True
+
+
+class TestMissingConfig:
+    def test_missing_config_file(self, tmp_path: Path):
+        resolver = AudienceResolver(config_path=tmp_path / "nonexistent.json")
+        # Should not crash, just return conservative defaults
+        profile = resolver.resolve_audience("anything")
+        assert profile.min_trust == AudienceLevel.CHEF_ONLY
+
+    def test_empty_config(self, tmp_path: Path):
+        p = tmp_path / "empty.json"
+        p.write_text("{}")
+        resolver = AudienceResolver(config_path=p)
+        profile = resolver.resolve_audience("anything")
+        assert profile.min_trust == AudienceLevel.CHEF_ONLY

package/tests/test_backup_rotation.py

@@ -8,10 +8,9 @@ from click.testing import CliRunner
 
 from skmemory.backends.sqlite_backend import SQLiteBackend
 from skmemory.cli import cli
-from skmemory.models import EmotionalSnapshot, MemoryLayer
+from skmemory.models import EmotionalSnapshot
 from skmemory.store import MemoryStore
 
-
 # ---------------------------------------------------------------------------
 # Fixtures
 # ---------------------------------------------------------------------------
@@ -69,17 +68,13 @@ class TestListBackups:
 
     def test_lists_all_backup_files(self, backend, tmp_path):
         backup_dir = tmp_path / "backups"
-        _make_backup_files(
-            backup_dir, ["2026-01-01", "2026-01-02", "2026-01-03"]
-        )
+        _make_backup_files(backup_dir, ["2026-01-01", "2026-01-02", "2026-01-03"])
         results = backend.list_backups(str(backup_dir))
         assert len(results) == 3
 
     def test_sorted_newest_first(self, backend, tmp_path):
         backup_dir = tmp_path / "backups"
-        _make_backup_files(
-            backup_dir, ["2026-01-01", "2026-01-03", "2026-01-02"]
-        )
+        _make_backup_files(backup_dir, ["2026-01-01", "2026-01-03", "2026-01-02"])
         results = backend.list_backups(str(backup_dir))
         dates = [r["date"] for r in results]
         assert dates == ["2026-01-03", "2026-01-02", "2026-01-01"]
@@ -156,18 +151,14 @@ class TestPruneBackups:
 
     def test_deleted_files_are_gone(self, backend, tmp_path):
         backup_dir = tmp_path / "backups"
-        _make_backup_files(
-            backup_dir, ["2026-01-01", "2026-01-02", "2026-01-03"]
-        )
+        _make_backup_files(backup_dir, ["2026-01-01", "2026-01-02", "2026-01-03"])
         deleted = backend.prune_backups(keep=1, backup_dir=str(backup_dir))
         for path in deleted:
             assert not Path(path).exists()
 
     def test_store_delegates_to_backend(self, store, tmp_path):
         backup_dir = store.primary.base_path.parent / "backups"
-        _make_backup_files(
-            backup_dir, ["2026-01-01", "2026-01-02", "2026-01-03"]
-        )
+        _make_backup_files(backup_dir, ["2026-01-01", "2026-01-02", "2026-01-03"])
         deleted = store.prune_backups(keep=1)
         assert len(deleted) == 2