Skip to content

Commit 7784be8

Browse files
authored
Merge pull request #22 from kagenti/fix/inference-scoped-provider-lookup
Fix: Use scoped provider lookup in inference set
2 parents 0a01b7b + 0f7113b commit 7784be8

1 file changed

Lines changed: 11 additions & 7 deletions

File tree

crates/openshell-server/src/inference.rs

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -297,13 +297,17 @@ async fn upsert_cluster_inference_route(
297297
return Err(Status::invalid_argument("model_id is required"));
298298
}
299299

300-
let provider = store
301-
.get_message_by_name::<Provider>(provider_name)
302-
.await
303-
.map_err(|e| Status::internal(format!("fetch provider failed: {e}")))?
304-
.ok_or_else(|| {
305-
Status::failed_precondition(format!("provider '{provider_name}' not found"))
306-
})?;
300+
let provider_record = crate::auth::ownership::resolve_scoped_name(
301+
store,
302+
Provider::object_type(),
303+
provider_name,
304+
principal,
305+
admin_role,
306+
)
307+
.await?
308+
.ok_or_else(|| Status::failed_precondition(format!("provider '{provider_name}' not found")))?;
309+
let provider = Provider::decode(provider_record.payload.as_slice())
310+
.map_err(|e| Status::internal(format!("decode provider failed: {e}")))?;
307311

308312
let empty = std::collections::HashMap::new();
309313
let provider_labels = provider.metadata.as_ref().map_or(&empty, |m| &m.labels);

0 commit comments

Comments
 (0)