cat-stack 2.1.0__tar.gz → 2.2.0__tar.gz

{cat_stack-2.1.0 → cat_stack-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-stack
-Version: 2.1.0
+Version: 2.2.0
 Summary: Domain-agnostic text, image, PDF, and DOCX 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

{cat_stack-2.1.0 → cat_stack-2.2.0}/src/catstack/__about__.py

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2025-present Christopher Soria <chrissoria@berkeley.edu>
 #
 # SPDX-License-Identifier: GPL-3.0-or-later
-__version__ = "2.1.0"
+__version__ = "2.2.0"
 __author__ = "Chris Soria"
 __email__ = "chrissoria@berkeley.edu"
 __title__ = "cat-stack"

{cat_stack-2.1.0 → cat_stack-2.2.0}/src/catstack/_batch.py

@@ -65,7 +65,7 @@ BATCH_ENDPOINTS = {
     },
 }
 
-UNSUPPORTED_BATCH_PROVIDERS = {"huggingface", "huggingface-together", "perplexity", "ollama"}
+UNSUPPORTED_BATCH_PROVIDERS = {"huggingface", "huggingface-together", "perplexity", "ollama", "claude-code", "claude-agent"}
 
 # Terminal states per provider
 _TERMINAL_STATES = {

{cat_stack-2.1.0 → cat_stack-2.2.0}/src/catstack/_providers.py

@@ -729,6 +729,24 @@ PROVIDER_CONFIG = {
 }
 
 
+# Providers that route through complete() with no HTTP endpoint of their own
+# (Claude subscription / CLI). Features that build a direct HTTP request
+# (image, PDF) can't use them — guard with a clear error, not a deep crash.
+_SUBSCRIPTION_PROVIDERS = ("claude-code", "claude-agent")
+
+
+def _require_http_provider(model_source, feature):
+    """Raise a clear error when an HTTP-only feature is used with a
+    subscription/CLI provider (claude-code / claude-agent)."""
+    if model_source in _SUBSCRIPTION_PROVIDERS:
+        raise ValueError(
+            f"{feature} is not supported with model_source='{model_source}'. "
+            "The Claude subscription/CLI backend supports text classification, "
+            "extraction, and summarization, but not " + feature.lower() + ". "
+            "Use an API-key provider (e.g. model_source='anthropic') instead."
+        )
+
+
 # =============================================================================
 # Unified API Client
 # =============================================================================

{cat_stack-2.1.0 → cat_stack-2.2.0}/src/catstack/_utils.py

@@ -361,7 +361,8 @@ def _get_stepback_insight(model_source, stepback, api_key, user_model, creativit
         get_stepback_insight_openai,
         get_stepback_insight_anthropic,
         get_stepback_insight_google,
-        get_stepback_insight_mistral
+        get_stepback_insight_mistral,
+        get_stepback_insight_via_complete
     )
 
     stepback_functions = {
@@ -373,6 +374,8 @@ def _get_stepback_insight(model_source, stepback, api_key, user_model, creativit
         "anthropic": get_stepback_insight_anthropic,
         "google": get_stepback_insight_google,
         "mistral": get_stepback_insight_mistral,
+        "claude-code": get_stepback_insight_via_complete,
+        "claude-agent": get_stepback_insight_via_complete,
     }
 
     func = stepback_functions.get(model_source)

{cat_stack-2.1.0 → cat_stack-2.2.0}/src/catstack/calls/stepback.py

@@ -187,3 +187,34 @@ def get_stepback_insight_mistral(
 
     except Exception as e:
         return None, False
+
+
+
+def get_stepback_insight_via_complete(
+    stepback,
+    api_key,
+    user_model,
+    model_source,
+    creativity=None,
+):
+    """Stepback insight via the central UnifiedLLMClient.complete().
+
+    For providers with no direct HTTP endpoint (claude-agent / claude-code),
+    which route through complete() rather than a provider-specific requests.post.
+    Returns (insight_text, True) on success, (None, False) otherwise.
+    """
+    from cat_stack._providers import UnifiedLLMClient
+    try:
+        client = UnifiedLLMClient(
+            provider=model_source, api_key=api_key or "", model=user_model
+        )
+        insight, error = client.complete(
+            messages=[{"role": "user", "content": stepback}],
+            creativity=creativity,
+            force_json=False,  # stepback insight is free text, not JSON
+        )
+        if error or not insight:
+            return None, False
+        return insight, True
+    except Exception:
+        return None, False

{cat_stack-2.1.0 → cat_stack-2.2.0}/src/catstack/image_functions.py

@@ -1,6 +1,7 @@
 import warnings
 
 from .text_functions import _detect_model_source
+from ._providers import _require_http_provider
 from ._utils import _clean_label, _extract_balanced_json
 from .calls.image_stepback import get_image_stepback_insight
 
@@ -147,6 +148,7 @@ def image_multi_class(
         raise FileNotFoundError(f"Directory {save_directory} doesn't exist")
 
     model_source = _detect_model_source(user_model, model_source)
+    _require_http_provider(model_source, "Image classification")
 
     image_files = _load_image_files(image_input)
 

{cat_stack-2.1.0 → cat_stack-2.2.0}/src/catstack/pdf_functions.py

@@ -1,6 +1,7 @@
 import warnings
 
 from .text_functions import _detect_model_source
+from ._providers import _require_http_provider
 from ._utils import _clean_label, _extract_balanced_json
 from .calls.pdf_stepback import get_pdf_stepback_insight
 
@@ -392,6 +393,7 @@ def pdf_multi_class(
         raise ValueError(f"mode must be 'image', 'text', or 'both', got: {mode}")
 
     model_source = _detect_model_source(user_model, model_source)
+    _require_http_provider(model_source, "PDF classification")
 
     # Providers with native PDF support (only used in image/both modes)
     native_pdf_providers = {"anthropic", "google"}

{cat_stack-2.1.0 → cat_stack-2.2.0}/src/catstack/text_functions.py

@@ -40,7 +40,8 @@ from .calls.stepback import (
     get_stepback_insight_openai,
     get_stepback_insight_anthropic,
     get_stepback_insight_google,
-    get_stepback_insight_mistral
+    get_stepback_insight_mistral,
+    get_stepback_insight_via_complete
 )
 from .calls.CoVe import (
     chain_of_verification_openai,
@@ -90,6 +91,8 @@ def _get_stepback_insight(model_source, stepback, api_key, user_model, creativit
         "anthropic": get_stepback_insight_anthropic,
         "google": get_stepback_insight_google,
         "mistral": get_stepback_insight_mistral,
+        "claude-code": get_stepback_insight_via_complete,
+        "claude-agent": get_stepback_insight_via_complete,
     }
 
     func = stepback_functions.get(model_source)