ComfyUI OpenAI 接口服务调用的自定义节点

🌐 语言选择 / Language Selection

一个自定义的ComfyUI节点，集成了OpenAI兼容的API，用于本地或云端部署的大模型API服务（Ollama、LM Studio,Vllm, Mindie等）调用，用于绘画提示词生成和增强，专为Stable Diffusion，FLux等模型工作流程设计。支持中文输入

功能特性

• 🤖 OpenAI兼容API集成：支持任何OpenAI兼容的API端点
• 🎨 Stable Diffusion提示词增强：自动扩展和改进提示词以获得更好的图像生成效果
• 🧠 DeepSeek R1支持：预配置了DeepSeek-R1-Distill-Qwen-32B模型
• 🧹 清洁输出：自动移除响应中的思考标签（<think>块）
• ⚙️ 灵活配置：可自定义模型、温度、令牌数和API设置

安装方法

1. 导航到ComfyUI自定义节点目录：

   cd ComfyUI/custom_nodes/
   

2. 克隆此仓库：

   git clone https://github.com/PICOPON/ComfyUI-API-OpenAI-Node.git
   

3. 安装必需的依赖：

   pip install -r requirements.txt
   

4. 重启ComfyUI

使用方法

1. 将OpenAI Node添加到您的ComfyUI工作流程中
2. 根据需要配置节点参数
3. 将输出连接到所需的下游节点

节点参数

| 参数 | 类型 | 默认值 | 描述 | |------|------|--------|------| | prompt | STRING | "user prompts" | 要处理的输入提示词 | | system_prompt | STRING | Stable Diffusion提示词生成器 | AI的系统指令 | | model | STRING | "DeepSeek-R1-Distill-Qwen-32B" | 要使用的模型名称 | | api_url | STRING | "http://127.0.0.1:1935/v1" | API端点URL | | api_key | STRING | "123" | API认证密钥 | | temperature | FLOAT | 0.6 | 创造性水平（0.0-1.0） | | max_tokens | INT | 1024 | 最大响应长度 |

默认系统提示词

节点预配置了针对Stable Diffusion优化的系统提示词：

You are a prompt generation AI. your task is to take a user input for a stable difusion prompt and output and expand the supplied prompt in a stable difusion format to provide better output. Do not deviate from the format. Do not output anything other than a stable diffusion prompt.Output English whatever the input language is.

（您是一个提示词生成AI。您的任务是接收用户输入的stable diffusion提示词，并输出扩展后的stable diffusion格式提示词以提供更好的输出。不要偏离格式。除了stable diffusion提示词之外不要输出任何其他内容。）

使用示例

基本用法

1. 输入："一只猫"
2. 输出："一只美丽蓬松的猫，详细的毛发纹理，明亮的眼睛，坐姿，柔和的光线，高质量，逼真，8k分辨率"

高级工作流程

文本输入 → OpenAI节点 → 文本处理 → 图像生成

参考流程文件AI_generated_Flux_flow.json （支持中文输入）

<img width="1912" height="962" alt="image" src="https://github.com/user-attachments/assets/c26761ca-4493-40e8-bf47-c5800f6913f0" />

API兼容性

此节点支持：

• ✅ OpenAI API
• ✅ DeepSeek API
• ✅ 本地LLM服务器（Ollama、LM Studio,Vllm, Mindie等）
• ✅ 任何OpenAI兼容的端点

配置示例

本地服务器（Ollama）

• API URL：http://localhost:1935/v1
• 模型：DeepSeek-R1-Distill-Qwen-32B或任何已安装的模型
• API密钥：保持为"123"或留空

DeepSeek API

• API URL：https://api.deepseek.com/v1
• 模型：deepseek-chat
• API密钥：您的DeepSeek API密钥

OpenAI API

• API URL：https://api.openai.com/v1
• 模型：gpt-4或gpt-3.5-turbo
• API密钥：您的OpenAI API密钥

详细功能

自动标签清理

节点自动移除AI响应中的<think>和</think>标签及其内容，确保下游处理的输出干净。

错误处理

• 网络超时保护（100秒）
• HTTP错误状态处理
• 异常捕获并提供描述性错误消息

灵活的API集成

• OpenAI兼容端点的自动URL补全
• 可选的API密钥认证
• 可配置的模型参数

故障排除

常见问题

API连接错误

• 检查API URL是否正确且可访问
• 如需认证，请验证您的API密钥
• 确保本地服务器正在运行（对于本地API）

找不到模型

• 验证您的API提供商的模型名称是否正确
• 检查您的账户/服务器中是否有该模型

超时问题

• 如需要，在代码中增加超时值
• 检查您的网络连接
• 验证API服务器状态

贡献

1. Fork此仓库
2. 创建您的功能分支（git checkout -b feature/AmazingFeature）
3. 提交您的更改（git commit -m 'Add some AmazingFeature'）
4. 推送到分支（git push origin feature/AmazingFeature）
5. 打开Pull Request

许可证

本项目采用GNU 3.0许可证 - 详情请参阅LICENSE文件。

致谢

• ComfyUI团队提供的优秀框架
• OpenAI提供的API标准
• DeepSeek提供的强大模型

支持

如果您遇到任何问题或有疑问：

1. 检查Issues页面
2. 创建新的issue并提供详细信息
3. 包括您的ComfyUI版本和错误日志

---

⭐ 如果这个节点对您的工作流程有帮助，请考虑给它一个星标！