From 6cc4e21044e22b465e40012d3fe6ac0c0afc77ca Mon Sep 17 00:00:00 2001
From: imeepos <imeepos@sker.com>
Date: Fri, 8 Aug 2025 15:39:25 +0800
Subject: [PATCH] fix: cargo check --lib error

---
 .../src/business/services/comfyui_manager.rs         | 12 +++---------
 .../src/business/services/template_engine.rs         |  4 ++--
 2 files changed, 5 insertions(+), 11 deletions(-)

diff --git a/apps/desktop/src-tauri/src/business/services/comfyui_manager.rs b/apps/desktop/src-tauri/src/business/services/comfyui_manager.rs
index dfa2953..ec7c66a 100644
--- a/apps/desktop/src-tauri/src/business/services/comfyui_manager.rs
+++ b/apps/desktop/src-tauri/src/business/services/comfyui_manager.rs
@@ -205,16 +205,10 @@ impl ComfyUIManager {
         self.config.read().await.clone()
     }
 
-    /// 获取客户端引用（用于高级操作）
-    pub async fn get_client(&self) -> Result<ComfyUIClient> {
+    /// 检查客户端是否连接
+    pub async fn is_connected(&self) -> bool {
         let client_guard = self.client.read().await;
-        match client_guard.as_ref() {
-            Some(client) => {
-                // 返回客户端的克隆
-                Ok(client.clone())
-            }
-            None => Err(anyhow!("客户端未连接")),
-        }
+        client_guard.is_some()
     }
 
     /// 自动重连
diff --git a/apps/desktop/src-tauri/src/business/services/template_engine.rs b/apps/desktop/src-tauri/src/business/services/template_engine.rs
index 11672a2..0b269b1 100644
--- a/apps/desktop/src-tauri/src/business/services/template_engine.rs
+++ b/apps/desktop/src-tauri/src/business/services/template_engine.rs
@@ -182,7 +182,7 @@ impl TemplateEngine {
     /// 预览模板实例（不执行）
     pub async fn preview_instance(&self, template_id: &str, parameters: ParameterValues) -> Result<ComfyUIWorkflow> {
         let instance = self.create_instance(template_id, parameters).await?;
-        Ok(instance.get_workflow().clone())
+        Ok(instance.workflow().clone())
     }
 
     /// 获取模板的参数定义
@@ -227,7 +227,7 @@ impl TemplateEngine {
 
     /// 预热模板缓存
     pub async fn warm_cache(&self) -> Result<()> {
-        let templates = self.repository.list_templates(true).await?;
+        let templates = self.repository.list_templates(true)?;
         
         {
             let mut cache = self.template_cache.write().await;