cat-llm 0.0.76__py3-none-any.whl → 0.0.78__py3-none-any.whl

{cat_llm-0.0.76.dist-info → cat_llm-0.0.78.dist-info}/METADATA

@@ -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.dist-info → cat_llm-0.0.78.dist-info}/RECORD

@@ -1,10 +1,10 @@
 catllm/CERAD_functions.py,sha256=q4HbP5e2Yu8NnZZ-2eX4sImyj6u3i8xWcq0pYU81iis,22676
-catllm/__about__.py,sha256=yXzP4t-1ifCb-n2qXIRLt4j8v0AsNmsQgXS3fMChAzo,430
+catllm/__about__.py,sha256=lOe7499O5Oy36HGSwHffLVJxyMRC1XJKMBy51_I2V3o,430
 catllm/__init__.py,sha256=sf02zp7N0NW0mAQi7eQ4gliWR1EwoqvXkHN2HwwjcTE,372
 catllm/build_web_research.py,sha256=880dfE2bEQb-FrXP-42JoLLtyc9ox_sBULDr38xiTiQ,22655
 catllm/image_functions.py,sha256=8_FftRU285x1HT-AgNkaobefQVD-5q7ZY_t7JFdL3Sg,36177
 catllm/model_reference_list.py,sha256=37pWwMcgnf4biE3BVRluH5oz2P6ccdJJiCVNHodBH8k,2307
-catllm/text_functions.py,sha256=XF6aGuUyihnCKwGnGyLM1PbFQg3fF6nhJ_PoSX2zLaY,36101
+catllm/text_functions.py,sha256=GilnjNlmnyvLfX1_jzx41q--ZzABuYtrGMTOFmioreY,35750
 catllm/calls/CoVe.py,sha256=Y9OGJbaeJ3Odwira92cPXUlnm_ADFqvpOSFSNjFzMMU,10847
 catllm/calls/__init__.py,sha256=fWuMwLeSGa6zXJYd4s8IyNblsD62G-1NMUsOKrNIkoI,725
 catllm/calls/all_calls.py,sha256=AeN1QocOvL3Z36lDkq6bO0LB3ruz6pXyedvdci0YCxQ,16627
@@ -13,7 +13,7 @@ catllm/images/cube.png,sha256=nFec3e5bmRe4zrBCJ8QK-HcJLrG7u7dYdKhmdMfacfE,77275
 catllm/images/diamond.png,sha256=rJDZKtsnBGRO8FPA0iHuA8FvHFGi9PkI_DWSFdw6iv0,99568
 catllm/images/overlapping_pentagons.png,sha256=VO5plI6eoVRnjfqinn1nNzsCP2WQhuQy71V0EASouW4,71208
 catllm/images/rectangles.png,sha256=2XM16HO9EYWj2yHgN4bPXaCwPfl7iYQy0tQUGaJX9xg,40692
-cat_llm-0.0.76.dist-info/METADATA,sha256=EGlOhrerEtwgdk98DPhSCSshmOKnhXHw67-25V8wrJs,23214
-cat_llm-0.0.76.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-cat_llm-0.0.76.dist-info/licenses/LICENSE,sha256=Vje2sS5WV4TnIwY5uQHrF4qnBAM3YOk1pGpdH0ot-2o,34969
-cat_llm-0.0.76.dist-info/RECORD,,
+cat_llm-0.0.78.dist-info/METADATA,sha256=yXiZ4pFba0gfp4m2yU6TRfFO0kzWoHoYcxTzchjz2Xc,23214
+cat_llm-0.0.78.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+cat_llm-0.0.78.dist-info/licenses/LICENSE,sha256=Vje2sS5WV4TnIwY5uQHrF4qnBAM3YOk1pGpdH0ot-2o,34969
+cat_llm-0.0.78.dist-info/RECORD,,

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"

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