cat-llm 0.0.6__tar.gz → 0.0.8__tar.gz

{cat_llm-0.0.6 → cat_llm-0.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-llm
-Version: 0.0.6
+Version: 0.0.8
 Summary: A tool for categorizing text data and images using LLMs and vision models
 Project-URL: Documentation, https://github.com/Christopher Soria/cat-llm#readme
 Project-URL: Issues, https://github.com/chrissoria/cat-llm/issues

{cat_llm-0.0.6 → cat_llm-0.0.8}/src/catllm/__about__.py

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

{cat_llm-0.0.6 → cat_llm-0.0.8}/src/catllm/__init__.py

@@ -11,4 +11,4 @@ from .__about__ import (
     __license__,
 )
 
-from .catllm import *
+from .cat_llm import *

{cat_llm-0.0.6 → cat_llm-0.0.8}/src/catllm/cat_llm.py

@@ -937,7 +937,14 @@ Number your categories from 1 through {cat_num} and be concise with the category
             raise ValueError(f"Unsupported model_source: {model_source}")
         
         # Extract just the text as a list
-        items = [line.split('. ', 1)[1] for line in responses[i].split('\n') if '. ' in line]
+        items = []
+        for line in responses[i].split('\n'):
+            if '. ' in line:
+                try:
+                    items.append(line.split('. ', 1)[1])
+                except IndexError:
+                    pass
+
         responses_list.append(items)
 
     flat_list = [item.lower() for sublist in responses_list for item in sublist]