cat-llm 0.0.72__tar.gz → 0.0.73__tar.gz

{cat_llm-0.0.72 → cat_llm-0.0.73}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-llm
-Version: 0.0.72
+Version: 0.0.73
 Summary: A tool for categorizing text data and images using LLMs and vision models
 Project-URL: Documentation, https://github.com/chrissoria/cat-llm#readme
 Project-URL: Issues, https://github.com/chrissoria/cat-llm/issues
@@ -189,16 +189,22 @@ Performs multi-label classification of text responses into user-defined categori
 Processes each text response individually, assigning one or more categories from the provided list. Supports flexible output formatting and optional saving of results to CSV for easy integration with data analysis workflows.
 
 **Parameters:**
-- `survey_question` (str): The survey question being analyzed
 - `survey_input` (list): List of text responses to classify
 - `categories` (list): List of predefined categories for classification
 - `api_key` (str): API key for the LLM service
-- `user_model` (str, default="gpt-4o"): Specific model to use
-- `creativity` (float, default=0): Temperature/randomness setting (0.0-1.0)
+- `user_model` (str, default="gpt-5"): Specific model to use
+- `user_prompt` (str, optional): Custom prompt template to override default prompting
+- `survey_question` (str, default=""): The survey question being analyzed
+- `example1` through `example6` (dict, optional): Few-shot learning examples (format: {"response": "...", "categories": [...]})
+- `creativity` (float, optional): Temperature/randomness setting (0.0-1.0, varies by model)
 - `safety` (bool, default=False): Enable safety checks on responses and saves to CSV at each API call step
+- `to_csv` (bool, default=False): Whether to save results to CSV
+- `chain_of_verification` (bool, default=False): Enable Chain-of-Verification prompting technique for improved accuracy
+- `step_back_prompt` (bool, default=False): Enable step-back prompting to analyze higher-level context before classification
+- `context_prompt` (bool, default=False): Add expert role and behavioral guidelines to the prompt
 - `filename` (str, default="categorized_data.csv"): Filename for CSV output
 - `save_directory` (str, optional): Directory path to save the CSV file
-- `model_source` (str, default="OpenAI"): Model provider ("OpenAI", "Anthropic", "Perplexity", "Mistral")
+- `model_source` (str, default="auto"): Model provider ("auto", "OpenAI", "Anthropic", "Google", "Mistral", "Perplexity", "Huggingface")
 
 **Returns:**
 - `pandas.DataFrame`: DataFrame with classification results, columns formatted as specified

{cat_llm-0.0.72 → cat_llm-0.0.73}/README.md

@@ -160,16 +160,22 @@ Performs multi-label classification of text responses into user-defined categori
 Processes each text response individually, assigning one or more categories from the provided list. Supports flexible output formatting and optional saving of results to CSV for easy integration with data analysis workflows.
 
 **Parameters:**
-- `survey_question` (str): The survey question being analyzed
 - `survey_input` (list): List of text responses to classify
 - `categories` (list): List of predefined categories for classification
 - `api_key` (str): API key for the LLM service
-- `user_model` (str, default="gpt-4o"): Specific model to use
-- `creativity` (float, default=0): Temperature/randomness setting (0.0-1.0)
+- `user_model` (str, default="gpt-5"): Specific model to use
+- `user_prompt` (str, optional): Custom prompt template to override default prompting
+- `survey_question` (str, default=""): The survey question being analyzed
+- `example1` through `example6` (dict, optional): Few-shot learning examples (format: {"response": "...", "categories": [...]})
+- `creativity` (float, optional): Temperature/randomness setting (0.0-1.0, varies by model)
 - `safety` (bool, default=False): Enable safety checks on responses and saves to CSV at each API call step
+- `to_csv` (bool, default=False): Whether to save results to CSV
+- `chain_of_verification` (bool, default=False): Enable Chain-of-Verification prompting technique for improved accuracy
+- `step_back_prompt` (bool, default=False): Enable step-back prompting to analyze higher-level context before classification
+- `context_prompt` (bool, default=False): Add expert role and behavioral guidelines to the prompt
 - `filename` (str, default="categorized_data.csv"): Filename for CSV output
 - `save_directory` (str, optional): Directory path to save the CSV file
-- `model_source` (str, default="OpenAI"): Model provider ("OpenAI", "Anthropic", "Perplexity", "Mistral")
+- `model_source` (str, default="auto"): Model provider ("auto", "OpenAI", "Anthropic", "Google", "Mistral", "Perplexity", "Huggingface")
 
 **Returns:**
 - `pandas.DataFrame`: DataFrame with classification results, columns formatted as specified

{cat_llm-0.0.72 → cat_llm-0.0.73}/src/catllm/__about__.py

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2025-present Christopher Soria <chrissoria@berkeley.edu>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.0.72"
+__version__ = "0.0.73"
 __author__ = "Chris Soria"
 __email__ = "chrissoria@berkeley.edu"
 __title__ = "cat-llm"

{cat_llm-0.0.72 → cat_llm-0.0.73}/src/catllm/text_functions.py

@@ -410,7 +410,6 @@ def multi_class(
                 When uncertain, prioritize precision over recall."""
 
                 prompt = context + prompt
-                print(prompt)
 
             if chain_of_verification:
                 step2_prompt = f"""You provided this initial categorization: