chore: commit local changes

This commit is contained in:
吉鑫升 committed 2026-06-04 19:35:56 +08:00
1 parent c1336ad5b3
commit e6e0832c8a
15 files changed
+291 -24

No files matched your search

@@ -217,6 +217,29 @@ async def test_sql_store_filters_errors_only(sql_store):
assert only_errors[0]["error_type"] == "APITimeoutError"
@pytest.mark.asyncio
async def test_sql_store_filters_by_user_ids(sql_store):
"""``user_ids`` (set by a fuzzy username search) matches several accounts."""
base = datetime.now(UTC)
for uid in ("alice", "alicia", "bob"):
await sql_store.record(
LlmMetricRecord(
id=uuid.uuid4().hex,
created_at=base,
user_id=uid,
duration_ms=1,
status="success",
)
)
# Fuzzy "ali" → {alice, alicia}; bob must be excluded.
rows = await sql_store.list(LlmMetricsQuery(user_ids=["alice", "alicia"], limit=10))
assert {r["user_id"] for r in rows} == {"alice", "alicia"}
assert await sql_store.count(LlmMetricsQuery(user_ids=["alice", "alicia"])) == 2
# user_ids takes precedence over a single user_id.
rows = await sql_store.list(LlmMetricsQuery(user_id="bob", user_ids=["alice"], limit=10))
assert {r["user_id"] for r in rows} == {"alice"}
@pytest.mark.asyncio
async def test_sql_store_iter_all_paginates(sql_store):
base = datetime.now(UTC)