Files
ai-agent-book/chapter3/memobase/test_evaluate_generic_empty_query.py
liqiang b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
ai-agent-book 精选快照(<2MB 代码与文档,来自 github.com/bojieli/ai-agent-book)
2026-08-20 13:12:50 +00:00

39 lines
1.3 KiB
Python

"""Regression tests: benchmark evaluation must not raise ZeroDivisionError on
tasks with an empty/whitespace query (e.g. loaded from tasks.json) that fall
through to the generic evaluator."""
import os
import sys
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from locomo_benchmark import BenchmarkTask, LOCOMOBenchmark
LONG_RESPONSE = (
"This is a reasonably long agent response. It has several sentences. "
"And it keeps going with more detail. Even more detail here."
)
def _suite():
return LOCOMOBenchmark.__new__(LOCOMOBenchmark)
def test_evaluate_generic_empty_query():
task = BenchmarkTask(id="t1", category="custom_category", query="")
score = _suite()._evaluate_generic(task, LONG_RESPONSE)
assert 0.0 <= score <= 1.0
def test_evaluate_response_whitespace_query_unknown_category():
task = BenchmarkTask(id="t2", category="custom_category", query=" ")
result = _suite().evaluate_response(
task, response=LONG_RESPONSE, execution_time=1.0, memory_usage={}
)
assert 0.0 <= result.score <= 1.0
def test_evaluate_generic_normal_query_unchanged():
task = BenchmarkTask(id="t3", category="custom_category", query="weather forecast analysis")
score = _suite()._evaluate_generic(task, LONG_RESPONSE.replace("agent", "weather"))
assert score > 0.0