mirror/nanobot
1
0
Fork 0
mirror of https://github.com/HKUDS/nanobot.git synced 2026-09-04 02:01:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

refactor(providers): define typed usage contract

Browse Source
This commit is contained in:
chengyongru
2026-08-25 01:04:25 +08:00
committed by chengyongru
parent 89c94d8744
commit 9895c23cb5
84 changed files with 1643 additions and 726 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_build_status.py
+10 -9
View File
@@ -1,5 +1,6 @@
"""Tests for build_status_content cache hit rate display."""
from nanobot.providers.base import LLMUsage
from nanobot.utils.helpers import build_status_content
@@ -8,7 +9,7 @@ def test_status_shows_cache_hit_rate():
version="0.1.0",
model="glm-4-plus",
start_time=1000000.0,
last_usage={"prompt_tokens": 2000, "completion_tokens": 300, "cached_tokens": 1200},
last_usage=LLMUsage.reported(input_tokens=2000, output_tokens=300, cache_read_tokens=1200),
context_window_tokens=128000,
session_msg_count=10,
context_tokens_estimate=5000,
@@ -19,12 +20,12 @@ def test_status_shows_cache_hit_rate():
def test_status_no_cache_info():
"""Without cached_tokens, display should not show cache percentage."""
"""Without a reported cache-read count, omit the cache percentage."""
content = build_status_content(
version="0.1.0",
model="glm-4-plus",
start_time=1000000.0,
last_usage={"prompt_tokens": 2000, "completion_tokens": 300},
last_usage=LLMUsage.reported(input_tokens=2000, output_tokens=300),
context_window_tokens=128000,
session_msg_count=10,
context_tokens_estimate=5000,
@@ -34,13 +35,13 @@ def test_status_no_cache_info():
assert "Tasks: 0 active" in content
def test_status_zero_cached_tokens():
"""cached_tokens=0 should not show cache percentage."""
def test_status_zero_cache_read_tokens():
"""An explicit zero cache-read count should not show cache percentage."""
content = build_status_content(
version="0.1.0",
model="glm-4-plus",
start_time=1000000.0,
last_usage={"prompt_tokens": 2000, "completion_tokens": 300, "cached_tokens": 0},
last_usage=LLMUsage.reported(input_tokens=2000, output_tokens=300, cache_read_tokens=0),
context_window_tokens=128000,
session_msg_count=10,
context_tokens_estimate=5000,
@@ -53,7 +54,7 @@ def test_status_100_percent_cached():
version="0.1.0",
model="glm-4-plus",
start_time=1000000.0,
last_usage={"prompt_tokens": 1000, "completion_tokens": 100, "cached_tokens": 1000},
last_usage=LLMUsage.reported(input_tokens=1000, output_tokens=100, cache_read_tokens=1000),
context_window_tokens=128000,
session_msg_count=5,
context_tokens_estimate=3000,
@@ -67,7 +68,7 @@ def test_status_context_pct_uses_budget_not_total():
version="0.1.0",
model="test",
start_time=1000000.0,
last_usage={"prompt_tokens": 2000, "completion_tokens": 300},
last_usage=LLMUsage.reported(input_tokens=2000, output_tokens=300),
context_window_tokens=128000,
session_msg_count=10,
context_tokens_estimate=120000,
@@ -83,7 +84,7 @@ def test_status_context_pct_capped_at_999():
version="0.1.0",
model="test",
start_time=1000000.0,
last_usage={"prompt_tokens": 2000, "completion_tokens": 300},
last_usage=LLMUsage.reported(input_tokens=2000, output_tokens=300),
context_window_tokens=10000,
session_msg_count=10,
context_tokens_estimate=100000,