This commit is contained in:
Authensor 2026-11-06 22:06:20 +00:00 • committed by GitHub
commit c141cc3143
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2381,6 +2381,16 @@ def get_model_from_request(
if vertex_match:
model = vertex_match.group(1)
# Watsonx routes carry the deployment name under "model_id" (the native
# Watsonx ML API shape). The generic "model" body key can also be set by
# the caller, but Watsonx executes model_id; both must be authorized,
# so model_id is checked whenever present on a /watsonx route,
# regardless of whether "model" already resolved.
if route.lower().startswith("/watsonx"):
watsonx_model: Final = request_data.get("model_id")
if isinstance(watsonx_model, str) and watsonx_model:
return watsonx_model
if route.lower().startswith("/bedrock"):
bedrock_model: Final = _model_from_bedrock_route(route)
return model if bedrock_model is None else bedrock_model