cat-llm 0.0.50__tar.gz → 0.0.52__tar.gz

{cat_llm-0.0.50 → cat_llm-0.0.52}/PKG-INFO

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

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

{cat_llm-0.0.50 → cat_llm-0.0.52}/src/catllm/__init__.py

@@ -13,4 +13,5 @@ from .__about__ import (
 
 from .text_functions import *
 from .CERAD_functions import * 
-from .image_functions import * 
+from .image_functions import * 
+from .build_web_research import * 

{cat_llm-0.0.50 → cat_llm-0.0.52}/src/catllm/build_web_research.py

@@ -36,7 +36,7 @@ def build_web_research_dataset(
     extracted_jsons = []
 
     for idx, item in enumerate(tqdm(search_input, desc="Building dataset")):
-        if idx == 0:  # delay the first item just to be safe
+        if idx > 0:  # Skip delay for first item only
             time.sleep(time_delay)
         reply = None  
 
@@ -88,13 +88,11 @@ def build_web_research_dataset(
                         if getattr(block, "type", "") == "text"
                     ).strip()
                     link1.append(reply)
-                    time.sleep(time_delay)
                     print(reply)
                     
                 except Exception as e:
                     print(f"An error occurred: {e}")
                     link1.append(f"Error processing input: {e}")
-                    time.sleep(time_delay)
             else:
                 raise ValueError("Unknown source! Currently this function only supports 'Anthropic' as model_source.")
             # in situation that no JSON is found