From 628eb67e73cc754e56a94c1292a3da50681873c3 Mon Sep 17 00:00:00 2001 From: Junji Takakura Date: Sat, 9 Aug 2025 20:58:29 +0900 Subject: [PATCH] fix(wadm-client): update model list topic to use 'get' instead of 'list' Signed-off-by: Junji Takakura --- crates/wadm-client/src/lib.rs | 11 +++++++---- crates/wadm-client/src/topics.rs | 2 +- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/crates/wadm-client/src/lib.rs b/crates/wadm-client/src/lib.rs index 5f59f85f..76cfa1f2 100644 --- a/crates/wadm-client/src/lib.rs +++ b/crates/wadm-client/src/lib.rs @@ -10,8 +10,8 @@ use wadm_types::{ api::{ DeleteModelRequest, DeleteModelResponse, DeleteResult, DeployModelRequest, DeployModelResponse, DeployResult, GetModelRequest, GetModelResponse, GetResult, - ModelSummary, PutModelResponse, PutResult, Status, StatusResponse, StatusResult, - VersionInfo, VersionResponse, + ListModelsResponse, ModelSummary, PutModelResponse, PutResult, Status, StatusResponse, + StatusResult, VersionInfo, VersionResponse, }, Manifest, }; @@ -129,9 +129,12 @@ impl Client { .client .request(topic, Vec::with_capacity(0).into()) .await?; - let body: Vec = + let body: ListModelsResponse = serde_json::from_slice(&resp.payload).map_err(SerializationError::from)?; - Ok(body) + if matches!(body.result, GetResult::Error) { + return Err(ClientError::ApiError(body.message)); + } + Ok(body.models) } /// Gets a manifest from the lattice by name and optionally its version. If no version is set, diff --git a/crates/wadm-client/src/topics.rs b/crates/wadm-client/src/topics.rs index 297581e9..6b60d92c 100644 --- a/crates/wadm-client/src/topics.rs +++ b/crates/wadm-client/src/topics.rs @@ -49,7 +49,7 @@ impl TopicGenerator { /// Returns the full topic for a model list operation pub fn model_list_topic(&self) -> String { - format!("{}.list", self.model_prefix()) + format!("{}.get", self.model_prefix()) } /// Returns the full topic for listing the versions of a model