cat-stack 0.1.0__py3-none-any.whl

cat_stack-0.1.0.dist-info/METADATA

@@ -0,0 +1,150 @@
+Metadata-Version: 2.4
+Name: cat-stack
+Version: 0.1.0
+Summary: Domain-agnostic text, image, and PDF classification engine powered by LLMs
+Project-URL: Documentation, https://github.com/chrissoria/cat-stack#readme
+Project-URL: Issues, https://github.com/chrissoria/cat-stack/issues
+Project-URL: Source, https://github.com/chrissoria/cat-stack
+Author-email: Chris Soria <chrissoria@berkeley.edu>
+License-Expression: GPL-3.0-or-later
+License-File: LICENSE
+Keywords: ensemble,image classification,llm,pdf classification,structured output,text classification
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Requires-Python: >=3.8
+Requires-Dist: anthropic
+Requires-Dist: openai
+Requires-Dist: pandas
+Requires-Dist: perplexityai
+Requires-Dist: requests
+Requires-Dist: tqdm
+Provides-Extra: embeddings
+Requires-Dist: sentence-transformers>=2.2.0; extra == 'embeddings'
+Provides-Extra: formatter
+Requires-Dist: accelerate>=0.27.0; extra == 'formatter'
+Requires-Dist: torch>=2.0.0; extra == 'formatter'
+Requires-Dist: transformers>=4.40.0; extra == 'formatter'
+Provides-Extra: pdf
+Requires-Dist: pymupdf>=1.23.0; extra == 'pdf'
+Description-Content-Type: text/markdown
+
+# cat-stack
+
+**Domain-agnostic text, image, and PDF classification engine powered by LLMs.**
+
+`cat-stack` is the shared base package for the [CatLLM](https://github.com/chrissoria/cat-llm) ecosystem. It provides the core classification, extraction, exploration, and summarization engine that all domain-specific CatLLM packages build on.
+
+## Installation
+
+```bash
+pip install cat-stack
+```
+
+Optional extras:
+
+```bash
+pip install cat-stack[pdf]         # PDF support (PyMuPDF)
+pip install cat-stack[embeddings]  # Embedding similarity scoring
+pip install cat-stack[formatter]   # JSON formatter fallback model
+```
+
+## Ecosystem
+
+`cat-stack` is independently useful for classifying any text column. Domain-specific packages extend it with tuned prompts and workflows:
+
+| Package | Domain |
+|---------|--------|
+| **cat-stack** | General-purpose text, image, PDF classification (this package) |
+| **cat-survey** | Survey response classification |
+| **cat-vader** | Social media text (Reddit, Twitter/X) |
+| **cat-ademic** | Academic papers, PDFs, citations |
+| **cat-cog** | Cognitive assessment & visual scoring (CERAD) |
+| **cat-pol** | Political text (manifestos, speeches, legislation) |
+
+Installing `cat-llm` pulls in all of the above.
+
+## Quick Start
+
+```python
+import cat_stack as cat
+
+# Classify text into predefined categories
+result = cat.classify(
+    input_data=df["text_column"],
+    categories=["Positive", "Negative", "Neutral"],
+    models=[("gpt-4o", "openai", OPENAI_KEY)],
+    filename="classified.csv"
+)
+```
+
+## Core API
+
+### `classify()`
+Assign predefined categories to text, images, or PDFs. Supports single-model and multi-model ensemble classification with consensus voting.
+
+```python
+cat.classify(
+    input_data=df["text"],
+    categories=["Cat A", "Cat B", "Cat C"],
+    models=[("gpt-4o", "openai", key1), ("claude-sonnet-4-20250514", "anthropic", key2)],
+    filename="results.csv"
+)
+```
+
+### `extract()`
+Discover categories from a corpus using LLM-driven exploration.
+
+```python
+cat.extract(
+    input_data=df["text"],
+    survey_question="What is this text about?",
+    models=[("gpt-4o", "openai", key)],
+)
+```
+
+### `explore()`
+Raw category extraction for saturation analysis.
+
+```python
+cat.explore(
+    input_data=df["text"],
+    description="Describe the main themes",
+    models=[("gpt-4o", "openai", key)],
+)
+```
+
+### `summarize()`
+Summarize text or PDF documents, with optional multi-model ensemble.
+
+```python
+cat.summarize(
+    input_data=df["text"],
+    models=[("gpt-4o", "openai", key)],
+    filename="summaries.csv"
+)
+```
+
+## Supported Providers
+
+OpenAI, Anthropic, Google (Gemini), Mistral, Perplexity, xAI (Grok), HuggingFace, Ollama (local models).
+
+All providers use the same `(model_name, provider, api_key)` tuple format. Provider is auto-detected from model name if omitted.
+
+## Features
+
+- **Multi-model ensemble** with consensus voting and agreement scores
+- **Batch API support** for OpenAI, Anthropic, Google, Mistral, and xAI
+- **Prompt strategies**: Chain-of-Thought, Chain-of-Verification, step-back prompting, few-shot examples
+- **Text, image, and PDF** input auto-detection
+- **Embedding similarity** tiebreaker for ensemble consensus ties
+
+## License
+
+GPL-3.0-or-later

cat_stack-0.1.0.dist-info/RECORD

@@ -0,0 +1,38 @@
+cat_stack/__about__.py,sha256=2yQWde3xin26jZeTfxWT9NpZPTNh75T7fpCjxPB-Jt8,411
+cat_stack/__init__.py,sha256=WeM8X2AGCbhqumWgfliDVlUguB_8TCuOtEffgeTp7Yc,3542
+cat_stack/_batch.py,sha256=2oJs1CcEWgVpP09TK-yJnEEwKq7lgHeXHI3HlBKJQUc,52703
+cat_stack/_category_analysis.py,sha256=LYp40Pw1-0uT9cruhnn6kpjxr23xm4XItHpl4hZwyCw,12692
+cat_stack/_chunked.py,sha256=1ZYb_G1anMnbtGV5zSfBTgIRlk78O1Viwj4bAmXDcOU,15356
+cat_stack/_embeddings.py,sha256=FQYQYwG2cQ8kraLLwjysFBk-2S5f0dZylZ9HWGnGEwo,6659
+cat_stack/_formatter.py,sha256=VgYn26zPVknPjEYE2iHEiygwol6iDrgS9kq4xgBMZ3A,5310
+cat_stack/_providers.py,sha256=Ek9G7nW6pgAjgEy87jcLFIsc8tBlyfZ_gjiSdhVdI4Y,39047
+cat_stack/_tiebreaker.py,sha256=gYP7hGbQcZ9uOPHIhzuCgFi46Kwu7ViAH0c3QlB-w1Y,9770
+cat_stack/_utils.py,sha256=YX8UX9Z8VAAeN7BO1RClgg8XK_YelvO1RG-Yb5tND4c,15712
+cat_stack/_web_fetch.py,sha256=zr86MXwq6jOuTmEJCLiKzbNtW36IfBaSp956lx3jkp0,5441
+cat_stack/classify.py,sha256=5GkYX8R3yezBkKeSXwF9K1EoLRIRFCPkG1v_u-aRnSM,31951
+cat_stack/explore.py,sha256=_LkdRkSIIXx0ixZbXGEfiA9irD96HtJC3dTZ7tQ_2MA,3943
+cat_stack/extract.py,sha256=PmFlrGaXOPOfonXOkl4WHzFAdmTOLva-7ZLpEaBSC34,8047
+cat_stack/image_functions.py,sha256=2O2vF7VEaeUesCKLXI-YHWqdCcBcgnX4b3LUegwS8bQ,89027
+cat_stack/model_reference_list.py,sha256=37pWwMcgnf4biE3BVRluH5oz2P6ccdJJiCVNHodBH8k,2307
+cat_stack/pdf_functions.py,sha256=KrEmpL_blY4J0VIr9W-4xFZTQjKCFsj1aWlC9Ckw5wM,88692
+cat_stack/summarize.py,sha256=aRBXU95hnlhcx4I0sjvLUa9yV5n6K3tnYQQHh-0FAmU,11462
+cat_stack/text_functions.py,sha256=lDv7dn8z3FoEoAzuZa5CiVWT2gP2yidXWHfHHyoRgKI,50932
+cat_stack/text_functions_ensemble.py,sha256=GIVtfAgAefTTOTCSzhP0nAHcO6wFzRc9uNmpX_I318U,141882
+cat_stack/calls/CoVe.py,sha256=92h-HATm_Sa9QT0nmuyURpkzm4wemSCwFhOZfBD_v_A,9819
+cat_stack/calls/__init__.py,sha256=j5n25HE7_E4imYgM8VcjP5FSRGUvASq4Uq0AghL2-2E,738
+cat_stack/calls/all_calls.py,sha256=SFqnbHZi8mjz2IUteSEsyz7z1DrTUTwNH2xlf64X4d8,20318
+cat_stack/calls/image_CoVe.py,sha256=d6l0cvZFRWt2b669A9_voNu0ofxiF00l9CP9pA0Uzto,12827
+cat_stack/calls/image_stepback.py,sha256=Wr_BIco0y0ReK93vJa4iiMZKtVT8NO1-B_irZn2Gyfc,5950
+cat_stack/calls/pdf_CoVe.py,sha256=3hvq-W-We6OZeKBygF6n0VVqlRJxwF0coWZx40no0c4,12934
+cat_stack/calls/pdf_stepback.py,sha256=UYCwVH074S6n6FqfpCCRVoHEJ-Y7DED9FN6gTWen8QI,5925
+cat_stack/calls/stepback.py,sha256=0VIRreH3q1cQ2HPrgGm13A_n0gcCMfzBJculuvAOQuQ,4754
+cat_stack/calls/top_n.py,sha256=mdPFLg3mJg9B3NjngR5aQjxM__B1br7uC-pSJtvWYXM,6283
+cat_stack/images/circle.png,sha256=JWujAWAh08-TajAoEr_TAeFNLlfbryOLw6cgIBREBuQ,86202
+cat_stack/images/cube.png,sha256=nFec3e5bmRe4zrBCJ8QK-HcJLrG7u7dYdKhmdMfacfE,77275
+cat_stack/images/diamond.png,sha256=rJDZKtsnBGRO8FPA0iHuA8FvHFGi9PkI_DWSFdw6iv0,99568
+cat_stack/images/overlapping_pentagons.png,sha256=VO5plI6eoVRnjfqinn1nNzsCP2WQhuQy71V0EASouW4,71208
+cat_stack/images/rectangles.png,sha256=2XM16HO9EYWj2yHgN4bPXaCwPfl7iYQy0tQUGaJX9xg,40692
+cat_stack-0.1.0.dist-info/METADATA,sha256=yYd_6IYCIj05YehBPugI208ENDTWnjqgyr3y3VWM-Jg,4818
+cat_stack-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+cat_stack-0.1.0.dist-info/licenses/LICENSE,sha256=Vje2sS5WV4TnIwY5uQHrF4qnBAM3YOk1pGpdH0ot-2o,34969
+cat_stack-0.1.0.dist-info/RECORD,,

cat_stack-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any