chore: telemetry test (#3405)

# What does this PR do?
- removed fixed-duration sleeps

## Test Plan
This commit is contained in:
ehhuang 2025-09-10 13:19:36 -07:00 committed by GitHub
parent d4e45cd5f1
commit d2f88a10fb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 10 additions and 17 deletions

View file

@ -46,10 +46,7 @@ def setup_telemetry_metrics_data(openai_client, client_with_models, text_model_i
break
except Exception:
pass
time.sleep(1)
# Wait additional time to ensure all metrics are processed
time.sleep(5)
time.sleep(0.1)
# Return the token lists for use in tests
return {"prompt_tokens": prompt_tokens, "completion_tokens": completion_tokens, "total_tokens": total_tokens}