predikit

让 LLM 智能体直接调用你的 ML 模型做预测

Predikit 是一个 Python 开发框架，定位为机器学习模型与 AI 智能体之间的桥梁。它解决的核心问题是：传统 ML 模型（如 scikit-learn、XGBoost 训练的模型）通常只能通过 API 或批处理调用，难以被 LangChain、OpenAI 等 LLM 驱动的智能体直接理解和使用。Predikit 通过 Pydantic 定义统一的模型接口，将模型封装成智能体可调用的工具，让 LLM 能够像调用函数一样调用预测模型，并自动处理输入校验、类型转换和结果解析。它支持 Snowflake 等数据源，可把模型部署为服务，同时保持与现有 ML 生态兼容。核心能力包括：模型包装为工具、智能体调用预测、结构化输入输出、多模型编排，适合需要将预测能力嵌入对话或自动化流程的场景。