mellea: 一个用于编写具有类型注释输出和自动修复功能的结构化、可测试生成程序的 Python 库
mellea: 一个用于编写具有类型注释输出和自动修复功能的结构化、可测试生成程序的 Python 库
它解决了什么问题
它解决了 AI 流水线中 LLM 调用不可预测和脆弱的问题,在这些场景中,静默失败和不可测试的输出往往使工作流变得脆弱。Mellea 用结构化、可测试的生成式编程方法取代了猜测和手动提示工程。
它是如何工作的
Mellea 使用 @generative 装饰器将标准 Python 函数转换为结构化的 LLM 调用。它利用 Python 类型提示和 Pydantic schema 来确保输出具有正确的类型和格式。该库还允许用户为调用附加自然语言要求,系统随后会对这些要求进行验证,如果未满足则会自动重试。
它是面向谁的
构建 AI 驱动的流水线和智能体(agents)的开发者,他们需要保证输出 schema、可验证的要求以及来自 LLM 的可预测结果。
亮点
- 结构化输出:使用 Pydantic schema 在生成时强制执行数据类型。
- 自动修复:根据自然语言要求验证输出并执行自动重试。
- 采样策略:支持多种提高可靠性的方法,例如多数投票(majority voting)和拒绝采样(rejection sampling)。
- 广泛的后端支持:兼容 OpenAI, Ollama, HuggingFace, WatsonX, LiteLLM, 和 Bedrock。
- MCP 兼容性:允许将生成式程序作为 Model Context Protocol (MCP) 工具进行暴露。
Sources
- undefinedgenerative-computing/mellea