speedy-utils 1.0.16__tar.gz → 1.0.21__tar.gz

{speedy_utils-1.0.16 → speedy_utils-1.0.21}/PKG-INFO

@@ -1,11 +1,12 @@
 Metadata-Version: 2.3
 Name: speedy-utils
-Version: 1.0.16
+Version: 1.0.21
 Summary: Fast and easy-to-use package for data science
 Author: AnhVTH
 Author-email: anhvth.226@gmail.com
-Requires-Python: >=3.9
+Requires-Python: >=3.8
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
@@ -19,7 +20,7 @@ Requires-Dist: fastprogress
 Requires-Dist: freezegun (>=1.5.1,<2.0.0)
 Requires-Dist: ipdb
 Requires-Dist: ipywidgets
-Requires-Dist: json-repair (>=0.40.0,<0.41.0)
+Requires-Dist: json-repair (>=0.25.0,<0.31.0)
 Requires-Dist: jupyterlab
 Requires-Dist: loguru
 Requires-Dist: matplotlib

{speedy_utils-1.0.16 → speedy_utils-1.0.21}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "speedy-utils"
-version = "1.0.16"
+version = "1.0.21"
 description = "Fast and easy-to-use package for data science"
 authors = ["AnhVTH <anhvth.226@gmail.com>"]
 readme = "README.md"
@@ -11,7 +11,7 @@ packages = [
 ]
 
 [build-system]
-requires = ["poetry-core>=1.0.16"]
+requires = ["poetry-core>=1.0.15"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.black]
@@ -36,7 +36,7 @@ no_implicit_optional = true
 strict_optional = true
 
 [tool.poetry.dependencies]
-python = ">=3.9"
+python = ">=3.8"
 numpy = "*"
 requests = "*"
 xxhash = "*"
@@ -54,7 +54,7 @@ pydantic = "*"
 tqdm = "*"
 cachetools = "*"
 bump2version = "*"
-json-repair = ">=0.40.0,<0.41.0"
+json-repair = ">=0.25.0,<0.31.0"
 fastprogress = "*"
 freezegun = "^1.5.1"
 packaging = ">=23.2,<25"
@@ -64,3 +64,18 @@ mpython = "speedy_utils.scripts.mpython:main"
 svllm = "llm_utils.scripts.vllm_serve:main"
 svllm-lb = "llm_utils.scripts.vllm_load_balancer:run_load_balancer"
 openapi_client_codegen = "speedy_utils.scripts.openapi_client_codegen:main"
+
+
+[tool.ruff.format]
+quote-style = "double"
+line-ending = "lf"
+docstring-code-format = true
+[tool.ruff]
+exclude = ["**/*.ipynb", "poly_frontend_controler/*", "poly_client/", "legacy"]
+ignore = [
+    "E501", # Line too long
+    "F401", # Unused import
+    "F403", # Wildcard import
+    "F841", # Local variable is assigned to but never used
+    "T201", # Use of `print` statement
+]

{speedy_utils-1.0.16 → speedy_utils-1.0.21}/src/llm_utils/__init__.py

@@ -9,8 +9,8 @@ from .chat_format import (
     transform_messages,
     transform_messages_to_chatml,
 )
-from .lm.lm import LM, LLMTask
-from .lm.alm import AsyncLM
+from .lm.async_lm import AsyncLLMTask, AsyncLM
+from .lm.sync_lm import LM, LLMTask
 
 __all__ = [
     "transform_messages",
@@ -26,4 +26,5 @@ __all__ = [
     "AsyncLM",
     "display_chat_messages_as_html",
     "LLMTask",
+    "AsyncLLMTask",
 ]

speedy_utils-1.0.21/src/llm_utils/lm/__init__.py

@@ -0,0 +1,12 @@
+# from .async_lm import AsyncLLMTask, AsyncLM
+# from .lm import LM, LLMTask
+
+# OAI_LM = LM
+
+# __all__ = [
+#     "LM",
+#     "OAI_LM",
+#     "AsyncLM",
+#     "LLMTask",
+#     "AsyncLLMTask",
+# ]

{speedy_utils-1.0.16 → speedy_utils-1.0.21}/src/llm_utils/lm/chat_html.py

@@ -1,5 +1,5 @@
 from typing import Any, Optional, cast
-from .lm import LM, Messages, LegacyMsgs, RawMsgs
+from .sync_lm import LM, Messages, LegacyMsgs, RawMsgs
 import sys
 
 # Configuration

{speedy_utils-1.0.16 → speedy_utils-1.0.21}/src/llm_utils/lm/lm_json.py

@@ -1,7 +1,7 @@
 from typing import Any, Optional
 
 
-from llm_utils.lm.lm import LM, RawMsgs
+from llm_utils.lm.sync_lm import LM, RawMsgs
 
 
 class LMJson(LM):

{speedy_utils-1.0.16 → speedy_utils-1.0.21}/src/speedy_utils/common/utils_cache.py

@@ -14,8 +14,8 @@ import xxhash
 from loguru import logger
 from pydantic import BaseModel
 
-from src.speedy_utils.common.utils_io import dump_json_or_pickle, load_json_or_pickle
-from src.speedy_utils.common.utils_misc import mkdir_or_exist
+from speedy_utils.common.utils_io import dump_json_or_pickle, load_json_or_pickle
+from speedy_utils.common.utils_misc import mkdir_or_exist
 
 SPEED_CACHE_DIR = osp.join(osp.expanduser("~"), ".cache/speedy_cache")
 LRU_MEM_CACHE = cachetools.LRUCache(maxsize=128_000)

speedy_utils-1.0.16/src/llm_utils/lm/__init__.py

@@ -1,9 +0,0 @@
-from .lm import LM
-from .alm import AsyncLM
-OAI_LM = LM
-
-__all__ = [
-    "LM",
-    "OAI_LM",
-    "AsyncLM",
-]