Use websocket v2 as model-preferred websocket protocol (#12838)

This commit is contained in:
pakrym-oai
2026-02-25 16:35:53 -08:00
committed by GitHub
parent a1cd78c818
commit 4fedef88e0
2 changed files with 30 additions and 10 deletions

View File

@@ -395,8 +395,8 @@ impl ModelClient {
/// This combines provider capability and feature gating; both must be true for websocket paths
/// to be eligible.
///
/// If websockets are only enabled via model preference (no explicit feature flag), default to
/// v1 behavior.
/// If websockets are only enabled via model preference (no explicit feature flag), prefer the
/// current v2 behavior.
pub fn active_ws_version(&self, model_info: &ModelInfo) -> Option<ResponsesWebsocketVersion> {
if !self.state.provider.supports_websockets
|| self.state.disable_websockets.load(Ordering::Relaxed)
@@ -406,7 +406,7 @@ impl ModelClient {
match self.state.responses_websocket_version {
Some(version) => Some(version),
None if model_info.prefer_websockets => Some(ResponsesWebsocketVersion::V1),
None if model_info.prefer_websockets => Some(ResponsesWebsocketVersion::V2),
None => None,
}
}