speedy-utils 1.0.16__py3-none-any.whl → 1.0.21__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- llm_utils/__init__.py +3 -2
- llm_utils/lm/__init__.py +11 -8
- llm_utils/lm/chat_html.py +1 -1
- llm_utils/lm/lm_json.py +1 -1
- speedy_utils/common/utils_cache.py +2 -2
- {speedy_utils-1.0.16.dist-info → speedy_utils-1.0.21.dist-info}/METADATA +4 -3
- {speedy_utils-1.0.16.dist-info → speedy_utils-1.0.21.dist-info}/RECORD +11 -11
- /llm_utils/lm/{alm.py → async_lm.py} +0 -0
- /llm_utils/lm/{lm.py → sync_lm.py} +0 -0
- {speedy_utils-1.0.16.dist-info → speedy_utils-1.0.21.dist-info}/WHEEL +0 -0
- {speedy_utils-1.0.16.dist-info → speedy_utils-1.0.21.dist-info}/entry_points.txt +0 -0
llm_utils/__init__.py
CHANGED
|
@@ -9,8 +9,8 @@ from .chat_format import (
|
|
|
9
9
|
transform_messages,
|
|
10
10
|
transform_messages_to_chatml,
|
|
11
11
|
)
|
|
12
|
-
from .lm.
|
|
13
|
-
from .lm.
|
|
12
|
+
from .lm.async_lm import AsyncLLMTask, AsyncLM
|
|
13
|
+
from .lm.sync_lm import LM, LLMTask
|
|
14
14
|
|
|
15
15
|
__all__ = [
|
|
16
16
|
"transform_messages",
|
|
@@ -26,4 +26,5 @@ __all__ = [
|
|
|
26
26
|
"AsyncLM",
|
|
27
27
|
"display_chat_messages_as_html",
|
|
28
28
|
"LLMTask",
|
|
29
|
+
"AsyncLLMTask",
|
|
29
30
|
]
|
llm_utils/lm/__init__.py
CHANGED
|
@@ -1,9 +1,12 @@
|
|
|
1
|
-
from .
|
|
2
|
-
from .
|
|
3
|
-
OAI_LM = LM
|
|
1
|
+
# from .async_lm import AsyncLLMTask, AsyncLM
|
|
2
|
+
# from .lm import LM, LLMTask
|
|
4
3
|
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
4
|
+
# OAI_LM = LM
|
|
5
|
+
|
|
6
|
+
# __all__ = [
|
|
7
|
+
# "LM",
|
|
8
|
+
# "OAI_LM",
|
|
9
|
+
# "AsyncLM",
|
|
10
|
+
# "LLMTask",
|
|
11
|
+
# "AsyncLLMTask",
|
|
12
|
+
# ]
|
llm_utils/lm/chat_html.py
CHANGED
llm_utils/lm/lm_json.py
CHANGED
|
@@ -14,8 +14,8 @@ import xxhash
|
|
|
14
14
|
from loguru import logger
|
|
15
15
|
from pydantic import BaseModel
|
|
16
16
|
|
|
17
|
-
from
|
|
18
|
-
from
|
|
17
|
+
from speedy_utils.common.utils_io import dump_json_or_pickle, load_json_or_pickle
|
|
18
|
+
from speedy_utils.common.utils_misc import mkdir_or_exist
|
|
19
19
|
|
|
20
20
|
SPEED_CACHE_DIR = osp.join(osp.expanduser("~"), ".cache/speedy_cache")
|
|
21
21
|
LRU_MEM_CACHE = cachetools.LRUCache(maxsize=128_000)
|
|
@@ -1,11 +1,12 @@
|
|
|
1
1
|
Metadata-Version: 2.3
|
|
2
2
|
Name: speedy-utils
|
|
3
|
-
Version: 1.0.
|
|
3
|
+
Version: 1.0.21
|
|
4
4
|
Summary: Fast and easy-to-use package for data science
|
|
5
5
|
Author: AnhVTH
|
|
6
6
|
Author-email: anhvth.226@gmail.com
|
|
7
|
-
Requires-Python: >=3.
|
|
7
|
+
Requires-Python: >=3.8
|
|
8
8
|
Classifier: Programming Language :: Python :: 3
|
|
9
|
+
Classifier: Programming Language :: Python :: 3.8
|
|
9
10
|
Classifier: Programming Language :: Python :: 3.9
|
|
10
11
|
Classifier: Programming Language :: Python :: 3.10
|
|
11
12
|
Classifier: Programming Language :: Python :: 3.11
|
|
@@ -19,7 +20,7 @@ Requires-Dist: fastprogress
|
|
|
19
20
|
Requires-Dist: freezegun (>=1.5.1,<2.0.0)
|
|
20
21
|
Requires-Dist: ipdb
|
|
21
22
|
Requires-Dist: ipywidgets
|
|
22
|
-
Requires-Dist: json-repair (>=0.
|
|
23
|
+
Requires-Dist: json-repair (>=0.25.0,<0.31.0)
|
|
23
24
|
Requires-Dist: jupyterlab
|
|
24
25
|
Requires-Dist: loguru
|
|
25
26
|
Requires-Dist: matplotlib
|
|
@@ -1,14 +1,14 @@
|
|
|
1
|
-
llm_utils/__init__.py,sha256=
|
|
1
|
+
llm_utils/__init__.py,sha256=SlaCMArn_uKVw4r0psz0q0IOQ1VFGdgCFOAKxQ81WTI,694
|
|
2
2
|
llm_utils/chat_format/__init__.py,sha256=8dBIUqFJvkgQYedxBtcyxt-4tt8JxAKVap2JlTXmgaM,737
|
|
3
3
|
llm_utils/chat_format/display.py,sha256=qaEGADGP8iQFzWOuzEO7_HyrqAFdEnUfkHAH28b0ym0,9772
|
|
4
4
|
llm_utils/chat_format/transform.py,sha256=8TZhvUS5DrjUeMNtDIuWY54B_QZ7jjpXEL9c8F5z79w,5400
|
|
5
5
|
llm_utils/chat_format/utils.py,sha256=xTxN4HrLHcRO2PfCTR43nH1M5zCa7v0kTTdzAcGkZg0,1229
|
|
6
6
|
llm_utils/group_messages.py,sha256=8CU9nKOja3xeuhdrX5CvYVveSqSKb2zQ0eeNzA88aTQ,3621
|
|
7
|
-
llm_utils/lm/__init__.py,sha256=
|
|
8
|
-
llm_utils/lm/
|
|
9
|
-
llm_utils/lm/chat_html.py,sha256=
|
|
10
|
-
llm_utils/lm/
|
|
11
|
-
llm_utils/lm/
|
|
7
|
+
llm_utils/lm/__init__.py,sha256=rX36_MsnekM5GHwWS56XELbm4W5x2TDwnPERDTfo0eU,194
|
|
8
|
+
llm_utils/lm/async_lm.py,sha256=06QVT7iWAa8Rz48oRmnzkS-dJYOiJ0zSfKqEhZY11S8,30825
|
|
9
|
+
llm_utils/lm/chat_html.py,sha256=FkGo0Dv_nAHYBMZzXfMu_bGQKaCx302goh3XaT-_ETc,8674
|
|
10
|
+
llm_utils/lm/lm_json.py,sha256=fMt42phzFV2f6ulrtWcDXsWHi8WcG7gGkCzpIq8VSSM,1975
|
|
11
|
+
llm_utils/lm/sync_lm.py,sha256=FMiAhltqUNZdzGejMBPqGfQ3PE-fUoIGW8hfjTO_fNo,31035
|
|
12
12
|
llm_utils/lm/utils.py,sha256=GMvs64DRzVnXAki4SZ-A6mx2Fi9IgeF11BA-5FB-CYg,4777
|
|
13
13
|
llm_utils/scripts/README.md,sha256=yuOLnLa2od2jp4wVy3rV0rESeiV3o8zol5MNMsZx0DY,999
|
|
14
14
|
llm_utils/scripts/vllm_load_balancer.py,sha256=qsVhQ3ubJ3JnOAu4psHB1PyKy9muX-mZZkFqepH8WcU,17507
|
|
@@ -21,7 +21,7 @@ speedy_utils/common/function_decorator.py,sha256=BspJ0YuGL6elS7lWBAgELZ-sCfED_1N
|
|
|
21
21
|
speedy_utils/common/logger.py,sha256=JqW9gG4ujfq4RldNeYP2p52BYgCwjkYeGGYyzLn6mfY,6422
|
|
22
22
|
speedy_utils/common/notebook_utils.py,sha256=7tFXMaE365ubrD9VCwDoZe7oBqZ2qG5t_Vla4Qvg1M8,2062
|
|
23
23
|
speedy_utils/common/report_manager.py,sha256=sEuwLcyYpYlj9uxIsC8yiAu1eirydkQIldKRtFStswI,3894
|
|
24
|
-
speedy_utils/common/utils_cache.py,sha256=
|
|
24
|
+
speedy_utils/common/utils_cache.py,sha256=NsPf6rWV-Y4pFvrhKXtTQpfcxpSNAejE8JHzmzbb-Q4,8416
|
|
25
25
|
speedy_utils/common/utils_io.py,sha256=d7PKz5tOPrwHDr7GXuYmILvjXJOFEwfzAEIuUcYaI60,4790
|
|
26
26
|
speedy_utils/common/utils_misc.py,sha256=cdEuBBpiB1xpuzj0UBDHDuTIerqsMIw37ENq6EXliOw,1795
|
|
27
27
|
speedy_utils/common/utils_print.py,sha256=iQqnOYw2EFC8TqeSDbrcnIQAUKT7FbB8Mec8b2aGAzw,4833
|
|
@@ -31,7 +31,7 @@ speedy_utils/multi_worker/thread.py,sha256=9pXjvgjD0s0Hp0cZ6I3M0ndp1OlYZ1yvqbs_b
|
|
|
31
31
|
speedy_utils/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
32
32
|
speedy_utils/scripts/mpython.py,sha256=73PHm1jqbCt2APN4xuNjD0VDKwzOj4EZsViEMQiZU2g,3853
|
|
33
33
|
speedy_utils/scripts/openapi_client_codegen.py,sha256=f2125S_q0PILgH5dyzoKRz7pIvNEjCkzpi4Q4pPFRZE,9683
|
|
34
|
-
speedy_utils-1.0.
|
|
35
|
-
speedy_utils-1.0.
|
|
36
|
-
speedy_utils-1.0.
|
|
37
|
-
speedy_utils-1.0.
|
|
34
|
+
speedy_utils-1.0.21.dist-info/METADATA,sha256=6OzFN9FlcbRmdnnYeDffIt9EYEZvPzZT6uxupCXT56k,7442
|
|
35
|
+
speedy_utils-1.0.21.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
|
|
36
|
+
speedy_utils-1.0.21.dist-info/entry_points.txt,sha256=T1t85jwx8fK6m5msdkBGIXH5R5Kd0zSL0S6erXERPzg,237
|
|
37
|
+
speedy_utils-1.0.21.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|