Refactor rate limit handling (#280)

* - Refactored retry for rate limits into the base class
- ConsumerProducer is derived from Consumer to simplify code
- Added rate_limit_count metrics for rate limit events

* Add rate limit events to VertexAI and Google AI Studio

* Added Grafana rate limit dashboard

* Add rate limit handling to all LLMs
This commit is contained in:
cybermaggedon 2025-01-27 17:04:49 +00:00 committed by GitHub
parent 26a586034c
commit 0e03bc05a4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
14 changed files with 174 additions and 298 deletions

View file

@ -577,7 +577,7 @@
"disableTextWrap": false,
"editorMode": "builder",
"exemplar": false,
"expr": "increase(processing_count_total{status!=\"success\"}[$__rate_interval])",
"expr": "sum by(job) (increase(rate_limit_count_total[$__rate_interval]))",
"format": "time_series",
"fullMetaSearch": false,
"includeNullMetadata": true,
@ -588,7 +588,7 @@
"useBackend": false
}
],
"title": "Errors",
"title": "Rate limit events",
"type": "timeseries"
},
{