From 4704d57e17b62ac347efc7eb9b9b8d5e178de2f6 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Mon, 23 Feb 2026 16:27:16 +0700 Subject: [PATCH] rename client --- crates/lib/artcraft_router/src/client/mod.rs | 4 +-- .../src/client/router_client.rs | 13 ---------- .../src/client/router_client_builder.rs | 25 ------------------- .../generate_video/generate_video_request.rs | 4 +-- 4 files changed, 4 insertions(+), 42 deletions(-) delete mode 100644 crates/lib/artcraft_router/src/client/router_client.rs delete mode 100644 crates/lib/artcraft_router/src/client/router_client_builder.rs diff --git a/crates/lib/artcraft_router/src/client/mod.rs b/crates/lib/artcraft_router/src/client/mod.rs index 83363dd43e..c7357a2280 100644 --- a/crates/lib/artcraft_router/src/client/mod.rs +++ b/crates/lib/artcraft_router/src/client/mod.rs @@ -1,4 +1,4 @@ +pub mod multi_router_client; +pub mod multi_router_client_builder; pub mod request_mismatch_mitigation_strategy; pub mod router_artcraft_client; -pub mod router_client; -pub mod router_client_builder; diff --git a/crates/lib/artcraft_router/src/client/router_client.rs b/crates/lib/artcraft_router/src/client/router_client.rs deleted file mode 100644 index f9925fbebb..0000000000 --- a/crates/lib/artcraft_router/src/client/router_client.rs +++ /dev/null @@ -1,13 +0,0 @@ -use crate::client::router_artcraft_client::RouterArtcraftClient; -use crate::errors::client_error::{ClientError, ClientType}; - -pub struct RouterClient { - pub(crate) artcraft_client: Option, -} - -impl RouterClient { - pub fn get_artcraft_client_ref(&self) -> Result<&RouterArtcraftClient, ClientError> { - self.artcraft_client.as_ref() - .ok_or(ClientError::ClientNotConfigured(ClientType::Artcraft)) - } -} diff --git a/crates/lib/artcraft_router/src/client/router_client_builder.rs b/crates/lib/artcraft_router/src/client/router_client_builder.rs deleted file mode 100644 index 207c7f483d..0000000000 --- a/crates/lib/artcraft_router/src/client/router_client_builder.rs +++ /dev/null @@ -1,25 +0,0 @@ -use crate::client::router_artcraft_client::RouterArtcraftClient; -use crate::client::router_client::RouterClient; - -pub struct RouterClientBuilder { - artcraft_client: Option, -} - -impl RouterClientBuilder { - pub fn new() -> Self { - Self { - artcraft_client: None, - } - } - - pub fn set_artcraft_client(mut self, client: RouterArtcraftClient) -> Self { - self.artcraft_client = Some(client); - self - } - - pub fn build(self) -> RouterClient { - RouterClient { - artcraft_client: self.artcraft_client, - } - } -} diff --git a/crates/lib/artcraft_router/src/generate/generate_video/generate_video_request.rs b/crates/lib/artcraft_router/src/generate/generate_video/generate_video_request.rs index a162908f9b..64cead2dc5 100644 --- a/crates/lib/artcraft_router/src/generate/generate_video/generate_video_request.rs +++ b/crates/lib/artcraft_router/src/generate/generate_video/generate_video_request.rs @@ -4,7 +4,7 @@ use crate::api::common_video_model::CommonVideoModel; use crate::api::image_list_ref::ImageListRef; use crate::api::image_ref::ImageRef; use crate::api::provider::Provider; -use crate::client::router_client::RouterClient; +use crate::client::multi_router_client::MultiRouterClient; use crate::client::request_mismatch_mitigation_strategy::RequestMismatchMitigationStrategy; pub struct GenerateVideoRequest<'a> { @@ -53,7 +53,7 @@ pub struct GenerateVideoRequest<'a> { pub idempotency_token: Option<&'a str>, /// The polymorphic router client, which can dispatch to multiple providers. - pub client: &'a RouterClient, + pub client: &'a MultiRouterClient, } impl <'a> GenerateVideoRequest<'a> {