cat-llm 0.0.31__tar.gz → 0.0.32__tar.gz

{cat_llm-0.0.31 → cat_llm-0.0.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-llm
-Version: 0.0.31
+Version: 0.0.32
 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.31 → cat_llm-0.0.32}/src/catllm/__about__.py

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

{cat_llm-0.0.31 → cat_llm-0.0.32}/src/catllm/__init__.py

@@ -11,6 +11,6 @@ from .__about__ import (
     __license__,
 )
 
-from .cat_llm import *
+from .text_functions import *
 from .CERAD_functions import * 
 from .image_functions import * 

{cat_llm-0.0.31 → cat_llm-0.0.32}/src/catllm/image_functions.py

@@ -72,7 +72,7 @@ def image_multi_class(
     
     # Handle extension safely
         ext = Path(img_path).suffix.lstrip(".").lower()
-        if model_source == "OpenAI":
+        if model_source == "OpenAI" or model_source == "Mistral":
             encoded_image = f"data:image/{ext};base64,{encoded}"
             prompt = [
                 {
@@ -309,7 +309,7 @@ def image_score(
         ext = Path(img_path).suffix.lstrip(".").lower()
         encoded_image = f"data:image/{ext};base64,{encoded}"
 
-        if model_source == "OpenAI":
+        if model_source == "OpenAI" or model_source == "Mistral":
             prompt = [
                 {
                     "type": "text",
@@ -575,7 +575,7 @@ def image_features(
         ext = Path(img_path).suffix.lstrip(".").lower()
         encoded_image = f"data:image/{ext};base64,{encoded}"
 
-        if model_source == "OpenAI":
+        if model_source == "OpenAI" or model_source == "Mistral":
             prompt = [
                 {
                     "type": "text",