mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-06 07:51:53 +00:00
fix(compressor): ABC compliance — total_tokens, api_mode, logger consistency
This commit is contained in:
parent
75643a6154
commit
8b2adead78
5 changed files with 29 additions and 5 deletions
22
tests/agent/test_last_total_tokens.py
Normal file
22
tests/agent/test_last_total_tokens.py
Normal file
|
|
@ -0,0 +1,22 @@
|
|||
"""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
|
||||
|
|
@ -87,5 +87,4 @@ def test_plugin_engine_update_model_args():
|
|||
assert kw["context_length"] == 131_072
|
||||
assert "model" in kw
|
||||
assert "provider" in kw
|
||||
# Should NOT pass api_mode — the ABC doesn't accept it
|
||||
assert "api_mode" not in kw
|
||||
assert "api_mode" in kw
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue