红尘
d73ff9b0fb
Deploy Site / deploy-vercel (push) Has been cancelled
Deploy Site / deploy-docs (push) Has been cancelled
Docker / shell lint / Lint Dockerfile (hadolint) (push) Has been cancelled
Docker / shell lint / Lint docker/ shell scripts (shellcheck) (push) Has been cancelled
Docker Build and Publish / build-amd64 (push) Has been cancelled
Docker Build and Publish / build-arm64 (push) Has been cancelled
Lint (ruff + ty) / ruff + ty diff (push) Has been cancelled
Lint (ruff + ty) / ruff enforcement (blocking) (push) Has been cancelled
Lint (ruff + ty) / Windows footguns (blocking) (push) Has been cancelled
Nix Lockfile Fix / auto-fix-main (push) Has been cancelled
Nix Lockfile Fix / fix (push) Has been cancelled
Nix / nix (macos-latest) (push) Has been cancelled
Nix / nix (ubuntu-latest) (push) Has been cancelled
OSV-Scanner / Scan lockfiles (push) Has been cancelled
Build Skills Index / build-index (push) Has been cancelled
Tests / test (1) (push) Has been cancelled
Tests / test (2) (push) Has been cancelled
Tests / test (3) (push) Has been cancelled
Tests / test (4) (push) Has been cancelled
Tests / test (5) (push) Has been cancelled
Tests / test (6) (push) Has been cancelled
Tests / e2e (push) Has been cancelled
uv.lock check / uv lock --check (push) Has been cancelled
Docker Build and Publish / merge (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
Tests / save-durations (push) Has been cancelled
23 lines
937 B
Python
23 lines
937 B
Python
"""Test that last_total_tokens is correctly set by ContextCompressor."""
|
|
|
|
from agent.context_compressor import ContextCompressor
|
|
|
|
|
|
def test_update_from_response_sets_total_tokens():
|
|
"""ABC contract: last_total_tokens must be set from API response."""
|
|
c = ContextCompressor(model="test", quiet_mode=True, config_context_length=200000)
|
|
|
|
c.update_from_response({"prompt_tokens": 100, "completion_tokens": 30, "total_tokens": 130})
|
|
assert c.last_total_tokens == 130
|
|
|
|
c.update_from_response({"prompt_tokens": 100, "completion_tokens": 30})
|
|
assert c.last_total_tokens == 130
|
|
|
|
|
|
def test_session_reset_clears_total_tokens():
|
|
"""on_session_reset must zero total_tokens."""
|
|
c = ContextCompressor(model="test", quiet_mode=True, config_context_length=200000)
|
|
c.update_from_response({"prompt_tokens": 100, "completion_tokens": 30, "total_tokens": 130})
|
|
c.on_session_reset()
|
|
assert c.last_total_tokens == 0
|