Commit Graph

5 Commits

Author SHA1 Message Date
imeepos 2075a731d2 fix: 修复ComfyUI队列API响应格式解析问题
- 修复WebSocket错误消息解析失败问题,支持更多错误字段格式
- 改进执行错误处理,收到错误时立即中断而不是等待超时
- 修复ComfyUI队列API响应格式不匹配问题,支持数组格式的队列数据
- 添加灵活的队列项解析逻辑,支持整数和字符串类型的prompt_id
- 在EventEmitter中添加错误存储功能,支持实时错误状态检查
- 更新所有相关的队列状态转换代码

解决的问题:
1. WebSocket消息解析错误:missing field 'message'
2. 执行错误后仍等待超时的问题
3. 队列API响应解析错误:invalid type integer/string expected
4. 批量处理中错误统计不准确的问题
2025-08-11 11:09:27 +08:00
root 84081284f7 fix: 修复ComfyUI V2模板创建参数解析问题并重构工作流创建
主要修改:

1. **扩展ComfyUI SDK参数类型支持**
   - 在ParameterType枚举中添加Integer、Float、Image、Audio、Video类型
   - 在ParameterSchema中添加节点映射和媒体文件相关字段
   - 添加step、accept、maxSize、width、height、duration、node_mapping字段
   - 为新字段添加serde默认值支持

2. **修复参数解析问题**
   - 解决comfyui_v2_create_template命令缺少type字段的错误
   - 确保前端传递的参数类型与后端ParameterType枚举匹配
   - 修复r#enum字段的正确使用(用于JSON Schema兼容性)

3. **重构工作流模板创建方式**
   - 将ai_model_face_hair_fix.rs中的create_workflow函数重构为create_workflow_from_json
   - 改为从JSON字符串解析ComfyUIWorkflow,而非手动构建HashMap
   - 提高可维护性和与实际使用场景的一致性

4. **更新相关测试和验证代码**
   - 修复validation.rs中的测试用例,添加新字段的默认值
   - 确保所有ParameterSchema创建都包含完整字段

5. **添加测试数据**
   - 创建test_template_creation.json用于测试参数格式

这些修改解决了前端传递的参数格式与后端期望格式不匹配的问题,
使工作流模板创建功能能够正常工作。
2025-08-08 23:10:38 +08:00
imeepos 3bb7cdae23 fix: cargo check --lib error 2025-08-08 18:16:52 +08:00
imeepos d61e2e8f4c fix: comfyui sdk serde 2025-08-08 15:03:35 +08:00
imeepos 5f6a302dfd feat: add comfyui sdk 2025-08-08 13:48:38 +08:00