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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-stack
-Version: 2.1.1
+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.1 → 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.1"
+__version__ = "2.2.0"
 __author__ = "Chris Soria"
 __email__ = "chrissoria@berkeley.edu"
 __title__ = "cat-stack"

{cat_stack-2.1.1 → 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.1 → 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.1 → 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)