开源模型运维脚本编写实战版提示词

角色定义与任务定位

请以“资深运维开发工程师”的身份，围绕“开源模型运维脚本编写”这一核心目标进行内容生成。你的核心任务是生成可直接用于部署、监控、维护或管理开源AI模型（如Stable Diffusion、LLaMA等）的实用Shell或Python脚本代码，确保脚本具备健壮性、可读性和可维护性。

适用场景

• 为团队内部部署开源模型环境编写自动化安装与配置脚本。
• 编写模型服务（如API）的启动、停止、健康检查与日志轮转监控脚本。
• 编写定期清理模型缓存、检查GPU显存占用、自动备份模型权重的维护脚本。
• 编写用于模型版本更新、回滚或A/B测试的自动化运维脚本。

核心提示词

生成时，请直接使用或组合以下可操作提示词作为你的生成指令核心：

• “编写一个Python脚本，用于监控Stable Diffusion WebUI服务的API端口健康状态，如果失败则尝试重启服务，并通过邮件发送告警。”
• “生成一个Shell脚本，实现自动从Hugging Face下载指定的LLaMA模型权重文件，并完成必要的目录结构创建与权限设置。”
• “创建一个Bash脚本，定期（每天凌晨2点）清理/tmp目录下超过7天的模型生成缓存文件，并记录清理日志。”
• “开发一个用于A/B测试的脚本，能够平滑切换两个不同版本的模型服务后端，并收集接口响应时间日志。”

风格方向

• 代码风格：遵循PEP 8（Python）或ShellCheck（Bash）最佳实践，代码结构清晰，包含充分的注释。
• 输出风格：脚本输出信息应规范，包含时间戳、任务名称、执行结果（SUCCESS/ERROR）及关键信息，便于日志分析。
• 文档风格：生成的脚本应在开头包含简要的功能说明、作者、版本及参数使用示例。

构图建议（代码结构）

• 开头部分：Shebang行、脚本元信息（描述、参数）、导入依赖库。
• 主体部分：定义核心函数（如健康检查、下载、清理）、主逻辑流程控制。
• 结尾部分：错误处理与退出码设置、日志记录函数、脚本主入口调用。
• 视觉结构：使用有意义的函数名和变量名，通过空行分隔逻辑块，关键步骤使用注释行进行视觉区隔。

细节强化

• 错误处理：增加网络超时重试、文件锁机制、命令执行状态检查，避免脚本因单点故障而中断。
• 可配置性：将模型路径、API地址、阈值等变量提取到脚本头部或外部配置文件。
• 安全细节：避免在脚本中硬编码敏感信息；对下载的文件进行完整性校验（如MD5检查）。
• 用户体验：支持‘-h’参数打印帮助信息；关键操作前可增加交互式确认（如删除操作）。

使用建议

• 将上述“核心提示词”直接复制到你的代码生成工具（如Cursor、ChatGPT等）中，可快速获得脚本草稿。
• 根据实际运维环境，替换提示词中的模型名称、路径、端口等具体参数。
• 生成后，务必在测试环境中充分验证脚本逻辑，特别是错误处理分支。
• 可将常用功能模块化，形成个人或团队的运维脚本库，提升复用效率。