Commit b4c1c54
feat(inference): expose httpx connection pool limits via network config (#6306)
# What does this PR do?
All remote inference providers use httpx internally but never exposed
httpx.Limits, so every provider was stuck with the hardcoded defaults of
100 max connections and 20 max keepalive connections regardless of the
platform's actual concurrency needs. Connection pooling is a standard
HTTP client concern platform admins should be able to tune.
Adds a LimitsConfig model (mirroring httpx.Limits) as a new field on the
existing NetworkConfig, and wires it into build_network_client_kwargs(),
the single shared function already used by OpenAIMixin (and therefore
every OpenAI-compatible remote provider), Bedrock, and the interactions
provider to build their httpx clients from TLS/proxy/timeout/headers
config. VertexAI builds its httpx client separately via the Google GenAI
SDK, so its build_httpx_kwargs() helper got the same handling.
Regenerated provider docs via provider_codegen.py to reflect the new
network.limits.* fields across all remote inference providers.
Closes #5910
---------
Signed-off-by: elina priyadarshinee <elina.priyadarshinee1@ibm.com>
Signed-off-by: Matthew Farrellee <matt@cs.wisc.edu>
Co-authored-by: Matthew Farrellee <matt@cs.wisc.edu>1 parent ebaaff8 commit b4c1c54
33 files changed
Lines changed: 451 additions & 149 deletions
File tree
- docs/docs/providers/inference
- src/ogx/providers
- inline/interactions
- remote/inference
- vertexai
- vllm
- utils/inference
- tests/unit/providers
- inference
- vertexai
- utils/inference
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
41 | 41 | | |
42 | 42 | | |
43 | 43 | | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
| 47 | + | |
44 | 48 | | |
45 | 49 | | |
46 | 50 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
0 commit comments