[Bugfix] Wire log_stats to AsyncOmni and add missing token metrics for non-text requests (#4482)

Signed-off-by: blonde <164845202+blondeCS@users.noreply.github.com>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
blondeandClaude Opus 4.6 authored and GitHub committed 2026-06-17 21:31:44 +08:00
1 parent ed37d8e7a2
commit 8fa0394d29
4 files changed
+36 -3

No files matched your search

@@ -646,6 +646,7 @@ async def build_async_omni_from_stage_config(
try:
kwargs = args.get_explicit_kwargs_dict()
model = kwargs.pop("model", None) or args.model
kwargs.setdefault("log_stats", not args.disable_log_stats)
async_omni = AsyncOmni(model=model, **kwargs)
# # Don't keep the dummy data in memory