540b4f5b01
First-PR scope from #1. Single-process Python daemon that relays between Claude Code instances and chat-Claude (Anthropic API). Components: * relay.config — .env + config.yaml loader. Auto-generates ntfy topic on first run and persists it back to .env. * relay.state — atomic file I/O via tempfile + rename, advisory flock at state/.lock to enforce single-instance. * relay.conversation — append-only history with summarization. Triggers a summarize call when total chars exceed HISTORY_CHAR_CAP (default 400k); replaces history with the summary plus the most recent 10 turns. * relay.anthropic_client — SDK wrapper. Marks the system prompt cacheable (5-min ephemeral cache); concatenates text blocks; estimates per-call cost from the Anthropic price table with cache-write/read accounted for. * relay.queue — JSON envelope intake; oldest-by-mtime; malformed envelopes moved to queue/.rejected/. * relay.dispatch — one-input-at-a-time per session (dispatch/<session_id>/input.txt). Won't overwrite a pending dispatch; queues internally and waits for CC to delete. * relay.ntfy — best-effort POST to https://ntfy.sh/<topic>; failures logged but never block the main loop. * relay.daemon — main loop. Polls jc_input.txt (priority) then queue/. Detects [NEEDS-JC] in the first 200 chars of any response and pauses dispatch until JC writes jc_input.txt. JC override supports @session-N: prefix for direct dispatch without an API call. * relay.__main__ — CLI: relay run / relay status / relay topic. Tests: 57 unit tests pass (config, state, conversation, queue, dispatch, anthropic_client, ntfy, full daemon loop with a fake client). One real-API smoke test marked real_api, opt-in via pytest -m real_api; skips cleanly on credit-balance errors. Out of scope for this PR (deferred to follow-ups): Flask status endpoint, multi-session config in production, exponential backoff, systemd unit, cost-tracking aggregation. Closes #1. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
49 lines
1.9 KiB
Python
49 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from relay.dispatch import DispatchManager
|
|
|
|
|
|
def test_writes_input_when_session_dir_empty(tmp_path: Path) -> None:
|
|
mgr = DispatchManager(tmp_path / "dispatch")
|
|
delivered = mgr.queue_or_write("session-1", "hello")
|
|
assert delivered is True
|
|
assert (tmp_path / "dispatch" / "session-1" / "input.txt").read_text() == "hello"
|
|
|
|
|
|
def test_does_not_overwrite_pending_input(tmp_path: Path) -> None:
|
|
mgr = DispatchManager(tmp_path / "dispatch")
|
|
mgr.queue_or_write("session-1", "first")
|
|
delivered = mgr.queue_or_write("session-1", "second")
|
|
assert delivered is False
|
|
# First file unchanged
|
|
assert (tmp_path / "dispatch" / "session-1" / "input.txt").read_text() == "first"
|
|
assert mgr.has_pending("session-1")
|
|
|
|
|
|
def test_flush_all_delivers_after_consumer_deletes(tmp_path: Path) -> None:
|
|
mgr = DispatchManager(tmp_path / "dispatch")
|
|
mgr.queue_or_write("session-1", "first")
|
|
mgr.queue_or_write("session-1", "second")
|
|
# Consumer "consumes" first
|
|
(tmp_path / "dispatch" / "session-1" / "input.txt").unlink()
|
|
delivered = mgr.flush_all()
|
|
assert delivered == 1
|
|
assert (tmp_path / "dispatch" / "session-1" / "input.txt").read_text() == "second"
|
|
assert not mgr.has_pending("session-1")
|
|
|
|
|
|
def test_independent_sessions_do_not_block_each_other(tmp_path: Path) -> None:
|
|
mgr = DispatchManager(tmp_path / "dispatch")
|
|
mgr.queue_or_write("session-1", "a") # blocks on session-1
|
|
delivered = mgr.queue_or_write("session-2", "b") # session-2 unaffected
|
|
assert delivered is True
|
|
assert (tmp_path / "dispatch" / "session-2" / "input.txt").read_text() == "b"
|
|
|
|
|
|
def test_input_path_resolution(tmp_path: Path) -> None:
|
|
mgr = DispatchManager(tmp_path / "dispatch")
|
|
p = mgr.input_path("session-1")
|
|
assert p == tmp_path / "dispatch" / "session-1" / "input.txt"
|