predikit 0.4.1__tar.gz → 0.4.3__tar.gz

predikit-0.4.3/.claude/settings.local.json

@@ -0,0 +1,15 @@
+{
+  "permissions": {
+    "allow": [
+      "Bash(git add *)",
+      "Bash(git commit -m ' *)",
+      "Bash(python -m pytest tests/ -v --tb=short --cov=src/predikit --cov-report=term-missing)",
+      "WebFetch(domain:github.com)",
+      "PowerShell(python -m pip install mypy -q)",
+      "PowerShell(python -m mypy src/predikit)",
+      "Bash(python -m pytest --tb=short -q)",
+      "Bash(uv run *)",
+      "PowerShell(C:\\\\AI\\\\predikit\\\\.venv\\\\Scripts\\\\python.exe -m ensurepip --upgrade)"
+    ]
+  }
+}

{predikit-0.4.1 → predikit-0.4.3}/.gitignore

@@ -10,3 +10,4 @@ htmlcov/
 .venv/
 venv/
 *.egg
+file.py

{predikit-0.4.1 → predikit-0.4.3}/CHANGELOG.md

@@ -6,6 +6,20 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 
 ## [Unreleased]
 
+## [0.4.2] - 2026-06-13
+
+### Changed
+- Redesigned PyPI/README hero: logo, centered tagline, and badges in a unified `<p align="center">` block
+- Tagline moved from a `##` heading to a proper descriptive paragraph
+- Badges converted to centered HTML `<img>` links for consistent rendering on PyPI
+- Quick code teaser repositioned directly below badges (before Table of Contents)
+- "Field naming rule" added to Table of Contents
+- `ainvoke()` added to `ModelTool` Core API reference table
+- `ModelEnsemble` Core API subsection added with constructor signature and full strategy table
+- Project Traffic / download badge moved to bottom of README
+- Development Status classifier bumped from `3 - Alpha` to `4 - Beta` in `pyproject.toml`
+- Removed CI test status badge from README
+
 ## [0.4.1] - 2026-06-02
 
 ### Added

{predikit-0.4.1 → predikit-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: predikit
-Version: 0.4.1
+Version: 0.4.3
 Summary: Turn any trained sklearn/XGBoost model into an LLM-callable tool with auto-generated schemas and typed I/O.
 Project-URL: Homepage, https://github.com/Tejas-TA/predikit
 Project-URL: Repository, https://github.com/Tejas-TA/predikit
@@ -10,7 +10,7 @@ Author-email: Tejas Tumakuru Ashok <tejasta@gmail.com>
 License: MIT
 License-File: LICENSE
 Keywords: agents,function-calling,llm,ml-tools,sklearn,xgboost
-Classifier: Development Status :: 3 - Alpha
+Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
@@ -48,34 +48,46 @@ Provides-Extra: xgboost
 Requires-Dist: xgboost>=1.7; extra == 'xgboost'
 Description-Content-Type: text/markdown
 
-# predikit
-[![PyPI version](https://img.shields.io/pypi/v/predikit.svg)](https://pypi.org/project/predikit/)
-[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-green.svg)](LICENSE)
-[![CI](https://github.com/Tejas-TA/predikit/actions/workflows/test.yml/badge.svg)](https://github.com/Tejas-TA/predikit/actions/workflows/test.yml)
-[![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
+<p align="center">
+  <picture>
+    <source srcset="https://raw.githubusercontent.com/Tejas-TA/predikit/main/docs/logo.gif">
+    <img src="https://raw.githubusercontent.com/Tejas-TA/predikit/main/docs/logo.png" alt="predikit" width="500"/>
+  </picture>
+</p>
+
+<p align="center">
+  Turn any trained scikit-learn or XGBoost model into an LLM-callable tool —<br/>
+  auto-generated JSON schemas, typed I/O, zero boilerplate.
+</p>
+
+<p align="center">
+  <a href="https://pypi.org/project/predikit/"><img src="https://img.shields.io/pypi/v/predikit.svg" alt="PyPI version"/></a>
+  <a href="https://www.python.org/"><img src="https://img.shields.io/badge/python-3.10+-blue.svg" alt="Python 3.10+"/></a>
+  <a href="LICENSE"><img src="https://img.shields.io/badge/License-MIT-green.svg" alt="License: MIT"/></a>
+  <a href="https://github.com/astral-sh/ruff"><img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json" alt="Ruff"/></a>
+</p>
+
+<p align="center">
+  <a href="https://pepy.tech/project/predikit"><img src="https://static.pepy.tech/personalized-badge/predikit?period=week&units=international_system&left_color=grey&right_color=blue&left_text=weekly+downloads" alt="Weekly Downloads"/></a>
+  <a href="https://pepy.tech/project/predikit"><img src="https://static.pepy.tech/personalized-badge/predikit?period=month&units=international_system&left_color=grey&right_color=blue&left_text=monthly+downloads" alt="Monthly Downloads"/></a>
+  <a href="https://pepy.tech/project/predikit"><img src="https://static.pepy.tech/personalized-badge/predikit?period=total&units=international_system&left_color=grey&right_color=blue&left_text=total+downloads" alt="Total Downloads"/></a>
+</p>
 
-### 📈 Project Traffic
-Detailed breakdown of downloads by version, region, and platform:
-
-[![Downloads](https://pepy.tech/badge/predikit?style=for-the-badge)](https://pepy.tech/project/predikit)
+```python
+tool = ModelTool(model=clf, name="classify_iris", ...)
+tool.to_openai()              # OpenAI function schema, ready to pass to the API
+tool.invoke({"sqft": 2200})   # → {"price_usd": 370730}
+```
 
 ## Table of Contents
 - [Install](#install)
 - [30-second example](#30-second-example)
 - [Core API](#core-api)
+- [Field naming rule](#field-naming-rule)
 - [Cookbook](#cookbook)
 - [Contributing](#contributing)
 - [License](#license)
 
-## Turn any trained scikit-learn or XGBoost model into an LLM-callable tool — auto-generated JSON schemas, typed I/O, zero boilerplate.
-
-```python
-tool = ModelTool(model=clf, name="classify_iris", ...)
-tool.to_openai()              # OpenAI function schema, ready to pass to the API
-tool.invoke({"sqft": 2200})   # → {"price_usd": 370730}
-```
-
 ## Install
 
 ```bash
@@ -153,6 +165,7 @@ ModelTool(
 | Method | Returns | What it does |
 |--------|---------|--------------|
 | `.invoke(input_dict)` | `dict` | Validates → predicts → returns `{output_name: value}` |
+| `.ainvoke(input_dict)` | `dict` | Async version of `.invoke()` |
 | `.to_openai()` | `dict` | OpenAI function-calling schema |
 | `.to_langchain()` | `StructuredTool` | LangChain tool |
 | `.to_callable()` | `Callable` | Plain Python function |
@@ -168,6 +181,30 @@ registry.to_langchain()  # → list[StructuredTool]
 registry.get("name")     # → ModelTool
 ```
 
+### `ModelEnsemble`
+
+Call multiple models and reconcile their outputs in one step:
+
+```python
+ModelEnsemble(
+    tools: list[ModelTool],   # models to run in parallel
+    name: str,                # ensemble tool name the LLM sees
+    description: str,
+    strategy: str,            # "collect" | "mean" | "vote" | "weighted_mean" | "weighted_vote"
+    weights: list[float],     # optional, for weighted strategies
+)
+```
+
+| Strategy | Behaviour |
+|----------|-----------|
+| `"collect"` | Merges all outputs into one dict (tools can have different `output_name`) |
+| `"mean"` | Averages numeric outputs (all tools must share `output_name`) |
+| `"vote"` | Majority class vote (all tools must share `output_name`) |
+| `"weighted_mean"` | Weighted average — provide a `weights` list |
+| `"weighted_vote"` | Weighted majority vote — provide a `weights` list |
+
+`ModelEnsemble` exposes the same `.invoke()`, `.ainvoke()`, `.to_openai()`, and `.to_langchain()` interface as `ModelTool`.
+
 ## Field naming rule
 
 **Your Pydantic schema field names must exactly match the column names the model was trained on.**
@@ -287,8 +324,6 @@ Only applies to classifiers that implement `predict_proba`. Regressors are unaff
 
 ### Multi-model ensemble
 
-Call multiple models and reconcile their outputs in one step:
-
 ```python
 from predikit import ModelEnsemble, ToolRegistry
 
@@ -303,12 +338,6 @@ result  = ensemble.invoke(inputs)  # → {"price_usd": 370112}
 schema  = ensemble.to_openai()     # works exactly like ModelTool
 ```
 
-| strategy | behaviour |
-|----------|-----------|
-| `"collect"` | merges all outputs into one dict (tools can have different `output_name`) |
-| `"mean"` | averages numeric outputs (all tools must share `output_name`) |
-| `"vote"` | majority class vote (all tools must share `output_name`) |
-
 Register ensembles alongside individual tools:
 
 ```python
@@ -379,3 +408,4 @@ See [CONTRIBUTING.md](CONTRIBUTING.md) for development setup, code style, and PR
 ## License
 
 MIT © Tejas Tumakuru Ashok
+