speedy-utils 1.1.31__py3-none-any.whl → 1.1.33__py3-none-any.whl

llm_utils/lm/llm.py

@@ -63,9 +63,13 @@ class LLM(
         vllm_cmd: str | None = None,
         vllm_timeout: int = 1200,
         vllm_reuse: bool = True,
+        verbose=False,
         **model_kwargs,
     ):
         """Initialize LLMTask."""
+        if verbose:
+            available_models = LLM.list_models(client=client)
+            logger.info(f'Available models: {available_models}')
         self.instruction = instruction
         self.input_model = input_model
         self.output_model = output_model

speedy_utils/__imports.py

@@ -34,7 +34,6 @@ import re
 import sys
 import textwrap
 import threading
-import time
 import traceback
 import types
 import uuid
@@ -77,6 +76,9 @@ from typing import (
 )
 
 import cachetools
+
+# Direct imports (previously lazy-loaded)
+import numpy as np
 import psutil
 from fastcore.parallel import parallel
 from json_repair import loads as jloads
@@ -84,52 +86,61 @@ from loguru import logger
 from tqdm import tqdm
 
 
-# Direct imports (previously lazy-loaded)
-import numpy as np
 tabulate = __import__('tabulate').tabulate
 import xxhash
 
+
 # Optional imports - lazy loaded for performance
 def _get_pandas():
     """Lazy import pandas."""
     try:
         import pandas as pd
+
         return pd
     except ImportError:
         return None
 
+
 def _get_ray():
     """Lazy import ray."""
     try:
         import ray
+
         return ray
     except ImportError:
         return None
 
+
 def _get_matplotlib():
     """Lazy import matplotlib."""
     try:
         import matplotlib
+
         return matplotlib
     except ImportError:
         return None
 
+
 def _get_matplotlib_pyplot():
     """Lazy import matplotlib.pyplot."""
     try:
         import matplotlib.pyplot as plt
+
         return plt
     except ImportError:
         return None
 
+
 def _get_ipython_core():
     """Lazy import IPython.core.getipython."""
     try:
         from IPython.core.getipython import get_ipython
-        return get_ipython
+
+        return get_ipython()
     except ImportError:
         return None
 
+
 # Cache for lazy imports
 _pandas_cache = None
 _ray_cache = None
@@ -137,9 +148,11 @@ _matplotlib_cache = None
 _plt_cache = None
 _get_ipython_cache = None
 
+
 # Lazy import classes for performance-critical modules
 class _LazyModule:
     """Lazy module loader that imports only when accessed."""
+
     def __init__(self, import_func, cache_var_name):
         self._import_func = import_func
         self._cache_var_name = cache_var_name
@@ -168,9 +181,10 @@ class _LazyModule:
 
     def __repr__(self):
         if self._module is None:
-            return f"<LazyModule: not loaded>"
+            return '<LazyModule: not loaded>'
         return repr(self._module)
 
+
 # Create lazy loaders for top slow imports (import only when accessed)
 pd = _LazyModule(_get_pandas, '_pandas_cache')
 ray = _LazyModule(_get_ray, '_ray_cache')
@@ -200,11 +214,12 @@ try:
 except ImportError:
     BaseModel = None
 if TYPE_CHECKING:
+    import matplotlib.pyplot as plt
     import numpy as np
     import pandas as pd
     import ray
     import torch
-    import matplotlib.pyplot as plt
+
     # xxhash
     import xxhash  # type: ignore
     from IPython.core.getipython import get_ipython  # type: ignore

{speedy_utils-1.1.31.dist-info → speedy_utils-1.1.33.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: speedy-utils
-Version: 1.1.31
+Version: 1.1.33
 Summary: Fast and easy-to-use package for data science
 Project-URL: Homepage, https://github.com/anhvth/speedy
 Project-URL: Repository, https://github.com/anhvth/speedy

{speedy_utils-1.1.31.dist-info → speedy_utils-1.1.33.dist-info}/RECORD

@@ -6,7 +6,7 @@ llm_utils/chat_format/transform.py,sha256=PJ2g9KT1GSbWuAs7giEbTpTAffpU9QsIXyRlbf
 llm_utils/chat_format/utils.py,sha256=M2EctZ6NeHXqFYufh26Y3CpSphN0bdZm5xoNaEJj5vg,1251
 llm_utils/lm/__init__.py,sha256=lFE2DZRpj6eRMo11kx7oRLyYOP2FuDmz08mAcq-cYew,730
 llm_utils/lm/base_prompt_builder.py,sha256=_TzYMsWr-SsbA_JNXptUVN56lV5RfgWWTrFi-E8LMy4,12337
-llm_utils/lm/llm.py,sha256=Ryne4why4VgRCvTJW1SLJciv2pcYYROGP0AwkLVdGPg,16299
+llm_utils/lm/llm.py,sha256=C8Z8l6Ljs7uVX-zabLcDCdTf3fpGxfljaYRM0patHUQ,16469
 llm_utils/lm/llm_signature.py,sha256=vV8uZgLLd6ZKqWbq0OPywWvXAfl7hrJQnbtBF-VnZRU,1244
 llm_utils/lm/lm_base.py,sha256=Bk3q34KrcCK_bC4Ryxbc3KqkiPL39zuVZaBQ1i6wJqs,9437
 llm_utils/lm/mixins.py,sha256=on83g-JO2SpZ0digOpU8mooqFBX6w7Bc-DeGzVoVCX8,14536
@@ -27,7 +27,7 @@ llm_utils/vector_cache/cli.py,sha256=MAvnmlZ7j7_0CvIcSyK4TvJlSRFWYkm4wE7zSq3KR8k
 llm_utils/vector_cache/core.py,sha256=VXuYJy1AX22NHKvIXRriETip5RrmQcNp73-g-ZT774o,30950
 llm_utils/vector_cache/types.py,sha256=CpMZanJSTeBVxQSqjBq6pBVWp7u2-JRcgY9t5jhykdQ,438
 llm_utils/vector_cache/utils.py,sha256=OsiRFydv8i8HiJtPL9hh40aUv8I5pYfg2zvmtDi4DME,1446
-speedy_utils/__imports.py,sha256=PhHqZWwVOKAbbXoWxZLVVyurGmZhui3boQ7Nji002cQ,7795
+speedy_utils/__imports.py,sha256=V0YzkDK4-QkK_IDXY1be6C6_STuNhXAKIp4_dM0coQs,7800
 speedy_utils/__init__.py,sha256=VkKqS4eHXd8YeDu2TAQ3Osqy70RSufUL1sECDoYzqvM,2685
 speedy_utils/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 speedy_utils/common/clock.py,sha256=raLtMGIgzrRej5kUt7hOUm2ZZw2THVPo-q8dMvdZOxw,7354
@@ -50,7 +50,7 @@ vision_utils/README.md,sha256=AIDZZj8jo_QNrEjFyHwd00iOO431s-js-M2dLtVTn3I,5740
 vision_utils/__init__.py,sha256=XsLxy1Fn33Zxu6hTFl3NEWfxGjuQQ-0Wmoh6lU9NZ_o,257
 vision_utils/io_utils.py,sha256=q41pffN632HbMmzcBzfg2Z7DvZZgoAQCdD9jHLqDgjc,26603
 vision_utils/plot.py,sha256=v73onfH8KbGHigw5KStUPqbLyJqIEOvvJaqtaoGKrls,12032
-speedy_utils-1.1.31.dist-info/METADATA,sha256=I_yQTL1iRuVJe_4Fy8ypBgjbzxl6XAszm58Bmz9R2oQ,8048
-speedy_utils-1.1.31.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-speedy_utils-1.1.31.dist-info/entry_points.txt,sha256=1rrFMfqvaMUE9hvwGiD6vnVh98kmgy0TARBj-v0Lfhs,244
-speedy_utils-1.1.31.dist-info/RECORD,,
+speedy_utils-1.1.33.dist-info/METADATA,sha256=QaZU14x_OlpExaRMZp3RnhkxEdRfBlhf0mqhaPTr6x4,8048
+speedy_utils-1.1.33.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+speedy_utils-1.1.33.dist-info/entry_points.txt,sha256=1rrFMfqvaMUE9hvwGiD6vnVh98kmgy0TARBj-v0Lfhs,244
+speedy_utils-1.1.33.dist-info/RECORD,,