feat: add uni-comfyui-sdk - Rust SDK for ComfyUI API

- Complete Rust SDK implementation based on OpenAPI specification - All 23 endpoints implemented with type-safe interfaces - Comprehensive error handling and async support - Full documentation and examples included - Client renamed to UniComfyUIClient for clear identification
2025-08-15 15:31:33 +08:00
parent 5380d9973f
commit 08b981c31f
12 changed files with 2193 additions and 0 deletions
--- a/cargos/uni-comfyui-sdk/README.md
+++ b/cargos/uni-comfyui-sdk/README.md
@@ -0,0 +1,124 @@
+# uni-comfyui-sdk
+
+A Rust SDK for ComfyUI Workflow Service & Management API.
+
+## Features
+
+- **Status Management**: Get metrics, server status, and file listings
+- **Workflow Management**: Create, retrieve, and delete workflows  
+- **Workflow Execution**: Run workflows and monitor their status
+- **Server Management**: Register, unregister, and manage ComfyUI servers
+- **Monitoring**: Health checks, system stats, and task monitoring
+- **Type Safety**: Strongly typed API with comprehensive error handling
+
+## Installation
+
+Add this to your `Cargo.toml`:
+
+```toml
+[dependencies]
+uni-comfyui-sdk = "0.1.0"
+```
+
+## Quick Start
+
+```rust
+use uni_comfyui_sdk::{UniComfyUIClient, Result};
+use std::collections::HashMap;
+
+#[tokio::main]
+async fn main() -> Result<()> {
+    // Create a client
+    let client = UniComfyUIClient::new("http://localhost:8000")?;
+
+    // Get system metrics
+    let metrics = client.get_metrics().await?;
+    println!("Metrics: {:#?}", metrics);
+
+    // Get all workflows
+    let workflows = client.get_all_workflows().await?;
+    println!("Found {} workflows", workflows.len());
+
+    // Run a workflow
+    let mut data = HashMap::new();
+    data.insert("input".to_string(), serde_json::json!("test input"));
+    let result = client.run_workflow("my_workflow", None, data).await?;
+    println!("Workflow result: {:#?}", result);
+
+    Ok(())
+}
+```
+
+## API Coverage
+
+### Status Endpoints
+- `get_metrics()` - Get queue status overview
+- `get_servers_status()` - Get all configured ComfyUI servers status
+- `list_server_files(server_index)` - List files in server input/output folders
+
+### Workflow Endpoints
+- `get_all_workflows()` - Get all workflows
+- `publish_workflow(data)` - Publish a new workflow
+- `delete_workflow(workflow_name)` - Delete a workflow
+- `get_one_workflow(base_name, version)` - Get workflow specification
+
+### Run Endpoints
+- `run_workflow(name, version, data)` - Execute workflow asynchronously
+- `get_run_status(workflow_run_id)` - Get workflow execution status
+
+### RunX Endpoints
+- `model_with_multi_dress(version, data)` - Model with multiple dresses workflow
+- `model_with_multi_dress_json_schema()` - Get workflow definition
+
+### Monitor Endpoints
+- `monitor_dashboard()` - Get monitoring dashboard HTML
+- `get_system_stats()` - Get system statistics
+- `get_task_stats()` - Get task statistics
+- `get_monitor_server_status()` - Get server status information
+- `get_recent_tasks(limit)` - Get recent tasks list
+- `health_check()` - Health check endpoint
+
+### ComfyUI Server Management Endpoints
+- `register_server(request)` - Register ComfyUI server
+- `update_heartbeat(request)` - Update server heartbeat
+- `unregister_server(request)` - Unregister ComfyUI server
+- `get_server_status(server_name)` - Get specific server status
+- `list_all_servers()` - Get all servers list
+- `get_system_health()` - Get system overall health status
+- `force_health_check()` - Force health check execution
+
+## Error Handling
+
+The SDK provides comprehensive error handling through the `ComfyUIError` enum:
+
+```rust
+use uni_comfyui_sdk::{UniComfyUIClient, ComfyUIError};
+
+match client.get_metrics().await {
+    Ok(metrics) => println!("Success: {:#?}", metrics),
+    Err(ComfyUIError::Http(e)) => println!("HTTP error: {}", e),
+    Err(ComfyUIError::Api { status, message }) => {
+        println!("API error {}: {}", status, message)
+    }
+    Err(e) => println!("Other error: {}", e),
+}
+```
+
+## Data Models
+
+The SDK includes strongly typed models for all API requests and responses:
+
+- `ServerRegistrationRequest` - Server registration data
+- `ServerHeartbeatRequest` - Server heartbeat data
+- `ServerStatusResponse` - Server status information
+- `ServerFiles` - Server file listings
+- `FileDetails` - Individual file information
+- And more...
+
+## Examples
+
+See the `examples/` directory for more usage examples.
+
+## License
+
+MIT