cat-llm 0.0.76__tar.gz → 0.0.78__tar.gz

{cat_llm-0.0.76 → cat_llm-0.0.78}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-llm
-Version: 0.0.76
+Version: 0.0.78
 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

{cat_llm-0.0.76 → cat_llm-0.0.78}/src/catllm/__about__.py

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

{cat_llm-0.0.76 → cat_llm-0.0.78}/src/catllm/text_functions.py

@@ -114,6 +114,7 @@ Number your categories from 1 through {cat_num} and be concise with the category
     
     return df
 
+#extract top categories from corpus
 #extract top categories from corpus
 def explore_common_categories(
     survey_question, 
@@ -313,14 +314,7 @@ Number your categories from 1 through {cat_num} and be concise with the category
     return top_categories_final
 
 #multi-class text classification
-# what this function does:
-# does context prompting, giving the model a background on the task at hand and the user's survey question
-# system prompting, overall context and purpose for the language model
-# role prompting, assings a spacific identity to the model
-# also enables few shot prompting, allowing the user to input a few examples
-# provides POSITIVE INSTRUCTIONS reather than limitations/restrictions
-# GOAL: enable step-back prompting
-# GOAL 2: enable self-consistency
+# GOAL: enable self-consistency
 def multi_class(
     survey_input,
     categories,
@@ -408,12 +402,14 @@ def multi_class(
     if categories == "auto":
         if survey_question == "": # step back requires the survey question to function well
             raise TypeError("survey_question is required when using step_back_prompt. Please provide the survey question you are analyzing.")
-            
+
         categories = explore_common_categories(
             survey_question=survey_question,
             survey_input=survey_input,
             research_question=research_question,
             api_key=api_key,
+            model_source=model_source,
+            user_model=user_model,
             top_n=top_n,
             cat_num=cat_num,
             divisions=divisions