speedy-utils 1.1.19__tar.gz → 1.1.21__tar.gz

{speedy_utils-1.1.19 → speedy_utils-1.1.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: speedy-utils
-Version: 1.1.19
+Version: 1.1.21
 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
@@ -18,26 +18,26 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
 Requires-Python: >=3.8
-Requires-Dist: aiohttp>=3.10.11
+Requires-Dist: aiohttp
 Requires-Dist: bump2version
 Requires-Dist: cachetools
 Requires-Dist: debugpy
 Requires-Dist: fastcore
 Requires-Dist: fastprogress
-Requires-Dist: freezegun>=1.5.1
+Requires-Dist: freezegun
 Requires-Dist: ipdb
 Requires-Dist: ipywidgets
-Requires-Dist: json-repair<0.31.0,>=0.25.0
+Requires-Dist: json-repair
 Requires-Dist: jupyterlab
 Requires-Dist: loguru
 Requires-Dist: matplotlib
 Requires-Dist: numpy
-Requires-Dist: openai>=1.106.0
-Requires-Dist: packaging<25,>=23.2
+Requires-Dist: openai
+Requires-Dist: packaging
 Requires-Dist: pandas
 Requires-Dist: pydantic
-Requires-Dist: pytest>=8.3.5
-Requires-Dist: ray>=2.36.1
+Requires-Dist: pytest
+Requires-Dist: ray
 Requires-Dist: requests
 Requires-Dist: scikit-learn
 Requires-Dist: tabulate

{speedy_utils-1.1.19 → speedy_utils-1.1.21}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "speedy-utils"
-version = "1.1.19"
+version = "1.1.21"
 description = "Fast and easy-to-use package for data science"
 authors = [{ name = "AnhVTH", email = "anhvth.226@gmail.com" }]
 readme = "README.md"
@@ -24,14 +24,14 @@ dependencies = [
     "tqdm",
     "cachetools",
     "bump2version",
-    "json-repair>=0.25.0,<0.31.0",
+    "json-repair",
     "fastprogress",
-    "freezegun>=1.5.1",
-    "packaging>=23.2,<25",
-    "openai>=1.106.0",
-    "ray>=2.36.1",
-    "aiohttp>=3.10.11",
-    "pytest>=8.3.5",
+    "freezegun",
+    "packaging",
+    "openai",
+    "ray",
+    "aiohttp",
+    "pytest",
 ]
 classifiers = [
     "Development Status :: 4 - Beta",

{speedy_utils-1.1.19 → speedy_utils-1.1.21}/src/llm_utils/lm/openai_memoize.py

@@ -1,4 +1,5 @@
 from openai import OpenAI, AsyncOpenAI
+from typing import Any, Callable
 
 from speedy_utils.common.utils_cache import memoize
 
@@ -30,6 +31,8 @@ class MOpenAI(OpenAI):
     - If you need a shared cache across instances, or more advanced cache controls,
         modify `memoize` or wrap at a class/static level instead of assigning to the
         bound method.
+    - Type information is now fully preserved by the memoize decorator, eliminating
+        the need for type casting.
 
     Example
             m = MOpenAI(api_key="...", model="gpt-4")
@@ -40,7 +43,12 @@ class MOpenAI(OpenAI):
     def __init__(self, *args, cache=True, **kwargs):
         super().__init__(*args, **kwargs)
         if cache:
-            self.post = memoize(self.post) # type: ignore
+            # Create a memoized wrapper for the instance's post method.
+            # The memoize decorator now preserves exact type information,
+            # so no casting is needed.
+            orig_post = self.post
+            memoized = memoize(orig_post)
+            self.post = memoized
 
 
 class MAsyncOpenAI(AsyncOpenAI):