wehub-resource-sync
151 lines
4.7 KiB
Python
151 lines
4.7 KiB
Python
from __future__ import annotations
|
|
|
|
from opensquilla.provider import ContentBlockCompaction
|
|
from opensquilla.session.context_view import (
|
|
build_compaction_context_items,
|
|
build_compaction_context_records,
|
|
build_provider_compaction_context,
|
|
)
|
|
from opensquilla.session.models import SessionContextState, SessionSummary
|
|
|
|
|
|
def test_provider_compaction_context_dropped_for_non_anthropic_provider() -> None:
|
|
# An Anthropic-native state must not replay when a different provider is
|
|
# active (the live-provider gate). A fork or mid-session switch can leave
|
|
# such state in the session; replaying it cross-provider corrupts the
|
|
# request, so the builder returns empty for non-anthropic providers.
|
|
anthropic_state = SessionContextState(
|
|
session_id="session",
|
|
session_key="agent:main:ctx",
|
|
provider="anthropic",
|
|
model="claude-opus-4-7",
|
|
state_kind="anthropic_compaction_block",
|
|
payload={"content": "native state"},
|
|
covered_through_id=9,
|
|
created_at=3000,
|
|
portable=False,
|
|
cacheable=True,
|
|
)
|
|
for provider_kind in ("openai", "gemini", "openai_responses"):
|
|
context = build_provider_compaction_context(
|
|
context_states=[anthropic_state],
|
|
provider_kind=provider_kind,
|
|
now_ms=4000,
|
|
)
|
|
assert context.messages == []
|
|
assert context.covered_through_ids == set()
|
|
|
|
|
|
def test_provider_compaction_context_prefers_latest_state_independent_of_input_order() -> None:
|
|
newer = SessionContextState(
|
|
session_id="session",
|
|
session_key="agent:main:ctx",
|
|
provider="anthropic",
|
|
model="claude-opus-4-7",
|
|
state_kind="anthropic_compaction_block",
|
|
payload={"content": "new native state"},
|
|
covered_through_id=9,
|
|
created_at=3000,
|
|
portable=False,
|
|
cacheable=True,
|
|
)
|
|
older = SessionContextState(
|
|
session_id="session",
|
|
session_key="agent:main:ctx",
|
|
provider="anthropic",
|
|
model="claude-opus-4-7",
|
|
state_kind="anthropic_compaction_block",
|
|
payload={"content": "old native state"},
|
|
covered_through_id=7,
|
|
created_at=1000,
|
|
portable=False,
|
|
cacheable=True,
|
|
)
|
|
|
|
context = build_provider_compaction_context(
|
|
context_states=[newer, older],
|
|
provider_kind="anthropic",
|
|
now_ms=4000,
|
|
)
|
|
|
|
assert context.covered_through_ids == {9}
|
|
assert len(context.messages) == 1
|
|
block = context.messages[0].content[0]
|
|
assert isinstance(block, ContentBlockCompaction)
|
|
assert block.content == "new native state"
|
|
|
|
|
|
def test_compaction_context_items_deduplicate_structured_state_by_latest_coverage() -> None:
|
|
newer = SessionContextState(
|
|
session_id="session",
|
|
session_key="agent:main:ctx",
|
|
provider="portable",
|
|
state_kind="structured_summary_v1",
|
|
payload={
|
|
"schema_version": 1,
|
|
"current_status": "new structured state",
|
|
},
|
|
covered_through_id=7,
|
|
created_at=3000,
|
|
portable=True,
|
|
cacheable=True,
|
|
)
|
|
older = SessionContextState(
|
|
session_id="session",
|
|
session_key="agent:main:ctx",
|
|
provider="portable",
|
|
state_kind="structured_summary_v1",
|
|
payload={
|
|
"schema_version": 1,
|
|
"current_status": "old structured state",
|
|
},
|
|
covered_through_id=7,
|
|
created_at=1000,
|
|
portable=True,
|
|
cacheable=True,
|
|
)
|
|
summary = SessionSummary(
|
|
session_id="session",
|
|
session_key="agent:main:ctx",
|
|
summary_text="plain summary fallback",
|
|
covered_through_id=7,
|
|
)
|
|
|
|
items = build_compaction_context_items(
|
|
context_states=[newer, older],
|
|
summaries=[summary],
|
|
now_ms=4000,
|
|
)
|
|
|
|
rendered = "\n".join(items)
|
|
assert "new structured state" in rendered
|
|
assert "old structured state" not in rendered
|
|
assert "plain summary fallback" not in rendered
|
|
|
|
|
|
def test_compaction_context_records_expose_correlation_metadata() -> None:
|
|
state = SessionContextState(
|
|
session_id="session",
|
|
session_key="agent:main:ctx",
|
|
provider="portable",
|
|
state_kind="structured_summary_v1",
|
|
payload={
|
|
"schema_version": 1,
|
|
"current_status": "structured state",
|
|
"compaction_id": "cmp_state_1",
|
|
},
|
|
covered_through_id=9,
|
|
portable=True,
|
|
cacheable=True,
|
|
)
|
|
|
|
records = build_compaction_context_records(
|
|
context_states=[state],
|
|
summaries=[],
|
|
)
|
|
|
|
assert len(records) == 1
|
|
assert records[0].compaction_id == "cmp_state_1"
|
|
assert records[0].source == "context_state"
|
|
assert records[0].covered_through_id == 9
|