cat-llm 0.0.37__py3-none-any.whl → 0.0.38__py3-none-any.whl

{cat_llm-0.0.37.dist-info → cat_llm-0.0.38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-llm
-Version: 0.0.37
+Version: 0.0.38
 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
@@ -368,10 +368,10 @@ Processes each image individually, evaluating the drawn shapes based on CERAD cr
 - `user_model` (str, default="gpt-4o"): Specific model to use
 - `creativity` (float, default=0): Temperature/randomness setting (0.0-1.0)
 - `reference_in_image` (bool, default=False): Whether a reference shape is present in the image for comparison
-- `provide_reference` (bool, default=False): Whether to provide a reference example image or description
+- `provide_reference` (bool, default=False): Whether to provide a reference example image (built in reference image)
 - `safety` (bool, default=False): Enable safety checks and save results at each API call step
 - `filename` (str, default="categorized_data.csv"): Filename for CSV output
-- `model_source` (str, default="OpenAI"): Model provider ("OpenAI", "Anthropic", "Perplexity", "Mistral")
+- `model_source` (str, default="OpenAI"): Model provider ("OpenAI", "Anthropic", "Mistral")
 
 **Returns:**
 - `pandas.DataFrame`: DataFrame with image paths, CERAD scores, and analysis details

{cat_llm-0.0.37.dist-info → cat_llm-0.0.38.dist-info}/RECORD

@@ -1,5 +1,5 @@
-catllm/CERAD_functions.py,sha256=NEf8HX1MNMjlMGuS42MaEcVE7XbQR9nGRYyPLQf91IU,19961
-catllm/__about__.py,sha256=WFqW7Tej2l4jCqQSDnklndjw7xQvVxk0mnWP4OXCnzw,404
+catllm/CERAD_functions.py,sha256=O2q36tDyTfx2VTaIb_aj4FZj1kOUOcHbHjXT1QLzFa0,20167
+catllm/__about__.py,sha256=8OdI51U4Os6bh-_fY-0OtVldFeKxgKbsA4THA6Egh8M,404
 catllm/__init__.py,sha256=BpAG8nPhM3ZQRd0WqkubI_36-VCOs4eCYtGVgzz48Bs,337
 catllm/image_functions.py,sha256=7nw0HDvacYUVo_VLcy-6Pi8QcmDbPtKVKPJITAK19RQ,31311
 catllm/text_functions.py,sha256=K6oetWYk25PwsllWSZP4cFrz7kyxJg0plPRvpmQkCsU,16846
@@ -8,7 +8,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.37.dist-info/METADATA,sha256=DTJBwpMhd7VU54SoTRplhN1ZsCR_efye6RdA2yL2VKw,17261
-cat_llm-0.0.37.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-cat_llm-0.0.37.dist-info/licenses/LICENSE,sha256=YYp5RSBpti1KKIJO0aV0GTRgQNLIsB7TEYivC3QkCOo,788
-cat_llm-0.0.37.dist-info/RECORD,,
+cat_llm-0.0.38.dist-info/METADATA,sha256=SmlnMw-RQX_S9VM_tNJprF85yWydYqRfQ3yUt7miyMo,17259
+cat_llm-0.0.38.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+cat_llm-0.0.38.dist-info/licenses/LICENSE,sha256=YYp5RSBpti1KKIJO0aV0GTRgQNLIsB7TEYivC3QkCOo,788
+cat_llm-0.0.38.dist-info/RECORD,,

catllm/CERAD_functions.py

@@ -229,12 +229,18 @@ def cerad_drawn_score(
                         f"Example:\n"
                         f"{example_JSON}"
                     ),
-                },
-                {
-                    "type": "image_url",
-                    "image_url": f"data:image/jpeg;base64,{encoded}"
                 }
             ]
+            if provide_reference:
+                prompt.append({
+                    "type": "image_url",
+                    "image_url": f"data:image/{ext};base64,{encoded_ref}"
+                })
+
+            prompt.append({
+                "type": "image_url",
+                "image_url": f"data:image/{ext};base64,{encoded_image}"
+            })
             
         if model_source == "OpenAI":
             from openai import OpenAI

catllm/__about__.py

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