virtual-context 0.2.8__tar.gz → 0.2.9__tar.gz

{virtual_context-0.2.8 → virtual_context-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: virtual-context
-Version: 0.2.8
+Version: 0.2.9
 Summary: OS-style virtual memory for LLM session context management
 Project-URL: Homepage, https://virtual-context.com
 Project-URL: Repository, https://github.com/virtual-context/virtual-context
@@ -180,10 +180,16 @@ proxy:
 **Daemon mode:** run as a background service:
 
 ```bash
-virtual-context onboard --install-daemon --upstream https://api.anthropic.com
+# Creates config if needed, installs + starts daemon
+virtual-context daemon install --upstream https://api.anthropic.com
+
+# Or: guided interactive setup with daemon
+virtual-context onboard --wizard --install-daemon
 ```
 
-Daemon setup docs (macOS `launchd`, Linux `systemd --user`, Windows Task Scheduler): [`docs/install.md`](docs/install.md)
+Daemon lifecycle: `daemon status | start | stop | restart | uninstall`
+
+Full setup docs (macOS `launchd`, Linux `systemd --user`, Windows Task Scheduler): [`docs/install.md`](docs/install.md)
 
 ### Python SDK
 
@@ -639,10 +645,14 @@ virtual-context transform -m "What about auth?"# tag + retrieve + assemble
 virtual-context aliases list                   # show all tag aliases
 virtual-context aliases suggest                # auto-detect potential aliases
 virtual-context aliases add db database        # register alias manually
+virtual-context init coding                    # create config from preset
+virtual-context onboard                        # guided setup (create/validate config)
+virtual-context onboard --wizard               # interactive multi-instance setup
 virtual-context proxy -u https://api.anthropic.com  # single-instance proxy
 virtual-context proxy                               # multi-instance (from config)
 virtual-context presets list                   # list available config presets
 virtual-context presets show coding            # dump preset config as YAML
+virtual-context daemon install                 # install daemon/service
 virtual-context daemon status                  # service status (platform-specific)
 virtual-context daemon start                   # start/enable daemon
 virtual-context daemon stop                    # stop daemon

{virtual_context-0.2.8 → virtual_context-0.2.9}/README.md

@@ -133,10 +133,16 @@ proxy:
 **Daemon mode:** run as a background service:
 
 ```bash
-virtual-context onboard --install-daemon --upstream https://api.anthropic.com
+# Creates config if needed, installs + starts daemon
+virtual-context daemon install --upstream https://api.anthropic.com
+
+# Or: guided interactive setup with daemon
+virtual-context onboard --wizard --install-daemon
 ```
 
-Daemon setup docs (macOS `launchd`, Linux `systemd --user`, Windows Task Scheduler): [`docs/install.md`](docs/install.md)
+Daemon lifecycle: `daemon status | start | stop | restart | uninstall`
+
+Full setup docs (macOS `launchd`, Linux `systemd --user`, Windows Task Scheduler): [`docs/install.md`](docs/install.md)
 
 ### Python SDK
 
@@ -592,10 +598,14 @@ virtual-context transform -m "What about auth?"# tag + retrieve + assemble
 virtual-context aliases list                   # show all tag aliases
 virtual-context aliases suggest                # auto-detect potential aliases
 virtual-context aliases add db database        # register alias manually
+virtual-context init coding                    # create config from preset
+virtual-context onboard                        # guided setup (create/validate config)
+virtual-context onboard --wizard               # interactive multi-instance setup
 virtual-context proxy -u https://api.anthropic.com  # single-instance proxy
 virtual-context proxy                               # multi-instance (from config)
 virtual-context presets list                   # list available config presets
 virtual-context presets show coding            # dump preset config as YAML
+virtual-context daemon install                 # install daemon/service
 virtual-context daemon status                  # service status (platform-specific)
 virtual-context daemon start                   # start/enable daemon
 virtual-context daemon stop                    # stop daemon

{virtual_context-0.2.8 → virtual_context-0.2.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "virtual-context"
-version = "0.2.8"
+version = "0.2.9"
 description = "OS-style virtual memory for LLM session context management"
 readme = "README.md"
 license = "AGPL-3.0-or-later"

{virtual_context-0.2.8 → virtual_context-0.2.9}/tests/test_cli_init.py

@@ -97,6 +97,46 @@ def test_daemon_restart_in_help(tmp_cwd):
     assert "restart" in result.stdout
 
 
+def test_daemon_not_installed_guard(tmp_cwd):
+    """daemon commands should fail with a helpful message when no daemon is installed."""
+    env = {**__import__("os").environ, "HOME": str(tmp_cwd)}
+    for action in ("status", "start", "stop", "restart"):
+        result = subprocess.run(
+            [sys.executable, "-m", "virtual_context.cli.main", "daemon", action],
+            capture_output=True,
+            text=True,
+            env=env,
+        )
+        assert result.returncode != 0, f"daemon {action} should fail when not installed"
+        assert "not installed" in result.stdout.lower(), (
+            f"daemon {action} should mention 'not installed'"
+        )
+        assert "daemon install" in result.stdout, (
+            f"daemon {action} should suggest 'daemon install'"
+        )
+
+
+def test_daemon_install_creates_config(tmp_cwd):
+    """daemon install should auto-create config from agentic preset if none exists."""
+    config_path = tmp_cwd / "virtual-context.yaml"
+    assert not config_path.exists()
+    # daemon install will fail on launchctl/systemctl, but should create config first
+    result = _run_cli("daemon", "install")
+    assert config_path.exists(), "daemon install should create config"
+    assert "Created config" in result.stdout
+    content = config_path.read_text()
+    assert "tag_generator" in content
+
+
+def test_daemon_install_preserves_existing_config(tmp_cwd):
+    """daemon install should not overwrite an existing config."""
+    config_path = tmp_cwd / "virtual-context.yaml"
+    config_path.write_text("custom: true\n")
+    result = _run_cli("daemon", "install")
+    assert config_path.read_text() == "custom: true\n"
+    assert "Created config" not in result.stdout
+
+
 def test_presets_list(tmp_cwd):
     result = _run_cli("presets", "list")
     assert result.returncode == 0

virtual_context-0.2.9/tests/test_fact_redesign.py

@@ -0,0 +1,95 @@
+"""Tests for fact extraction redesign."""
+from virtual_context.types import CompactorConfig
+
+
+def test_compactor_config_has_code_mode():
+    cc = CompactorConfig()
+    assert hasattr(cc, "code_mode")
+    assert cc.code_mode is True
+
+
+def test_config_loader_reads_code_mode():
+    from virtual_context.config import load_config
+    import tempfile, os, yaml
+    cfg = {"compaction": {"code_mode": False}, "context_window": 100000}
+    with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
+        yaml.dump(cfg, f)
+        tmp_path = f.name
+    try:
+        config = load_config(tmp_path)
+        assert config.compactor.code_mode is False
+    finally:
+        os.unlink(tmp_path)
+
+
+def test_replace_facts_for_segment_default():
+    from virtual_context.core.store import ContextStore
+
+    class MinimalStore(ContextStore):
+        """Minimal concrete subclass that only implements abstract methods as no-ops."""
+        def store_segment(self, segment): return ""
+        def get_segment(self, ref, *, conversation_id=None): return None
+        def get_summary(self, ref, *, conversation_id=None): return None
+        def get_summaries_by_tags(self, tags, min_overlap=1, limit=10, before=None, after=None, conversation_id=None): return []
+        def search(self, query, tags=None, limit=5, conversation_id=None): return []
+        def get_all_tags(self, conversation_id=None): return []
+        def get_conversation_stats(self): return []
+        def get_tag_aliases(self, conversation_id=None): return {}
+        def set_tag_alias(self, alias, canonical, conversation_id=""): pass
+        def delete_segment(self, ref): return False
+        def cleanup(self, max_age=None, max_total_tokens=None): return 0
+        def save_tag_summary(self, tag_summary, conversation_id=""): pass
+        def get_tag_summary(self, tag, conversation_id=""): return None
+        def get_all_tag_summaries(self, *, conversation_id=None): return []
+        def search_full_text(self, query, limit=5, conversation_id=None): return []
+        def get_segments_by_tags(self, tags, min_overlap=1, limit=20, conversation_id=None): return []
+
+    store = MinimalStore()
+    deleted, inserted = store.replace_facts_for_segment("conv1", "seg1", [])
+    assert deleted == 0
+    assert inserted == 0
+
+
+def test_compaction_uses_replace_facts_for_segment():
+    """Verify the compaction pipeline calls replace_facts_for_segment."""
+    import inspect
+    from virtual_context.core import compaction_pipeline
+    source = inspect.getsource(compaction_pipeline)
+    assert "replace_facts_for_segment" in source, \
+        "compaction_pipeline must call replace_facts_for_segment"
+
+
+def test_code_mode_prompt_appended():
+    """When code_mode is True, the compactor prompt should include the code mode block."""
+    from virtual_context.core.compactor import CODE_MODE_FACT_PROMPT
+    assert "Do NOT extract intermediary" in CODE_MODE_FACT_PROMPT
+    assert "conclusions, findings, discoveries" in CODE_MODE_FACT_PROMPT
+
+
+def test_tag_generator_accepts_code_mode():
+    from virtual_context.core.tag_generator import LLMTagGenerator
+    from virtual_context.types import TagGeneratorConfig
+    from unittest.mock import MagicMock
+    gen = LLMTagGenerator(llm_provider=MagicMock(), config=TagGeneratorConfig(), code_mode=True)
+    assert gen._code_mode is True
+
+
+def test_build_tag_generator_passes_code_mode():
+    from virtual_context.core.tag_generator import build_tag_generator, LLMTagGenerator
+    from virtual_context.types import TagGeneratorConfig
+    from unittest.mock import MagicMock
+    config = TagGeneratorConfig(type="llm", provider="openrouter", model="test")
+    gen = build_tag_generator(config, MagicMock(), code_mode=True)
+    assert isinstance(gen, LLMTagGenerator)
+    assert gen._code_mode is True
+    gen2 = build_tag_generator(config, MagicMock())
+    assert gen2._code_mode is False
+
+
+def test_engine_passes_code_mode_to_tag_generator():
+    import inspect
+    from virtual_context import engine
+    source = inspect.getsource(engine)
+    assert "code_mode" in source, "engine.py must reference code_mode"
+    assert "compactor.code_mode" in source or "config.compactor.code_mode" in source, \
+        "engine.py must read code_mode from compactor config"

{virtual_context-0.2.8 → virtual_context-0.2.9}/tests/test_format_agnostic.py

@@ -450,7 +450,7 @@ class TestCollapseChainsCrossFormat:
         fmt = detect_format(body)
         tti = TurnTagIndex()
         original_count = len(body["messages"])
-        result, collapsed, refs = collapse_turn_chains(
+        result, collapsed, refs, _ = collapse_turn_chains(
             body, fmt, protected_recent_turns=3, turn_tag_index=tti,
             store=None, conversation_id="test",
         )
@@ -480,7 +480,7 @@ class TestCollapseChainsCrossFormat:
         fmt = detect_format(body)
         tti = TurnTagIndex()
         original_count = len(body["input"])
-        result, collapsed, refs = collapse_turn_chains(
+        result, collapsed, refs, _ = collapse_turn_chains(
             body, fmt, protected_recent_turns=3, turn_tag_index=tti,
             store=None, conversation_id="test",
         )
@@ -507,7 +507,7 @@ class TestCollapseChainsCrossFormat:
         fmt = detect_format(body)
         tti = TurnTagIndex()
         original_count = len(body["messages"])
-        result, collapsed, refs = collapse_turn_chains(
+        result, collapsed, refs, _ = collapse_turn_chains(
             body, fmt, protected_recent_turns=3, turn_tag_index=tti,
             store=None, conversation_id="test",
         )

virtual_context-0.2.9/tests/test_store_recovery.py

@@ -0,0 +1,192 @@
+"""Tests for store-backed pipeline recovery."""
+
+from virtual_context.types import MonitorConfig
+
+
+def test_monitor_config_has_store_recovery_threshold():
+    mc = MonitorConfig()
+    assert hasattr(mc, "store_recovery_threshold")
+    assert mc.store_recovery_threshold == 0.70
+
+
+def _make_dummy_store():
+    """Create a minimal concrete ContextStore for testing default method implementations."""
+    from virtual_context.core.store import ContextStore
+
+    class DummyStore(ContextStore):
+        def store_segment(self, segment): return ""
+        def get_segment(self, ref, *, conversation_id=None): return None
+        def get_summary(self, ref, *, conversation_id=None): return None
+        def get_summaries_by_tags(self, tags, min_overlap=1, limit=10, before=None, after=None, conversation_id=None): return []
+        def search(self, query, tags=None, limit=5, conversation_id=None): return []
+        def get_all_tags(self, conversation_id=None): return []
+        def get_conversation_stats(self): return []
+        def get_tag_aliases(self, conversation_id=None): return {}
+        def set_tag_alias(self, alias, canonical, conversation_id=""): pass
+        def delete_segment(self, ref): return False
+        def cleanup(self, max_age=None, max_total_tokens=None): return 0
+        def save_tag_summary(self, tag_summary, conversation_id=""): pass
+        def get_tag_summary(self, tag, conversation_id=""): return None
+        def get_all_tag_summaries(self, *, conversation_id=None): return []
+        def search_full_text(self, query, limit=5, conversation_id=None): return []
+        def get_segments_by_tags(self, tags, min_overlap=1, limit=20, conversation_id=None): return []
+
+    return DummyStore()
+
+
+def test_get_chain_snapshots_for_conversation_abstract():
+    s = _make_dummy_store()
+    assert s.get_chain_snapshots_for_conversation("conv1") == []
+    assert s.get_chain_snapshots_for_conversation("conv1", min_turn=100) == []
+
+
+def test_get_tool_names_for_refs_abstract():
+    s = _make_dummy_store()
+    assert s.get_tool_names_for_refs(["ref1", "ref2"]) == []
+
+
+def test_collapse_turn_chains_recovers_from_store():
+    from unittest.mock import MagicMock
+    from virtual_context.proxy.message_filter import collapse_turn_chains
+    from virtual_context.proxy.formats import detect_format
+    from virtual_context.core.turn_tag_index import TurnTagIndex, TurnTagEntry
+
+    body = {
+        "system": "test",
+        "messages": [
+            {"role": "user", "content": "hello"},
+            {"role": "assistant", "content": [{"type": "text", "text": "hi"}]},
+            {"role": "user", "content": "current question"},
+        ],
+        "model": "claude-opus-4-6",
+    }
+    fmt = detect_format(body)
+    tti = TurnTagIndex()
+    for i in range(100):
+        tti.entries.append(TurnTagEntry(turn_number=i, message_hash=f"hash_{i}", tags=[f"tag_{i}"]))
+
+    mock_store = MagicMock()
+    mock_store.get_chain_snapshots_for_conversation.return_value = [
+        {"ref": "chain_50_abc123", "turn_number": 50, "tool_output_refs": "tool_ref1,tool_ref2", "message_count": 4},
+        {"ref": "chain_80_def456", "turn_number": 80, "tool_output_refs": "tool_ref3", "message_count": 3},
+    ]
+    mock_store.get_tool_names_for_refs.return_value = ["web_search", "Read"]
+
+    result, count, refs, recovered = collapse_turn_chains(
+        body, fmt,
+        protected_recent_turns=6,
+        turn_tag_index=tti,
+        store=mock_store,
+        conversation_id="test-conv",
+        client_truncated=True,
+    )
+
+    mock_store.get_chain_snapshots_for_conversation.assert_called_once()
+    assert recovered == 2
+    assert "chain_50_abc123" in refs
+    assert "chain_80_def456" in refs
+
+    import json
+    result_json = json.dumps(result)
+    assert "Compacted turn 50" in result_json
+    assert "Compacted turn 80" in result_json
+    assert "vc_restore_tool" in result_json
+
+
+def test_fill_pass_restores_from_store_on_truncation():
+    from unittest.mock import MagicMock
+    from virtual_context.proxy.message_filter import fill_pass
+    from virtual_context.proxy.formats import detect_format
+    from virtual_context.types import AssembledContext, RetrievalResult
+    from virtual_context.core.turn_tag_index import TurnTagIndex, TurnTagEntry
+    import copy
+
+    body = {
+        "system": "test",
+        "messages": [
+            {"role": "user", "content": "recent question"},
+            {"role": "assistant", "content": [{"type": "text", "text": "recent reply"}]},
+            {"role": "user", "content": "current question"},
+        ],
+        "model": "claude-opus-4-6",
+    }
+    fmt = detect_format(body)
+
+    mock_store = MagicMock()
+    mock_store.get_all_tag_summaries.return_value = []
+    mock_store.load_recent_turn_messages.return_value = [
+        (10, "older question about cooking", "I explained Italian techniques"),
+        (11, "what about baking?", "Bread baking involves..."),
+    ]
+    mock_store.get_tool_outputs_for_turn.return_value = []
+
+    assembled = AssembledContext(
+        presented_segment_refs=set(),
+        presented_tags=set(),
+        tag_sections={},
+        retrieval_result=RetrievalResult(),
+    )
+
+    tti = TurnTagIndex()
+    for i in range(50):
+        tti.entries.append(TurnTagEntry(turn_number=i, message_hash=f"hash_{i}", tags=[f"tag_{i}"]))
+
+    result_body, summaries, turns = fill_pass(
+        body=body, fmt=fmt,
+        outbound_tokens=70000, target_tokens=90000,
+        assembled=assembled, pre_filter_body=copy.deepcopy(body),
+        store=mock_store, conversation_id="test-conv",
+        summary_ratio=0.0,
+        client_truncated=True,
+        turn_tag_index=tti,
+    )
+
+    mock_store.load_recent_turn_messages.assert_called_once()
+    assert turns >= 1
+
+
+def test_fill_pass_skips_tool_turns_from_store():
+    from unittest.mock import MagicMock
+    from virtual_context.proxy.message_filter import fill_pass
+    from virtual_context.proxy.formats import detect_format
+    from virtual_context.types import AssembledContext, RetrievalResult
+    from virtual_context.core.turn_tag_index import TurnTagIndex, TurnTagEntry
+    import copy, json
+
+    body = {
+        "system": "test",
+        "messages": [{"role": "user", "content": "current question"}],
+        "model": "claude-opus-4-6",
+    }
+    fmt = detect_format(body)
+
+    mock_store = MagicMock()
+    mock_store.get_all_tag_summaries.return_value = []
+    mock_store.load_recent_turn_messages.return_value = [
+        (10, "run the tests", "Here are the results"),
+        (11, "what about baking?", "Bread baking involves..."),
+    ]
+    mock_store.get_tool_outputs_for_turn.side_effect = lambda cid, tn: ["ref1"] if tn == 10 else []
+
+    assembled = AssembledContext(
+        presented_segment_refs=set(), presented_tags=set(),
+        tag_sections={}, retrieval_result=RetrievalResult(),
+    )
+
+    tti = TurnTagIndex()
+    for i in range(50):
+        tti.entries.append(TurnTagEntry(turn_number=i, message_hash=f"hash_{i}", tags=[f"tag_{i}"]))
+
+    result_body, summaries, turns = fill_pass(
+        body=body, fmt=fmt,
+        outbound_tokens=70000, target_tokens=90000,
+        assembled=assembled, pre_filter_body=copy.deepcopy(body),
+        store=mock_store, conversation_id="test-conv",
+        summary_ratio=0.0,
+        client_truncated=True,
+        turn_tag_index=tti,
+    )
+
+    result_json = json.dumps(result_body)
+    assert "Bread baking" in result_json
+    assert "run the tests" not in result_json

virtual_context-0.2.9/virtual-context.yaml

@@ -0,0 +1,109 @@
+# virtual-context configuration — agentic preset
+# Generated by: virtual-context init agentic
+# Docs: https://github.com/virtual-context/virtual-context
+#
+# Balanced defaults for autonomous agent conversations across any domain.
+# Tags emerge from conversation — no predefined domains to configure.
+
+version: "0.2"
+storage_root: ".virtualcontext"
+context_window: 120000
+token_counter: "estimate"
+
+paging:
+  enabled: true
+  auto_promote: true
+  auto_evict: true
+  max_tool_loops: 10
+
+tag_generator:
+  type: "llm"
+  provider: "ollama"
+  model: "qwen3:4b-instruct-2507-fp16"
+  max_tags: 10
+  min_tags: 5
+  max_tokens: 8192
+  prompt_mode: "compact"
+  context_lookback_pairs: 5
+  context_bleed_threshold: 0.1
+  keyword_fallback:
+    tag_keywords:
+      legal: [court, filing, motion, attorney, judge, case]
+      medical: [medication, doctor, lab, symptom, dose, blood]
+      code: [function, bug, deploy, git, API, database, error]
+    tag_patterns:
+      legal:
+        - "\\b\\d{2}-cv-\\d+\\b"
+        - "\\bN\\.?J\\.?S\\.?A\\.?\\b"
+      code:
+        - "\\bdef \\w+\\("
+        - "\\bclass \\w+[:(]"
+
+tag_rules:
+  - match: "legal*"
+    priority: 9
+    ttl_days: null
+  - match: "medical*"
+    priority: 8
+    ttl_days: 90
+  - match: "debug*"
+    priority: 7
+    ttl_days: 7
+  - match: "*"
+    priority: 5
+    ttl_days: 30
+
+compaction:
+  soft_threshold: 0.70
+  hard_threshold: 0.85
+  protected_recent_turns: 6
+  overflow_buffer: 1.2
+  summary_ratio: 0.15
+  min_summary_tokens: 200
+  max_summary_tokens: 2000
+  max_concurrent_summaries: 4
+  llm_token_overhead: 2000
+
+summarization:
+  provider: "ollama"
+  model: "qwen3:4b-instruct-2507-fp16"
+  max_tokens: 1000
+  temperature: 0.3
+
+providers:
+  ollama:
+    type: "generic_openai"
+    base_url: "http://127.0.0.1:11434/v1"
+
+storage:
+  backend: "sqlite"
+  sqlite:
+    path: ".virtualcontext/store.db"
+
+facts:
+  graph_links: true
+
+assembly:
+  core_context_max_tokens: 18000
+  tag_context_max_tokens: 30000
+  facts_max_tokens: 20000
+  context_hint_enabled: true
+  context_hint_max_tokens: 2000
+  core_files: []
+
+retrieval:
+  skip_active_tags: true
+  active_tag_lookback: 4
+  inbound_tagger_type: "embedding"
+  embedding_model: "all-MiniLM-L6-v2"
+  embedding_threshold: 0.3
+  strategy_config:
+    default:
+      min_overlap: 1
+      max_results: 10
+      max_budget_fraction: 0.25
+      include_related: true
+
+telemetry:
+  enabled: true
+  models_file: "models.yaml"

{virtual_context-0.2.8 → virtual_context-0.2.9}/virtual_context/__init__.py

@@ -14,7 +14,7 @@ from .types import (
     VirtualContextConfig,
 )
 
-__version__ = "0.2.8"
+__version__ = "0.2.9"
 
 __all__ = [
     "VirtualContextEngine",